260607更新
This commit is contained in:
parent
e353616756
commit
dfd6b63abf
308
utils/agent.py
308
utils/agent.py
|
|
@ -5,25 +5,108 @@
|
|||
|
||||
# 列举导入模块
|
||||
from pathlib import Path
|
||||
from sys import path
|
||||
from typing import List, Optional, Union, cast
|
||||
from typing import TypeVar
|
||||
import time
|
||||
from typing import AsyncGenerator, List, Optional
|
||||
from uuid import uuid4
|
||||
|
||||
from pydantic_ai import Agent as BaseAgent, AgentRunResult
|
||||
from pydantic_ai import Agent as PydanticAIAgent, ModelMessage
|
||||
from pydantic_ai.capabilities import AgentCapability
|
||||
from pydantic_ai.messages import ModelMessagesTypeAdapter
|
||||
from pydantic_ai.models.openai import OpenAIChatModel
|
||||
from pydantic_ai.native_tools import AbstractNativeTool
|
||||
from pydantic_ai.output import OutputSpec
|
||||
from pydantic_ai.providers.openai import OpenAIProvider
|
||||
from starlette.applications import Starlette
|
||||
from starlette.routing import Mount, Route
|
||||
|
||||
path.append(Path(__file__).resolve().parent.as_posix())
|
||||
from memory import Memory
|
||||
from sqlite import SQLite
|
||||
|
||||
AgentDepsT = TypeVar("AgentDepsT")
|
||||
OutputDataT = TypeVar("OutputDataT")
|
||||
|
||||
class AgentMemory(SQLite):
|
||||
"""
|
||||
智能体的记忆体，支持：
|
||||
create：新增对话消息
|
||||
read：查询会话历史消息
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
"""
|
||||
初始化智能体的记忆体
|
||||
"""
|
||||
# 构建智能体的记忆体的数据库路径
|
||||
super().__init__(database=Path(__file__).parent.resolve() / "agent_memory.db")
|
||||
|
||||
try:
|
||||
with self:
|
||||
self.execute(
|
||||
sql="""
|
||||
CREATE TABLE IF NOT EXISTS new_messages
|
||||
(
|
||||
--唯一标识
|
||||
id TEXT PRIMARY KEY,
|
||||
--会话唯一标识
|
||||
session_id TEXT NOT NULL,
|
||||
--新对话消息
|
||||
new_messages TEXT NOT NULL,
|
||||
--时间戳（毫秒）
|
||||
timestamp INTEGER NOT NULL
|
||||
)
|
||||
"""
|
||||
)
|
||||
except Exception as exception:
|
||||
raise RuntimeError(
|
||||
f"初始化智能体的记忆体发生异常：{str(exception)}"
|
||||
) from exception
|
||||
|
||||
def create_new_messages(
|
||||
self, session_id: str, new_messages: List[ModelMessage]
|
||||
) -> bool:
|
||||
"""
|
||||
新增新对话消息
|
||||
:param session_id: 会话唯一标识
|
||||
:param new_messages: 新对话消息
|
||||
:return: 新增是否成功
|
||||
"""
|
||||
try:
|
||||
with self:
|
||||
return self.execute(
|
||||
sql="""
|
||||
INSERT INTO new_messages (id, session_id, new_messages, timestamp) VALUES (?, ?, ?, ?)
|
||||
""",
|
||||
parameters=(
|
||||
uuid4().hex.lower(),
|
||||
session_id,
|
||||
ModelMessagesTypeAdapter.dump_json(new_messages),
|
||||
int(time.time() * 1000),
|
||||
),
|
||||
)
|
||||
except Exception as exception:
|
||||
raise RuntimeError(f"新增对话消息发生异常：{str(exception)}") from exception
|
||||
|
||||
def read_message_history(self, session_id: str) -> List[ModelMessage]:
|
||||
"""
|
||||
查询对话消息历史
|
||||
:param session_id: 会话唯一标识
|
||||
:return: 对话消息历史
|
||||
"""
|
||||
try:
|
||||
with self:
|
||||
result = self.query_all(
|
||||
sql="""
|
||||
SELECT new_messages
|
||||
FROM new_messages
|
||||
WHERE session_id = ?
|
||||
ORDER BY timestamp ASC
|
||||
""",
|
||||
parameters=(session_id,),
|
||||
)
|
||||
message_history = []
|
||||
for row in result:
|
||||
message_history.extend(
|
||||
ModelMessagesTypeAdapter.validate_json(row["new_message"])
|
||||
)
|
||||
return message_history
|
||||
except Exception as exception:
|
||||
raise RuntimeError(
|
||||
f"查询会话历史消息发生异常：{str(exception)}"
|
||||
) from exception
|
||||
|
||||
|
||||
class Agent:
|
||||
|
|
@ -49,31 +132,11 @@ class Agent:
|
|||
# 生成会话唯一标识
|
||||
self.session_id = uuid4().hex.lower()
|
||||
|
||||
# 创建智能体
|
||||
self.agent = self._create_agent(
|
||||
instructions=instructions,
|
||||
capabilities=capabilities,
|
||||
output_type=output_type,
|
||||
)
|
||||
a = self.agent.to_web()
|
||||
# 实例智能体的记忆体
|
||||
self.agent_memory = AgentMemory()
|
||||
|
||||
# 实例记忆体
|
||||
self.memory = Memory()
|
||||
|
||||
def _create_agent(
|
||||
self,
|
||||
instructions: str,
|
||||
capabilities: Optional[List[AgentCapability]],
|
||||
output_type: OutputSpec,
|
||||
) -> BaseAgent:
|
||||
"""
|
||||
创建智能体
|
||||
:param instructions: 指令
|
||||
:param capabilities: 智能体能力列表
|
||||
:param output_type: 输出类型
|
||||
:return: Agent 实例
|
||||
"""
|
||||
agent = BaseAgent(
|
||||
# 实例智能体
|
||||
self.agent = PydanticAIAgent(
|
||||
model=OpenAIChatModel(
|
||||
model_name="deepseek-v4-flash",
|
||||
provider=OpenAIProvider(
|
||||
|
|
@ -86,165 +149,28 @@ class Agent:
|
|||
output_type=output_type,
|
||||
retries=1,
|
||||
)
|
||||
return agent
|
||||
|
||||
async def run(self, user_prompt: str | List[str]) -> AgentRunResult:
|
||||
async def output_message_streamed(
|
||||
self, user_prompt: str | List[str]
|
||||
) -> AsyncGenerator[str, None]:
|
||||
"""
|
||||
异步运行
|
||||
:param user_prompt: 用户提示词
|
||||
:return: 智能体回复
|
||||
智能体流式输出消息
|
||||
:param user_prompt: 用户提示词（用户输入消息）
|
||||
:return: 流式消息
|
||||
"""
|
||||
# 查询会话历史消息
|
||||
message_history = self.memory.read(session_id=self.session_id)
|
||||
result = await self.agent.run(
|
||||
"""定义：一次会话（session）包含若干论对话（turn），每一轮对话由用户输入消息（message）和智能体输出消息组成"""
|
||||
# 查询该会话历史对话消息列表
|
||||
message_history = self.agent_memory.read_message_history(
|
||||
session_id=self.session_id
|
||||
)
|
||||
|
||||
async with self.agent.run_stream(
|
||||
user_prompt=user_prompt, message_history=message_history
|
||||
)
|
||||
# 记录会话历史消息
|
||||
self.memory.create(
|
||||
) as result:
|
||||
async for chunk in result.stream_text():
|
||||
yield chunk
|
||||
|
||||
self.agent_memory.create_new_messages(
|
||||
session_id=self.session_id,
|
||||
dialogue_message=result.new_messages(),
|
||||
new_messages=result.new_messages(),
|
||||
)
|
||||
return result
|
||||
|
||||
def run_agent_application(
|
||||
self,
|
||||
native_tools: Optional[List[AbstractNativeTool]] = None,
|
||||
html_source: Optional[Union[str, Path]] = None,
|
||||
) -> Starlette:
|
||||
"""
|
||||
启动智能体应用
|
||||
:param native_tools: 前端可选工具列表
|
||||
:return: Starlette 实例
|
||||
"""
|
||||
from starlette.requests import Request
|
||||
from starlette.responses import JSONResponse, HTMLResponse, Response
|
||||
|
||||
def api(
|
||||
agent: BaseAgent[AgentDepsT, OutputDataT],
|
||||
native_tools: Optional[List[AbstractNativeTool]] = None,
|
||||
) -> Starlette:
|
||||
"""
|
||||
创建处理请求应用
|
||||
:param agent: 智能体
|
||||
:param native_tools: 前端可选原生工具列表
|
||||
:return: Starlette 实例
|
||||
"""
|
||||
from pydantic_ai.models import Model
|
||||
from pydantic_ai.ui._web.api import ModelInfo
|
||||
from pydantic_ai.ui._web.api import (
|
||||
ConfigureFrontend,
|
||||
BuiltinToolInfo,
|
||||
ChatRequestExtra,
|
||||
validate_request_options,
|
||||
)
|
||||
from pydantic_ai.ui.vercel_ai import VercelAIAdapter
|
||||
from pydantic_ai.capabilities import NativeTool
|
||||
|
||||
async def chat_options(request: Request) -> Response:
|
||||
"""处理跨域预检请求"""
|
||||
return Response()
|
||||
|
||||
agent_model = cast(Model, agent.model)
|
||||
# 前端可选原生工具列表
|
||||
frontend_native_tools = [
|
||||
t
|
||||
for t in (native_tools or [])
|
||||
if t.unique_id
|
||||
not in {
|
||||
t.unique_id
|
||||
for t in agent._cap_native_tools
|
||||
if isinstance(t, AbstractNativeTool)
|
||||
}
|
||||
]
|
||||
|
||||
async def configurations(request: Request) -> Response:
|
||||
"""处理前端模型与工具配置请求"""
|
||||
configurations = ConfigureFrontend(
|
||||
models=[
|
||||
ModelInfo(
|
||||
id=agent_model.model_id,
|
||||
name=agent_model.label,
|
||||
builtin_tools=[
|
||||
t.unique_id
|
||||
for t in frontend_native_tools
|
||||
if type(t) in agent_model.profile.supported_native_tools
|
||||
],
|
||||
)
|
||||
], # 前端仅可选择智能体已配置的模型
|
||||
builtin_tools=[
|
||||
BuiltinToolInfo(id=t.unique_id, name=t.label)
|
||||
for t in frontend_native_tools
|
||||
],
|
||||
)
|
||||
return JSONResponse(content=configurations.model_dump(by_alias=True))
|
||||
|
||||
async def chat_post(request: Request) -> Response:
|
||||
"""处理对话请求"""
|
||||
# 实例 Vercel AI 适配器
|
||||
adapter = await VercelAIAdapter[
|
||||
TypeVar("AgentDepsT"), TypeVar("OutputDataT")
|
||||
].from_request(request=request, agent=self.agent)
|
||||
|
||||
# 解析请求中额外数据，包括前端选择的模型标识、原生工具标识和其它配置等
|
||||
extra_data = ChatRequestExtra.model_validate(
|
||||
adapter.run_input.__pydantic_extra__
|
||||
)
|
||||
if error := validate_request_options(
|
||||
extra_data=extra_data,
|
||||
model_ids={agent_model.model_id}, # 前端仅可选择智能体已配置的模型
|
||||
builtin_tool_ids={t.unique_id for t in frontend_native_tools},
|
||||
):
|
||||
return JSONResponse(content={"error": error}, status_code=400)
|
||||
|
||||
streaming_response = await VercelAIAdapter[
|
||||
TypeVar("AgentDepsT"), TypeVar("OutputDataT")
|
||||
].dispatch_request(
|
||||
request=request,
|
||||
agent=self.agent,
|
||||
capabilities=[
|
||||
NativeTool(t)
|
||||
for t in frontend_native_tools
|
||||
if t.unique_id in extra_data.builtin_tools
|
||||
],
|
||||
)
|
||||
return streaming_response
|
||||
|
||||
async def health(request: Request) -> Response:
|
||||
"""处理健康检查请求"""
|
||||
return JSONResponse(content={"ok": True})
|
||||
|
||||
return Starlette(
|
||||
routes=[
|
||||
Route(path="/chat", endpoint=chat_options, methods=["OPTIONS"]),
|
||||
Route(path="/configure", endpoint=configurations, methods=["GET"]),
|
||||
Route(path="/chat", endpoint=chat_post, methods=["POST"]),
|
||||
Route(path="/health", endpoint=health, methods=["GET"]),
|
||||
]
|
||||
)
|
||||
|
||||
application = Starlette(
|
||||
routes=[
|
||||
Mount(
|
||||
"/api",
|
||||
app=api(agent=self.agent, native_tools=native_tools),
|
||||
)
|
||||
]
|
||||
)
|
||||
|
||||
async def index(request: Request) -> Response:
|
||||
"""S处理聊天界面请求"""
|
||||
from pydantic_ai.ui._web.app import _get_ui_html
|
||||
|
||||
content = await _get_ui_html(html_source)
|
||||
|
||||
return HTMLResponse(
|
||||
content=content,
|
||||
headers={
|
||||
"Cache-Control": "public, max-age=3600",
|
||||
},
|
||||
)
|
||||
|
||||
application.router.add_route(path="/", endpoint=index, methods=["GET"])
|
||||
application.router.add_route(path="/{id}", endpoint=index, methods=["GET"])
|
||||
|
||||
return application
|
||||
|
|
|
|||
105
utils/memory.py
105
utils/memory.py
|
|
@ -1,105 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
记忆模块
|
||||
"""
|
||||
|
||||
# 列举导入模块
|
||||
from pathlib import Path
|
||||
import sys
|
||||
import time
|
||||
from typing import List
|
||||
from uuid import uuid4
|
||||
|
||||
from pydantic_ai import ModelMessage
|
||||
from pydantic_ai.messages import ModelMessagesTypeAdapter
|
||||
|
||||
sys.path.append(Path(__file__).resolve().parent.as_posix())
|
||||
from sqlite import SQLite
|
||||
|
||||
|
||||
class Memory(SQLite):
|
||||
"""
|
||||
记忆体，支持：
|
||||
create：新增对话消息
|
||||
read：查询会话历史消息
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
"""
|
||||
初始化记忆体
|
||||
"""
|
||||
# 构建数据库路径
|
||||
super().__init__(
|
||||
database=Path(__file__).parent.resolve() / "memory_database.db"
|
||||
)
|
||||
|
||||
try:
|
||||
with self:
|
||||
self.execute(
|
||||
sql="""
|
||||
CREATE TABLE IF NOT EXISTS messages
|
||||
(
|
||||
--唯一标识
|
||||
id TEXT PRIMARY KEY,
|
||||
--会话唯一标识
|
||||
session_id TEXT NOT NULL,
|
||||
--对话消息
|
||||
dialogue_message TEXT NOT NULL,
|
||||
--时间戳（毫秒）
|
||||
timestamp INTEGER NOT NULL
|
||||
)
|
||||
"""
|
||||
)
|
||||
except Exception as exception:
|
||||
raise RuntimeError(f"初始化记忆体发生异常：{str(exception)}") from exception
|
||||
|
||||
def create(self, session_id: str, dialogue_message: List[ModelMessage]) -> bool:
|
||||
"""
|
||||
新增对话消息
|
||||
:param session_id: 会话唯一标识
|
||||
:param dialogue_message: 对话消息
|
||||
:return: 新增是否成功
|
||||
"""
|
||||
try:
|
||||
with self:
|
||||
return self.execute(
|
||||
sql="""
|
||||
INSERT INTO messages (id, session_id, dialogue_message, timestamp) VALUES (?, ?, ?, ?)
|
||||
""",
|
||||
parameters=(
|
||||
uuid4().hex.lower(),
|
||||
session_id,
|
||||
ModelMessagesTypeAdapter.dump_json(dialogue_message),
|
||||
int(time.time() * 1000),
|
||||
),
|
||||
)
|
||||
except Exception as exception:
|
||||
raise RuntimeError(f"新增对话消息发生异常：{str(exception)}") from exception
|
||||
|
||||
def read(self, session_id: str) -> List[ModelMessage]:
|
||||
"""
|
||||
查询会话历史消息
|
||||
:param session_id: 会话唯一标识
|
||||
:return: 会话历史消息
|
||||
"""
|
||||
try:
|
||||
with self:
|
||||
result = self.query_all(
|
||||
sql="""
|
||||
SELECT dialogue_message
|
||||
FROM messages
|
||||
WHERE session_id = ?
|
||||
ORDER BY timestamp ASC
|
||||
""",
|
||||
parameters=(session_id,),
|
||||
)
|
||||
message_history = []
|
||||
for row in result:
|
||||
message_history.extend(
|
||||
ModelMessagesTypeAdapter.validate_json(row["dialogue_message"])
|
||||
)
|
||||
return message_history
|
||||
except Exception as exception:
|
||||
raise RuntimeError(
|
||||
f"查询会话历史消息发生异常：{str(exception)}"
|
||||
) from exception
|
||||
Binary file not shown.
|
|
@ -1,282 +0,0 @@
|
|||
import reflex as rx
|
||||
import reflex as rx
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 全局状态（已修复所有类型错误）
|
||||
# ==============================================
|
||||
class State(rx.State):
|
||||
# 登录状态
|
||||
is_logged_in: bool = False
|
||||
username: str = ""
|
||||
password: str = ""
|
||||
|
||||
# 当前页面
|
||||
current_page: str = "chat"
|
||||
|
||||
# 会话列表
|
||||
sessions: list[dict] = [
|
||||
{"id": 1, "title": "会话 1"},
|
||||
{"id": 2, "title": "会话 2"},
|
||||
{"id": 3, "title": "会话 3"},
|
||||
]
|
||||
current_session_id: int | None = 1
|
||||
|
||||
# 登录弹窗控制
|
||||
show_login_modal: bool = True
|
||||
|
||||
# ------------------------------
|
||||
# 显式定义 setter（修复报错）
|
||||
# ------------------------------
|
||||
@rx.event
|
||||
def set_username(self, value: str):
|
||||
self.username = value
|
||||
|
||||
@rx.event
|
||||
def set_password(self, value: str):
|
||||
self.password = value
|
||||
|
||||
# ------------------------------
|
||||
# 登录 / 退出
|
||||
# ------------------------------
|
||||
@rx.event
|
||||
def login(self):
|
||||
if self.username.strip() and self.password.strip():
|
||||
self.is_logged_in = True
|
||||
self.show_login_modal = False
|
||||
|
||||
@rx.event
|
||||
def logout(self):
|
||||
self.is_logged_in = False
|
||||
self.show_login_modal = True
|
||||
self.username = ""
|
||||
self.password = ""
|
||||
|
||||
# ------------------------------
|
||||
# 页面切换
|
||||
# ------------------------------
|
||||
@rx.event
|
||||
def set_page(self, page: str):
|
||||
self.current_page = page
|
||||
|
||||
# ------------------------------
|
||||
# 会话操作
|
||||
# ------------------------------
|
||||
@rx.event
|
||||
def new_session(self):
|
||||
new_id = max([s["id"] for s in self.sessions], default=0) + 1
|
||||
self.sessions.append({"id": new_id, "title": f"会话 {new_id}"})
|
||||
self.current_session_id = new_id
|
||||
|
||||
@rx.event
|
||||
def delete_session(self, session_id: int):
|
||||
self.sessions = [s for s in self.sessions if s["id"] != session_id]
|
||||
if self.current_session_id == session_id:
|
||||
if self.sessions:
|
||||
self.current_session_id = self.sessions[0]["id"]
|
||||
else:
|
||||
self.current_session_id = None
|
||||
|
||||
@rx.event
|
||||
def set_current_session_id(self, sid: int):
|
||||
self.current_session_id = sid
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 登录弹窗（已修复）
|
||||
# ==============================================
|
||||
def login_modal():
|
||||
return rx.cond(
|
||||
State.show_login_modal,
|
||||
rx.box(
|
||||
rx.box(
|
||||
rx.vstack(
|
||||
rx.heading("登录"),
|
||||
rx.input(
|
||||
placeholder="用户名",
|
||||
value=State.username,
|
||||
on_change=State.set_username,
|
||||
width="100%",
|
||||
),
|
||||
rx.input(
|
||||
placeholder="密码",
|
||||
type="password",
|
||||
value=State.password,
|
||||
on_change=State.set_password,
|
||||
width="100%",
|
||||
),
|
||||
rx.button(
|
||||
"登录",
|
||||
on_click=State.login,
|
||||
width="100%",
|
||||
bg="blue",
|
||||
color="white",
|
||||
),
|
||||
spacing="4",
|
||||
padding="6",
|
||||
width="350px",
|
||||
),
|
||||
bg="white",
|
||||
padding="8",
|
||||
border_radius="lg",
|
||||
shadow="lg",
|
||||
),
|
||||
position="fixed",
|
||||
inset="0",
|
||||
bg="rgba(0,0,0,0.5)",
|
||||
display="flex",
|
||||
align_items="center",
|
||||
justify_content="center",
|
||||
z_index="1000",
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 左侧边栏
|
||||
# ==============================================
|
||||
def sidebar():
|
||||
return rx.box(
|
||||
rx.vstack(
|
||||
rx.heading("AI 智能体", font_size="lg"),
|
||||
rx.button(
|
||||
"💬 聊天",
|
||||
on_click=State.set_page("chat"),
|
||||
width="100%",
|
||||
bg=rx.cond(State.current_page == "chat", "blue", "gray"),
|
||||
color="white",
|
||||
),
|
||||
rx.button(
|
||||
"📚 知识库",
|
||||
on_click=State.set_page("knowledge"),
|
||||
width="100%",
|
||||
bg=rx.cond(State.current_page == "knowledge", "blue", "gray"),
|
||||
color="white",
|
||||
),
|
||||
rx.button(
|
||||
"⚙️ 设置",
|
||||
on_click=State.set_page("settings"),
|
||||
width="100%",
|
||||
bg=rx.cond(State.current_page == "settings", "blue", "gray"),
|
||||
color="white",
|
||||
),
|
||||
rx.spacer(),
|
||||
rx.button(
|
||||
"🚪 退出登录",
|
||||
on_click=State.logout,
|
||||
width="100%",
|
||||
bg="red",
|
||||
color="white",
|
||||
),
|
||||
padding="4",
|
||||
height="100vh",
|
||||
spacing="3",
|
||||
),
|
||||
width="220px",
|
||||
bg="white",
|
||||
shadow="md",
|
||||
)
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 聊天页面
|
||||
# ==============================================
|
||||
def chat_page():
|
||||
return rx.hstack(
|
||||
rx.box(
|
||||
rx.vstack(
|
||||
rx.button(
|
||||
"➕ 新建会话",
|
||||
on_click=State.new_session,
|
||||
width="100%",
|
||||
bg="green",
|
||||
color="white",
|
||||
),
|
||||
rx.foreach(
|
||||
State.sessions,
|
||||
lambda s: rx.hstack(
|
||||
rx.text(
|
||||
s["title"],
|
||||
flex=1,
|
||||
cursor="pointer",
|
||||
on_click=lambda: State.set_current_session_id(s["id"]),
|
||||
),
|
||||
rx.icon(
|
||||
"trash",
|
||||
size=16,
|
||||
color="red",
|
||||
cursor="pointer",
|
||||
on_click=lambda: State.delete_session(s["id"]),
|
||||
),
|
||||
width="100%",
|
||||
padding="2",
|
||||
bg=rx.cond(
|
||||
State.current_session_id == s["id"],
|
||||
"gray.100",
|
||||
"transparent",
|
||||
),
|
||||
border_radius="sm",
|
||||
),
|
||||
),
|
||||
spacing="2",
|
||||
align_items="start",
|
||||
width="100%",
|
||||
),
|
||||
width="220px",
|
||||
padding="4",
|
||||
border_right="1px solid #e2e8f0",
|
||||
),
|
||||
rx.box(
|
||||
rx.heading("聊天内容"),
|
||||
flex=1,
|
||||
padding="4",
|
||||
),
|
||||
flex=1,
|
||||
)
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 内容区域
|
||||
# ==============================================
|
||||
def content_area():
|
||||
return rx.box(
|
||||
rx.cond(
|
||||
State.current_page == "chat",
|
||||
chat_page(),
|
||||
rx.cond(
|
||||
State.current_page == "knowledge",
|
||||
rx.heading("知识库管理"),
|
||||
rx.heading("系统设置"),
|
||||
),
|
||||
),
|
||||
flex=1,
|
||||
)
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 主页面
|
||||
# ==============================================
|
||||
def index():
|
||||
return rx.box(
|
||||
login_modal(),
|
||||
rx.cond(
|
||||
State.is_logged_in,
|
||||
rx.hstack(
|
||||
sidebar(),
|
||||
content_area(),
|
||||
),
|
||||
rx.center(
|
||||
rx.heading("请登录使用系统"),
|
||||
height="100vh",
|
||||
),
|
||||
),
|
||||
min_height="100vh",
|
||||
bg="gray.50",
|
||||
)
|
||||
|
||||
|
||||
# ==============================================
|
||||
# 启动
|
||||
# ==============================================
|
||||
app = rx.App()
|
||||
app.add_page(index)
|
||||
|
|
@ -0,0 +1,28 @@
|
|||
"""The main Chat app."""
|
||||
|
||||
import reflex as rx
|
||||
from application.components import chat, navbar
|
||||
|
||||
|
||||
def index() -> rx.Component:
|
||||
"""The main app."""
|
||||
return rx.vstack(
|
||||
navbar.navbar(),
|
||||
chat.chat(),
|
||||
chat.action_bar(),
|
||||
background_color=rx.color("mauve", 1),
|
||||
color=rx.color("mauve", 12),
|
||||
height="100dvh",
|
||||
align_items="stretch",
|
||||
spacing="0",
|
||||
)
|
||||
|
||||
|
||||
# Add state and page to the app.
|
||||
app = rx.App(
|
||||
theme=rx.theme(
|
||||
appearance="dark",
|
||||
accent_color="purple",
|
||||
),
|
||||
)
|
||||
app.add_page(index)
|
||||
|
|
@ -0,0 +1,112 @@
|
|||
import reflex as rx
|
||||
|
||||
from application.state import Turn, State
|
||||
from reflex.constants.colors import ColorType
|
||||
|
||||
|
||||
def message_content(text: str, color: ColorType) -> rx.Component:
|
||||
"""Create a message content component.
|
||||
|
||||
Args:
|
||||
text: The text to display.
|
||||
color: The color of the message.
|
||||
|
||||
Returns:
|
||||
A component displaying the message.
|
||||
"""
|
||||
return rx.markdown(
|
||||
text,
|
||||
background_color=rx.color(color, 4),
|
||||
color=rx.color(color, 12),
|
||||
display="inline-block",
|
||||
padding_inline="1em",
|
||||
border_radius="8px",
|
||||
)
|
||||
|
||||
|
||||
def message(turn: Turn) -> rx.Component:
|
||||
"""A single question/answer message.
|
||||
|
||||
Args:
|
||||
turn: The question/answer pair.
|
||||
|
||||
Returns:
|
||||
A component displaying the question/answer pair.
|
||||
"""
|
||||
return rx.box(
|
||||
rx.box(
|
||||
message_content(turn["question"], "mauve"),
|
||||
text_align="right",
|
||||
margin_bottom="8px",
|
||||
),
|
||||
rx.box(
|
||||
message_content(turn["answer"], "accent"),
|
||||
text_align="left",
|
||||
margin_bottom="8px",
|
||||
),
|
||||
max_width="50em",
|
||||
margin_inline="auto",
|
||||
)
|
||||
|
||||
|
||||
def chat() -> rx.Component:
|
||||
"""List all the messages in a single conversation."""
|
||||
return rx.auto_scroll(
|
||||
rx.foreach(State.selected_chat, message),
|
||||
flex="1",
|
||||
padding="8px",
|
||||
)
|
||||
|
||||
|
||||
def action_bar() -> rx.Component:
|
||||
"""The action bar to send a new message."""
|
||||
return rx.center(
|
||||
rx.vstack(
|
||||
rx.form(
|
||||
rx.hstack(
|
||||
rx.input(
|
||||
rx.input.slot(
|
||||
rx.tooltip(
|
||||
rx.icon("info", size=18),
|
||||
content="Enter a question to get a response.",
|
||||
)
|
||||
),
|
||||
placeholder="Type something...",
|
||||
id="question",
|
||||
flex="1",
|
||||
),
|
||||
rx.button(
|
||||
"Send",
|
||||
loading=State.processing,
|
||||
disabled=State.processing,
|
||||
type="submit",
|
||||
),
|
||||
max_width="50em",
|
||||
margin="0 auto",
|
||||
align_items="center",
|
||||
),
|
||||
reset_on_submit=True,
|
||||
on_submit=State.process_question,
|
||||
),
|
||||
rx.text(
|
||||
"ReflexGPT may return factually incorrect or misleading responses. Use discretion.",
|
||||
text_align="center",
|
||||
font_size=".75em",
|
||||
color=rx.color("mauve", 10),
|
||||
),
|
||||
rx.logo(margin_block="-1em"),
|
||||
width="100%",
|
||||
padding_x="16px",
|
||||
align="stretch",
|
||||
),
|
||||
position="sticky",
|
||||
bottom="0",
|
||||
left="0",
|
||||
padding_y="16px",
|
||||
backdrop_filter="auto",
|
||||
backdrop_blur="lg",
|
||||
border_top=f"1px solid {rx.color('mauve', 3)}",
|
||||
background_color=rx.color("mauve", 2),
|
||||
align="stretch",
|
||||
width="100%",
|
||||
)
|
||||
|
|
@ -0,0 +1,115 @@
|
|||
import reflex as rx
|
||||
from application.state import State
|
||||
|
||||
|
||||
def sidebar_chat(chat: str) -> rx.Component:
|
||||
"""A sidebar chat item.
|
||||
|
||||
Args:
|
||||
chat: The chat item.
|
||||
"""
|
||||
return rx.drawer.close(
|
||||
rx.hstack(
|
||||
rx.button(
|
||||
chat,
|
||||
on_click=lambda: State.set_chat(chat),
|
||||
width="80%",
|
||||
variant="surface",
|
||||
),
|
||||
rx.button(
|
||||
rx.icon(
|
||||
tag="trash",
|
||||
on_click=State.delete_chat(chat),
|
||||
stroke_width=1,
|
||||
),
|
||||
width="20%",
|
||||
variant="surface",
|
||||
color_scheme="red",
|
||||
),
|
||||
width="100%",
|
||||
),
|
||||
key=chat,
|
||||
)
|
||||
|
||||
|
||||
def sidebar(trigger) -> rx.Component:
|
||||
"""The sidebar component."""
|
||||
return rx.drawer.root(
|
||||
rx.drawer.trigger(trigger),
|
||||
rx.drawer.overlay(),
|
||||
rx.drawer.portal(
|
||||
rx.drawer.content(
|
||||
rx.vstack(
|
||||
rx.heading("Chats", color=rx.color("mauve", 11)),
|
||||
rx.divider(),
|
||||
rx.foreach(State.chat_titles, lambda chat: sidebar_chat(chat)),
|
||||
align_items="stretch",
|
||||
width="100%",
|
||||
),
|
||||
top="auto",
|
||||
right="auto",
|
||||
height="100%",
|
||||
width="20em",
|
||||
padding="2em",
|
||||
background_color=rx.color("mauve", 2),
|
||||
outline="none",
|
||||
)
|
||||
),
|
||||
direction="left",
|
||||
)
|
||||
|
||||
|
||||
def modal(trigger) -> rx.Component:
|
||||
"""A modal to create a new chat."""
|
||||
return rx.dialog.root(
|
||||
rx.dialog.trigger(trigger),
|
||||
rx.dialog.content(
|
||||
rx.form(
|
||||
rx.hstack(
|
||||
rx.input(
|
||||
placeholder="Chat name",
|
||||
name="new_chat_name",
|
||||
flex="1",
|
||||
min_width="20ch",
|
||||
),
|
||||
rx.button("Create chat"),
|
||||
spacing="2",
|
||||
wrap="wrap",
|
||||
width="100%",
|
||||
),
|
||||
on_submit=State.create_chat,
|
||||
),
|
||||
background_color=rx.color("mauve", 1),
|
||||
),
|
||||
open=State.is_modal_open,
|
||||
on_open_change=State.set_is_modal_open,
|
||||
)
|
||||
|
||||
|
||||
def navbar():
|
||||
return rx.hstack(
|
||||
rx.badge(
|
||||
State.current_chat,
|
||||
rx.tooltip(
|
||||
rx.icon("info", size=14),
|
||||
content="The current selected chat.",
|
||||
),
|
||||
size="3",
|
||||
variant="soft",
|
||||
margin_inline_end="auto",
|
||||
),
|
||||
modal(
|
||||
rx.icon_button("message-square-plus"),
|
||||
),
|
||||
sidebar(
|
||||
rx.icon_button(
|
||||
"messages-square",
|
||||
background_color=rx.color("mauve", 6),
|
||||
)
|
||||
),
|
||||
justify_content="space-between",
|
||||
align_items="center",
|
||||
padding="12px",
|
||||
border_bottom=f"1px solid {rx.color('mauve', 3)}",
|
||||
background_color=rx.color("mauve", 2),
|
||||
)
|
||||
|
|
@ -0,0 +1,189 @@
|
|||
import os
|
||||
from typing import Any, Optional, TypedDict
|
||||
import reflex
|
||||
from openai import OpenAI
|
||||
from openai.types.chat import ChatCompletionMessageParam
|
||||
from sys import path
|
||||
from pathlib import Path
|
||||
from dataclasses import dataclass
|
||||
|
||||
path.append(str(Path(__file__).resolve().parent.parent.parent))
|
||||
from utils.agent import Agent
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
class Turn(BaseModel):
|
||||
"""一轮对话中用户输入消息和智能体输出消息"""
|
||||
|
||||
input_message: str = Field(description="用户输入的消息")
|
||||
output_message: str
|
||||
|
||||
|
||||
class Session(BaseModel):
|
||||
"""会话数据模型"""
|
||||
|
||||
uuid: str = Field(..., description="会话唯一标识")
|
||||
agent: Optional[Agent] = Field(default=None, description="智能体实例") # 鼓励
|
||||
message_history: List[Turn] = Field(default=[], description="会话消息历史")
|
||||
|
||||
|
||||
class State(reflex.State):
|
||||
"""实例状态管理"""
|
||||
|
||||
agent: Agent | None = None
|
||||
|
||||
def _init_agent(self) -> Agent:
|
||||
"""初始化智能体"""
|
||||
if not self.agent:
|
||||
self.agent = Agent(
|
||||
instructions="You are a friendly chatbot named Reflex. Respond in markdown."
|
||||
)
|
||||
return self.agent
|
||||
|
||||
current_chat_name = "Intros" # 当前聊天名称
|
||||
|
||||
sessions: dict[str, list[Turn]] = {
|
||||
current_chat_name: [],
|
||||
} # 会话列表
|
||||
|
||||
# 智能体是否正在输出
|
||||
outputting: bool = False
|
||||
|
||||
# 新会话弹窗是否打开
|
||||
is_modal_open: bool = False
|
||||
|
||||
@reflex.event
|
||||
def create_chat(self, form_data: dict[str, Any]):
|
||||
"""Create a new chat."""
|
||||
# Add the new chat to the list of chats.
|
||||
new_chat_name = form_data["new_chat_name"]
|
||||
self.current_chat = new_chat_name
|
||||
self._chats[new_chat_name] = []
|
||||
self.is_modal_open = False
|
||||
|
||||
@reflex.event
|
||||
def set_is_modal_open(self, is_open: bool):
|
||||
"""Set the new chat modal open state.
|
||||
|
||||
Args:
|
||||
is_open: Whether the modal is open.
|
||||
"""
|
||||
self.is_modal_open = is_open
|
||||
|
||||
@reflex.var
|
||||
def selected_chat(self) -> list[TurnMessages]:
|
||||
"""Get the list of turns for the current chat.
|
||||
|
||||
Returns:
|
||||
The list of turns.
|
||||
"""
|
||||
return (
|
||||
self._chats[self.current_chat] if self.current_chat in self._chats else []
|
||||
)
|
||||
|
||||
@reflex.event
|
||||
def delete_chat(self, chat_name: str):
|
||||
"""Delete the current chat."""
|
||||
if chat_name not in self._chats:
|
||||
return
|
||||
del self._chats[chat_name]
|
||||
if len(self._chats) == 0:
|
||||
self._chats = {
|
||||
"Intros": [],
|
||||
}
|
||||
if self.current_chat not in self._chats:
|
||||
self.current_chat = list(self._chats.keys())[0]
|
||||
|
||||
@reflex.event
|
||||
def set_chat(self, chat_name: str):
|
||||
"""Set the name of the current chat.
|
||||
|
||||
Args:
|
||||
chat_name: The name of the chat.
|
||||
"""
|
||||
self.current_chat = chat_name
|
||||
|
||||
@reflex.event
|
||||
def set_new_chat_name(self, new_chat_name: str):
|
||||
"""Set the name of the new chat.
|
||||
|
||||
Args:
|
||||
new_chat_name: The name of the new chat.
|
||||
"""
|
||||
self.new_chat_name = new_chat_name
|
||||
|
||||
@reflex.var
|
||||
def chat_titles(self) -> list[str]:
|
||||
"""Get the list of chat titles.
|
||||
|
||||
Returns:
|
||||
The list of chat names.
|
||||
"""
|
||||
return list(self._chats.keys())
|
||||
|
||||
@reflex.event
|
||||
async def process_question(self, form_data: dict[str, Any]):
|
||||
# Get the question from the form
|
||||
question = form_data["question"]
|
||||
|
||||
# Check if the question is empty
|
||||
if not question:
|
||||
return
|
||||
|
||||
async for value in self.openai_process_question(question):
|
||||
yield value
|
||||
|
||||
@reflex.event
|
||||
async def openai_process_question(self, question: str):
|
||||
"""Get the response from the API.
|
||||
|
||||
Args:
|
||||
form_data: A dict with the current question.
|
||||
"""
|
||||
|
||||
# Add the question to the list of questions.
|
||||
qa = TurnMessages(input_message=question, output_message="")
|
||||
self._chats[self.current_chat].append(qa)
|
||||
|
||||
# Clear the input and start the processing.
|
||||
self.processing = True
|
||||
yield
|
||||
|
||||
# Build the messages.
|
||||
messages: list[ChatCompletionMessageParam] = [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a friendly chatbot named Reflex. Respond in markdown.",
|
||||
}
|
||||
]
|
||||
for qa in self._chats[self.current_chat]:
|
||||
messages.append({"role": "user", "content": qa["input_message"]})
|
||||
messages.append({"role": "assistant", "content": qa["output_message"]})
|
||||
|
||||
# Remove the last mock answer.
|
||||
messages = messages[:-1]
|
||||
|
||||
# Start a new session to answer the question.
|
||||
session = OpenAI().chat.completions.create(
|
||||
model=os.getenv("OPENAI_MODEL", "gpt-3.5-turbo"),
|
||||
messages=messages,
|
||||
stream=True,
|
||||
)
|
||||
|
||||
# Stream the results, yielding after every word.
|
||||
for item in session:
|
||||
if hasattr(item.choices[0].delta, "content"):
|
||||
answer_text = item.choices[0].delta.content
|
||||
# Ensure answer_text is not None before concatenation
|
||||
if answer_text is not None:
|
||||
self._chats[self.current_chat][-1]["answer"] += answer_text
|
||||
else:
|
||||
# Handle the case where answer_text is None, perhaps log it or assign a default value
|
||||
# For example, assigning an empty string if answer_text is None
|
||||
answer_text = ""
|
||||
self._chats[self.current_chat][-1]["answer"] += answer_text
|
||||
self._chats = self._chats
|
||||
yield
|
||||
|
||||
# Toggle the processing flag.
|
||||
self.processing = False
|
||||
|
|
@ -1,5 +1,5 @@
|
|||
import reflex as rx
|
||||
|
||||
config = rx.Config(
|
||||
app_name="ai_agent",
|
||||
app_name="application",
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue