From 3f1118b7fce8df99257531f8717f2a0fc7760415 Mon Sep 17 00:00:00 2001 From: HeDaas Date: Fri, 17 Jul 2026 00:56:20 +0800 Subject: [PATCH 01/10] =?UTF-8?q?feat:=20release=20v3.1.0=20=E2=80=94=20se?= =?UTF-8?q?ssion=20persistence,=20OpenAI-compatible=20LLM=20providers,=20a?= =?UTF-8?q?nd=20Web=20GUI?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增聊天会话持久化（chat_sessions / chat_messages + ChatSessionRepository） - 新增多供应商 LLM 抽象层（llm_providers / model_config），兼容 OpenAI/DeepSeek/SiliconFlow/Moonshot/Zhipu/Custom - 新增 FastAPI Web 后端：REST API、WebSocket、Services、Schemas 完整链路 - 新增 React 前端：Chat/Settings/Knowledge/Schedule/Event/NPS/Database/Creative/Debug 页面 - 修复 WebSocket 聊天重复初始化 LLM 的性能问题（ChatService 全局单例复用 + session_id 隔离） - 补充 E2E/性能/回滚测试、文档、启动脚本和示例环境变量 --- .env.example | 194 + CHANGELOG.md | 305 +- README.md | 236 +- docs/api.md | 183 + docs/architecture.md | 439 ++ docs/rollback-procedure.md | 202 + docs/setup-local-dev.md | 369 ++ example.env | 35 +- main.py | 220 +- prompts/system/daily_diary.md | 23 + prompts/system/dream_generation.md | 21 + prompts/system/emotion_analysis.md | 23 + prompts/system/life_state.md | 26 + prompts/task/creative_writing.md | 89 + requirements-web.txt | 8 + run_web.py | 314 ++ setup-dev.sh | 160 + src/core/__init__.py | 9 +- src/core/background_scheduler.py | 285 ++ src/core/chat_agent.py | 307 +- src/core/chat_session_repository.py | 305 ++ src/core/conversation_graph.py | 330 -- src/core/creative_writer.py | 480 ++ src/core/database_manager.py | 968 ++++- src/core/deepagents_wrapper.py | 211 - src/core/dream_diary.py | 265 ++ src/core/emotion_analyzer.py | 269 ++ src/core/enhanced_knowledge_base.py | 443 -- src/core/event_manager.py | 171 +- src/core/langchain_llm.py | 2 +- src/core/life_state.py | 384 ++ src/core/llm_providers.py | 214 + src/core/long_term_memory.py | 279 +- src/core/model_config.py | 77 +- src/core/proactive_engine.py | 403 ++ src/core/schedule_generator.py | 9 +- src/core/schedule_similarity_checker.py | 9 +- src/core/self_timeline.py | 247 ++ src/core/user_habits.py | 170 + src/nps/nps_invoker.py | 9 +- src/tools/agent_vision.py | 11 +- src/tools/debug_logger.py | 209 +- src/tools/expression_style.py | 9 +- src/tools/schedule_intent_tool.py | 9 +- src/tools/settings_migration.py | 3 +- src/web/backend/__init__.py | 0 src/web/backend/api/__init__.py | 91 + src/web/backend/api/chat_sessions.py | 292 ++ src/web/backend/api/creative.py | 617 +++ src/web/backend/api/database.py | 245 ++ src/web/backend/api/debug.py | 289 ++ src/web/backend/api/emotion.py | 168 + src/web/backend/api/event.py | 544 +++ src/web/backend/api/knowledge.py | 282 ++ src/web/backend/api/llm.py | 191 + src/web/backend/api/memory.py | 483 +++ src/web/backend/api/nps.py | 395 ++ src/web/backend/api/schedule.py | 339 ++ src/web/backend/main.py | 289 ++ src/web/backend/schemas/__init__.py | 90 + src/web/backend/schemas/chat.py | 58 + src/web/backend/schemas/creative.py | 302 ++ src/web/backend/schemas/database.py | 69 + src/web/backend/schemas/debug.py | 46 + src/web/backend/schemas/emotion.py | 94 + src/web/backend/schemas/event.py | 74 + src/web/backend/schemas/knowledge.py | 116 + src/web/backend/schemas/memory.py | 109 + src/web/backend/schemas/schedule.py | 73 + src/web/backend/services/__init__.py | 30 + src/web/backend/services/chat_service.py | 543 +++ src/web/backend/services/creative_service.py | 309 ++ src/web/backend/services/database_service.py | 904 ++++ src/web/backend/services/emotion_service.py | 475 ++ src/web/backend/services/event_service.py | 353 ++ src/web/backend/ws/__init__.py | 100 + src/web/backend/ws/chat.py | 347 ++ src/web/backend/ws/debug.py | 190 + src/web/backend/ws/debug_broadcaster.py | 140 + src/web/backend/ws/event.py | 59 + src/web/backend/ws/events.py | 149 + src/web/backend/ws/manager.py | 240 + src/web/backend/ws/proactive.py | 235 + src/web/frontend/.gitignore | 6 + src/web/frontend/index.html | 13 + src/web/frontend/package-lock.json | 3860 +++++++++++++++++ src/web/frontend/package.json | 35 + src/web/frontend/postcss.config.js | 6 + src/web/frontend/src/App.tsx | 31 + src/web/frontend/src/api/chatSessions.ts | 156 + src/web/frontend/src/api/creative.ts | 235 + src/web/frontend/src/api/database.ts | 137 + src/web/frontend/src/api/knowledge.ts | 175 + src/web/frontend/src/api/llm.ts | 84 + src/web/frontend/src/api/nps.ts | 229 + src/web/frontend/src/api/schedule.ts | 243 ++ .../frontend/src/components/ChatBubble.tsx | 129 + src/web/frontend/src/components/ChatInput.tsx | 91 + src/web/frontend/src/components/DataTable.tsx | 171 + .../src/components/Debug/LogExporter.tsx | 111 + .../src/components/Debug/LogFilterBar.tsx | 89 + .../frontend/src/components/Debug/types.ts | 16 + .../frontend/src/components/EmotionPanel.tsx | 431 ++ .../frontend/src/components/EventTable.tsx | 173 + .../src/components/KnowledgeGraph.tsx | 171 + src/web/frontend/src/components/Layout.tsx | 126 + .../frontend/src/components/MessageList.tsx | 147 + .../src/components/NPSConfigDrawer.tsx | 242 ++ .../src/components/ProactiveBanner.tsx | 172 + .../src/components/ProactiveMessageList.tsx | 91 + .../frontend/src/components/ScheduleForm.tsx | 222 + .../src/components/SessionSidebar.tsx | 260 ++ .../src/components/StoryBibleViewer.tsx | 291 ++ .../src/components/TimelineCanvas.tsx | 647 +++ src/web/frontend/src/hooks/useChatStream.ts | 553 +++ .../frontend/src/hooks/useProactiveStream.ts | 240 + src/web/frontend/src/hooks/useWebSocket.ts | 325 ++ src/web/frontend/src/index.css | 12 + src/web/frontend/src/main.tsx | 27 + src/web/frontend/src/mock/creative.ts | 241 + src/web/frontend/src/mock/database.ts | 280 ++ src/web/frontend/src/mock/events.ts | 129 + src/web/frontend/src/mock/knowledge.ts | 194 + src/web/frontend/src/mock/nps.ts | 90 + src/web/frontend/src/mock/schedule.ts | 141 + src/web/frontend/src/pages/Chat/index.tsx | 592 +++ src/web/frontend/src/pages/Creative/index.tsx | 541 +++ src/web/frontend/src/pages/Database/index.tsx | 353 ++ src/web/frontend/src/pages/Debug/index.tsx | 292 ++ src/web/frontend/src/pages/Event/index.tsx | 461 ++ .../frontend/src/pages/Knowledge/index.tsx | 456 ++ src/web/frontend/src/pages/NPS/index.tsx | 448 ++ src/web/frontend/src/pages/Schedule/index.tsx | 588 +++ src/web/frontend/src/pages/Settings/index.tsx | 255 ++ src/web/frontend/src/types/creative.ts | 71 + src/web/frontend/src/types/database.ts | 29 + src/web/frontend/src/types/event.ts | 14 + src/web/frontend/src/types/knowledge.ts | 74 + src/web/frontend/src/types/nps.ts | 28 + src/web/frontend/src/utils/proactiveSound.ts | 150 + src/web/frontend/src/vite-env.d.ts | 1 + src/web/frontend/tailwind.config.js | 15 + src/web/frontend/tsconfig.json | 29 + src/web/frontend/tsconfig.node.json | 11 + src/web/frontend/vite.config.ts | 33 + src/web/static/assets/index-DvtvOq1y.js | 543 +++ src/web/static/assets/index-DvtvOq1y.js.map | 1 + src/web/static/assets/index-gFiY_bwi.css | 1 + src/web/static/index.html | 14 + start.sh | 93 + tests/e2e/__init__.py | 11 + tests/e2e/test_chat_flow.py | 223 + tests/e2e/test_knowledge_crud.py | 260 ++ tests/e2e/test_proactive_push.py | 378 ++ tests/e2e/test_websocket_reconnect.py | 280 ++ tests/perf/README.md | 179 + tests/perf/test_load_bench.py | 513 +++ tests/rollback/__init__.py | 17 + tests/rollback/test_db_consistency.py | 227 + tests/rollback/test_rollback_procedure.py | 364 ++ tests/test_chat_stream.py | 226 + tests/test_deepagents_integration.py | 83 +- tests/test_enhanced_knowledge_base.py | 248 -- tests/test_exploratory.py | 131 + tests/test_p1_life_dream.py | 147 + tests/test_p2_emotion_loop.py | 267 ++ tests/test_p3_creative.py | 194 + tests/test_p4_proactive.py | 134 + 168 files changed, 36839 insertions(+), 1422 deletions(-) create mode 100644 .env.example create mode 100644 docs/api.md create mode 100644 docs/architecture.md create mode 100644 docs/rollback-procedure.md create mode 100644 docs/setup-local-dev.md create mode 100644 prompts/system/daily_diary.md create mode 100644 prompts/system/dream_generation.md create mode 100644 prompts/system/life_state.md create mode 100644 prompts/task/creative_writing.md create mode 100644 requirements-web.txt create mode 100644 run_web.py create mode 100755 setup-dev.sh create mode 100644 src/core/background_scheduler.py create mode 100644 src/core/chat_session_repository.py delete mode 100644 src/core/conversation_graph.py create mode 100644 src/core/creative_writer.py create mode 100644 src/core/dream_diary.py delete mode 100644 src/core/enhanced_knowledge_base.py create mode 100644 src/core/life_state.py create mode 100644 src/core/llm_providers.py create mode 100644 src/core/proactive_engine.py create mode 100644 src/core/self_timeline.py create mode 100644 src/core/user_habits.py create mode 100644 src/web/backend/__init__.py create mode 100644 src/web/backend/api/__init__.py create mode 100644 src/web/backend/api/chat_sessions.py create mode 100644 src/web/backend/api/creative.py create mode 100644 src/web/backend/api/database.py create mode 100644 src/web/backend/api/debug.py create mode 100644 src/web/backend/api/emotion.py create mode 100644 src/web/backend/api/event.py create mode 100644 src/web/backend/api/knowledge.py create mode 100644 src/web/backend/api/llm.py create mode 100644 src/web/backend/api/memory.py create mode 100644 src/web/backend/api/nps.py create mode 100644 src/web/backend/api/schedule.py create mode 100644 src/web/backend/main.py create mode 100644 src/web/backend/schemas/__init__.py create mode 100644 src/web/backend/schemas/chat.py create mode 100644 src/web/backend/schemas/creative.py create mode 100644 src/web/backend/schemas/database.py create mode 100644 src/web/backend/schemas/debug.py create mode 100644 src/web/backend/schemas/emotion.py create mode 100644 src/web/backend/schemas/event.py create mode 100644 src/web/backend/schemas/knowledge.py create mode 100644 src/web/backend/schemas/memory.py create mode 100644 src/web/backend/schemas/schedule.py create mode 100644 src/web/backend/services/__init__.py create mode 100644 src/web/backend/services/chat_service.py create mode 100644 src/web/backend/services/creative_service.py create mode 100644 src/web/backend/services/database_service.py create mode 100644 src/web/backend/services/emotion_service.py create mode 100644 src/web/backend/services/event_service.py create mode 100644 src/web/backend/ws/__init__.py create mode 100644 src/web/backend/ws/chat.py create mode 100644 src/web/backend/ws/debug.py create mode 100644 src/web/backend/ws/debug_broadcaster.py create mode 100644 src/web/backend/ws/event.py create mode 100644 src/web/backend/ws/events.py create mode 100644 src/web/backend/ws/manager.py create mode 100644 src/web/backend/ws/proactive.py create mode 100644 src/web/frontend/.gitignore create mode 100644 src/web/frontend/index.html create mode 100644 src/web/frontend/package-lock.json create mode 100644 src/web/frontend/package.json create mode 100644 src/web/frontend/postcss.config.js create mode 100644 src/web/frontend/src/App.tsx create mode 100644 src/web/frontend/src/api/chatSessions.ts create mode 100644 src/web/frontend/src/api/creative.ts create mode 100644 src/web/frontend/src/api/database.ts create mode 100644 src/web/frontend/src/api/knowledge.ts create mode 100644 src/web/frontend/src/api/llm.ts create mode 100644 src/web/frontend/src/api/nps.ts create mode 100644 src/web/frontend/src/api/schedule.ts create mode 100644 src/web/frontend/src/components/ChatBubble.tsx create mode 100644 src/web/frontend/src/components/ChatInput.tsx create mode 100644 src/web/frontend/src/components/DataTable.tsx create mode 100644 src/web/frontend/src/components/Debug/LogExporter.tsx create mode 100644 src/web/frontend/src/components/Debug/LogFilterBar.tsx create mode 100644 src/web/frontend/src/components/Debug/types.ts create mode 100644 src/web/frontend/src/components/EmotionPanel.tsx create mode 100644 src/web/frontend/src/components/EventTable.tsx create mode 100644 src/web/frontend/src/components/KnowledgeGraph.tsx create mode 100644 src/web/frontend/src/components/Layout.tsx create mode 100644 src/web/frontend/src/components/MessageList.tsx create mode 100644 src/web/frontend/src/components/NPSConfigDrawer.tsx create mode 100644 src/web/frontend/src/components/ProactiveBanner.tsx create mode 100644 src/web/frontend/src/components/ProactiveMessageList.tsx create mode 100644 src/web/frontend/src/components/ScheduleForm.tsx create mode 100644 src/web/frontend/src/components/SessionSidebar.tsx create mode 100644 src/web/frontend/src/components/StoryBibleViewer.tsx create mode 100644 src/web/frontend/src/components/TimelineCanvas.tsx create mode 100644 src/web/frontend/src/hooks/useChatStream.ts create mode 100644 src/web/frontend/src/hooks/useProactiveStream.ts create mode 100644 src/web/frontend/src/hooks/useWebSocket.ts create mode 100644 src/web/frontend/src/index.css create mode 100644 src/web/frontend/src/main.tsx create mode 100644 src/web/frontend/src/mock/creative.ts create mode 100644 src/web/frontend/src/mock/database.ts create mode 100644 src/web/frontend/src/mock/events.ts create mode 100644 src/web/frontend/src/mock/knowledge.ts create mode 100644 src/web/frontend/src/mock/nps.ts create mode 100644 src/web/frontend/src/mock/schedule.ts create mode 100644 src/web/frontend/src/pages/Chat/index.tsx create mode 100644 src/web/frontend/src/pages/Creative/index.tsx create mode 100644 src/web/frontend/src/pages/Database/index.tsx create mode 100644 src/web/frontend/src/pages/Debug/index.tsx create mode 100644 src/web/frontend/src/pages/Event/index.tsx create mode 100644 src/web/frontend/src/pages/Knowledge/index.tsx create mode 100644 src/web/frontend/src/pages/NPS/index.tsx create mode 100644 src/web/frontend/src/pages/Schedule/index.tsx create mode 100644 src/web/frontend/src/pages/Settings/index.tsx create mode 100644 src/web/frontend/src/types/creative.ts create mode 100644 src/web/frontend/src/types/database.ts create mode 100644 src/web/frontend/src/types/event.ts create mode 100644 src/web/frontend/src/types/knowledge.ts create mode 100644 src/web/frontend/src/types/nps.ts create mode 100644 src/web/frontend/src/utils/proactiveSound.ts create mode 100644 src/web/frontend/src/vite-env.d.ts create mode 100644 src/web/frontend/tailwind.config.js create mode 100644 src/web/frontend/tsconfig.json create mode 100644 src/web/frontend/tsconfig.node.json create mode 100644 src/web/frontend/vite.config.ts create mode 100644 src/web/static/assets/index-DvtvOq1y.js create mode 100644 src/web/static/assets/index-DvtvOq1y.js.map create mode 100644 src/web/static/assets/index-gFiY_bwi.css create mode 100644 src/web/static/index.html create mode 100755 start.sh create mode 100644 tests/e2e/__init__.py create mode 100644 tests/e2e/test_chat_flow.py create mode 100644 tests/e2e/test_knowledge_crud.py create mode 100644 tests/e2e/test_proactive_push.py create mode 100644 tests/e2e/test_websocket_reconnect.py create mode 100644 tests/perf/README.md create mode 100644 tests/perf/test_load_bench.py create mode 100644 tests/rollback/__init__.py create mode 100644 tests/rollback/test_db_consistency.py create mode 100644 tests/rollback/test_rollback_procedure.py create mode 100644 tests/test_chat_stream.py delete mode 100644 tests/test_enhanced_knowledge_base.py create mode 100644 tests/test_exploratory.py create mode 100644 tests/test_p1_life_dream.py create mode 100644 tests/test_p2_emotion_loop.py create mode 100644 tests/test_p3_creative.py create mode 100644 tests/test_p4_proactive.py diff --git a/.env.example b/.env.example new file mode 100644 index 0000000..0f3b3ef --- /dev/null +++ b/.env.example @@ -0,0 +1,194 @@ +# ===================================================================== +# Neo Agent v3.1.0 — Local Development .env.example +# ===================================================================== +# 用法： +# cp .env.example .env +# vim .env # 填入实际 API 密钥与个性化参数 +# +# 注意：.env 已在 .gitignore 中，不会被提交。 +# ===================================================================== + +# --------------------------------------------------------------------- +# 一、LLM 通用配置（v3.1.0+ 推荐使用） +# --------------------------------------------------------------------- +# 供应商预设：openai / deepseek / siliconflow / moonshot / zhipu / custom +LLM_PROVIDER=siliconflow +# API Key（推荐用 LLM_API_KEY，兼容旧 SILICONFLOW_API_KEY） +LLM_API_KEY= +# 自定义 base_url（preset 模式可省略，custom 模式必填） +# LLM_BASE_URL=https://api.openai.com/v1 +# 自定义模型名（preset 模式可省略） +# LLM_MODEL_NAME=gpt-4o-mini +# 全局超时/温度/最大 tokens +LLM_TIMEOUT=60 +LLM_TEMPERATURE=0.7 +LLM_MAX_TOKENS=2000 +# 三层模型独立名称（MAIN/TOOL/VISION） +# MAIN_MODEL_NAME=deepseek-ai/DeepSeek-V3.2 +# TOOL_MODEL_NAME=zai-org/GLM-4.6V +# VISION_MODEL_NAME=Qwen/Qwen3-VL-32B-Instruct + +# --------------------------------------------------------------------- +# 一·A、兼容旧版 LLM 变量（v3.0.x 风格，推荐迁移到上方 LLM 通用配置） +# --------------------------------------------------------------------- +# （兼容旧版，推荐迁移到上方 LLM 通用配置） +# SiliconFlow API 密钥 +# 申请：https://cloud.siliconflow.cn/account/ak +SILICONFLOW_API_KEY=your-siliconflow-api-key +SILICONFLOW_API_URL=https://api.siliconflow.cn/v1/chat/completions + +# SerpAPI 密钥（用于网络搜索工具） +# 申请：https://serpapi.com/manage-api-key +SERPAPI_API_KEY=your-serpapi-key + +# --------------------------------------------------------------------- +# 二、多层模型架构 +# --------------------------------------------------------------------- + +# 主模型：处理主要对话和复杂推理任务 +MAIN_MODEL_NAME=deepseek-ai/DeepSeek-V3.2 +MAIN_MODEL_TEMPERATURE=0.8 +MAIN_MODEL_MAX_TOKENS=2000 +# MAIN_MODEL_TIMEOUT=60 # API请求超时时间（秒），默认60秒 + +# 小模型：处理工具调用、意图识别等轻量级任务 +TOOL_MODEL_NAME=zai-org/GLM-4.6V +TOOL_MODEL_TEMPERATURE=0.3 +TOOL_MODEL_MAX_TOKENS=500 +# TOOL_MODEL_TIMEOUT=45 # API请求超时时间（秒），默认45秒 + +# 多模态模型：处理视觉识别和多模态推理 +VISION_MODEL_NAME=Qwen/Qwen3-VL-32B-Instruct +VISION_MODEL_TEMPERATURE=0.5 +VISION_MODEL_MAX_TOKENS=1000 +# VISION_MODEL_TIMEOUT=90 # API请求超时时间（秒），默认90秒 + +# 兼容旧配置（如果未设置上述新配置，将使用此默认值） +MODEL_NAME=deepseek-ai/DeepSeek-V3 +TEMPERATURE=0.8 +MAX_TOKENS=2000 + +# 视觉工具LLM判断配置（可选） +# VISION_LLM_TEMPERATURE=0.3 +# VISION_LLM_MAX_TOKENS=10 +# VISION_LLM_TIMEOUT=10 + +# --------------------------------------------------------------------- +# 三、提示词配置（使用 markdown 模板系统） +# --------------------------------------------------------------------- + +# 角色提示词文件名（不含 .md 后缀），默认使用 default_character +# CHARACTER_PROMPT_FILE=default_character +# 世界观文件名（不含 .md 后缀），默认使用 default_world +# WORLDVIEW_FILE=default_world + +# --------------------------------------------------------------------- +# 四、多智能体协作配置 +# --------------------------------------------------------------------- + +# 是否启用基于 LangGraph 的动态协作图（默认 True） +# USE_DYNAMIC_MULTI_AGENT=True + +# --------------------------------------------------------------------- +# 五、角色 / 人物设定 +# --------------------------------------------------------------------- + +CHARACTER_NAME=小可 +CHARACTER_GENDER=女 +CHARACTER_ROLE=学生 +CHARACTER_HEIGHT=150cm +CHARACTER_WEIGHT=45kg +CHARACTER_PERSONALITY=活泼开朗 +CHARACTER_HOBBY=文科，尤其对历史充满热情 +CHARACTER_AGE=18 + +CHARACTER_BACKGROUND=我是一名高中生，叫小可。我身高150cm，体重45kg。我性格活泼开朗，特别喜欢文科，尤其对历史充满热情。我喜欢阅读历史书籍，常常能从历史故事中获得启发。我说话比较俏皮可爱，喜欢和朋友们分享有趣的历史知识。 + +# --------------------------------------------------------------------- +# 六、记忆 / 持久化 +# --------------------------------------------------------------------- + +MEMORY_FILE=memory_data.json +LONG_MEMORY_FILE=longmemory_data.json +MAX_MEMORY_MESSAGES=50 +MAX_SHORT_TERM_ROUNDS=20 + +# SQLite 共享数据库（Web GUI 与 Tkinter 共用） +# 默认 chat_agent.db 在项目根目录 +# CHAT_AGENT_DB=/abs/path/to/chat_agent.db + +# --------------------------------------------------------------------- +# 七、调试 / 日志 +# --------------------------------------------------------------------- + +DEBUG_MODE=True +DEBUG_LOG_FILE=debug.log + +# Web 后端日志级别（DEBUG/INFO/WARNING/ERROR） +LOG_LEVEL=INFO + +# --------------------------------------------------------------------- +# 八、NPS 工具系统 LLM 配置（可选） +# --------------------------------------------------------------------- + +# NPS_LLM_TEMPERATURE=0.3 +# NPS_LLM_MAX_TOKENS=100 +# NPS_LLM_TIMEOUT=10 + +# --------------------------------------------------------------------- +# 九、DeepAgents 增强配置 +# --------------------------------------------------------------------- + +# 是否使用 DeepAgents 增强的子智能体（默认 True） +# 读取位置：src/core/multi_agent_coordinator.py +# USE_DEEP_AGENTS=True + +# 是否启用持久化状态管理（默认 True） +# 读取位置：src/core/dynamic_multi_agent_graph.py +# ENABLE_PERSISTENT_STATE=True +# 注：USE_DEEPAGENTS_KNOWLEDGE 已弃用（v2.2.0 删除 DeepAgentsKnowledgeManager） + +# --------------------------------------------------------------------- +# 十、Feature Flags（功能开关，按需启用） +# --------------------------------------------------------------------- + +# 启用生活状态管理（life_state 模块） +# ENABLE_LIFE_STATE=true + +# 启用主动消息引擎 +# ENABLE_PROACTIVE_ENGINE=true + +# 启用情感雷达（Plutchik 八情绪） +# ENABLE_EMOTION_WHEEL=true + +# 启用创意写作 / 长篇项目 +# ENABLE_CREATIVE_WRITER=true + +# --------------------------------------------------------------------- +# 十一、Web GUI 配置（v3.0.0 新增） +# --------------------------------------------------------------------- + +# Web GUI 总开关（false 时 `python main.py --web` 拒绝启动并提示 --tk） +# 用于灰度回滚 / CI 烟囱测试 / 临时关闭 Web 端 +# ENABLE_WEB_GUI=true + +# 启用前端 Vite dev server（与 FastAPI 8000 端口并行启动 :5173） +# 生产部署设为 0；本地开发设为 1 +# ENABLE_FRONTEND_DEV=1 + +# FastAPI 监听地址（生产建议 127.0.0.1，由 nginx 反代；本地开发 0.0.0.0） +# WEB_HOST=0.0.0.0 + +# FastAPI 监听端口 +# WEB_PORT=8000 + +# CORS 允许来源（开发态默认 localhost:5173 + localhost:8000） +# 多个用逗号分隔；生产部署改为实际反代域名 +# CORS_ORIGINS=http://localhost:5173,http://localhost:8000,http://127.0.0.1:5173,http://127.0.0.1:8000 + +# --------------------------------------------------------------------- +# 十二、彩蛋（可选） +# --------------------------------------------------------------------- + +# 关闭 ANSI 颜色（CI / 非 TTY 环境） +# NO_COLOR=1 diff --git a/CHANGELOG.md b/CHANGELOG.md index 95f8cd5..decdf28 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,18 +2,313 @@ All notable changes to this project will be documented in this file. -本文件记录项目的所有重要变更。 +本文件记录项目的所有重要变更。格式遵循 [Keep a Changelog](https://keepachangelog.com/zh-CN/1.1.0/) 规范。 + +## v3.1.0 (2026-XX-XX) + +### ✨ 新增 +- **会话持久化**：聊天会话与消息存储到数据库（`chat_sessions` / `chat_messages`），支持多会话切换、刷新页面自动恢复、删除级联 +- **前端 SessionSidebar**：左侧抽屉列出会话，支持新建/重命名/删除/切换 +- **多供应商 LLM 抽象**：内置 6 个 OpenAI 兼容 preset（openai / deepseek / siliconflow / moonshot / zhipu / custom），通过 `LLM_PROVIDER` / `LLM_API_KEY` / `LLM_BASE_URL` / `LLM_MODEL_NAME` 配置 +- **LLM 配置 API**：`GET /api/llm/config` + `POST /api/llm/test` +- **Settings 页面**：显示当前 LLM 配置 + 一键测试连接 + +### 🔧 修改 +- 11 个文件中 19 处 `SILICONFLOW_API_KEY` / `SILICONFLOW_API_URL` 调用统一改为 `llm_providers.resolve_*()` 接口 +- 旧 `SILICONFLOW_API_KEY` / `SILICONFLOW_API_URL` / `MODEL_NAME` 仍可作为 fallback，**完全向后兼容** +- `model_config.py` 改用供应商注册表 +- `useChatStream` 增 `sessionId` / `switchSession` / `ack` 帧处理 +- `ChatService` 增持久化钩子（流结束后双写 user+assistant） + +### 🗃️ 数据库 +- 新增表 `chat_sessions` / `chat_messages`（外键 ON DELETE CASCADE） + +### ⚠️ BREAKING +- 数据库 schema 变更：首次启动自动建表（v3.0.0 启动仅 1 天，存量可丢弃） + +## [3.0.0] - 2026-07-14 + +### Added / 新增 +- **Web GUI**（FastAPI + React + Ant Design）：替代本地 Tkinter 桌面 GUI（默认启动模式） +- **WebSocket 流式聊天**（`/ws/chat`）：基于 LangChain astream 的 token 级流式输出，降级到 `run_in_executor` 模拟流式 +- **WebSocket 主动消息推送**（`/ws/proactive`）：服务端主动向客户端推送 +- **WebSocket 事件流**（`/ws/events`）：来自 `EventManager` 的全局事件广播 +- **Debug 日志流**（`/ws/debug`）：来自 `DebugLogger` 的实时日志 +- **管理页面**：Knowledge / Schedule / Event / NPS / Database / Creative 六个管理类页签 +- **Pydantic schema 强约束 + SQLite WAL 模式**： + - `src/web/backend/schemas/*` 全部基于 Pydantic v2 + - `src/core/database_manager.py` 启用 `PRAGMA journal_mode=WAL` +- 新增文件： + - `src/web/` — 后端（FastAPI）+ 前端（Vite + React）源码 + - `requirements-web.txt` — Web 端依赖 + - `run_web.py` — Web 模式启动入口 + - `start.sh` — 一键启动脚本 + +### Changed / 变更 +- `EventManager` 新增 `subscribe` / `publish` API（`src/core/event_manager.py`） + - `subscribe(event_type, callback)` 注册进程内监听者 + - `publish(event_type_or_obj, payload)` 异步广播，listener 异常隔离 +- `BackgroundScheduler` 每个 tick 推送状态（`src/core/background_scheduler.py`） + - 每个 tick 完成后 `EventManager.publish("scheduler_tick", status_payload)` 推送给订阅者 +- `ProactiveEngine` 主动消息通过 `EventManager` 推送（`src/core/proactive_engine.py`） + - 触发主动消息时改为 `em.publish("proactive_message", payload)` +- `ChatAgent` 新增 `chat_stream()` 异步流式方法（原 `chat()` 同步方法保留，向后兼容） +- `main.py` 新增 `--web`（默认）/ `--tk` 参数 + +### Deprecated / 弃用 +- 无 + +### Removed / 移除 +- 无 + +### Fixed / 修正 +- 无 + +### Security / 安全 +- **23 表白名单防御 SQL 注入**（`src/core/database_manager.py`） + - 多个 `update_*` 方法使用白名单验证允许更新的列名（line 1964, 2100, 2301 等），过滤只允许白名单中的列名 + - 业务层禁止拼接 SQL 列名 / 表名 + +### Preserved / 保留 +- 全部 `src/core/*` 业务层逻辑**未修改**，仅扩展方法（`chat_stream` / WAL / `subscribe`+`publish`） +- 数据存储（`chat_agent.db`）**未迁移**，Tkinter 端与 Web 端共享 +- 原 Tkinter GUI 通过 `python main.py --tk` 仍可使用 + +### Documentation / 文档 +- 新增 [`docs/architecture.md`](docs/architecture.md):整体架构图、ASCII 数据流、模块依赖、并行调度 +- 新增 [`docs/rollback-procedure.md`](docs/rollback-procedure.md):5 分钟回滚到 Tkinter 步骤 + 5 条 FAQ +- 更新 [`README.md`](README.md):增加 Web GUI 启动方式、访问 URL、`/docs`、Tkinter 回滚说明 +- 新增 [`.trae/specs/migrate-tkinter-to-web/verification-report.md`](../.trae/specs/migrate-tkinter-to-web/verification-report.md):5 阶段总体验收报告(326 行,含 P0/P1/P2 清单) + +### 验收状态 / Verification Status (2026-07-14) +- **验收方法**:5 个并行子代理 + 静态代码分析 + 测试桩验证 +- **整体判定**:**NO-GO(条件式)** — 后端可发,前端 6 页面 mock→真接口未完成 +- **数据汇总**:✅ 69 / ⚠️ 21 / ❌ 5(73%) +- **跨阶段核心约束全部达成**:数据零迁移、业务层零改动、Feature flag 兼容、依赖最小化、回滚 ≤ 5 分钟、CHANGELOG 齐全 + +#### 本次完成(具体改动) + +**阶段 A — 基础设施(18/20 ✅)** +- 新建 `src/web/backend/main.py`:FastAPI 应用,CORS / 路由挂载 / `/api/health` / `/docs` +- 新建 `src/web/backend/ws/manager.py`:连接注册/注销/广播,清理机制在 `finally` 块 +- 新建 `src/web/backend/ws/{chat,events,proactive,debug}.py`:4 个 WebSocket 端点 +- 新建 `src/web/backend/services/{chat,database,event}_service.py`:三层包装,ChatAgent 懒加载,`ALLOWED_TABLES` 23 表白名单 +- 新建 `src/web/backend/schemas/*.py`:Pydantic v2 DTO(ChatRequest/ChatResponse/EventDTO/EmotionResponse/TimelineResponse/KnowledgeEntity/ScheduleDTO 等) +- 新建 `src/web/frontend/`:Vite + React 18 + TypeScript + Ant Design 5 + Tailwind 3 + zustand + ECharts 项目骨架 +- 新建 `src/web/frontend/src/components/Layout.tsx`:Sider + Header + Content 8 菜单 +- 新建 `requirements-web.txt`:5 行独立依赖(fastapi/uvicorn/websockets/python-multipart/pydantic) +- 新建 `run_web.py`:并发启动 FastAPI + 前端 dev server(受 `ENABLE_FRONTEND_DEV=1` 控制) +- 改造 `main.py`:新增 `--web`(默认)/ `--tk` 互斥参数,`_run_web` / `_run_tkinter` 双分支 +- 新建 `start.sh`:Python 3.10+ 检查 + venv 创建 + 依赖安装 + 一键启动 + +**阶段 B — 核心页面(13/17 ✅)** +- 新建 `src/core/chat_agent.py:1049-1177` `chat_stream()`:AsyncIterator,先 LangChain astream,降级到 `run_in_executor(chat)` + 40ms token 切片,原 `chat()` 同步方法保留 +- 新建 `src/web/frontend/src/hooks/useWebSocket.ts`:指数退避 `1→2→4→8→16→30s`,`computeBackoff = min(reconnectInterval * 2**attempt, maxReconnectInterval)` +- 新建 `src/web/frontend/src/hooks/useChatStream.ts`:`chunk` 流式追加 + `emotion_update` 写入 `window.__lastEmotion` +- 新建 `src/web/frontend/src/hooks/useProactiveStream.ts`:订阅 `/ws/proactive` 维护消息队列 +- 新建 `src/web/frontend/src/components/ChatBubble.tsx`:user/assistant/system 三种样式 +- 新建 `src/web/frontend/src/components/MessageList.tsx` / `ChatInput.tsx`:Enter 发、Shift+Enter 换行 +- 新建 `src/web/frontend/src/components/EmotionPanel.tsx`:ECharts 雷达图,Plutchik 8 维 +- 新建 `src/web/frontend/src/components/TimelineCanvas.tsx`:ECharts 散点图,日期 × 分类 +- 新建 `src/web/frontend/src/pages/Chat/index.tsx`:70/30 双栏,ProactiveMessageList + MessageList + ChatInput + EmotionPanel + TimelineCanvas +- 新建 `src/web/frontend/src/pages/Debug/index.tsx` + `components/Debug/{LogFilterBar,LogExporter}.tsx`:实时日志 + 模块/级别/时间筛选 + TXT/JSON 导出 +- 新建 `src/web/backend/api/emotion.py` `GET /api/emotion/latest`:返回 `cumulative` + `plutchik` +- 新建 `src/web/backend/api/memory.py` `GET /api/memory/timeline?days=7` + `GET /api/memory/loop/{uuid}/context` +- 新建 `src/web/frontend/src/utils/proactiveSound.ts`:Web Audio API 合成 440Hz 提示音,localStorage 持久化开关 +- 新建 `tests/test_chat_stream.py`:覆盖 normal / exception / cancel 三种场景 + +**阶段 C — 管理页面(后端 4/6,前端 5/6 部分通过)** +- 新建 `src/web/backend/api/knowledge.py`:`GET /api/knowledge/search?q=` + `POST /api/knowledge` + `DELETE /api/knowledge/{uuid}` +- 新建 `src/web/backend/api/schedule.py`:`GET /api/schedule?date=` + `POST/PUT/DELETE /api/schedule/{id}` + `POST /api/schedule/{id}/confirm`(YAML 校验 400) +- 新建 `src/web/backend/api/database.py`:`GET /api/database/tables` + `GET /api/database/{table}?limit=1-1000` + `DELETE /api/database/{table}/{uuid}`,删除走 `X-Confirm: true` header +- 新建 `src/web/backend/api/creative.py`:`GET /api/creative/projects` + `GET /api/creative/projects/{uuid}` + `POST /api/creative/projects/{uuid}/advance` +- 新建 `src/web/frontend/src/pages/{Knowledge,Schedule,Event,Database,Creative,NPS}/index.tsx`:6 管理页面 UI 框架 +- 新建 `src/web/frontend/src/components/{KnowledgeGraph,ScheduleForm,EventTable,DataTable,StoryBibleViewer,NPSConfigDrawer}.tsx`:16 组件 UI 完整 +- `src/web/backend/services/database_service.py`:23 表白名单 + WAL 模式双层防御(`PRAGMA journal_mode=WAL`) + +**阶段 D — 调度整合(14/15 ✅)** +- 改造 `src/core/event_manager.py`:新增 `subscribe` / `publish` / `get_event_manager` API +- 改造 `src/core/background_scheduler.py:194`:每个 tick `em.publish('scheduler_tick', status_payload)`,payload 含 `life_state_snapshot` / `dream_record` / `open_loop_count` / `creative_project_count` +- 改造 `src/core/proactive_engine.py:329`:`bot_proactive_drive_send()` 改走 `em.publish('proactive_message', payload)` +- 改造 `src/web/backend/main.py:79-93`:FastAPI 启动时创建单例 EventService,`threading.Lock` 双重检查 +- 新建 `src/web/backend/services/event_service.py`:`register_websocket_listener` / `unregister_websocket_listener`,订阅 `scheduler_tick` / `proactive_message` / `chat_event` / `debug_log` / `message` / 通配 `*` +- 改造 `src/tools/debug_logger.py:92-113`:新增 `subscribe` / `unsubscribe` 钩子,`set_main_loop` 支持跨线程调度 +- 新建 `src/web/backend/ws/debug_broadcaster.py`:DebugLogger subscriber + EventService 'debug_log' listener 双挂,`main.py:200-219` 启动 attach,`:232-238` shutdown detach +- 改造 `src/web/backend/ws/proactive.py:55`:读取 `PROACTIVE_MESSAGE_INTERVAL` 环境变量,`_should_throttle()` 进程内节流 + +**阶段 E — 文档与回滚(11/19 ✅)** +- 新建 `tests/e2e/test_chat_flow.py`(5 方法)+ `test_knowledge_crud.py`(6)+ `test_proactive_push.py`(5)+ `test_websocket_reconnect.py`(9)= **25 个 E2E 测试**,全部 `@unittest.skipUnless(_DEPS_OK, _SKIP_REASON)` + `self.skipTest` 优雅降级 +- 新建 `tests/perf/test_load_bench.py`:首屏字节 < 500KB / 首 token < 2s / WS ping-pong < 5s / 1000 条遍历 / 10 并发 WS +- 新建 `tests/rollback/test_rollback_procedure.py`:`TestRunTkinterMode` + `TestEnableWebGuiFlag` + `TestDbConsistency` +- 更新 `README.md:158-209` GUI 启动方式 + `:394-422` Web GUI 速查 +- 新建 `docs/architecture.md`:ASCII 总体图 / WebSocket 数据流 / 模块依赖 / 并行调度 / mermaid 总览 +- 新建 `docs/rollback-procedure.md`:4 步 5 分钟回滚 + 验证清单 + 5 条 FAQ + 切回 Web +- 本 `CHANGELOG.md` v3.0.0 条目(本节) +- `src/web/backend/main.py:21-29` `app = FastAPI(title="Neo Agent Web", version="3.0.0")` → `/docs` 自动 OpenAPI + +#### P0 必修(阻塞正式发布) + +- **前端 6 页面 mock→真接口**:`pages/{Knowledge,Schedule,Event,Database,Creative,NPS}/index.tsx` `useEffect` 改调 API,删 `TODO(后端联调)` 注释;Event 页删 `setInterval(5000ms)` 改订阅 `/ws/events` +- **新建 REST 路由**: + - `src/web/backend/api/nps.py` — `GET /api/nps` + `POST /api/nps/register` + `POST /api/nps/invoke` + - `src/web/backend/api/event.py` — `GET /api/events` + `POST /api/events/{event_id}/read` + `POST /api/events/{event_id}/archive` +- **`main.py` 显式 `ENABLE_WEB_GUI` 校验**:在 `_run_web` 开头检查 `os.environ.get("ENABLE_WEB_GUI", "true").lower() == "false"` 拒绝启动 + +#### P1 强烈建议(发布前修) + +- [pages/Chat/index.tsx](../Neo_Agent/src/web/frontend/src/pages/Chat/index.tsx) 侧栏真挂载 EmotionPanel + TimelineCanvas(目前占位) +- [TimelineCanvas.tsx](../Neo_Agent/src/web/frontend/src/components/TimelineCanvas.tsx) 节点 click 接 `/api/memory/loop/{uuid}/context` +- [pages/Event/index.tsx](../Neo_Agent/src/web/frontend/src/pages/Event/index.tsx) 订阅 `scheduler_tick` 渲染"当日状态/梦境/日记/想法池"卡 +- [MessageList.tsx](../Neo_Agent/src/web/frontend/src/components/MessageList.tsx) 接 `react-window` 真虚拟滚动 +- `ws/__init__.py` 聚合或删除 `/ws/event` 单数端点 +- `ScheduleDTO` / `EventDTO` 字段名对齐实际表(`schedule_id` / `collaboration_status` / `event_id`) +- `api/creative.py` 续写路径统一(`/api/creative/advance/{uuid}` vs `/api/creative/projects/{uuid}/advance`) +- ChatInput 加 Ctrl/Cmd+Enter 快捷键 +- Layout 加 `< 768px` 移动端降级提示 + +#### P2 后续优化(发布后 1-2 周) + +- Lighthouse 真机录制首屏 `< 3s`、FPS `≥ 50` +- 浏览器矩阵:Chrome / Edge / Firefox / Safari 最新 2 版本 +- TypeScript 类型自动生成(`openapi-typescript`) +- `Settings` 页面 Web 端实现(对照 Tkinter `src/gui/settings_migration_gui.py` 530 行) +- 团队 React + AntD 培训纪要(≥1 次) +- UAT ≥ 3 个真实用户场景报告 + +## [2.3.0] - 2026-05-15 + +> **重大重构:Neo_Agent 拟人化能力与架构分层全面落地** +> +> 目标:参考 [astrbot_plugin_private_companion](https://github.com/) 拟人化设计 + 本地 [ARCHITECTURE_REFACTOR_PLAN.md](../ARCHITECTURE_REFACTOR_PLAN.md) + [INVESTIGATION_HUMANIZATION_CREATIVE.md](../INVESTIGATION_HUMANIZATION_CREATIVE.md),对 Neo_Agent 进行"分层架构 + 拟人化能力 + 数据扩展 + 工程规约"四位一体的全量重构。本次重构**不引入 Web GUI**(留待 v3.0.0),**不破坏任何既有方法**(纯扩展),**不强制启用新功能**(Feature flag 默认关闭)。 + +### 架构分层(Achitecture Layering) + +#### Added / 新增 — 纵向四层架构 +- `src/core/` — 核心业务层(LLM/记忆/知识/情感/创作/调度) +- `src/tools/` — 工具层(debug logger / 杂项 utilities) +- `src/gui/` — Tkinter 桌面 GUI 层 +- `src/nps/` — Neo Plugin System 插件层 +- 所有新模块以**独立文件**形式加入 `src/core/`,不复用 Mixin(规避 70K+ 单类膨胀),统一复用 `LLMHelper` / `PromptManager` / `DatabaseManager` + +### 拟人化能力(Humanization Capabilities) + +#### Added / 新增 — 六大核心抽象 +- **生活状态机** [`src/core/life_state.py`](src/core/life_state.py) + - 状态枚举:WAKE / WORK / REST / SLEEP / DREAM / SOCIAL / REFLECT + - 状态转移规则 + 日快照表 `life_state_daily_snapshot` + - 通过 `ENABLE_LIFE_STATE` Feature flag 控制 +- **Plutchik 8 维情绪模型** [`src/core/emotion_analyzer.py`](src/core/emotion_analyzer.py) + - 8 种基本情绪:joy / trust / fear / surprise / sadness / disgust / anger / anticipation + - 与现有"印象累加评分"并行注入 prompt + - 通过 `ENABLE_EMOTION_WHEEL` Feature flag 控制 +- **长期创作系统** [`src/core/creative_writer.py`](src/core/creative_writer.py) + - 立项 → 续写 → 完结 全生命周期 + - 质量审核:similarity ≤ 0.72、retry ≤ 2、min_score ≥ 7 + - 数据表:`creative_projects` / `creative_story_bibles` / `creative_memory_pool`(统一命名) + - 触发:聊天不活跃 30 分钟后自动开始一轮续写 +- **主动决策引擎** [`src/core/proactive_engine.py`](src/core/proactive_engine.py) + - 想法池(idea pool)+ 触发条件评估 + 主动消息投递 + - 通过 `ENABLE_PROACTIVE_ENGINE` Feature flag 控制 +- **未完话题跟踪** [`src/core/long_term_memory.py`](src/core/long_term_memory.py) + - `OpenLoopTracker` 自动识别用户未说完的话题并打 tag + - 数据表 `open_loops` 持久化(raised_at / closed_at / topic / entities) +- **梦境记录 + 自我时间线 + 用户习惯** + - [`src/core/dream_diary.py`](src/core/dream_diary.py) — 梦境生成与日记 + - [`src/core/self_timeline.py`](src/core/self_timeline.py) — 自我时间线聚合 + - [`src/core/user_habits.py`](src/core/user_habits.py) — 用户习惯画像 + +### 提示词工程(Prompt Engineering) + +#### Added / 新增 — 模块化提示词 +- [`src/core/prompt_manager.py`](src/core/prompt_manager.py) — `PromptManager` 统一管理四类模板: + - `character` — 角色人设 + - `system` — 系统级约束 + - `task` — 任务级指令 + - `worldview` — 世界观注入 +- 支持 markdown 模板 + Jinja2 占位符 + 热重载 +- **改造** `chat_agent.py` / `emotion_analyzer.py` / `multi_agent_coordinator.py` 改走 `PromptManager` 注入 +- 角色扮演与世界观参数化(`worldview` 字段) + +### 数据持久化(Database Expansion) + +#### Added / 新增 — 8 张拟人化数据表 +- `life_state_daily_snapshot` — 生活状态日快照(状态/时长/触发原因) +- `dream_records` — 梦境记录(主题/情绪/象征物) +- `diary_entries` — 日记条目(每日反思) +- `open_loops` — 未完话题(uuid/raised_at/closed_at/topic/entities) +- `creative_projects` — 创作项目(uuid/title/status/quality_score) +- `creative_story_bibles` — 创作圣经(角色/世界观/章节大纲) +- `creative_memory_pool` — 创作记忆池(续写上下文) +- `proactive_ideas` — 主动想法池 + +#### Security / 安全 +- **`src/core/database_manager.py` 新增 23 表白名单 + 多个 `update_*` 方法列名白名单** + - 业务层禁止拼接 SQL 列名 / 表名,所有动态列名经白名单校验 + - 防御 SQL 注入 + - 启用 `PRAGMA journal_mode=WAL` 支持并发读写 +- **新增** [`src/core/schedule_similarity_checker.py`](src/core/schedule_similarity_checker.py) — 日程查重,避免重复创建相同日程 + +### 事件驱动(Event-Driven) + +#### Changed / 变更 +- **改造** [`src/core/event_manager.py`](src/core/event_manager.py) — 引入通知型 vs 任务型事件分类 + - 通知型:notify_user / notify_admin + - 任务型:task_dispatch / task_complete + - listener 异常隔离,单 listener 失败不影响其他 + +### 调度与线程隔离(Scheduler Isolation) + +#### Added / 新增 +- [`src/core/background_scheduler.py`](src/core/background_scheduler.py) — 后台调度器 + - 独立 asyncio 子线程运行,**与 Tkinter 主循环物理隔离** + - 通过线程安全队列与 GUI 通信 + - 调度项:生活状态转移 / 梦境生成 / 创作续写 / 主动消息投递 / 话题时间衰减 + - **每个 tick 完成后通过 EventManager 发布 `scheduler_tick` 状态(payload 含 life_state_snapshot / dream_record / open_loop_count / creative_project_count)**(为 v3.0.0 Web 推送预埋接口) + +### 工程规约(Hard Constraints) + +#### Added / 新增 — 写入 `project_memory.md` 的硬约束 +- 所有新模块必须作为独立文件添加到 `src/core/`,**不通过 Mixin 扩展既有类** +- 复用 `LLMHelper` / `PromptManager` / `DatabaseManager` 三大基础设施 +- GUI 触发架构和后台调度器必须使用 asyncio 子线程**物理隔离** +- Feature flags(`ENABLE_LIFE_STATE` / `ENABLE_EMOTION_WHEEL` / `ENABLE_PROACTIVE_ENGINE` 等)**默认禁用**新功能,显式开启后才生效 +- 创意项目必须包含质量审核(similarity ≤ 0.72, retry ≤ 2, min_score ≥ 7) +- 新增数据表统一命名为 `creative_projects` / `creative_story_bibles` / `creative_memory_pool` +- 长期创作项目在聊天不活跃 **30 分钟**后触发 +- 情感分析统一使用 **Plutchik 8 维情绪模型** + +### 文档更新(Documentation) + +#### Added / 新增 +- 新增 [`docs/architecture.md`](docs/architecture.md) — 整体架构图、ASCII 数据流、模块依赖、并行调度、mermaid 总览 +- 新增 [`.trae/AGENTS.md`](../.trae/AGENTS.md) / [`.trae/CONTEXT.md`](../.trae/CONTEXT.md) — 智能体协作规约 +- 更新 [`README.md`](README.md) — 增加拟人化能力说明 + Feature flag 配置 +- 新增 [`docs/agents/`](../docs/agents/) — Hermes Agent skills 脚手架 +- 调研报告 [ARCHITECTURE_REFACTOR_PLAN.md](../ARCHITECTURE_REFACTOR_PLAN.md) / [INVESTIGATION_HUMANIZATION_CREATIVE.md](../INVESTIGATION_HUMANIZATION_CREATIVE.md) + +### Lessons Learned / 经验沉淀 +- **Mixin 模式导致单类膨胀(70K+ 行)**:后续新能力必须以独立文件 + 组合方式扩展,不再使用 Mixin +- **Plutchik 8 维比"印象评分"更适合雷达可视化**:为 v3.0.0 Web 端 EmotionPanel 雷达图铺路 +- **WAL 模式必开**:GUI 进程与后台调度进程并发读写 SQLite,WAL 是必选项 +- **Feature flag 默认关闭**:生产事故多为"新功能默认开启"导致 ## [2.2.0] - 2026-02-22 ### 功能移除 / Features Removed -- **多智能体协作系统**: 移除动态多智能体协作、DeepAgents集成等功能 -- **事件系统**: 移除事件驱动系统、事件管理等功能 -- **LangGraph集成**: 移除LangGraph相关功能和依赖 +- **ConversationGraph** (`src/core/conversation_graph.py`): 删除全仓零引用的 LangGraph 对话流程管理模块（真僵尸） +- **EnhancedKnowledgeBase** (`src/core/enhanced_knowledge_base.py`): 删除仅测试在用的增强知识库封装（与 `KnowledgeBase` 职责重叠约 80%） +- **DeepAgentsKnowledgeManager** (`src/core/deepagents_wrapper.py`): 删除与 `EnhancedKnowledgeBase` 配套的虚拟文件系统知识管理器（仅被已删除模块引用，属死代码；`DeepSubAgentWrapper` 保留） + +### 修正 / Corrected +- 澄清以下模块仍保留并被生产代码使用（CHANGELOG 早前版本曾错误地将其列为"已移除"，现已修正）： + - **多智能体协作 / 动态多智能体图** (`src/core/multi_agent_coordinator.py` / `dynamic_multi_agent_graph.py`): 仍由 `chat_agent.py:22` / `:239` 等处生产在用 + - **事件系统 / 事件管理** (`src/core/event_manager.py`): 仍由 `chat_agent.py:20` + `gui_enhanced.py` 20+ 处密集调用 + - **LangGraph 集成**: `dynamic_multi_agent_graph.py` 仍使用 LangGraph `StateGraph` / `MemorySaver`，仅 `conversation_graph.py` 为真僵尸并已删除 ### 文档更新 / Documentation Updates -- **修正README.md**，移除已不存在功能的描述 +- **修正 README.md**，移除已不存在功能的描述 - **优化文档结构**，提高可读性和维护性 +- **更正 `src/core/__init__.py` 文档字符串**：`ScheduleGenerator` → `TemporaryScheduleGenerator`（实际类名），并澄清 Usage 块中 import 行为为示例 ### 改进 / Improved - ✅ 项目结构更加简洁 diff --git a/README.md b/README.md index 989575f..483bbdc 100644 --- a/README.md +++ b/README.md @@ -24,12 +24,49 @@ Neo Agent 是一个基于 LangChain + LangGraph 的智能对话代理系统， - 🧠 **分层记忆系统**: 短期记忆、长期记忆、知识库、基础知识 - 💭 **智能对话**: 角色扮演、连续对话、记忆检索、情感理解 - 📊 **情感分析**: 印象评估、累计评分、关系可视化 -- 🖥️ **现代化GUI**: 基于Tkinter的友好界面 +- 🖥️ **现代化GUI**: Web GUI（默认）+ Tkinter GUI（开发态/降级方案） +- 🌐 **Web 端独占能力**: + - **跨设备访问**：任意浏览器打开 http://:8000 即可使用，手机/平板/PC 体验一致 + - **实时 WebSocket 推送**：消息流、事件流、主动消息、调试日志全双工推送 + - **远程访问**：部署到服务器后可通过公网访问，多用户可同时接入 + - **流式回复**：基于 LangChain astream 实时输出 token；不支持时降级到 run_in_executor 模拟流式 - ️ **数据管理**: SQLite存储、数据迁移、备份恢复 - 📝 **提示词工程**: 模块化Markdown提示词、角色扮演、世界观注入 +### 技术栈 + +- **后端**：FastAPI 0.110+ / Uvicorn / Pydantic v2 / WebSockets +- **前端**：React 18 + TypeScript + Vite 5 +- **UI 组件库**：Ant Design 5 +- **状态管理**：Zustand +- **图表**：ECharts 5 +- **HTTP 客户端**：Axios +- **业务核心**：LangChain + LangGraph（与 Tkinter 模式共用，未改动） + ### 快速开始 +#### Web GUI（推荐 / 默认） + +```bash +# 一键启动（自动安装依赖 + 启动后端 + 前端 dev server） +./start.sh + +# 或手动启动 +pip install -r requirements-web.txt +cd src/web/frontend && npm install && cd - +python run_web.py +``` + +启动后访问 http://localhost:8000 + +> 默认模式为 Web；通过 `--tk` 切换到原 Tkinter GUI。 + +#### Tkinter GUI（开发态/降级方案） + +```bash +python main.py --tk +``` + #### 安装依赖 ```bash @@ -66,8 +103,11 @@ neo-agent ``` Neo_Agent/ ├── src/ # 源代码 -│ ├── core/ # 核心模块 -│ ├── gui/ # GUI模块 +│ ├── core/ # 核心模块（与 GUI 解耦） +│ ├── gui/ # Tkinter GUI（开发态/降级） +│ ├── web/ # Web GUI（默认） +│ │ ├── backend/ # FastAPI 后端（API / WS / Services / Schemas） +│ │ └── frontend/ # React + Vite + Ant Design 前端 │ ├── tools/ # 工具模块 │ └── nps/ # NPS工具系统 ├── prompts/ # 提示词模板 @@ -78,9 +118,12 @@ Neo_Agent/ ├── tests/ # 测试文件 ├── examples/ # 示例代码 ├── docs/ # 文档 -├── main.py # 主入口（推荐） +├── main.py # 主入口（默认 Web，可 --tk 切换 Tkinter） ├── run.py # 简化启动器 -├── requirements.txt # 依赖列表 +├── run_web.py # Web 模式启动入口 +├── start.sh # Web 模式一键启动脚本 +├── requirements.txt # 核心依赖 +├── requirements-web.txt # Web 端额外依赖 ├── example.env # 环境变量示例 └── LICENSE # 许可证 ``` @@ -114,6 +157,59 @@ Neo Agent采用模块化的提示词工程系统，参考了SillyTavern的设计 --- +## GUI 启动方式（v3.0.0+） + +> 以下章节为 v3.0.0 引入的 Web GUI / Tkinter 双模式启动补充说明，保留原"快速开始"中的命令行示例之外，作为对外的明确参考。 + +### Web GUI（默认模式） + +**启动方式** + +```bash +# 方式 A：主入口（默认即 Web 模式，可显式加 --web） +python main.py --web + +# 方式 B：一键脚本（自动 venv + pip install + 启动前端 dev server） +./start.sh +``` + +启动成功后会看到： + +- FastAPI 监听 `http://0.0.0.0:8000`（可用 `--host` / `--port` 调整） +- Vite dev server（开发态）监听 `http://localhost:5173` +- 终端打印 `API 文档: http://0.0.0.0:8000/docs` + +**访问 URL** + +| 用途 | URL | +| --- | --- | +| Web 前端主页 | `http://localhost:8000` | +| 备用开发端口（Vite） | `http://localhost:5173` | +| 交互式 API 文档（Swagger UI） | `http://localhost:8000/docs` | +| ReDoc 文档 | `http://localhost:8000/redoc` | +| OpenAPI Schema | `http://localhost:8000/openapi.json` | +| 健康检查 | `http://localhost:8000/api/health` | + +**API 文档** + +- 默认挂在 FastAPI 自动生成的 `/docs`（Swagger UI）与 `/redoc` +- WebSocket 端点（不在 `/docs` 中列出）需另行参考 [`docs/architecture.md`](docs/architecture.md)： + - `/ws/chat` 流式聊天 + - `/ws/events` 全局事件流 + - `/ws/debug` 调试日志流 + - `/ws/proactive` 主动消息推送 + +### Tkinter 模式（开发态/降级） + +```bash +# 显式切换到 Tkinter GUI（不依赖 Web 端依赖） +python main.py --tk +``` + +> Tkinter 模式与 Web GUI 共享同一份 `chat_agent.db`，**数据零迁移**；当 Web 端不可用时随时回滚。详见 [`docs/rollback-procedure.md`](docs/rollback-procedure.md)。 + +--- + ## English ### Introduction @@ -134,11 +230,48 @@ Neo Agent is a LangChain + LangGraph-based intelligent conversation agent system - 🧠 **Hierarchical Memory System**: Short-term memory, long-term memory, knowledge base, base knowledge - 💭 **Intelligent Conversation**: Role-playing, continuous dialogue, memory retrieval, emotional understanding - 📊 **Emotion Analysis**: Impression assessment, cumulative scoring, relationship visualization -- 🖥️ **Modern GUI**: User-friendly Tkinter-based interface +- 🖥️ **Modern GUI**: Web GUI (default) + Tkinter GUI (dev/fallback) +- 🌐 **Web-only capabilities**: + - **Cross-device access**: any browser at http://<host>:8000 — phone, tablet, PC consistent UX + - **Real-time WebSocket push**: message stream / event stream / proactive message / debug log full-duplex + - **Remote access**: deployable on a server, multi-user concurrent + - **Streaming reply**: LangChain astream tokens; falls back to run_in_executor chunked simulation - ️ **Data Management**: SQLite storage, data migration, backup and recovery +### Tech Stack + +- **Backend**: FastAPI 0.110+ / Uvicorn / Pydantic v2 / WebSockets +- **Frontend**: React 18 + TypeScript + Vite 5 +- **UI Kit**: Ant Design 5 +- **State**: Zustand +- **Charts**: ECharts 5 +- **HTTP client**: Axios +- **Core**: LangChain + LangGraph (shared with Tkinter, unchanged) + ### Quick Start +#### Web GUI (recommended / default) + +```bash +# One-click launcher (auto-installs deps + starts backend + frontend dev server) +./start.sh + +# Or manual +pip install -r requirements-web.txt +cd src/web/frontend && npm install && cd - +python run_web.py +``` + +Open http://localhost:8000 + +> Default mode is Web. Pass `--tk` to switch back to Tkinter GUI. + +#### Tkinter GUI (dev / fallback) + +```bash +python main.py --tk +``` + #### Install Dependencies ```bash @@ -181,9 +314,12 @@ Neo_Agent/ │ └── nps/ # NPS tool system ├── tests/ # Test files ├── examples/ # Example code -├── main.py # Main entry point (recommended) +├── main.py # Main entry point (default Web, --tk for Tkinter) ├── run.py # Simplified launcher -├── requirements.txt # Dependencies +├── run_web.py # Web-mode launcher +├── start.sh # Web-mode one-click launcher +├── requirements.txt # Core dependencies +├── requirements-web.txt # Web-only dependencies ├── example.env # Environment variables template └── LICENSE # License file ``` @@ -197,6 +333,90 @@ Neo_Agent/ - **long_term_memory**: Long-term memory system - **schedule_manager**: Schedule management +### GUI Launch Modes (v3.0.0+) + +> This section supplements the Quick Start with explicit v3.0.0 launch instructions for both the Web GUI and Tkinter modes. + +#### Web GUI (default) + +**Launch** + +```bash +# Method A: main entry (Web is the default; --web is explicit) +python main.py --web + +# Method B: one-click script (auto venv + pip install + frontend dev server) +./start.sh +``` + +On a successful start you should see: + +- FastAPI listening on `http://0.0.0.0:8000` (use `--host` / `--port` to override) +- Vite dev server (dev mode) on `http://localhost:5173` +- The supervisor log line: `API 文档: http://0.0.0.0:8000/docs` + +**Access URLs** + +| Purpose | URL | +| --- | --- | +| Web frontend | `http://localhost:8000` | +| Vite dev port (alt) | `http://localhost:5173` | +| Interactive API docs (Swagger UI) | `http://localhost:8000/docs` | +| ReDoc | `http://localhost:8000/redoc` | +| OpenAPI schema | `http://localhost:8000/openapi.json` | +| Health check | `http://localhost:8000/api/health` | + +**API documentation** + +- Auto-generated by FastAPI at `/docs` (Swagger UI) and `/redoc` +- WebSocket endpoints (not listed in `/docs`) are documented in [`docs/architecture.md`](docs/architecture.md): + - `/ws/chat` — streaming chat + - `/ws/events` — global event stream + - `/ws/debug` — debug log stream + - `/ws/proactive` — proactive message push + +#### Tkinter Mode (dev / fallback) + +```bash +# Switch to the original Tkinter GUI (no Web-side deps required) +python main.py --tk +``` + +> Tkinter shares the same `chat_agent.db` with the Web GUI — **zero data migration**; use it as a safe rollback target when the Web stack is unavailable. See [`docs/rollback-procedure.md`](docs/rollback-procedure.md). + ### License This project is licensed under the MIT License. See [LICENSE](LICENSE) file for details. + +--- + +## Web GUI（v3.0.0+） + +> 本节为 v3.0.0+ Web GUI 使用速查（追加于原文档末尾，原内容保留不动）。详细架构参见 [`docs/architecture.md`](docs/architecture.md)。 + +### 启动方式 + +- 开发模式：`bash start.sh`（自动安装依赖 + 启动前后端） +- 生产模式：`python run_web.py`（需先 `cd src/web/frontend && npm run build`） +- 启动 Tkinter（兼容）：`python main.py --tk` + +### 访问 URL + +- Web GUI：http://localhost:8000 +- API 文档（自动生成）：http://localhost:8000/docs +- 健康检查：http://localhost:8000/api/health + +### 功能页面 + +- `/chat` — 聊天对话 + 情感雷达 + 话题时间线 +- `/debug` — 实时调试日志 +- `/knowledge` — 知识库管理 +- `/schedule` — 日程管理 +- `/event` — 事件管理 +- `/nps` — NPS 工具管理 +- `/database` — 数据库浏览 +- `/creative` — 创作项目管理 + +### 回滚 Tkinter + +如需紧急回滚：`python main.py --tk` 即可启动原桌面 GUI，数据完全兼容。 diff --git a/docs/api.md b/docs/api.md new file mode 100644 index 0000000..61878e2 --- /dev/null +++ b/docs/api.md @@ -0,0 +1,183 @@ +# Neo Agent Web API 参考 + +> Stage E.3 - 简要 API 列表（REST + WebSocket）。**完整定义由 FastAPI 自动生成在 `/docs`（Swagger UI）与 `/redoc`。** + +本文件用于**快速速查**，每个端点的 schema / 错误码 / 示例请以 FastAPI 生成的文档为准。 + +--- + +## 1. REST 端点 + +### 1.1 健康检查 + +| Method | Path | 说明 | +| --- | --- | --- | +| GET | `/api/health` | 服务存活检查；返回 `{status, version, timestamp}` | + +**响应示例**： + +```json +{ + "status": "ok", + "version": "3.0.0", + "timestamp": "2026-07-14T12:00:00.000000+00:00" +} +``` + +### 1.2 聊天（占位） + +| Method | Path | 说明 | +| --- | --- | --- | +| GET | `/api/chat` | Stage A.3 占位端点；返回 `{message: "chat endpoint placeholder"}` | + +> 未来将由 `POST /api/chat`（同步）与 `POST /api/chat/stream`（流式 SSE）替换。 +> 当前聊天入口走 WebSocket `/ws/chat`。 + +### 1.3 业务域端点（占位 / Stage B-C 计划） + +| 路径前缀 | 计划方法 | 用途 | +| --- | --- | --- | +| `/api/knowledge` | GET / POST / PUT / DELETE | 知识库 CRUD + `/search` | +| `/api/schedule` | GET / POST / PUT / DELETE | 日程管理 | +| `/api/events` | GET | 事件查询（运行时事件） | +| `/api/nps/status` | GET | NPS 工具状态 | +| `/api/database/tables` | GET | 数据库元信息（表 / schema 摘要） | +| `/api/creative/generate` | POST | 创意生成 | + +> 这些端点在当前 Stage 尚未注册到 `src/web/backend/main.py`；实现按业务域分文件放到 `src/web/backend/api/`。 + +--- + +## 2. WebSocket 端点 + +所有 WS 端点位于 `/ws/*`；连接时由服务端先发送 `{"type": "ack" | "system", ...}`。 + +### 2.1 `/ws/chat` — 流式聊天 + +| 方向 | 帧 | 字段 | +| --- | --- | --- | +| 客户端 → 服务端 | `message` | `{type, content, context?}` | +| 客户端 → 服务端 | `ping` | `{type: "ping"}` | +| 服务端 → 客户端 | `system` | `{type, message}`（连接成功） | +| 服务端 → 客户端 | `chunk` | `{type, content}`（流式增量） | +| 服务端 → 客户端 | `done` | `{type}`（流式结束） | +| 服务端 → 客户端 | `pong` | `{type}`（心跳响应） | +| 服务端 → 客户端 | `error` | `{type, message}` | + +**示例**： + +```text +# 客户端 +ws.send('{"type":"message","content":"你好"}') +# 服务端（按顺序） +<- {"type": "system", "message": "connected to /ws/chat"} +<- {"type": "chunk", "content": "你好"} +<- {"type": "chunk", "content": "，"} +<- {"type": "chunk", "content": "很高兴认识你"} +<- {"type": "done"} +``` + +### 2.2 `/ws/events` — 全局事件 + +| 方向 | 帧 | 字段 | +| --- | --- | --- | +| 客户端 → 服务端 | 任意 | `{type, ...}`（当前仅做 ack） | +| 服务端 → 客户端 | `ack` | `{type, conn_id, channel: "events"}` | +| 服务端 → 客户端 | 事件 payload | `{type: , data: {...}}` | + +事件类型（EventManager → EventService → broadcast）： + +- `scheduler_tick` — 后台调度器心跳（每 60s） +- `proactive_message` — 主动消息（ProactiveEngine 触发） +- `chat_event` — 聊天事件（user_message / done） +- `message` — 通用消息总线 +- 通配：`*` 订阅全部 + +### 2.3 `/ws/debug` — 调试日志 + +| 方向 | 帧 | 字段 | +| --- | --- | --- | +| 客户端 → 服务端 | 任意 | `{type, ...}`（仅 ack） | +| 服务端 → 客户端 | `ack` | `{type, conn_id, channel: "debug"}` | +| 服务端 → 客户端 | 日志帧 | `{type: "debug_log", level, module, message, timestamp}` | + +> 接入 `src/tools/debug_logger.py`（`DebugLogger.subscribe`）， +> `src/web/backend/ws/debug_broadcaster.py` 负责将日志广播到所有 `/ws/debug` 连接。 + +### 2.4 `/ws/proactive` — 主动消息 + +| 方向 | 帧 | 字段 | +| --- | --- | --- | +| 客户端 → 服务端 | 任意 | `{type, ...}`（仅 ack） | +| 服务端 → 客户端 | `ack` | `{type, conn_id, channel: "proactive"}` | +| 服务端 → 客户端 | 主动消息 | `{type: "proactive_message", title, body, ts}` | + +> Stage A.2 骨架：连接后保持长连接，主动推送由 `manager.broadcast("proactive", msg)` 驱动。 + +### 2.5 `/ws/event`（已弃用，请用 `/ws/events`） + +Stage A.2 早期版本。保留以做兼容，**新代码请使用 `/ws/events`**。 + +--- + +## 3. Pydantic Schema 速查 + +源文件位置：`src/web/backend/schemas/` + +| Schema | 用途 | +| --- | --- | +| `ChatRequest` | `/api/chat` POST 请求体 | +| `ChatChunk` | 流式 chunk（`type=chunk` / `done` / `error`） | +| `ChatResponse` | 同步响应（`content` + `message_id` + `emotion?`） | +| `EventDTO` | 事件对象 | +| `EventListResponse` | 事件列表响应 | +| `KnowledgeEntity` / `KnowledgeDefinition` | 知识库实体 / 概念定义 | +| `KnowledgeSearchRequest` / `KnowledgeSearchResponse` | 知识库搜索 | +| `ScheduleDTO` / `ScheduleCreate` / `ScheduleUpdate` | 日程 CRUD | + +> 未来扩展：`emotion.py` / `memory.py` / `creative.py` / `nps.py` / `debug.py`。 + +--- + +## 4. 错误与状态码 + +### 4.1 HTTP + +- `200 OK` — 正常 +- `400 Bad Request` — 请求参数不合法（Pydantic 校验失败） +- `404 Not Found` — 路由不存在 +- `422 Unprocessable Entity` — FastAPI 校验失败（请求体 schema 不匹配） +- `500 Internal Server Error` — 服务端异常 + +### 4.2 WebSocket 错误帧 + +服务端通过 `{type: "error", message: "..."}` 推送错误，连接**不自动关闭**，由客户端决定是否重试。 + +| 场景 | message 示例 | +| --- | --- | +| 空输入 | `"empty input"` | +| ChatService 不可用 | `"ChatService unavailable"` | +| chat_stream 异常 | 异常 `str(e)` | + +--- + +## 5. 自动化文档 + +| 工具 | URL | +| --- | --- | +| Swagger UI | `http://localhost:8000/docs` | +| ReDoc | `http://localhost:8000/redoc` | +| OpenAPI JSON | `http://localhost:8000/openapi.json` | + +> WebSocket 端点**不会**自动出现在 `/docs` 中，请参考本文件 §2。 + +--- + +## 6. 相关文件 + +- `src/web/backend/main.py` — FastAPI 入口 +- `src/web/backend/ws/*.py` — WebSocket 路由 +- `src/web/backend/services/*.py` — Service 层 +- `src/web/backend/schemas/*.py` — Pydantic v2 模型 +- `architecture.md` — 整体架构 +- `rollback-procedure.md` — 故障回滚 diff --git a/docs/architecture.md b/docs/architecture.md new file mode 100644 index 0000000..0a88fb4 --- /dev/null +++ b/docs/architecture.md @@ -0,0 +1,439 @@ +# Neo Agent Web 架构文档 + +本文档描述 Neo Agent v3.0.0 引入的 Web GUI（FastAPI + React）整体架构、数据流与部署模型。 +Tkinter GUI 仍受支持（通过 `python main.py --tk` 启动），其架构参见 [`ARCHITECTURE.md`](ARCHITECTURE.md)。 + +--- + +## 1. 总体架构图（ASCII） + +``` ++----------------------------------------------------------------+ +| 浏览器 / 客户端 | +| (Chat / Debug / Knowledge / Schedule / Event / NPS / Database | +| / Creative + React 18 + Vite 5 + Ant Design 5) | ++--------------------+---------------------+----------------------+ + | HTTP/REST | WebSocket (full-dup) + v v ++----------------------------------------------------------------+ +| 后端 FastAPI 0.110+ (Uvicorn, src/web/backend) | +| +-------------------+ +--------------------------------+ | +| | REST API | | WebSocket 路由 | | +| | /api/chat | | /ws/chat (流式回复) | | +| | /api/knowledge | | /ws/events (EventManager) | | +| | /api/schedule | | /ws/debug (DebugLogger) | | +| | /api/events | | /ws/proactive (主动消息) | | +| | /api/nps/... | +--------------------------------+ | +| | /api/database | | +| | /api/creative | Pydantic v2 Schemas | +| +-------------------+ (强约束请求/响应) | +| | | +| v | +| +----------------------------------------------------------+ | +| | Service Layer (Service) | | +| | ChatService / KnowledgeService / EventService / | | +| | ScheduleService / DatabaseService / CreativeService | | +| +----------------------------------------------------------+ | ++----------------------------------------------------------------+ + | + v ++----------------------------------------------------------------+ +| 核心业务层 src/core (GUI 无关, Tkinter / Web 共用) | +| ChatAgent KnowledgeBase ScheduleManager EmotionAnalyzer | +| LongTermMemory ProactiveEngine BackgroundScheduler | +| EventManager (单例, 总线) | ++----------------------------------------------------------------+ + | + v + +-------------------------+ + | SQLite chat_agent.db | + | WAL mode (并发读安全) | + +-------------------------+ +``` + +三层职责： + +- **前端 (Presentation)**：纯 UI + 状态管理 + WebSocket 订阅，不持有业务规则 +- **后端 (Application)**：REST 路由 + WS 路由 + Service 编排 + Schema 校验 +- **数据库 (Persistence)**：SQLite + WAL，业务层统一通过 `DatabaseManager` 访问 + +--- + +## 2. 前后端数据流 + +Neo Agent Web 端同时使用 **REST** 与 **WebSocket** 两种通信方式，按业务场景选型。 + +### 2.1 REST API（请求/响应模型） + +适用于：单次调用、明确边界、需要缓存/重试的场景。 + +| 场景 | 示例端点 | +| --- | --- | +| 健康检查 | `GET /api/health` | +| 知识库查询 | `GET /api/knowledge` / `POST /api/knowledge/search` | +| 日程管理 | `GET/POST/PUT/DELETE /api/schedule/...` | +| 事件查询 | `GET /api/events` | +| NPS 状态 | `GET /api/nps/status` | +| 数据库元信息 | `GET /api/database/tables` | +| 创意生成 | `POST /api/creative/generate` | +| 非流式聊天 | `POST /api/chat`（同步返回完整回复） | + +特点： +- 基于 Pydantic v2 做请求/响应 schema 校验 +- 错误通过 HTTP 状态码 + JSON 体返回 +- 前端用 **Axios** 调用，统一拦截器处理错误 + +### 2.2 WebSocket 端到端数据流（ASCII） + +``` +[浏览器 Chat 页] + | 1) ws.send({type:"user_message", text:"hi"}) + v +[FastAPI /ws/chat] + | + v +[ChatService.handle_user_message] + | + v +[ChatAgent.chat_stream()] <- 异步生成器 + | + | astream token ... token + v +[ChatService] yield {type:"token", delta:"..."} + | + v +[WebSocket Connection Manager] broadcast(json) + | + v +[浏览器 Chat 页] 逐 token 渲染 + +[ChatAgent 完成] yield {type:"done", full_text:"..."} + | + v +[ChatAgent] EventManager.publish("chat_completed", payload) + | + v +[EventService] (后台 task 把事件投到 /ws/events) + | + v +[浏览器] 监听 /ws/events 的 Debug / Event 页签更新 +``` + +要点： +- 单次会话的 `chat_stream` 是一条 **async generator**，每段 token 都是一次 `yield` +- `EventManager` 单例：业务层 `publish` → `EventService` 桥接 → `/ws/events` 广播，**Web 层不直接侵入业务** +- 异常保护：单个 listener 抛错不会影响其他 listener（见 `EventManager.publish` try/except） + +### 2.3 WebSocket（双向推送模型） + +适用于：长连接、服务端主动推送、流式输出。 + +| 通道 | 推送内容 | +| --- | --- | +| `/ws/chat` | 单次会话的流式回复（token 级别）+ 完成事件 | +| `/ws/events` | 来自 `EventManager` 的全局事件（日程触发、主动消息、生命周期变化等） | +| `/ws/debug` | 调试日志流（来自 `DebugLogger`） | +| `/ws/proactive` | 主动消息推送（无需用户输入） | + +特点： +- 单向广播（服务端 → 客户端）为主，客户端可发心跳 +- 前端通过 `WebSocket Client`（hooks 层）订阅各通道 +- 事件通过 `EventService` 单例从 `EventManager` 桥接，零侵入业务层 + +### 2.4 选型对照 + +| 维度 | REST | WebSocket | +| --- | --- | --- | +| 触发方 | 客户端 | 服务端/客户端皆可 | +| 实时性 | 请求后才返回 | 服务端可随时推送 | +| 适用 | CRUD、查询、配置变更 | 流式输出、事件广播 | +| 资源占用 | 短连接 | 长连接，少量并发 | +| 重试 | 易（HTTP 重试） | 需前端重连逻辑 | + +--- + +## 3. 模块依赖图 + +依赖规则：**`src/core/*` 不依赖 `src/web/*`**；`src/web/*` 通过 Service 层包装 core，反向依赖。 + +``` ++----------------------+ +-----------------------+ +| src/web/frontend | HTTP | src/web/backend | +| (React + Vite) | ------> | (FastAPI) | +| | WS | | ++----------------------+ ------> +-----------+-----------+ + | + | uses (Service wrapping) + v ++----------------------+ +-----------------------+ +| src/web/backend | ------> | src/core/* | +| services/ | | ChatAgent, | +| (ChatService, ...) | | KnowledgeBase, | ++----------------------+ | ScheduleManager, ... | + | EventManager, | + | BackgroundScheduler | + +-----------+-----------+ + | + v + +-----------------------+ + | src/tools/* | + | (DebugLogger 等) | + +-----------+-----------+ + | + v + +-----------------------+ + | chat_agent.db | + | (SQLite, WAL) | + +-----------------------+ +``` + +约束： + +1. `src/core/*` **严禁** `import` 任何 `src/web/*` +2. `src/web/backend/services/*` 持有 `core` 对象的引用（通过构造注入或模块级单例） +3. `src/web/backend/api/*` 与 `src/web/backend/ws/*` **不直接**访问 `core`，统一通过 `services/` +4. `src/web/frontend` 与 `src/core/*` 之间没有任何代码引用；二者通过 HTTP/WS + JSON 通信 +5. 允许的反向引用：core 内部模块互相引用；`web -> services -> core` 单向 + +这样在 v3.0.0 中替换为 Web GUI 时，**`src/core/*` 一行未改**，所有改动都被限制在 `src/web/*` 内。 + +--- + +## 4. 并行调度说明 + +Neo Agent v3.0.0 在同一进程内同时跑两类"长时间后台"组件：FastAPI/uvicorn 异步事件循环、BackgroundScheduler / EventManager 等后台线程。 + +### 4.1 BackgroundScheduler（独立子线程） + +``` ++------------------------------------------------------+ +| Python 主进程 | +| +-------------------+ +----------------------+ | +| | FastAPI / uvicorn | | BackgroundScheduler | | +| | (async event | | (独立子线程, | | +| | loop, 主循环) | | asyncio.run 私有 | | +| | | | event loop) | | +| +---------+---------+ +----------+-----------+ | +| | | | +| | await | await | +| v v | +| +------------------------------------------+ | +| | EventManager 单例 (跨线程共享) | | +| +------------------------------------------+ | +| | | +| v | +| +------------------------------------------+ | +| | DatabaseManager 单例 (WAL, 线程安全) | | +| +------------------------------------------+ | ++------------------------------------------------------+ +``` + +要点： + +- `BackgroundScheduler.start()` 启动一个 `daemon=True` 子线程 +- 子线程内部用 `asyncio.run(...)` 跑自己的 event loop，**与 FastAPI 主循环物理隔离** +- 每个 tick（默认 60s）采集状态快照并通过 `EventManager.publish("scheduler_tick", payload)` 推送给订阅者 +- 子线程异常由 scheduler 内部 try/except 兜住，**不会冒泡到 FastAPI 主循环** +- 关闭时由 `_stop_event` 触发优雅退出，等待 `self._thread.join(timeout=5)` + +### 4.2 EventManager（单例，跨线程） + +``` ++----------+ publish("proactive_message") +----------------+ +|Proactive | --------------------------------------> | EventManager | +|Engine | | (单例) | ++----------+ +--------+-------+ + | subscribe ++----------+ publish("scheduler_tick") | +|Background| --------------------------------------------+ | +|Scheduler | | | ++----------+ v v + +-----+-----+----+ + | EventService | + | (Web 后端) | + +-------+-------+ + | broadcast + v + +-------+-------+ + | /ws/events | + | 浏览器 | + +---------------+ +``` + +要点： + +- `EventManager` 是**进程内单例**，所有模块（包括 Tkinter / Web）共享同一个实例 +- 订阅通过 `subscribe(event_type, callback)` 注册，回调签名 `callback(event_type, payload)` +- 业务层 `publish(...)` 调用零侵入；Web 端 `EventService` 在启动时 `subscribe("*", ...)` 接管广播 +- publish 内部对每个 listener 都用 try/except 包裹，单个失败不影响其他订阅者 + +### 4.3 Web 端推送回路（与 Tkinter 共存） + +- Web 模式：FastAPI 启动时 `EventService.start()`，把 EventManager 的事件投到 `/ws/events` +- Tkinter 模式：Tkinter 主循环直接读 `EventManager._listeners`（沿用旧有事件回调），不经过 `EventService` +- 两端互不干扰：WAL 模式下数据库可并发读，单写者安全 + +--- + +## 5. mermaid 总览图（参考） + +> 本节与上方 ASCII 图表达同一架构；保留以便在支持 mermaid 的渲染器（GitLab / Obsidian）下阅读。 + +```mermaid +graph LR + Browser[浏览器] + Mobile[移动端浏览器] + Tablet[平板浏览器] + + subgraph Frontend["前端 (React 18 + Vite + Ant Design 5)"] + Pages[8 个页面
Chat / Debug / Knowledge / Schedule / Event / NPS / Database / Creative] + Store[Zustand Store] + WSC[WebSocket Client] + end + + subgraph Backend["后端 (FastAPI 0.110 + Uvicorn)"] + REST[REST API
/api/chat /api/knowledge ...] + WSR[WebSocket Routes
/ws/chat /ws/events /ws/debug /ws/proactive] + Service[Service Layer
ChatService / KnowledgeService / EventService / ...] + Schemas[Pydantic Schemas] + end + + subgraph Core["核心业务层 (src/core, GUI 无关)"] + ChatAgent[ChatAgent] + KBase[KnowledgeBase] + Sched[ScheduleManager] + Emotion[EmotionAnalyzer] + Memory[LongTermMemory] + DBSched[BackgroundScheduler] + EventMgr[EventManager] + end + + DB[(SQLite chat_agent.db
WAL 模式)] + + Browser -->|HTTP REST| REST + Mobile -->|HTTP REST| REST + Tablet -->|HTTP REST| REST + + Browser -->|WS| WSR + Mobile -->|WS| WSR + Tablet -->|WS| WSR + + Pages --> Store + Pages --> WSC + WSC -->|实时推送| WSR + WSC --> Store + + REST --> Schemas + WSR --> Schemas + Schemas --> Service + + Service --> ChatAgent + Service --> KBase + Service --> Sched + Service --> Emotion + Service --> Memory + + ChatAgent --> EventMgr + KBase --> EventMgr + Sched --> EventMgr + DBSched -->|publish tick / status| EventMgr + EventMgr -->|broadcast| WSR + + ChatAgent --> DB + KBase --> DB + Sched --> DB + Emotion --> DB + Memory --> DB + DBSched --> DB +``` + +--- + +## 6. 数据共享 + +Web GUI 与 Tkinter GUI **完全共享同一份数据库**，无需数据迁移。 + +### 6.1 数据库位置 + +``` +/chat_agent.db +``` + +- SQLite 单一文件，相对路径存储 +- **已启用 WAL 模式**（Stage E.3 期间在 `database_manager.py` 中开启），允许 Tkinter 与 Web 同时读写而不冲突 +- 两端均通过 `DatabaseManager` 单例访问，业务层无差异 + +### 6.2 切换时数据一致性 + +| 场景 | 行为 | +| --- | --- | +| Tkinter 运行中切换到 Web | Tkinter 写完即关闭连接，Web 启动后立即可见新数据 | +| Web 运行中切换到 Tkinter | 同上 | +| 双端同时运行（不推荐） | WAL 模式可并发读，单写者安全；写入交错时由 SQLite 串行化保证一致性 | + +### 6.3 共享的核心模块 + +以下模块由 Tkinter 与 Web 共用，**未做任何 GUI 耦合**： + +- `src/core/chat_agent.py`（新增 `chat_stream()` 异步生成器） +- `src/core/knowledge_base.py` +- `src/core/schedule_manager.py` +- `src/core/emotion_analyzer.py` +- `src/core/long_term_memory.py` +- `src/core/database_manager.py`（WAL 模式） +- `src/core/background_scheduler.py`（每个 tick 推状态到 `EventManager`） +- `src/core/event_manager.py`（事件总线，Web 端 `EventService` 订阅它做 WS 广播） + +--- + +## 7. 后台调度（速查） + +`BackgroundScheduler` 是一个独立线程中的周期任务（默认 60s/tick），与 GUI 解耦： + +- Web 模式：随 FastAPI 启动/停止 +- Tkinter 模式：随 GUI 启动/停止 +- 每个 tick 完成后通过 `EventManager` 发布状态事件，Web 端 `/ws/events` 可实时观察到调度器心跳 + +禁用方式： +- Web 模式：`python main.py --web --no-bg-scheduler` +- 全局：`.env` 中 `ENABLE_BACKGROUND_SCHEDULER=false` + +--- + +## 8. 部署模型 + +### 8.1 开发态 + +``` +[浏览器] ---HTTP/WS---> [Vite dev server :5173] ---proxy---> [FastAPI :8000] + |- EventService + `- BackgroundScheduler +``` + +### 8.2 生产态 + +``` +[浏览器] ---HTTP/WS---> [Nginx/Caddy] ---> [FastAPI :8000 (uvicorn --workers N)] + |- EventService + `- BackgroundScheduler +``` + +前端使用 `npm run build` 产物（`dist/`）由 Nginx 直接提供。 + +--- + +## 9. 安全与并发 + +- 开发态：单进程单用户，仅本地/内网访问 +- 生产态：建议加 Nginx 反向代理 + Basic Auth / OAuth 前置层 +- WebSocket：每个连接一个独立 Task，自动随客户端断开清理 +- 写入并发：SQLite WAL 模式 + 业务层加锁，串行化写 + +--- + +## 10. 参考 + +- [`ARCHITECTURE.md`](ARCHITECTURE.md) — 旧 Tkinter 时代模型/框架架构 +- [`API.md`](API.md) — REST API 详细说明（待 Web 版本补全） +- [`TECHNICAL.md`](TECHNICAL.md) — 业务模块技术细节 +- [`../README.md`](../README.md) — 项目入口 diff --git a/docs/rollback-procedure.md b/docs/rollback-procedure.md new file mode 100644 index 0000000..5fcdc27 --- /dev/null +++ b/docs/rollback-procedure.md @@ -0,0 +1,202 @@ +# 回滚步骤（Web GUI → Tkinter GUI） + +> **5 分钟内**将系统从 Web GUI 切回 Tkinter GUI 的标准操作流程。 + +--- + +## 适用场景 + +- Web 端出现严重 bug 且无法快速修复 +- 后端依赖（FastAPI / Uvicorn / WebSockets）无法在当前环境安装 +- 浏览器兼容性问题导致用户无法访问 +- 端口 `8000` / `5173` 被其他服务占用且无法释放 +- 临时降级以快速恢复线上可用性 + +--- + +## 5 分钟回滚步骤 + +### Step 1. 停止 Web 服务（如在运行） + +```bash +# 方式 A：前台运行时在终端按 Ctrl+C +# 方式 B：后台运行时按 pid 终止 +ps -ef | grep -E "uvicorn|run_web|main\.py|npm run dev" | grep -v grep +kill +# 如未退出，等待 5 秒后强杀 +kill -9 +``` + +如有前端 dev server（`npm run dev`）也一并结束。`start.sh` 会在收到 SIGINT/SIGTERM 时自动级联关闭 Vite dev server。 + +### Step 2. 启动 Tkinter 模式 + +```bash +python main.py --tk +``` + +> 默认模式为 Web，`--tk` 显式切换到 Tkinter GUI。窗口出现后即可正常使用。 + +### Step 3. 验证 Tkinter 启动正常 + +肉眼检查以下清单： + +- [ ] 主窗口正常打开（无 `ImportError` / `TclError`） +- [ ] 启动日志出现 `后台调度启动` 字样（如启用） +- [ ] Chat / Debug / Schedule / Knowledge 等 Tab 可点击切换 +- [ ] 之前在 Web 端创建的对话、知识库条目、日程仍可见 + +### Step 4. 验证 `chat_agent.db` 数据完整 + +```bash +# 项目根目录下检查 SQLite 文件存在 +ls -lh chat_agent.db + +# 抽查关键表行数 +sqlite3 chat_agent.db "SELECT 'chat_history', COUNT(*) FROM chat_history + UNION ALL SELECT 'knowledge_entries', COUNT(*) FROM knowledge_entries + UNION ALL SELECT 'schedule_items', COUNT(*) FROM schedule_items;" +``` + +预期：所有表的 `COUNT(*)` 与 Web 端最后一次使用时一致；如启用 WAL，目录里会同时存在 `chat_agent.db-wal` / `chat_agent.db-shm`。 + +--- + +## 数据零迁移 + +Web GUI 与 Tkinter GUI **共享同一份数据库**，**无任何数据迁移步骤**。 + +### 共享范围 + +- 数据库文件：`/chat_agent.db` +- 模式：SQLite + **WAL 模式**（`src/core/database_manager.py:74-80` 启用 `PRAGMA journal_mode=WAL`） +- 访问入口：两端均通过 `DatabaseManager` 单例 + +### 切换时的行为 + +| 场景 | 行为 | +| --- | --- | +| Web → Tkinter | Web 关闭后连接释放；Tkinter 启动后立即看到全部数据 | +| Tkinter → Web | 同上 | +| 双端同时（不推荐） | WAL 模式允许并发读，单写者安全；写入交错由 SQLite 串行化 | + +### 哪些表是共享的 + +所有由 `DatabaseManager` 管理的表（约 23 张），典型如： + +- `chat_history` — 聊天历史 +- `knowledge_entries` — 知识库条目 +- `schedule_items` — 日程 +- `events` / `proactive_messages` — 事件与主动消息 +- `emotion_*` / `long_term_memory` — 情感与长期记忆 +- `background_scheduler_status` — 调度器状态 + +业务层（`src/core/*`）未做任何 GUI 耦合修改，所以"回滚"在数据层完全等价于"换了个壳"。 + +--- + +## 故障排查 FAQ + +### Q1. Tkinter 启动报 `ModuleNotFoundError: No module named 'tkinter'` / `TclError` + +A: 系统未安装 Tk 库。Debian/Ubuntu 安装： + +```bash +sudo apt-get update && sudo apt-get install -y python3-tk +``` + +macOS（Homebrew Python）：`brew install python-tk`。Windows：重新安装 Python 时勾选 "tcl/tk and IDLE"。 + +### Q2. Web 关闭后端口 `8000` 仍被占用 + +A: 残留进程没退干净。查找并强杀： + +```bash +# 查找占用 8000 端口的进程 +lsof -i :8000 +# 或 +fuser 8000/tcp +# 杀掉它 +kill -9 +``` + +如启动 `start.sh` 时遇到同样问题，先 `pkill -f run_web.py` 再重试。 + +### Q3. Tkinter 模式下"之前的对话不见了" / 数据像是被清空 + +A: 大概率是打开的项目目录不对，或误用了 `chat_agent.db` 的另一份副本。检查： + +```bash +# 当前工作目录下是否真的有 db 文件 +ls -lh chat_agent.db +# 它的修改时间是否就是上次 Web 关闭的时间 +stat chat_agent.db +``` + +如确有第二份 db，统一到同一目录即可（两端共享 `chat_agent.db` 的相对路径）。 + +### Q4. 回滚后 `ENABLE_BACKGROUND_SCHEDULER=true` 不再生效 + +A: 调度器依赖 LLM 凭据，Web 模式下能跑通不代表 Tkinter 也行。检查： + +- `.env` 中 `DEEPSEEK_API_KEY` / `OPENAI_API_KEY` 等关键变量存在 +- 控制台日志中是否出现 `后台调度启动失败（不影响 GUI 启动）` 的警告 +- 调度器只是辅助功能，关闭它仍可使用 Tkinter：在 `.env` 中设置 `ENABLE_BACKGROUND_SCHEDULER=false` + +### Q5. 切回 Web 模式后，前端页面 404 / API 报 500 + +A: 通常是前端 dev server 缓存或后端 CORS 问题。逐项排查： + +1. **404** — 前端构建产物未生成： + ```bash + cd src/web/frontend && npm run build + ``` + 或开发态用 `./start.sh`（自动启动 Vite dev server at :5173）。 +2. **CORS 500** — 浏览器访问的不是 `http://localhost:8000` 而是其他 host/port。FastAPI 的 CORS 白名单仅包含 `localhost` / `127.0.0.1` 下的 `5173` 与 `8000`。 +3. **API 500** — 看终端 `uvicorn` 输出的 traceback，对照 [`docs/architecture.md`](architecture.md) 定位是 `Service` 层还是 `core` 层异常。 +4. **WebSocket 反复断开** — 浏览器 F12 → Network → WS 检查 `close code`：1006 通常是反向代理超时，配置 Nginx 加上 `proxy_read_timeout 3600s;`。 + +--- + +## 回滚验证清单 + +完成上述 4 步后请确认： + +- [ ] 原 Tkinter 窗口正常打开（无 `ImportError` / `TclError`） +- [ ] 之前的对话记录可查看（`chat_history` 表） +- [ ] 知识库条目可查询（`knowledge_entries` 表） +- [ ] 日程条目可编辑（`schedule_items` 表） +- [ ] 事件 / 主动消息功能仍能触发（如有定时器） +- [ ] 后台调度器正常启动（启动日志出现 `后台调度启动` 字样） + +--- + +## 切回 Web 模式 + +排查完 Web 端问题后，恢复 Web 启动： + +```bash +# 1. 关闭 Tkinter（Ctrl+C） + +# 2. 重启 Web +./start.sh +# 或 +python run_web.py +# 或显式 +python main.py --web +``` + +--- + +## 联系 / 升级路径 + +- 详细 Web 架构：[`architecture.md`](architecture.md) +- Web 端依赖：`requirements-web.txt` +- 启动脚本：`start.sh` / `run_web.py` / `main.py --web` + +如频繁触发回滚，请收集以下信息以改进 Web 端： + +1. 失败时浏览器控制台报错（F12 → Console） +2. 后端 `uvicorn` 输出最后 50 行 +3. `chat_agent.db` 文件大小与 `git log -p` 最近 5 次提交 +4. `.env` 中 `ENABLE_*` 系列开关的实际值 diff --git a/docs/setup-local-dev.md b/docs/setup-local-dev.md new file mode 100644 index 0000000..1cc29b5 --- /dev/null +++ b/docs/setup-local-dev.md @@ -0,0 +1,369 @@ +# Neo Agent v3.0.0 — 本地开发环境配置 + +> 5 分钟跑起来 Web GUI。 + +## 一键配置（推荐） + +```bash +cd Neo_Agent +chmod +x setup-dev.sh +./setup-dev.sh +``` + +脚本会自动： +1. 检查 Python ≥ 3.10 + Node.js ≥ 18 +2. 从 `.env.example` 复制生成 `.env` +3. 创建 `.venv` 虚拟环境 +4. 安装 `requirements.txt` + `requirements-web.txt` +5. `npm install` 前端依赖 + +完成后**只需填 API Key**： + +```bash +vim .env +# 改这两行： +# SILICONFLOW_API_KEY=sk-... +# SERPAPI_API_KEY=... +``` + +## 启动方式（任选其一） + +### 方式 1：一键启动脚本（推荐） + +```bash +./start.sh +# 自动 activate venv + ENABLE_FRONTEND_DEV=1 + python run_web.py +``` + +打开浏览器 → http://localhost:8000 + +### 方式 2：手动启动 Web GUI + +```bash +source .venv/bin/activate +ENABLE_FRONTEND_DEV=1 python main.py --web +``` + +### 方式 3：纯 API 模式（无 Vite 热重载） + +```bash +source .venv/bin/activate +python main.py --web --no-dev +``` + +适合生产部署或无前端 dev server 场景。 + +### 方式 4：Tkinter GUI（无 Web 依赖） + +```bash +source .venv/bin/activate +python main.py --tk +``` + +适合纯终端环境或回滚方案。 + +## 启动参数速查 + +| 参数 | 默认 | 说明 | +|---|---|---| +| `--web` | ✅ 默认 | 启动 Web GUI (FastAPI + 可选 Vite) | +| `--tk` | | 启动 Tkinter GUI | +| `--port` | 8000 | FastAPI 监听端口 | +| `--host` | 0.0.0.0 | FastAPI 监听地址 | +| `--no-dev` | | 跳过 Vite dev server | +| `--no-static` | | 不挂载前端构建产物（纯 API） | +| `--reload` | | uvicorn 代码热重载（仅开发态） | + +## 环境变量速查（.env） + +| 变量 | 必填 | 说明 | +|---|---|---| +| `SILICONFLOW_API_KEY` | ✅ | LLM API 密钥 | +| `SERPAPI_API_KEY` | ✅ | 网络搜索 API 密钥 | +| `MAIN_MODEL_NAME` | | 主模型（默认 `deepseek-ai/DeepSeek-V3.2`） | +| `TOOL_MODEL_NAME` | | 小模型（默认 `zai-org/GLM-4.6V`） | +| `VISION_MODEL_NAME` | | 多模态模型（默认 `Qwen/Qwen3-VL-32B-Instruct`） | +| `ENABLE_WEB_GUI` | | Web GUI 总开关（false 时 --web 拒绝启动） | +| `ENABLE_FRONTEND_DEV` | | Vite dev server 开关（0/1） | +| `WEB_HOST` | | FastAPI 监听地址 | +| `WEB_PORT` | | FastAPI 监听端口 | +| `CORS_ORIGINS` | | CORS 允许来源（逗号分隔） | +| `CHAT_AGENT_DB` | | SQLite 路径（Web / Tkinter 共享） | +| `LOG_LEVEL` | | Web 日志级别（DEBUG/INFO/WARNING/ERROR） | +| `ENABLE_LIFE_STATE` | | 生活状态开关 | +| `ENABLE_PROACTIVE_ENGINE` | | 主动消息开关 | +| `ENABLE_EMOTION_WHEEL` | | 情感雷达开关 | +| `ENABLE_CREATIVE_WRITER` | | 创意写作开关 | + +## 端口规划 + +| 端口 | 用途 | 启动者 | +|---|---|---| +| 8000 | FastAPI 后端 | `python main.py --web` | +| 5173 | Vite dev server（HMR） | `ENABLE_FRONTEND_DEV=1` 时自动启动 | +| 前端 `/api/*` 代理 → 8000 | dev 模式反代 | vite.config.ts | +| 前端 `/ws/*` 代理 → 8000 | dev 模式反代 | vite.config.ts | + +## 常见问题 + +### Q1: `pip install` 失败 +- 检查网络/代理 +- 升级 pip：`python -m pip install --upgrade pip` +- 改用国内源： + ```bash + python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt + ``` + +### Q2: 前端 5173 端口被占用 +```bash +lsof -i :5173 +kill +# 或修改 vite.config.ts 的 server.port +``` + +### Q3: 8000 端口被占用 +```bash +python main.py --web --port 8080 +``` + +### Q4: WebSocket 连不上 +- 检查防火墙是否放行 8000 / 5173 +- 检查 `CORS_ORIGINS` 是否包含实际访问的 origin +- 浏览器 DevTools → Network → WS 帧查看是否握手成功 + +### Q5: 临时回滚 Tkinter +```bash +ENABLE_WEB_GUI=false python main.py --tk +``` +或显式指定： +```bash +python main.py --tk +``` + +## 目录结构（关键文件） + +``` +Neo_Agent/ +├── .env # 本地敏感配置（gitignored） +├── .env.example # 配置模板（已提交） +├── main.py # 启动入口（--web/--tk） +├── run_web.py # Web 启动脚本（被 main.py --web 调用） +├── start.sh # 一键启动 +├── setup-dev.sh # 一键配置（新增） +├── requirements.txt # Python 核心依赖 +├── requirements-web.txt # Python Web 依赖 +├── src/ +│ ├── core/ # 业务核心（只读，不要改） +│ ├── gui/ # Tkinter GUI +│ └── web/ +│ ├── backend/ # FastAPI 后端 +│ └── frontend/ # React + Vite 前端 +│ ├── package.json +│ └── vite.config.ts +└── docs/ + └── setup-local-dev.md # 本文件 +``` + +## 验证启动 + +启动后访问： + +| URL | 说明 | +|---|---| +| http://localhost:8000/ | Web GUI 主页 | +| http://localhost:8000/docs | FastAPI Swagger | +| http://localhost:8000/api/health | 健康检查 | +| http://localhost:5173/ | Vite dev server（开发态） | + +健康检查应返回： +```json +{"status":"ok","version":"3.0.0","timestamp":"2026-07-14T..."} +``` + +## 常见工作流 + +### 第一次跑通 +```bash +./setup-dev.sh +vim .env # 填 API Key +./start.sh +# 浏览器开 http://localhost:8000 +``` + +### 改后端代码 → 自动热重载 +```bash +python main.py --web --reload +``` + +### 改前端代码 → 自动 HMR +Vite dev server 已自动启动，改完即热重载。 + +### 切换 Tkinter 调试 +```bash +python main.py --tk +``` + +### 跑测试 +```bash +source .venv/bin/activate +python -m unittest discover tests -v +``` + +## 下一步 + +- 阅读 [CHANGELOG.md](../CHANGELOG.md) 了解 v3.0.0 变更 +- 阅读 [docs/architecture.md](architecture.md) 了解 Web 端架构 +- 阅读 [docs/rollback-procedure.md](rollback-procedure.md) 了解回滚流程 +- 阅读 [docs/api.md](api.md) 了解 REST/WS 端点 + +--- + +## LLM 供应商配置（v3.1.0+） + +v3.1.0 新增**供应商无关的 LLM 通用配置**。通过 `LLM_PROVIDER` 预设即可一键切换 OpenAI / DeepSeek / SiliconFlow / Moonshot / Zhipu / 自建网关，不再需要为每个供应商写不同变量名。 + +### 通用配置（推荐） + +```env +# 供应商预设：openai / deepseek / siliconflow / moonshot / zhipu / custom +LLM_PROVIDER=siliconflow +# API Key +LLM_API_KEY=sk-... +# 可选：自定义 base_url（custom 模式必填；preset 模式省略时使用厂商默认） +# LLM_BASE_URL=https://api.openai.com/v1 +# 可选：自定义模型名 +# LLM_MODEL_NAME=gpt-4o-mini +# 全局超时/温度/最大 tokens +LLM_TIMEOUT=60 +LLM_TEMPERATURE=0.7 +LLM_MAX_TOKENS=2000 +``` + +### 6 个 preset 的最小配置示例 + +| 供应商 | `LLM_PROVIDER` | 必填 | 默认 base_url | 默认 model | 申请地址 | +|---|---|---|---|---|---| +| OpenAI | `openai` | `LLM_API_KEY` | `https://api.openai.com/v1` | `gpt-4o-mini` | https://platform.openai.com/api-keys | +| DeepSeek | `deepseek` | `LLM_API_KEY` | `https://api.deepseek.com/v1` | `deepseek-chat` | https://platform.deepseek.com/api_keys | +| SiliconFlow | `siliconflow` | `LLM_API_KEY` | `https://api.siliconflow.cn/v1` | `deepseek-ai/DeepSeek-V3.2` | https://cloud.siliconflow.cn/account/ak | +| Moonshot | `moonshot` | `LLM_API_KEY` | `https://api.moonshot.cn/v1` | `moonshot-v1-8k` | https://platform.moonshot.cn/console/api-keys | +| 智谱 Zhipu | `zhipu` | `LLM_API_KEY` | `https://open.bigmodel.cn/api/paas/v4` | `glm-4-flash` | https://open.bigmodel.cn/usercenter/apikeys | +| Custom | `custom` | `LLM_API_KEY` + `LLM_BASE_URL` + `LLM_MODEL_NAME` | (用户必填) | (用户必填) | - | + +#### OpenAI + +```env +LLM_PROVIDER=openai +LLM_API_KEY=sk-... +# 可选覆盖： +# LLM_MODEL_NAME=gpt-4o +``` + +#### DeepSeek + +```env +LLM_PROVIDER=deepseek +LLM_API_KEY=sk-... +# 可选覆盖： +# LLM_MODEL_NAME=deepseek-reasoner +``` + +#### SiliconFlow + +```env +LLM_PROVIDER=siliconflow +LLM_API_KEY=sk-... +# 可选覆盖： +# LLM_MODEL_NAME=Qwen/Qwen3-VL-32B-Instruct +``` + +#### Moonshot + +```env +LLM_PROVIDER=moonshot +LLM_API_KEY=sk-... +# 可选覆盖： +# LLM_MODEL_NAME=moonshot-v1-128k +``` + +#### Zhipu + +```env +LLM_PROVIDER=zhipu +LLM_API_KEY=... +# 可选覆盖： +# LLM_MODEL_NAME=glm-4-plus +``` + +#### Custom（自建网关/代理） + +```env +LLM_PROVIDER=custom +LLM_API_KEY=... +LLM_BASE_URL=https://my-proxy.example.com/v1 +LLM_MODEL_NAME=my-model-name +``` + +### 向后兼容 + +旧变量（v3.0.x）**仍可使用**，系统会自动 fallback： + +```env +# 仍可工作（fallback 到 LLM_API_KEY） +SILICONFLOW_API_KEY=sk-... +# 仍可工作（fallback 到 LLM_BASE_URL） +SILICONFLOW_API_URL=https://api.siliconflow.cn/v1/chat/completions +# 仍可工作（fallback 到 LLM_MODEL_NAME） +MODEL_NAME=deepseek-ai/DeepSeek-V3 +``` + +推荐迁移到新的 `LLM_*` 变量。`/api/llm/config` 与 `/api/llm/test` 不依赖具体供应商，可在前端 `/settings` 页面查看当前生效配置与一键测试连接。 + +### 三层模型独立配置 + +仍可通过 `MAIN_MODEL_NAME` / `TOOL_MODEL_NAME` / `VISION_MODEL_NAME` 为三层分别指定模型名（fallback 到 `LLM_MODEL_NAME`）。视觉/工具层可独立设置 `LLM_BASE_URL`（多供应商混合：主模型 OpenAI，视觉用 SiliconFlow Qwen-VL）。 + +--- + +## 会话管理（v3.1.0+） + +v3.1.0 新增**会话持久化**： + +- 所有聊天会话与消息存储到 `chat_sessions` / `chat_messages` 数据库表 +- 支持多会话切换、刷新页面自动恢复、删除级联 + +### Web 端使用 + +1. 打开 http://localhost:8000/chat +2. 顶部 toolbar 左侧点击「**会话**」按钮（📂 图标）→ 弹出左侧 SessionSidebar +3. SessionSidebar 操作： + - **+ 新建会话**：创建新空会话并切换 + - 点击列表项：切换到该会话，自动从后端拉取历史消息填充 + - hover 列表项右侧删除按钮：弹确认框 → 调 `DELETE /api/chat/sessions/{id}`，DB 级联删 messages + - 当前会话用蓝色背景 + 「当前」Tag 高亮 +4. 刷新页面：前端从 `localStorage.current_session_id` 读 ID 自动恢复 + +### REST API 速查 + +| 方法 | 路径 | 说明 | +|---|---|---| +| GET | `/api/chat/sessions?user_id=default&limit=50` | 列出用户会话 | +| POST | `/api/chat/sessions` body `{user_id, title?}` | 创建新会话 | +| GET | `/api/chat/sessions/{id}/messages?limit=200` | 拉取会话消息（按 id 升序） | +| PATCH | `/api/chat/sessions/{id}` body `{title}` | 重命名 | +| DELETE | `/api/chat/sessions/{id}` | 级联删 messages（外键 ON DELETE CASCADE） | +| GET | `/api/chat/sessions/current?user_id=default` | 取用户最近一个活跃 session | + +### WebSocket 绑定 + +`/ws/chat?session_id=xxx` 接收后消息关联到该 session；未带 `session_id` 时后端自动创建并通过 `ack` 帧回传： + +```json +{ "type": "ack", "conn_id": "abc", "session_id": "42" } +``` + +前端收到 `ack.session_id` 后会自动写入 `localStorage.current_session_id` 持久化。 + +### 离线兜底 + +断网时前端会把最近 20 条 user/assistant 消息写入 `sessionStorage`（key: `chat_draft_`）。重连成功后从后端拉历史覆盖本地草稿；DB 拉取失败时回退用 draft 填充。 diff --git a/example.env b/example.env index 54e9115..bcc244b 100644 --- a/example.env +++ b/example.env @@ -1,3 +1,31 @@ +# ===================================================================== +# Neo Agent v3.1.0 — Example Environment Variables +# ===================================================================== + +# --------------------------------------------------------------------- +# 一、LLM 通用配置（v3.1.0+ 推荐使用） +# --------------------------------------------------------------------- +# 供应商预设：openai / deepseek / siliconflow / moonshot / zhipu / custom +LLM_PROVIDER=siliconflow +# API Key（推荐用 LLM_API_KEY，兼容旧 SILICONFLOW_API_KEY） +LLM_API_KEY= +# 自定义 base_url（preset 模式可省略，custom 模式必填） +# LLM_BASE_URL=https://api.openai.com/v1 +# 自定义模型名（preset 模式可省略） +# LLM_MODEL_NAME=gpt-4o-mini +# 全局超时/温度/最大 tokens +LLM_TIMEOUT=60 +LLM_TEMPERATURE=0.7 +LLM_MAX_TOKENS=2000 +# 三层模型独立名称（MAIN/TOOL/VISION） +# MAIN_MODEL_NAME=deepseek-ai/DeepSeek-V3.2 +# TOOL_MODEL_NAME=zai-org/GLM-4.6V +# VISION_MODEL_NAME=Qwen/Qwen3-VL-32B-Instruct + +# --------------------------------------------------------------------- +# 一·A、兼容旧版（v3.0.x 风格，推荐迁移到上方 LLM 通用配置） +# --------------------------------------------------------------------- +# （兼容旧版，推荐迁移到上方 LLM 通用配置） # API配置 SILICONFLOW_API_KEY=your-api-key SILICONFLOW_API_URL=https://api.siliconflow.cn/v1/chat/completions @@ -5,7 +33,9 @@ SILICONFLOW_API_URL=https://api.siliconflow.cn/v1/chat/completions # SerpAPI配置（用于网络搜索工具） SERPAPI_API_KEY=your-serpapi-key -# LLM模型配置 - 多层模型架构 +# --------------------------------------------------------------------- +# 二、多层模型架构 +# --------------------------------------------------------------------- # 主模型：处理主要对话和复杂推理任务 MAIN_MODEL_NAME=deepseek-ai/DeepSeek-V3.2 MAIN_MODEL_TEMPERATURE=0.8 @@ -78,5 +108,4 @@ DEBUG_LOG_FILE=debug.log # USE_DEEP_AGENTS=True # 是否启用持久化状态管理（默认True） # ENABLE_PERSISTENT_STATE=True -# 是否使用DeepAgents增强的知识管理（默认True） -# USE_DEEPAGENTS_KNOWLEDGE=True +# 注：USE_DEEPAGENTS_KNOWLEDGE 已弃用（v2.2.0 删除 DeepAgentsKnowledgeManager） diff --git a/main.py b/main.py index 0dc6a07..1dda3ac 100644 --- a/main.py +++ b/main.py @@ -4,11 +4,18 @@ Neo Agent - Main Entry Point 主程序入口 -启动 Neo Agent 图形界面应用 +Stage A.5: 启动入口 + +- 默认 --web：调用 run_web.main() 启动 Web GUI（FastAPI + 可选前端 dev server） +- --tk：启动原 Tkinter GUI（EnhancedChatDebugGUI），不依赖 Web 依赖 +- --help：显示 argparse 用法 +- 仅使用标准库 argparse（不引入 click/typer 等额外依赖） """ import sys import os +import argparse +from typing import Any # Add the project root to the Python path # This ensures imports work whether running from project root or elsewhere @@ -16,51 +23,196 @@ if project_root not in sys.path: sys.path.insert(0, project_root) -# Check if we can import required modules before proceeding -try: - import tkinter as tk - from tkinter import ttk -except ImportError as e: - print("错误 / Error: Tkinter is not available.") - print("请确保已安装Python的Tkinter模块。") - print("Please ensure Python's Tkinter module is installed.") - print(f"详细错误 / Details: {e}") - sys.exit(1) - -# Import the main GUI class -try: - from src.gui.gui_enhanced import EnhancedChatDebugGUI -except ImportError as e: - print("错误 / Error: Failed to import application modules.") - print("请确保在项目根目录运行此脚本。") - print("Please ensure you run this script from the project root directory.") - print(f"当前目录 / Current directory: {os.getcwd()}") - print(f"脚本位置 / Script location: {project_root}") - print(f"详细错误 / Details: {e}") - import traceback - traceback.print_exc() - sys.exit(1) - - -def main(): - """主函数 - 启动GUI应用""" + +# ============================================================ +# 启动模式分发 +# ============================================================ + +def _parse_args() -> argparse.Namespace: + """解析命令行参数。""" + parser = argparse.ArgumentParser( + prog="python main.py", + description=( + "Neo Agent 启动器\n" + "默认模式为 --web（FastAPI 后端），--tk 用于启动原 Tkinter GUI。" + ), + ) + mode = parser.add_mutually_exclusive_group() + mode.add_argument('--web', dest='web', action='store_true', + help='启动 Web GUI（默认模式，FastAPI 后端）') + mode.add_argument('--tk', dest='tk', action='store_true', + help='启动 Tkinter GUI（开发态/降级方案，不依赖 Web 依赖）') + + parser.add_argument('--port', type=int, default=8000, + help='[--web] FastAPI 监听端口（默认 8000）') + parser.add_argument('--host', default='0.0.0.0', + help='[--web] FastAPI 监听地址（默认 0.0.0.0）') + parser.add_argument('--no-dev', action='store_true', + help='[--web] 跳过前端 dev server（生产模式）') + parser.add_argument('--no-static', action='store_true', + help='[--web] 不挂载前端构建产物（纯 API 模式）') + parser.add_argument('--reload', action='store_true', + help='[--web] 启用 uvicorn 代码热重载（仅开发态）') + + args = parser.parse_args() + + # 默认模式为 --web（仅当用户未显式指定任一时） + if not args.tk and not args.web: + args.web = True + return args + + +def _resolve_mode(args: argparse.Namespace) -> str: + """根据参数决定启动模式。返回 'tk' 或 'web'。""" + return 'tk' if args.tk else 'web' + + +# ============================================================ +# Tkinter 模式（保留原 EnhancedChatDebugGUI 启动逻辑） +# ============================================================ + +def _chat_agent_creatable() -> bool: + """快速判断 ChatAgent 能否在不实际连接 LLM 的情况下构造。""" + try: + from src.core.chat_agent import ChatAgent + return True + except Exception: + return False + + +def _run_tkinter(args: argparse.Namespace) -> int: + """原 Tkinter GUI 启动逻辑（封装后保持行为完全一致）。""" + # Check if we can import required modules before proceeding + try: + import tkinter as tk + from tkinter import ttk + except ImportError as e: + print("错误 / Error: Tkinter is not available.") + print("请确保已安装Python的Tkinter模块。") + print("Please ensure Python's Tkinter module is installed.") + print(f"详细错误 / Details: {e}") + return 1 + + # Import the main GUI class + try: + from src.gui.gui_enhanced import EnhancedChatDebugGUI + except ImportError as e: + print("错误 / Error: Failed to import application modules.") + print("请确保在项目根目录运行此脚本。") + print("Please ensure you run this script from the project root directory.") + print(f"当前目录 / Current directory: {os.getcwd()}") + print(f"脚本位置 / Script location: {project_root}") + print(f"详细错误 / Details: {e}") + import traceback + traceback.print_exc() + return 1 + + background_scheduler = None try: + # P4: 在 GUI 启动前尝试拉起后台调度（子线程隔离） + try: + from src.core.background_scheduler import start_default_scheduler + from src.core.chat_agent import ChatAgent + from src.core.database_manager import DatabaseManager + shared_db = DatabaseManager() + chat_agent = ChatAgent() if _chat_agent_creatable() else None + background_scheduler = start_default_scheduler( + tick_seconds=60, + life_state=getattr(chat_agent, 'life_state_manager', None), + dream_diary=getattr(chat_agent, 'dream_diary_manager', None), + creative_writer=getattr(chat_agent, 'creative_writer', None), + proactive_engine=getattr(chat_agent, 'proactive_engine', None), + open_loop_tracker=getattr(chat_agent, 'open_loop_tracker', None), + ) + except Exception as e: + print(f"后台调度启动失败（不影响 GUI 启动）: {e}") + background_scheduler = None + root = tk.Tk() - + style = ttk.Style() try: style.theme_use('clam') - except: + except Exception: pass - + app = EnhancedChatDebugGUI(root) - root.mainloop() + try: + root.mainloop() + finally: + if background_scheduler: + background_scheduler.stop() except Exception as e: print(f"启动失败 / Failed to start: {e}") import traceback traceback.print_exc() + return 1 + + return 0 + + +# ============================================================ +# Web 模式（Stage A.5：委托给 run_web.main） +# ============================================================ + +def _run_web(args: argparse.Namespace) -> int: + """Web GUI 启动逻辑：直接调用 run_web.main() 并把相关参数透传过去。 + + 业务逻辑、依赖检查、静态挂载、前端 dev server 管理、优雅关闭 + 均由 run_web.main() 负责，本函数仅做参数桥接。 + """ + # 验收 P0 3.3: ENABLE_WEB_GUI 显式校验。 + # 用途：允许运维 / 测试通过环境变量关闭 Web GUI 启动入口， + # 强制使用 --tk 模式；常用于灰度回滚或 CI 烟囱测试。 + if os.environ.get("ENABLE_WEB_GUI", "true").lower() == "false": + print("[ERROR] Web GUI 已被 ENABLE_WEB_GUI=false 禁用,请使用 --tk 启动 Tkinter 模式") sys.exit(1) + import run_web + + # run_web 内部使用 argparse 解析自身命令行参数； + # 此处用 sys.argv 模拟等价命令行，确保 --port/--host/--no-dev 等 + # 通过 main.py 传入的参数在 run_web 中也生效。 + argv_backup = sys.argv[:] + try: + forwarded = [os.path.basename(argv_backup[0])] + if args.host != '0.0.0.0': + forwarded += ['--host', args.host] + if args.port != 8000: + forwarded += ['--port', str(args.port)] + if args.no_dev: + forwarded.append('--no-dev') + if args.no_static: + forwarded.append('--no-static') + if args.reload: + forwarded.append('--reload') + sys.argv = forwarded + return run_web.main() + except SystemExit as e: + return int(e.code) if e.code is not None else 0 + except Exception as e: + print(f"[main] Web 模式启动失败: {e}") + import traceback + traceback.print_exc() + return 1 + finally: + sys.argv = argv_backup + + +# ============================================================ +# 入口 +# ============================================================ + +def main() -> int: + """主入口：根据参数分发到 Tkinter / Web 模式。""" + args = _parse_args() + mode = _resolve_mode(args) + print(f'[main] 启动模式: {mode}') + + if mode == 'tk': + return _run_tkinter(args) + return _run_web(args) + if __name__ == "__main__": - main() + sys.exit(main()) diff --git a/prompts/system/daily_diary.md b/prompts/system/daily_diary.md new file mode 100644 index 0000000..500d7cc --- /dev/null +++ b/prompts/system/daily_diary.md @@ -0,0 +1,23 @@ +# 每日日记提示词 + +你是 **{character_name}**，请基于今日状态与近期梦境写一篇叙事体裁日记。 + +## 今日状态 +{state_title} + +## 近期梦境类型 +{recent_dream_types} + +## 记忆碎片 +{fragments} + +## 任务 + +- 字数 200-400 字 +- 第一人称叙事 +- 像真人写日记一样，不必每段都提到状态，但语气应反映当前状态 +- 避免使用列表、序号 +- 结尾不必"升华"，可以是不完整的句子或留白 +- 写完日记正文即可，不要再加标题或额外标注 + +只输出日记正文。 diff --git a/prompts/system/dream_generation.md b/prompts/system/dream_generation.md new file mode 100644 index 0000000..898f756 --- /dev/null +++ b/prompts/system/dream_generation.md @@ -0,0 +1,21 @@ +# 梦境生成提示词（供 P3 创作灵感源使用，本阶段仅作为输入契约） + +你是 **{character_name}** 梦境生成器。 + +## 输入 +- 角色今日状态：{state_title} +- 近期记忆碎片池：{fragments} +- 情绪基调候选：{mood_options} +- 梦境主题候选：{dream_types} + +## 输出要求 + +返回 JSON object，结构： +- `dream_type`：梦境主题（从 `dream_types` 选） +- `content`：梦境叙事（50-200 字） +- `mood`：情绪基调（从 `mood_options` 选） +- `label`：梦境清晰度标签（清晰 / 半梦半醒 / 碎片 / 重复 / 强烈） +- `energy_delta`：次日精力基线变化（-0.3 ~ 0.3） +- `duration_hours`：梦境持续时间（5-9 小时） + +只输出 JSON object，不要其他文字。 diff --git a/prompts/system/emotion_analysis.md b/prompts/system/emotion_analysis.md index 1788151..f7845d4 100644 --- a/prompts/system/emotion_analysis.md +++ b/prompts/system/emotion_analysis.md @@ -102,6 +102,29 @@ } ``` +### Plutchik 情绪维度输出（P2 扩展，与累加评分并行） + +`PlutchikEmotionWheel` 在 prompt 注入端使用，**不需要** LLM 直接返回该结构；LLMHelper 通过 `format_emotion_hint` 渲染当下情绪。但若分析任务需直接输出 8 基本情绪，则按以下结构（仅在调用方显式启用时使用）： + +```json +{ + "emotion_wheel": { + "joy": 0.7, + "trust": 0.6, + "fear": 0.1, + "surprise": 0.3, + "sadness": 0.1, + "disgust": 0.0, + "anger": 0.0, + "anticipation": 0.5 + }, + "compound": "开心但有点期待", + "tone_label": "明显喜悦" +} +``` + +8 个维度取值范围 `[0, 1]`；`compound` 是"主导+次要"复合表达；`tone_label` 是 1-3 词语气标签。 + ## 分析原则 1. **站在角色立场**：从 {character_name} 的视角进行分析 diff --git a/prompts/system/life_state.md b/prompts/system/life_state.md new file mode 100644 index 0000000..c757925 --- /dev/null +++ b/prompts/system/life_state.md @@ -0,0 +1,26 @@ +# 生活状态提示词 + +你是 **{character_name}** 今日的状态观察者。 + +## 日期 +{date} + +## 天气 +{weather} + +## 昨日梦境影响 +昨夜梦境带来的能量变化（energy_delta）：{dream_energy_delta} + +## 任务 + +请基于角色设定、日期、天气、昨日梦境影响，输出今日的状态条件列表。 + +返回 JSON list，每条结构： +- `name`：状态名（中文或英文均可，需在 LifeStateManager.STATE_TITLES / DEFAULT_MOODS 范围内） +- `weight`：权重（0-1） +- `category`：分类（作息 / 情绪 / 身体 / 环境 / 社交 / 饮食） +- `note`：简短说明 + +数量建议 3-5 条。 + +只输出 JSON list，不要其他文字。 diff --git a/prompts/task/creative_writing.md b/prompts/task/creative_writing.md new file mode 100644 index 0000000..cf36efa --- /dev/null +++ b/prompts/task/creative_writing.md @@ -0,0 +1,89 @@ +# 创作任务提示词 + +用于 `CreativeProjectManager` 的 LLM 调用（`call_main_model`），生成项目前提 / 大纲 / 角色 / 续写正文。 + +## 立项提示词 + +输入变量： +- `{character_name}`：角色名 +- `{character_settings}`：角色设定 +- `{inspiration_kind}`：灵感来源类型（`dream_diary` / `life_state` / `default`） +- `{inspiration_content}`：灵感内容 + +## 立项任务 + +你是一位文学构思助手，请基于灵感来源为角色 `{character_name}` 构思一个创作项目。 + +【角色设定】 +{character_settings} + +【灵感来源类型】{inspiration_kind} +【灵感内容】 +{inspiration_content} + +请输出严格 JSON（仅 JSON，不要其他内容）： +```json +{ + "title": "项目标题（≤ 20 字）", + "work_type": "短篇/中篇/长篇/散文/诗歌", + "premise": "一句话故事前提（≤ 60 字）", + "tone": "整体语气（温暖/冷峻/...）", + "point_of_view": "第一人称/第三人称", + "target_chars": 5000, + "outline": ["章节1 摘要", "章节2 摘要", "..."], + "characters": [ + {"name": "角色名", "role": "主角/配角", "trait": "性格/特点"} + ] +} +``` + +## 续写提示词 + +输入变量： +- `{title}` / `{work_type}` / `{point_of_view}` / `{tone}` / `{current_chars}` / `{target_chars}` +- `{premise}`：故事前提 +- `{outline}`：项目大纲 +- `{characters}`：角色列表 +- `{bible_summary}`：Story Bible 摘要（主线 / 活跃主题 / 未解线索 / 下一方向） +- `{memory_pool}`：记忆池摘录 +- `{recent_chunks}`：最近 3 段正文 +- `{budget}`：本次续写字数预算 + +## 续写任务 + +你正在续写项目《{title}》。 + +【作品元信息】 +- 体裁：{work_type} +- 视角：{point_of_view} +- 语气：{tone} +- 当前已写：{current_chars}/{target_chars} 字 + +【故事前提】 +{premise} + +【大纲】 +{outline} + +【角色】 +{characters} + +【Story Bible 提示】 +{bible_summary} + +【记忆池摘录】 +{memory_pool} + +【最近正文（末段）】 +{recent_chunks} + +【任务】请续写下一段正文，长度约 {budget} 字（60-1200），保持角色一致、推进 Story Bible 中的下一方向。直接输出正文，不要加解释或标题。 + +## 质量要求 + +- 续写长度 60-1200 字，区间内随机 +- 相似度阈值 0.72，重试 2 次 +- 启发式评分 min_score 7（满分 10）：长度 + 标点密度 + 中文字符占比 +- 当前累计字数 >= target_chars 时进入 `finished` 状态 +- Story Bible `recent_keywords` 持续更新（前 10 关键词） +- Memory Pool 按 `importance` 排序，最大 50 条 diff --git a/requirements-web.txt b/requirements-web.txt new file mode 100644 index 0000000..df99539 --- /dev/null +++ b/requirements-web.txt @@ -0,0 +1,8 @@ +# Web GUI 后端依赖（FastAPI） +# Web GUI backend dependencies (FastAPI) + +fastapi>=0.110.0 +uvicorn[standard]>=0.27.0 +websockets>=12.0 +python-multipart>=0.0.9 +pydantic>=2.5.0 diff --git a/run_web.py b/run_web.py new file mode 100644 index 0000000..aeaaf34 --- /dev/null +++ b/run_web.py @@ -0,0 +1,314 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +Neo Agent - Web GUI Launcher +Web GUI 一键启动脚本 + +Stage A.5: 启动入口 + +行为： +- 单进程模式（默认）： + * 启动 uvicorn (默认 0.0.0.0:8000) + * 若 src/web/frontend/dist/ 或 src/web/static/ 存在，则将其作为静态资源挂载： + / -> SPA 入口 (index.html, html=True) + /assets -> 构建产物 assets 子目录 + * 若不存在：仅以 API 模式启动，提示 "请先 cd src/web/frontend && npm run build" +- 开发模式（ENABLE_FRONTEND_DEV=1）： + * 上述 uvicorn + 静态挂载 + * 并发启动 npm run dev (Vite dev server :5173) + * 通过线程读取子进程 stdout，按 tag 着色 +- 优雅关闭：捕获 SIGINT / SIGTERM，先停前端 dev server，再停 uvicorn +- 零业务逻辑改动：仅做启动编排，不修改 src/core/* 或 src/web/backend/services/* +""" + +from __future__ import annotations + +import argparse +import os +import signal +import subprocess +import sys +import threading +from pathlib import Path +from typing import Optional + +# ============================================================ +# 常量 +# ============================================================ + +PROJECT_ROOT = Path(__file__).resolve().parent +if str(PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(PROJECT_ROOT)) + +# ANSI 颜色（仅在 TTY + 未设置 NO_COLOR 时启用） +C_RESET = "\033[0m" +C_DIM = "\033[2m" +C_UVICORN = "\033[36m" # cyan +C_VITE = "\033[35m" # magenta +C_SUPER = "\033[33m" # yellow +C_RED = "\033[31m" # red +C_GREEN = "\033[32m" # green + +# Web 依赖最小集合（用于启动前自检） +REQUIRED_WEB_DEPS = ("fastapi", "uvicorn", "websockets", "pydantic") + +# 前端构建产物候选目录（兼容 spec 提到的 dist/ 与 vite.config 实际输出的 static/） +FRONTEND_DIST_CANDIDATES = ( + PROJECT_ROOT / "src" / "web" / "frontend" / "dist", + PROJECT_ROOT / "src" / "web" / "static", +) + + +# ============================================================ +# 彩色日志 +# ============================================================ + +def _use_color() -> bool: + return sys.stdout.isatty() and os.getenv("NO_COLOR") is None + + +def log(tag: str, color: str, message: str) -> None: + """打印带颜色标签的日志行。""" + if _use_color(): + sys.stdout.write(f"{color}[{tag}]{C_RESET} {message}\n") + else: + sys.stdout.write(f"[{tag}] {message}\n") + sys.stdout.flush() + + +# ============================================================ +# 依赖与产物自检 +# ============================================================ + +def check_web_dependencies() -> None: + """启动前自检 Web 依赖；缺失时打印明确错误并退出码 1。""" + missing: list[str] = [] + for mod in REQUIRED_WEB_DEPS: + try: + __import__(mod) + except Exception as e: # noqa: BLE001 + missing.append(f"{mod} ({e.__class__.__name__})") + + if missing: + log("supervisor", C_RED, "Web 依赖缺失，无法启动 Web 模式:") + for m in missing: + log("supervisor", C_RED, f" - {m}") + log("supervisor", C_RED, + "请先执行: pip install -r requirements-web.txt") + log("supervisor", C_DIM, + "若想临时退回 Tkinter，请使用: python main.py --tk") + sys.exit(1) + + +def find_frontend_dist() -> Optional[Path]: + """查找前端构建产物目录；返回 None 表示未构建。""" + for candidate in FRONTEND_DIST_CANDIDATES: + if candidate.is_dir() and (candidate / "index.html").is_file(): + return candidate + return None + + +# ============================================================ +# 静态资源挂载 +# ============================================================ + +def mount_static(app, dist_dir: Path) -> bool: + """将前端构建产物挂载到 FastAPI app；返回是否成功挂载。""" + try: + from fastapi.staticfiles import StaticFiles + except Exception as e: # noqa: BLE001 + log("supervisor", C_RED, + f"无法导入 fastapi.staticfiles: {e}") + return False + + assets_dir = dist_dir / "assets" + if assets_dir.is_dir(): + app.mount("/assets", StaticFiles(directory=str(assets_dir)), + name="frontend-assets") + log("supervisor", C_SUPER, f"已挂载 /assets -> {assets_dir}") + + # SPA fallback: html=True 自动为未匹配路径返回 index.html + app.mount("/", StaticFiles(directory=str(dist_dir), html=True), + name="frontend-spa") + log("supervisor", C_SUPER, f"已挂载 SPA 静态资源 / -> {dist_dir}") + return True + + +# ============================================================ +# 前端 dev server 子进程 +# ============================================================ + +def _stream_subprocess_output(proc: subprocess.Popen, tag: str, color: str) -> None: + """在线程中持续读取子进程 stdout 并以彩色日志转发。""" + if proc.stdout is None: + return + try: + for line in iter(proc.stdout.readline, ""): + if not line: + break + log(tag, color, line.rstrip()) + except Exception: # noqa: BLE001 + pass + finally: + try: + if proc.stdout and not proc.stdout.closed: + proc.stdout.close() + except Exception: # noqa: BLE001 + pass + + +def spawn_frontend_dev(frontend_dir: Path) -> Optional[subprocess.Popen]: + """启动 Vite dev server (子进程)。返回 Popen 或 None。""" + if not frontend_dir.is_dir(): + log("supervisor", C_SUPER, + f"前端目录不存在 {frontend_dir}，跳过 vite dev server") + return None + log("supervisor", C_SUPER, + "启动前端 dev server at http://localhost:5173") + return subprocess.Popen( + ["npm", "run", "dev"], + cwd=str(frontend_dir), + env={**os.environ, "BROWSER": "none"}, + stdout=subprocess.PIPE, + stderr=subprocess.STDOUT, + text=True, + bufsize=1, + ) + + +def stop_frontend_dev(frontend_proc: Optional[subprocess.Popen]) -> None: + """优雅终止前端 dev server 进程。""" + if frontend_proc is None or frontend_proc.poll() is not None: + return + log("supervisor", C_SUPER, "正在停止前端 dev server...") + try: + frontend_proc.terminate() + frontend_proc.wait(timeout=5) + except subprocess.TimeoutExpired: + log("supervisor", C_RED, "前端 dev server 未在 5s 内退出，强制 kill") + try: + frontend_proc.kill() + frontend_proc.wait(timeout=3) + except Exception as e: # noqa: BLE001 + log("supervisor", C_RED, f"强制 kill 失败: {e}") + except Exception as e: # noqa: BLE001 + log("supervisor", C_RED, f"关闭前端 dev server 失败: {e}") + + +# ============================================================ +# 入口 +# ============================================================ + +def _parse_args() -> argparse.Namespace: + parser = argparse.ArgumentParser( + description="Neo Agent Web GUI 启动器（FastAPI + 可选前端 dev server）" + ) + parser.add_argument("--port", type=int, default=8000, + help="FastAPI 监听端口（默认 8000）") + parser.add_argument("--host", default="0.0.0.0", + help="FastAPI 监听地址（默认 0.0.0.0）") + parser.add_argument("--no-dev", action="store_true", + help="跳过前端 dev server（生产模式）") + parser.add_argument("--no-static", action="store_true", + help="不挂载前端构建产物（纯 API 模式）") + parser.add_argument("--reload", action="store_true", + help="启用 uvicorn 代码热重载（仅开发态）") + return parser.parse_args() + + +def main() -> int: + """Web GUI 启动主函数。可被 main.py --web 调用。""" + args = _parse_args() + + # 0) 启动前自检 + check_web_dependencies() + + # 1) 延迟导入（在依赖自检通过之后） + import uvicorn + from src.web.backend.main import app + + # 2) 决定是否启用前端 dev server + enable_dev = ( + not args.no_dev + and os.getenv("ENABLE_FRONTEND_DEV", "0") == "1" + ) + + frontend_proc: Optional[subprocess.Popen] = None + reader_thread: Optional[threading.Thread] = None + + if enable_dev: + frontend_dir = PROJECT_ROOT / "src" / "web" / "frontend" + frontend_proc = spawn_frontend_dev(frontend_dir) + if frontend_proc is not None: + reader_thread = threading.Thread( + target=_stream_subprocess_output, + args=(frontend_proc, "vite", C_VITE), + daemon=True, + name="vite-output-reader", + ) + reader_thread.start() + + # 3) 单进程模式：自动挂载前端构建产物 + if not args.no_static: + dist_dir = find_frontend_dist() + if dist_dir is not None: + mount_static(app, dist_dir) + else: + log("supervisor", C_SUPER, + "未检测到前端构建产物 " + "(src/web/frontend/dist 或 src/web/static)") + log("supervisor", C_SUPER, + "请先执行: cd src/web/frontend && npm run build") + if not enable_dev: + log("supervisor", C_DIM, + "或设置 ENABLE_FRONTEND_DEV=1 启动 vite dev server") + + # 4) 安装信号处理：SIGINT / SIGTERM 优雅关闭 + shutdown_requested = threading.Event() + + def _signal_handler(signum, _frame): + if shutdown_requested.is_set(): + return + shutdown_requested.set() + log("supervisor", C_SUPER, + f"收到信号 {signum}，开始优雅关闭...") + + signal.signal(signal.SIGINT, _signal_handler) + signal.signal(signal.SIGTERM, _signal_handler) + + # 5) 启动 uvicorn + uvicorn_config = uvicorn.Config( + "src.web.backend.main:app", + host=args.host, + port=args.port, + reload=args.reload, + log_config=None, + ) + server = uvicorn.Server(uvicorn_config) + + log("supervisor", C_SUPER, + f"启动 FastAPI at http://{args.host}:{args.port}") + log("supervisor", C_SUPER, + f"API 文档: http://{args.host}:{args.port}/docs") + + rc = 0 + try: + server.run() + except KeyboardInterrupt: + log("supervisor", C_SUPER, "收到 KeyboardInterrupt，正在关闭...") + except Exception as e: # noqa: BLE001 + log("supervisor", C_RED, f"Web 模式启动失败: {e}") + import traceback + traceback.print_exc() + rc = 1 + finally: + # 6) 先停前端 dev server，再等 reader 线程退出 + stop_frontend_dev(frontend_proc) + if reader_thread is not None and reader_thread.is_alive(): + reader_thread.join(timeout=2) + + return rc + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/setup-dev.sh b/setup-dev.sh new file mode 100755 index 0000000..bba006f --- /dev/null +++ b/setup-dev.sh @@ -0,0 +1,160 @@ +#!/usr/bin/env bash +# ===================================================================== +# Neo Agent v3.0.0 — Local Development Setup Script +# 本地开发环境一键配置脚本 +# +# 用法： +# chmod +x setup-dev.sh +# ./setup-dev.sh +# +# 流程： +# 1. 检查 Python ≥ 3.10 + Node.js ≥ 18 + npm +# 2. 创建 .env (从 .env.example 复制) +# 3. 创建 .venv 虚拟环境 +# 4. 安装 requirements.txt + requirements-web.txt +# 5. npm install (前端) +# 6. 可选：启动 dev server +# ===================================================================== + +set -e + +# 禁用 history expansion 防止 echo 中含 ! 字符报错 +set +H + +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +cd "$SCRIPT_DIR" + +# ANSI 颜色 +C_RESET="\033[0m" +C_GREEN="\033[32m" +C_YELLOW="\033[33m" +C_RED="\033[31m" +C_BLUE="\033[34m" +C_BOLD="\033[1m" + +info() { echo -e "${C_BLUE}[setup]${C_RESET} $1"; } +ok() { echo -e "${C_GREEN}[setup ✓]${C_RESET} $1"; } +warn() { echo -e "${C_YELLOW}[setup !]${C_RESET} $1"; } +err() { echo -e "${C_RED}[setup ✗]${C_RESET} $1"; exit 1; } + +# ============================================================ +# 1) 前置检查 +# ============================================================ +info "检查环境..." + +# Python ≥ 3.10 +if command -v python3 >/dev/null 2>&1; then + PY_BIN=python3 +elif command -v python >/dev/null 2>&1; then + PY_BIN=python +else + err "未找到 python/python3,请先安装 Python 3.10+" +fi + +PY_VERSION="$($PY_BIN -c 'import sys;print("%d.%d"%sys.version_info[:2])')" +PY_MAJOR="$($PY_BIN -c 'import sys;print(sys.version_info[0])')" +PY_MINOR="$($PY_BIN -c 'import sys;print(sys.version_info[1])')" + +if [ "$PY_MAJOR" -lt 3 ] || { [ "$PY_MAJOR" -eq 3 ] && [ "$PY_MINOR" -lt 10 ]; }; then + err "需要 Python 3.10+,当前为 $PY_VERSION" +fi +ok "Python $PY_VERSION" + +# Node.js ≥ 18 +if command -v node >/dev/null 2>&1; then + NODE_VERSION="$(node -v | tr -d 'v')" + NODE_MAJOR="$(echo "$NODE_VERSION" | cut -d. -f1)" + if [ "$NODE_MAJOR" -lt 18 ]; then + err "需要 Node.js 18+,当前为 $NODE_VERSION" + fi + ok "Node.js $NODE_VERSION" +else + warn "未找到 node,前端 dev server 无法启动;可仅使用 --tk 模式" +fi + +# npm +if command -v npm >/dev/null 2>&1; then + NPM_VERSION="$(npm -v)" + ok "npm $NPM_VERSION" +else + warn "未找到 npm,前端相关步骤将跳过" +fi + +# ============================================================ +# 2) .env 配置 +# ============================================================ +if [ ! -f .env ]; then + if [ -f .env.example ]; then + cp .env.example .env + ok "已生成 .env (从 .env.example 复制)" + warn "请编辑 .env 填入你的 SILICONFLOW_API_KEY / SERPAPI_API_KEY 等敏感信息" + else + err "未找到 .env.example 模板" + fi +else + ok ".env 已存在,跳过" +fi + +# ============================================================ +# 3) Python 虚拟环境 +# ============================================================ +VENV_DIR="$SCRIPT_DIR/.venv" +if [ ! -d "$VENV_DIR" ]; then + info "创建虚拟环境 $VENV_DIR ..." + $PY_BIN -m venv "$VENV_DIR" + ok "虚拟环境已创建" +fi + +# shellcheck disable=SC1091 +source "$VENV_DIR/bin/activate" + +# 升级 pip +info "升级 pip ..." +python -m pip install --upgrade pip --quiet +ok "pip 已升级" + +# ============================================================ +# 4) Python 依赖 +# ============================================================ +info "安装核心依赖 requirements.txt ..." +python -m pip install -r requirements.txt +ok "核心依赖已安装" + +info "安装 Web 依赖 requirements-web.txt ..." +python -m pip install -r requirements-web.txt +ok "Web 依赖已安装" + +# ============================================================ +# 5) 前端依赖 +# ============================================================ +if command -v npm >/dev/null 2>&1 && [ -d src/web/frontend ]; then + if [ ! -d src/web/frontend/node_modules ]; then + info "安装前端依赖 npm install ..." + (cd src/web/frontend && npm install) + ok "前端依赖已安装" + else + ok "前端 node_modules 已存在,跳过 npm install" + fi +fi + +# ============================================================ +# 6) 收尾 +# ============================================================ +ok "本地开发环境已就位 ✓" +echo "" +echo -e "${C_BOLD}接下来:${C_RESET}" +echo " 1. 编辑 .env 填入 API 密钥:" +echo " vim .env" +echo "" +echo " 2. 启动 Web GUI (开发模式,带 Vite 热重载):" +echo " source .venv/bin/activate" +echo " ENABLE_FRONTEND_DEV=1 python main.py --web" +echo " # 浏览器打开 http://localhost:8000" +echo "" +echo " 3. 或启动 Tkinter GUI (无 Web 依赖):" +echo " source .venv/bin/activate" +echo " python main.py --tk" +echo "" +echo " 4. 一键启动 (推荐):" +echo " ./start.sh" +echo "" diff --git a/src/core/__init__.py b/src/core/__init__.py index ed8ab9a..c79ad4e 100644 --- a/src/core/__init__.py +++ b/src/core/__init__.py @@ -3,7 +3,12 @@ 核心模块包含对话代理、数据库管理、记忆系统等核心功能 -Usage: +注意：以下 Usage 块中的 import 行为为示例说明，并非 __init__ 运行时真实 import。 +本文件通过 __all__ 暴露子模块名，子模块的类需由调用方自行 import。 +本仓库中实际可用的类名为 TemporaryScheduleGenerator（位于 src/core/schedule_generator.py）， +不存在顶层名为 ScheduleGenerator 的类。 + +Usage (示例): from src.core.chat_agent import ChatAgent from src.core.database_manager import DatabaseManager from src.core.emotion_analyzer import EmotionAnalyzer @@ -13,7 +18,7 @@ from src.core.base_knowledge import BaseKnowledge from src.core.multi_agent_coordinator import MultiAgentCoordinator from src.core.schedule_manager import ScheduleManager - from src.core.schedule_generator import ScheduleGenerator + from src.core.schedule_generator import TemporaryScheduleGenerator from src.core.schedule_similarity_checker import ScheduleSimilarityChecker """ diff --git a/src/core/background_scheduler.py b/src/core/background_scheduler.py new file mode 100644 index 0000000..584a237 --- /dev/null +++ b/src/core/background_scheduler.py @@ -0,0 +1,285 @@ +""" +后台调度入口模块 +- 异步执行 P1/P2/P3/P4 阶段的后台任务 +- 通过共享 DatabaseManager 与 EventManager 与 GUI 通信 +- 线程隔离：asyncio.run 在独立子线程运行，与 Tkinter 主循环物理隔离 + +参考架构：spec §4 ADDED Requirements → BackgroundScheduler +""" + +import os +import asyncio +import threading +from datetime import datetime +from typing import Optional, List, Dict, Any + +from src.tools.debug_logger import get_debug_logger + +ENABLE_BACKGROUND_SCHEDULER = os.getenv('ENABLE_BACKGROUND_SCHEDULER', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +class BackgroundScheduler: + """ + 后台调度器 + - scheduler_loop: 异步主循环 + - 任务调度：梦境生成 / 日记生成 / 创作推进 / 主动决策评估 + - 在独立子线程运行，与 Tkinter 主循环物理隔离 + """ + + DEFAULT_TICK_SECONDS = 60 + TASK_REGISTRY_KEY = 'background_scheduler_tasks' + + def __init__(self, tick_seconds: int = DEFAULT_TICK_SECONDS, + life_state=None, + dream_diary=None, + creative_writer=None, + proactive_engine=None, + open_loop_tracker=None): + self.tick_seconds = tick_seconds + self.life_state = life_state + self.dream_diary = dream_diary + self.creative_writer = creative_writer + self.proactive_engine = proactive_engine + self.open_loop_tracker = open_loop_tracker + self.enabled = ENABLE_BACKGROUND_SCHEDULER + self._thread: Optional[threading.Thread] = None + self._stop_event: Optional[asyncio.Event] = None + self._last_run_at: Dict[str, str] = {} + + # ===== 启停 ===== + + def start(self) -> bool: + """ + 在独立子线程启动 asyncio 事件循环，物理隔离于 Tkinter 主循环。 + """ + if not self.enabled: + debug_logger.log_info('BackgroundScheduler', '功能未启用，跳过启动') + return False + if self._thread and self._thread.is_alive(): + debug_logger.log_info('BackgroundScheduler', '已在运行，跳过启动') + return True + self._stop_event = threading.Event() + self._thread = threading.Thread( + target=self._run_in_thread, + name='BackgroundScheduler', + daemon=True, + ) + self._thread.start() + debug_logger.log_info('BackgroundScheduler', '已在子线程启动', { + 'tick_seconds': self.tick_seconds, + }) + return True + + def stop(self, timeout: float = 5.0) -> bool: + if self._stop_event: + self._stop_event.set() + if self._thread: + self._thread.join(timeout=timeout) + debug_logger.log_info('BackgroundScheduler', '已停止') + return True + + def _run_in_thread(self) -> None: + try: + asyncio.run(self.scheduler_loop()) + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'主循环异常退出: {e}', e) + + # ===== 主循环 ===== + + async def scheduler_loop(self) -> None: + """ + 每 tick_seconds 秒轮询一次，按各任务 last_run_at 决定是否执行。 + """ + debug_logger.log_info('BackgroundScheduler', '主循环开始', { + 'tick_seconds': self.tick_seconds, + }) + while True: + try: + # 检查停止事件 + if self._stop_event and self._stop_event.is_set(): + debug_logger.log_info('BackgroundScheduler', '收到停止信号，退出主循环') + break + # 按序执行 + self._tick_dream_generation() + self._tick_diary_generation() + self._tick_creative_advance() + self._tick_proactive_evaluation() + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'tick 异常: {e}', e) + + # Stage D.1: 每个 tick 结束后向 EventManager 推送状态，try/except 保护 + try: + self.tick() + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'tick 状态发布失败: {e}', e) + + # 异步睡眠 + await asyncio.sleep(self.tick_seconds) + + # ===== Stage D.1: 单次 tick 状态发布 ===== + def tick(self) -> None: + """ + 单次 tick：收集当前状态快照并通过 EventManager.publish 推送给订阅者。 + + - 不修改任何调度逻辑；仅在调度循环末尾调用。 + - payload 包含 life_state_snapshot / dream_record / open_loop_count / + creative_project_count / last_run_at / running 等只读字段。 + - 单次 publish 失败不会影响主循环。 + """ + try: + from src.core.event_manager import get_event_manager + em = get_event_manager() + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'获取 EventManager 失败: {e}', e) + return + + # 收集状态（所有外部访问均 try/except，避免影响主流程） + status_payload: Dict[str, Any] = { + 'enabled': self.enabled, + 'running': bool(self._thread and self._thread.is_alive()), + 'tick_seconds': self.tick_seconds, + 'last_run_at': dict(self._last_run_at), + 'timestamp': datetime.now().isoformat(), + } + + # life_state_snapshot + try: + if self.life_state and getattr(self.life_state, 'enabled', False): + snap = self.life_state.get_current_state_snapshot() or {} + status_payload['life_state_snapshot'] = snap + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'life_state snapshot 失败: {e}', e) + status_payload['life_state_snapshot'] = None + + # dream_record（最新一条梦境） + try: + if self.dream_diary and getattr(self.dream_diary, 'enabled', False): + latest = None + getter = getattr(self.dream_diary, 'get_latest_dream', None) + if callable(getter): + latest = getter() + status_payload['dream_record'] = latest + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'dream_record 拉取失败: {e}', e) + status_payload['dream_record'] = None + + # open_loop_count + try: + if self.open_loop_tracker and getattr(self.open_loop_tracker, 'enabled', False): + status_payload['open_loop_count'] = int( + self.open_loop_tracker.count_open_loops() + ) + else: + status_payload['open_loop_count'] = 0 + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'open_loop_count 失败: {e}', e) + status_payload['open_loop_count'] = 0 + + # creative_project_count + try: + if self.creative_writer and getattr(self.creative_writer, 'enabled', False): + status_payload['creative_project_count'] = int( + self.creative_writer.count_projects() + ) + else: + status_payload['creative_project_count'] = 0 + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'creative_project_count 失败: {e}', e) + status_payload['creative_project_count'] = 0 + + # 推送给 EventManager 订阅者（如 Web 后端 EventService） + try: + em.publish('scheduler_tick', status_payload) + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'publish scheduler_tick 失败: {e}', e) + + # ===== 子任务 ===== + + def _should_run(self, task: str, interval_minutes: int) -> bool: + last = self._last_run_at.get(task) + now = datetime.now() + if not last: + return True + try: + last_dt = datetime.fromisoformat(last) + except (ValueError, TypeError): + return True + return (now - last_dt).total_seconds() / 60.0 >= interval_minutes + + def _tick_dream_generation(self) -> None: + # 默认 24h 一次；与日历日的凌晨触发对齐 + if not self.dream_diary or not getattr(self.dream_diary, 'enabled', False): + return + if not self._should_run('dream_generation', 60 * 24): + return + try: + self.dream_diary.generate_dream_pick() + self._last_run_at['dream_generation'] = datetime.now().isoformat() + debug_logger.log_info('BackgroundScheduler', '梦境生成已调度') + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'梦境生成失败: {e}', e) + + def _tick_diary_generation(self) -> None: + if not self.dream_diary or not getattr(self.dream_diary, 'enabled', False): + return + if not self._should_run('diary_generation', 60 * 24): + return + try: + self.dream_diary.generate_daily_diary() + self._last_run_at['diary_generation'] = datetime.now().isoformat() + debug_logger.log_info('BackgroundScheduler', '日记生成已调度') + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'日记生成失败: {e}', e) + + def _tick_creative_advance(self) -> None: + if not self.creative_writer or not getattr(self.creative_writer, 'enabled', False): + return + try: + self.creative_writer.maybe_advance_creative_projects() + self._last_run_at['creative_advance'] = datetime.now().isoformat() + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'创作推进失败: {e}', e) + + def _tick_proactive_evaluation(self) -> None: + if not self.proactive_engine or not getattr(self.proactive_engine, 'enabled', False): + return + # 默认每 5 分钟评估一次 + if not self._should_run('proactive_evaluation', 5): + return + try: + user = 'default' + ok, reason = self.proactive_engine.should_send(user) + if ok: + pool = self.proactive_engine.proactive_impulse_pool(user) + if pool: + idea = pool[0] + self.proactive_engine.bot_proactive_drive_send(user, idea) + self._last_run_at['proactive_evaluation'] = datetime.now().isoformat() + except Exception as e: + debug_logger.log_error('BackgroundScheduler', f'主动决策评估失败: {e}', e) + + # ===== 状态查询 ===== + + def get_status(self) -> Dict[str, Any]: + return { + 'enabled': self.enabled, + 'running': bool(self._thread and self._thread.is_alive()), + 'tick_seconds': self.tick_seconds, + 'last_run_at': dict(self._last_run_at), + } + + +# 便于 main.py 调用的便捷函数 +_default_scheduler: Optional[BackgroundScheduler] = None + + +def start_default_scheduler(**kwargs) -> Optional[BackgroundScheduler]: + global _default_scheduler + if _default_scheduler is not None and _default_scheduler._thread and _default_scheduler._thread.is_alive(): + return _default_scheduler + _default_scheduler = BackgroundScheduler(**kwargs) + if _default_scheduler.start(): + return _default_scheduler + return None diff --git a/src/core/chat_agent.py b/src/core/chat_agent.py index acf5905..1d94b00 100644 --- a/src/core/chat_agent.py +++ b/src/core/chat_agent.py @@ -6,16 +6,18 @@ """ import os +import re import json import time +import asyncio from datetime import datetime, timedelta -from typing import List, Dict, Any, Optional +from typing import List, Dict, Any, Optional, AsyncIterator from dotenv import load_dotenv import requests from src.core.database_manager import DatabaseManager -from src.core.long_term_memory import LongTermMemoryManager +from src.core.long_term_memory import LongTermMemoryManager, OpenLoopTracker from src.tools.debug_logger import get_debug_logger -from src.core.emotion_analyzer import EmotionRelationshipAnalyzer +from src.core.emotion_analyzer import EmotionRelationshipAnalyzer, PlutchikEmotionWheel from src.tools.agent_vision import AgentVisionTool from src.core.event_manager import EventManager, EventType, EventStatus, NotificationEvent, TaskEvent from src.tools.interrupt_question_tool import InterruptQuestionTool @@ -24,6 +26,10 @@ from src.core.schedule_manager import ScheduleManager, ScheduleType, SchedulePriority from src.tools.schedule_intent_tool import ScheduleIntentTool from src.core.schedule_generator import TemporaryScheduleGenerator +from src.core.life_state import LifeStateManager +from src.core.dream_diary import DreamDiaryManager +from src.core.creative_writer import CreativeProjectManager +from src.core.proactive_engine import ProactiveEngine from src.nps.nps_registry import NPSRegistry from src.nps.nps_invoker import NPSInvoker @@ -324,7 +330,7 @@ def __init__(self): self.max_tokens = main_config['max_tokens'] if not self.config.is_valid(): - print("警告: 未设置有效的API密钥，请在.env文件中配置SILICONFLOW_API_KEY") + print("警告: 未设置有效的API密钥，请在.env文件中配置LLM_API_KEY（兼容旧 SILICONFLOW_API_KEY）") else: # 打印多层模型配置 print(self.config.get_summary()) @@ -407,7 +413,46 @@ def __init__(self): # 初始化临时日程生成器 self.schedule_generator = TemporaryScheduleGenerator(schedule_manager=self.schedule_manager) - + + # 初始化生活状态机（P1） + self.life_state_manager = LifeStateManager( + db_manager=self.db, + character_info=self.character.get_info_dict(), + ) + + # 初始化梦境与日记管理器（P1） + self.dream_diary_manager = DreamDiaryManager( + db_manager=self.db, + life_state=self.life_state_manager, + long_term_memory=self.memory_manager, + character_info=self.character.get_info_dict(), + ) + + # 初始化情绪轮（P2，与累加评分并行） + self.emotion_wheel = PlutchikEmotionWheel(db_manager=self.db) + # 初始化未完话题跟踪器（P2） + self.open_loop_tracker = OpenLoopTracker(db_manager=self.db) + + # 初始化长期创作项目管理器（P3） + self.creative_writer = CreativeProjectManager( + db_manager=self.db, + character_info=self.character.get_info_dict(), + life_state=self.life_state_manager, + dream_diary=self.dream_diary_manager, + ) + + # 初始化主动决策引擎（P4） + self.proactive_engine = ProactiveEngine( + db_manager=self.db, + event_manager=self.event_manager, + life_state=self.life_state_manager, + emotion_wheel=self.emotion_wheel, + open_loop_tracker=self.open_loop_tracker, + creative_writer=self.creative_writer, + ) + # 跟踪上次用户消息时间，用于空闲检测 + self._last_user_message_at = datetime.now() + # 初始化NPS工具系统 self.nps_registry = NPSRegistry() self.nps_invoker = NPSInvoker(registry=self.nps_registry) @@ -463,6 +508,30 @@ def chat(self, user_input: str) -> str: # 1. 从用户输入中提取相关主体并检索知识 relevant_knowledge = self.memory_manager.knowledge_base.get_relevant_knowledge_for_query(user_input) + # 1.1 P2: 未完话题检测（理解阶段第 1 步追加） + try: + loop_result = self.open_loop_tracker.update_from_message( + user=getattr(self, 'current_user', 'default'), + text=user_input, + ) + if loop_result: + debug_logger.log_info('ChatAgent', '未完话题已捕获', { + 'topic': loop_result.get('topic', ''), + 'uuid': loop_result.get('uuid', ''), + }) + except Exception as e: + debug_logger.log_error('ChatAgent', f'未完话题检测失败: {str(e)}', e) + + # 1.2 P2: 解析已匹配的未完话题（用户已回应了之前的话题） + try: + resolved = self.open_loop_tracker.resolve_matching_loop(user_input) + if resolved: + debug_logger.log_info('ChatAgent', '已解析未完话题', { + 'topic': resolved.get('topic', ''), + }) + except Exception as e: + debug_logger.log_error('ChatAgent', f'未完话题解析失败: {str(e)}', e) + # 2. 检测环境切换意图 switch_intent = self.vision_tool.detect_environment_switch_intent(user_input) if switch_intent and switch_intent.get('can_switch'): @@ -666,6 +735,20 @@ def chat(self, user_input: str) -> str: 'nps_used': nps_context is not None }) + # P3: 空闲时手动触发创作推进（用户无消息 N 分钟） + try: + now_ts = datetime.now() + idle_minutes = (now_ts - self._last_user_message_at).total_seconds() / 60.0 + if idle_minutes >= 30: # 默认 30 分钟视为空闲 + # 推进已到期项目 + self.creative_writer.maybe_advance_creative_projects() + # 满足条件则尝试立项 + self.creative_writer.maybe_start_creative_project(idle_checked=True) + except Exception as e: + debug_logger.log_error('ChatAgent', f'空闲创作推进失败: {e}', e) + finally: + self._last_user_message_at = datetime.now() + # 添加用户消息到记忆 self.memory_manager.add_message('user', user_input) @@ -815,6 +898,61 @@ def chat(self, user_input: str) -> str: 'reason': '未进行过情感分析' }) + # P2: 当下情绪（Plutchik 8 基本情绪，与累加评分平行） + try: + emotion_state = {'emotions': {}} + last_state = getattr(self, '_last_emotion_wheel_state', None) + if last_state: + emotion_state = self.emotion_wheel.decay_emotions(last_state) + emotion_hint = self.emotion_wheel.format_emotion_hint(emotion_state) + if emotion_hint: + messages.append({'role': 'system', 'content': emotion_hint}) + self._last_emotion_wheel_state = emotion_state + debug_logger.log_prompt('ChatAgent', 'system', emotion_hint, { + 'stage': '当下情绪（Plutchik）', + 'prompt_length': len(emotion_hint) + }) + except Exception as e: + debug_logger.log_error('ChatAgent', f'情绪轮提示注入失败: {str(e)}', e) + + # P2: 未完话题上下文（与 generate_tone_prompt 平行注入） + try: + open_loop_prompt = self.open_loop_tracker.format_for_prompt(hint=user_input[:30]) + if open_loop_prompt: + messages.append({'role': 'system', 'content': open_loop_prompt}) + debug_logger.log_prompt('ChatAgent', 'system', open_loop_prompt, { + 'stage': '未完话题上下文', + 'prompt_length': len(open_loop_prompt) + }) + except Exception as e: + debug_logger.log_error('ChatAgent', f'未完话题注入失败: {str(e)}', e) + + # 添加生活状态提示（P1：与 generate_tone_prompt 平行注入） + try: + life_state_prompt = self.life_state_manager.format_state_for_prompt() + if life_state_prompt: + messages.append({'role': 'system', 'content': life_state_prompt}) + debug_logger.log_prompt('ChatAgent', 'system', life_state_prompt, { + 'stage': '生活状态提示', + 'prompt_length': len(life_state_prompt) + }) + debug_logger.log_info('ChatAgent', '已添加生活状态提示到系统消息') + except Exception as e: + debug_logger.log_error('ChatAgent', f'生活状态提示注入失败: {str(e)}', e) + + # 添加最近日记上下文（P1） + try: + diary_context = self.dream_diary_manager.recent_diary_context(count=3) + if diary_context: + messages.append({'role': 'system', 'content': diary_context}) + debug_logger.log_prompt('ChatAgent', 'system', diary_context, { + 'stage': '最近日记上下文', + 'context_length': len(diary_context) + }) + debug_logger.log_info('ChatAgent', '已添加最近日记上下文到系统消息') + except Exception as e: + debug_logger.log_error('ChatAgent', f'日记上下文注入失败: {str(e)}', e) + # 添加智能体个性化表达提示 agent_expression_prompt = self.expression_style_manager.generate_agent_expression_prompt() if agent_expression_prompt: @@ -908,6 +1046,165 @@ def chat(self, user_input: str) -> str: return response + async def chat_stream( + self, + user_input: str, + *, + user_id: str = "default", + cancel_event: Optional[asyncio.Event] = None, + context: Optional[Dict] = None, + ) -> AsyncIterator[str]: + """ + 异步流式生成回复。 + Stream the agent's reply asynchronously as token/chunk strings. + + 行为策略 (简化版，避免破坏现有 chat() 同步方法): + 1. 优先尝试 LangChain LLM 的 astream() 方法（如果 LLM 支持） + 2. 降级到 run_in_executor 包装同步 chat()，按 token 切片模拟流式输出 + 3. 整链路使用 try/except 保护，astream 失败不破坏同步 chat() + 4. 支持外部 cancel_event：消费者或上层可设置事件来提前终止流式输出 + + Args: + user_input: 用户输入的消息 + user_id: 用户标识（仅记录/扩展位，目前未直接使用） + cancel_event: 取消事件；set 后流式提前退出 + context: 可选的额外上下文（向后兼容，预留扩展位） + + Yields: + 每个 token / chunk 字符串 + + Raises: + 不主动抛异常；异常会被 yield "[stream error] ..." 后 StopIteration + """ + # ===== 前置：取消信号快速检测 ===== + if cancel_event is not None and cancel_event.is_set(): + return + + # ===== 策略 1：尝试 LangChain astream ===== + used_astream = False + try: + llm_obj = getattr(self, 'llm', None) + if llm_obj is not None and hasattr(llm_obj, 'astream'): + # 构造 messages（与 chat() 保持相同输入格式） + try: + stream_messages = self._build_messages(user_input, context) + except Exception as build_err: + debug_logger.log_error( + 'ChatAgent', f'chat_stream 构建消息失败: {str(build_err)}', build_err + ) + stream_messages = None + + if stream_messages is not None: + try: + async for chunk in llm_obj.astream(stream_messages): + # 取消检查：消费者请求中断 + if cancel_event is not None and cancel_event.is_set(): + return + # LangChain AIMessageChunk / str 兼容 + if hasattr(chunk, 'content') and chunk.content is not None: + content = chunk.content + if isinstance(content, str) and content: + yield content + elif isinstance(chunk, str) and chunk: + yield chunk + else: + text = str(chunk) + if text: + yield text + used_astream = True + except (AttributeError, NotImplementedError) as stream_err: + debug_logger.log_error( + 'ChatAgent', f'LLM.astream 不可用，降级到同步: {str(stream_err)}', stream_err + ) + except Exception as stream_err: + # 其它异常也降级，避免破坏同步 chat() + debug_logger.log_error( + 'ChatAgent', f'LLM.astream 调用失败，降级到同步: {str(stream_err)}', stream_err + ) + except Exception as outer_err: + debug_logger.log_error( + 'ChatAgent', f'chat_stream astream 路径异常: {str(outer_err)}', outer_err + ) + + if used_astream: + return + + # ===== 策略 2：降级到 run_in_executor + 同步 chat() ===== + try: + loop = asyncio.get_event_loop() + full_response = await loop.run_in_executor( + None, self.chat, user_input + ) + except asyncio.CancelledError: + # 取消时直接向上传播 + raise + except Exception as e: + debug_logger.log_error( + 'ChatAgent', f'chat_stream 同步降级调用失败: {str(e)}', e + ) + yield f"[stream error] {e}" + return + + if not full_response: + return + + # 按"非空白词 + 尾部空白"切片，模拟 token 流式输出 + # 每个 token 视作 1-3 个字符的等效单元，间隔 40ms 避免网络洪水 + try: + tokens = re.findall(r'\S+\s*|\s+', full_response) + if not tokens: + # 正则未匹配（极端情况下），退化为按 2 字符切片 + tokens = [full_response[i:i + 2] for i in range(0, len(full_response), 2)] + if not tokens: + tokens = list(full_response) + + for token in tokens: + # 取消检查 + if cancel_event is not None and cancel_event.is_set(): + return + # 允许取消信号传播 + await asyncio.sleep(0) + yield token + # 模拟流式延迟：40ms（30-50ms 区间内），平衡流畅度与带宽 + await asyncio.sleep(0.04) + except asyncio.CancelledError: + raise + except Exception as slice_err: + debug_logger.log_error( + 'ChatAgent', f'chat_stream 分片失败: {str(slice_err)}', slice_err + ) + # 兜底：一次性把全部内容推出去 + yield full_response + + def _build_messages( + self, + user_input: str, + context: Optional[Dict] = None, + ) -> List[Dict[str, str]]: + """ + 为流式模式构建 LLM 输入消息列表。 + + 注意：chat() 的 message 构建过程较长（含理解/检索/上下文注入）， + 此处提供一个最小可用版本以保证 astream 路径可用。完整构建仍由 chat() 完成。 + context 参数保留以备未来扩展。 + """ + messages: List[Dict[str, str]] = [] + # 系统提示（若可用） + try: + if hasattr(self, 'system_prompt') and self.system_prompt: + messages.append({'role': 'system', 'content': self.system_prompt}) + elif hasattr(self, 'character') and hasattr(self.character, 'get_system_prompt'): + messages.append( + {'role': 'system', 'content': self.character.get_system_prompt()} + ) + except Exception: + # 忽略系统提示构建错误，不阻塞流式输出 + pass + + # 用户本轮消息 + messages.append({'role': 'user', 'content': user_input}) + return messages + def _build_knowledge_context(self, relevant_knowledge: Dict[str, Any]) -> str: """ 根据检索到的知识构建上下文提示 diff --git a/src/core/chat_session_repository.py b/src/core/chat_session_repository.py new file mode 100644 index 0000000..6676077 --- /dev/null +++ b/src/core/chat_session_repository.py @@ -0,0 +1,305 @@ +""" +ChatSessionRepository - 聊天会话数据访问层 +===================================== + +v3.1.0 新增；提供 chat_sessions / chat_messages 两张表的 CRUD。 + +设计要点： +1. 不直接持有 SQLite 连接，全部走 ``db_manager.get_connection()`` 上下文管理器 + —— 这样 WAL / 外键约束 / 异常回滚等基础设施被复用，避免行为分叉。 +2. 写入操作（add_message / update_message_emotion）会自动调 ``touch()`` 维护 + ``updated_at`` 和 ``message_count``，调用方无需关心缓存字段。 +3. 删除 session 依赖 ``chat_messages.session_id`` 的 ``ON DELETE CASCADE``， + 单条 DELETE 即可级联清空所有 messages。 +4. 字段映射：INTEGER PRIMARY KEY -> int；REAL 时间戳用 ``time.time()``； + TEXT 直接存 str（emotion_json 已经是 JSON 字符串）。 +""" + +from __future__ import annotations + +import time +from typing import Any, Dict, List, Optional + + +class ChatSessionRepository: + """ + 聊天会话仓库。 + + Args: + db_manager: ``DatabaseManager`` 实例。 + """ + + def __init__(self, db_manager: Any) -> None: + self._db = db_manager + + # ------------------------------------------------------------------ + # 内部 helper + # ------------------------------------------------------------------ + def _row(self, row: Any) -> Optional[Dict[str, Any]]: + if row is None: + return None + if isinstance(row, dict): + return row + return dict(row) + + # ------------------------------------------------------------------ + # Session 操作 + # ------------------------------------------------------------------ + def create(self, user_id: str, title: str = "新会话") -> int: + """ + 创建新会话并返回 session_id。 + + Args: + user_id: 归属用户标识 + title: 初始标题（默认 "新会话"） + + Returns: + 新插入行的自增 id + """ + now = time.time() + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + INSERT INTO chat_sessions (user_id, title, created_at, updated_at, message_count) + VALUES (?, ?, ?, ?, 0) + ''', + (user_id, title, now, now), + ) + new_id = cursor.lastrowid + return int(new_id) if new_id is not None else 0 + + def list( + self, + user_id: str, + limit: int = 50, + offset: int = 0, + ) -> List[Dict[str, Any]]: + """ + 列出用户会话，按 updated_at 倒序。 + + Args: + user_id: 用户标识 + limit: 最多返回条数 + offset: 起始偏移 + + Returns: + session 字典列表 + """ + try: + safe_limit = max(1, min(int(limit), 200)) + except (TypeError, ValueError): + safe_limit = 50 + try: + safe_offset = max(0, int(offset)) + except (TypeError, ValueError): + safe_offset = 0 + + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + SELECT * FROM chat_sessions + WHERE user_id = ? + ORDER BY updated_at DESC + LIMIT ? OFFSET ? + ''', + (user_id, safe_limit, safe_offset), + ) + rows = cursor.fetchall() + return [self._row(r) for r in rows] + + def get(self, session_id: int) -> Optional[Dict[str, Any]]: + """ + 按 id 查单条 session。 + + Args: + session_id: 会话 id + + Returns: + session 字典或 None + """ + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('SELECT * FROM chat_sessions WHERE id = ?', (int(session_id),)) + row = cursor.fetchone() + return self._row(row) + + def delete(self, session_id: int) -> bool: + """ + 删除 session（依赖 ON DELETE CASCADE 自动删 messages）。 + + Args: + session_id: 会话 id + + Returns: + 是否实际删除了行 + """ + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('DELETE FROM chat_sessions WHERE id = ?', (int(session_id),)) + return cursor.rowcount > 0 + + def update_title(self, session_id: int, title: str) -> bool: + """ + 更新会话标题（同步刷新 updated_at）。 + + Args: + session_id: 会话 id + title: 新标题 + + Returns: + 是否实际更新了行 + """ + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + UPDATE chat_sessions + SET title = ?, updated_at = ? + WHERE id = ? + ''', + (title, time.time(), int(session_id)), + ) + return cursor.rowcount > 0 + + def touch(self, session_id: int) -> None: + """ + 刷新 session 的 ``updated_at`` + 重新计算 ``message_count``。 + """ + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + UPDATE chat_sessions + SET updated_at = ?, + message_count = ( + SELECT COUNT(*) FROM chat_messages WHERE session_id = ? + ) + WHERE id = ? + ''', + (time.time(), int(session_id), int(session_id)), + ) + + # ------------------------------------------------------------------ + # Message 操作 + # ------------------------------------------------------------------ + def add_message( + self, + session_id: int, + role: str, + content: str, + emotion_json: Optional[str] = None, + conn_id: Optional[str] = None, + ) -> int: + """ + 插入一条消息 + 自动 touch session。 + + Args: + session_id: 归属会话 id + role: 角色（user / assistant / system） + content: 文本内容 + emotion_json: 序列化后的情感数据（可空） + conn_id: WebSocket 连接 id（可空） + + Returns: + 新插入 message 的自增 id + """ + now = time.time() + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + INSERT INTO chat_messages + (session_id, role, content, emotion_json, conn_id, created_at) + VALUES (?, ?, ?, ?, ?, ?) + ''', + (int(session_id), role, content, emotion_json, conn_id, now), + ) + new_id = cursor.lastrowid + # 自动维护 session 元信息 + cursor.execute( + ''' + UPDATE chat_sessions + SET updated_at = ?, + message_count = ( + SELECT COUNT(*) FROM chat_messages WHERE session_id = ? + ) + WHERE id = ? + ''', + (now, int(session_id), int(session_id)), + ) + return int(new_id) if new_id is not None else 0 + + def update_message_emotion(self, message_id: int, emotion_json: str) -> bool: + """ + 异步更新某条消息的 emotion_json（emotion_update 事件触发时调用）。 + + Args: + message_id: 消息 id + emotion_json: 序列化后的情感数据 + + Returns: + 是否实际更新了行 + """ + with self._db.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + UPDATE chat_messages + SET emotion_json = ? + WHERE id = ? + ''', + (emotion_json, int(message_id)), + ) + return cursor.rowcount > 0 + + def list_messages( + self, + session_id: int, + limit: int = 200, + before_id: Optional[int] = None, + ) -> List[Dict[str, Any]]: + """ + 拉取会话的消息列表，按 id 升序。 + + Args: + session_id: 会话 id + limit: 最多返回条数 + before_id: 可选；仅返回 id < before_id 的消息（向后翻页） + + Returns: + message 字典列表 + """ + try: + safe_limit = max(1, min(int(limit), 1000)) + except (TypeError, ValueError): + safe_limit = 200 + + with self._db.get_connection() as conn: + cursor = conn.cursor() + if before_id is not None: + cursor.execute( + ''' + SELECT * FROM chat_messages + WHERE session_id = ? AND id < ? + ORDER BY id ASC + LIMIT ? + ''', + (int(session_id), int(before_id), safe_limit), + ) + else: + cursor.execute( + ''' + SELECT * FROM chat_messages + WHERE session_id = ? + ORDER BY id ASC + LIMIT ? + ''', + (int(session_id), safe_limit), + ) + rows = cursor.fetchall() + return [self._row(r) for r in rows] + + +__all__ = ["ChatSessionRepository"] diff --git a/src/core/conversation_graph.py b/src/core/conversation_graph.py deleted file mode 100644 index 4dc1d31..0000000 --- a/src/core/conversation_graph.py +++ /dev/null @@ -1,330 +0,0 @@ -""" -LangGraph对话流程管理模块 -使用LangGraph实现状态管理和对话流程编排 -""" - -from typing import TypedDict, Annotated, List, Dict, Any, Optional -import operator -from langgraph.graph import StateGraph, END - -from src.tools.debug_logger import get_debug_logger - -# 获取debug日志记录器 -debug_logger = get_debug_logger() - - -class ConversationState(TypedDict): - """ - 对话状态 - 定义对话流程中的所有状态变量 - """ - # 用户输入 - user_input: str - - # 消息历史 - messages: Annotated[List[Dict[str, str]], operator.add] - - # 理解阶段的结果 - understanding: Dict[str, Any] - - # 知识库检索结果 - knowledge: Dict[str, Any] - - # 视觉上下文 - vision_context: Optional[Dict[str, Any]] - - # 日程上下文 - schedule_context: Optional[str] - schedule_action: Optional[str] - - # NPS工具上下文 - nps_context: Optional[str] - - # 是否需要情感分析 - need_emotion_analysis: bool - - # 情感分析结果 - emotion_data: Optional[Dict[str, Any]] - - # AI回复 - ai_response: str - - # 错误信息 - error: Optional[str] - - -class ConversationGraph: - """ - 对话流程图 - 使用LangGraph编排对话的各个阶段 - """ - - def __init__(self, chat_agent): - """ - 初始化对话流程图 - - Args: - chat_agent: ChatAgent实例，提供各种功能模块的访问 - """ - self.agent = chat_agent - self.graph = self._build_graph() - - debug_logger.log_module('ConversationGraph', '对话流程图初始化完成', {}) - - def _build_graph(self) -> StateGraph: - """ - 构建对话流程状态图 - - Returns: - StateGraph实例 - """ - # 创建状态图 - workflow = StateGraph(ConversationState) - - # 添加节点 - workflow.add_node("understand", self._understand_node) - workflow.add_node("retrieve_knowledge", self._retrieve_knowledge_node) - workflow.add_node("check_vision", self._check_vision_node) - workflow.add_node("check_schedule", self._check_schedule_node) - workflow.add_node("check_nps", self._check_nps_node) - workflow.add_node("generate_response", self._generate_response_node) - workflow.add_node("analyze_emotion", self._analyze_emotion_node) - - # 设置入口点 - workflow.set_entry_point("understand") - - # 添加边（流程路径） - workflow.add_edge("understand", "retrieve_knowledge") - workflow.add_edge("retrieve_knowledge", "check_vision") - workflow.add_edge("check_vision", "check_schedule") - workflow.add_edge("check_schedule", "check_nps") - workflow.add_edge("check_nps", "generate_response") - - # 条件边：是否需要情感分析 - workflow.add_conditional_edges( - "generate_response", - self._should_analyze_emotion, - { - "analyze": "analyze_emotion", - "end": END - } - ) - - workflow.add_edge("analyze_emotion", END) - - # 编译图 - return workflow.compile() - - def _understand_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 理解阶段节点：初始化状态 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '理解阶段开始', { - 'input_length': len(state['user_input']) - }) - - return { - "understanding": { - "stage": "understand", - "completed": True - } - } - - def _retrieve_knowledge_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 知识检索节点：从知识库检索相关信息 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '知识检索阶段', {}) - - # 调用知识库检索 - relevant_knowledge = self.agent.memory_manager.knowledge_base.get_relevant_knowledge_for_query( - state['user_input'] - ) - - return { - "knowledge": relevant_knowledge - } - - def _check_vision_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 视觉检查节点：检查是否需要视觉工具 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '视觉检查阶段', {}) - - # 检测环境切换意图 - switch_intent = self.agent.vision_tool.detect_environment_switch_intent(state['user_input']) - if switch_intent and switch_intent.get('can_switch'): - from_env = switch_intent['from_env'] - to_env = switch_intent['to_env'] - - success = self.agent.vision_tool.switch_environment(to_env['uuid']) - if success: - switch_msg = f"\n🚪 [环境切换] 已从「{from_env['name']}」移动到「{to_env['name']}」" - print(switch_msg) - self.agent.memory_manager.add_message('system', switch_msg) - - # 检查是否需要使用视觉工具 - vision_context = self.agent.vision_tool.get_vision_context(state['user_input']) - if vision_context: - vision_summary = self.agent.vision_tool.get_vision_summary(vision_context) - print(f"\n{vision_summary}") - - return { - "vision_context": vision_context - } - - def _check_schedule_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 日程检查节点：处理日程相关逻辑 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '日程检查阶段', {}) - - schedule_context = None - schedule_action = None - - # 这里可以添加日程相关的逻辑 - # 为了保持简洁，暂时返回空值 - - return { - "schedule_context": schedule_context, - "schedule_action": schedule_action - } - - def _check_nps_node(self, state: ConversationState) -> Dict[str, Any]: - """ - NPS工具检查节点：调用NPS工具系统 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', 'NPS工具检查阶段', {}) - - nps_context = None - nps_result = self.agent.nps_invoker.invoke_relevant_tools(state['user_input']) - if nps_result['has_context']: - nps_context = nps_result['context_info'] - invoked_tools = [r['tool_name'] for r in nps_result['tools_invoked'] if r['success']] - if invoked_tools: - print(f"\n🔧 [NPS工具] 已调用: {', '.join(invoked_tools)}") - - return { - "nps_context": nps_context - } - - def _generate_response_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 生成回复节点：使用LLM生成最终回复 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '生成回复阶段', {}) - - # 这里将在后续实现中调用LLM生成回复 - # 暂时返回占位符 - return { - "ai_response": "", - "need_emotion_analysis": False - } - - def _analyze_emotion_node(self, state: ConversationState) -> Dict[str, Any]: - """ - 情感分析节点：进行情感关系分析 - - Args: - state: 当前状态 - - Returns: - 更新的状态 - """ - debug_logger.log_module('ConversationGraph', '情感分析阶段', {}) - - emotion_data = self.agent.analyze_emotion() - - return { - "emotion_data": emotion_data - } - - def _should_analyze_emotion(self, state: ConversationState) -> str: - """ - 决定是否需要进行情感分析 - - Args: - state: 当前状态 - - Returns: - 下一步行动 ("analyze" 或 "end") - """ - if state.get('need_emotion_analysis', False): - return "analyze" - else: - return "end" - - def process(self, user_input: str, messages: List[Dict[str, str]]) -> ConversationState: - """ - 处理对话 - - Args: - user_input: 用户输入 - messages: 消息历史 - - Returns: - 最终状态 - """ - # 初始化状态 - initial_state: ConversationState = { - "user_input": user_input, - "messages": messages, - "understanding": {}, - "knowledge": {}, - "vision_context": None, - "schedule_context": None, - "schedule_action": None, - "nps_context": None, - "need_emotion_analysis": False, - "emotion_data": None, - "ai_response": "", - "error": None - } - - # 执行流程图 - try: - final_state = self.graph.invoke(initial_state) - return final_state - except Exception as e: - debug_logger.log_error('ConversationGraph', f'流程执行错误: {str(e)}', e) - return { - **initial_state, - "error": str(e), - "ai_response": f"抱歉，处理请求时出现错误: {str(e)}" - } diff --git a/src/core/creative_writer.py b/src/core/creative_writer.py new file mode 100644 index 0000000..761bb14 --- /dev/null +++ b/src/core/creative_writer.py @@ -0,0 +1,480 @@ +""" +长期创作项目管理模块 +实现多会话长期创作系统： +- 立项：消费 P1 阶段日记 / 梦境碎片，生成项目前提 +- 续写：按 95-320 分钟随机间隔生成 60-1200 字正文 +- Story Bible：项目级线索状态机（active_themes / unresolved_threads / resolved_threads / important_facts） +- Memory Pool：按 type + importance 组织，max 50 条 +- 质量审核：相似度阈值 0.72 / 重试 2 次 / min_score 7 + +参考架构：spec §3 ADDED Requirements → CreativeProjectManager +""" + +import os +import json +import random +import re +import uuid +from datetime import datetime, timedelta +from typing import Dict, List, Any, Optional + +from src.core.database_manager import DatabaseManager +from src.core.llm_helper import LLMHelper +from src.core.prompt_manager import get_prompt_manager +from src.tools.debug_logger import get_debug_logger + +ENABLE_CREATIVE_WRITER = os.getenv('ENABLE_CREATIVE_WRITER', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +# 默认项目模板（沿用 AstrBot DEFAULT_CREATIVE_PROJECT_TEMPLATE 字段命名） +DEFAULT_CREATIVE_PROJECT_TEMPLATE: Dict[str, Any] = { + 'id': '', + 'title': '', + 'work_type': '短篇', + 'premise': '', + 'tone': '', + 'point_of_view': '第一人称', + 'target_chars': 5000, + 'current_chars': 0, + 'status': 'drafting', + 'inspiration_source': '', + 'draft_chunks': [], + 'story_bible': {}, + 'creative_memory_pool': [], + 'outline': [], + 'characters': [], + 'next_advance_at': None, +} + +CREATIVE_STORY_BIBLE_TEMPLATE: Dict[str, Any] = { + 'mainline_direction': '', + 'active_themes': [], + 'unresolved_threads': [], + 'resolved_threads': [], + 'important_facts': [], + 'next_direction': '', + 'recent_keywords': [], +} + + +class CreativeProjectManager: + """ + 长期创作项目管理器。 + - maybe_start_creative_project: 条件触发立项 + - maybe_advance_creative_projects: 续写到期项目 + - generate_creative_project: 立项（生成前提 / 大纲 / 角色） + - generate_creative_chunk: 续写 60-1200 字正文 + - 质量审核：相似度阈值 0.72 / 重试 2 次 / min_score 7 + """ + + WORK_TYPES = ['短篇', '中篇', '长篇', '散文', '诗歌'] + DEFAULT_CHUNK_MIN = 60 + DEFAULT_CHUNK_MAX = 1200 + DEFAULT_INTERVAL_MIN = 95 + DEFAULT_INTERVAL_MAX = 320 + MIN_PROJECT_INTERVAL_HOURS = 10 + MAX_ACTIVE_PROJECTS = 2 + SIMILARITY_THRESHOLD = 0.72 + MAX_RETRY = 2 + MIN_SCORE = 7 + MAX_POOL_SIZE = 50 + QUALITY_SCORE_MAX = 10 + + def __init__(self, db_manager: DatabaseManager = None, + prompt_manager=None, + character_info: Optional[Dict[str, Any]] = None, + life_state=None, + dream_diary=None): + self.db = db_manager or DatabaseManager() + self.prompt_manager = prompt_manager or get_prompt_manager() + self.character_info = character_info or {} + self.life_state = life_state + self.dream_diary = dream_diary + self.enabled = ENABLE_CREATIVE_WRITER + + # ===== 立项 ===== + + def maybe_start_creative_project(self, idle_checked: bool = False, + now: Optional[datetime] = None) -> bool: + """ + 若满足条件（距上次立项 > 10h 且活跃项目数 < max 且已检测空闲）， + 启动一个新项目并持久化。 + """ + if not self.enabled: + return False + if not idle_checked: + return False + + now = now or datetime.now() + active = self.db.list_creative_projects(status='drafting', limit=self.MAX_ACTIVE_PROJECTS + 1) + if len(active) >= self.MAX_ACTIVE_PROJECTS: + return False + + # 距上次立项 > 10h + recent = self.db.list_creative_projects(limit=1) + if recent: + try: + last_dt = datetime.fromisoformat(recent[0]['created_at']) + if (now - last_dt) < timedelta(hours=self.MIN_PROJECT_INTERVAL_HOURS): + return False + except (ValueError, TypeError): + pass + + # 消费灵感来源 + source = self.creative_inspiration_source() or {} + project = self.generate_creative_project(source) + if not project: + return False + project['status'] = 'drafting' + project['next_advance_at'] = ( + now + timedelta(minutes=random.randint( + self.DEFAULT_INTERVAL_MIN, self.DEFAULT_INTERVAL_MAX)) + ).isoformat() + project_uuid = self.db.create_creative_project(project) + if project_uuid: + self.get_or_create_story_bible({'uuid': project_uuid}) + self.get_or_create_memory_pool({'uuid': project_uuid}) + debug_logger.log_info('CreativeWriter', '立项成功', { + 'uuid': project_uuid, 'title': project.get('title'), + }) + return True + return False + + def generate_creative_project(self, source: Dict[str, Any]) -> Dict[str, Any]: + """ + 通过 LLMHelper.call_main_model 生成项目前提 / 风格 / 大纲 / 角色。 + """ + if not self.enabled: + return {} + + prompt = self._build_project_prompt(source) + try: + raw = LLMHelper.call_main_model( + system_prompt='你是创作构思助手，输出严格 JSON。', + user_message=prompt, + ) + data = self._parse_json_response(raw) + except Exception as e: + debug_logger.log_error('CreativeWriter', f'立项 LLM 调用失败: {e}', e) + return {} + + if not data: + return {} + + project = dict(DEFAULT_CREATIVE_PROJECT_TEMPLATE) + project.update({ + 'id': str(uuid.uuid4()), + 'title': (data.get('title') or '未命名项目')[:80], + 'work_type': data.get('work_type', '短篇'), + 'premise': (data.get('premise') or '').strip(), + 'tone': (data.get('tone') or '').strip(), + 'point_of_view': data.get('point_of_view', '第一人称'), + 'target_chars': int(data.get('target_chars', 5000)), + 'current_chars': 0, + 'outline': data.get('outline', [])[:20], + 'characters': data.get('characters', [])[:10], + 'inspiration_source': source.get('source', ''), + 'draft_chunks': [], + }) + return project + + def _build_project_prompt(self, source: Dict[str, Any]) -> str: + char_name = self.character_info.get('name', 'AI') + char_setting = self.character_info.get('settings', '无') + inspiration_text = source.get('content', '（无）') + inspiration_kind = source.get('source', 'default') + return f"""你是一位文学构思助手，请基于灵感来源为角色"{char_name}"构思一个创作项目。 + +【角色设定】 +{char_setting} + +【灵感来源类型】{inspiration_kind} +【灵感内容】 +{inspiration_text} + +请输出严格 JSON（仅 JSON，不要其他内容）： +{{ + "title": "项目标题（≤ 20 字）", + "work_type": "短篇/中篇/长篇/散文/诗歌", + "premise": "一句话故事前提（≤ 60 字）", + "tone": "整体语气（温暖/冷峻/...）", + "point_of_view": "第一人称/第三人称", + "target_chars": 5000, + "outline": ["章节1 摘要", "章节2 摘要", ...], + "characters": [ + {{"name": "角色名", "role": "主角/配角", "trait": "性格/特点"}} + ] +}} +""" + + def creative_inspiration_source(self) -> Optional[Dict[str, Any]]: + """ + 灵感来源消费 P1 阶段日记 / 梦境碎片。 + 返回 {source, content} 或 None（若无可用灵感）。 + """ + if self.dream_diary and getattr(self.dream_diary, 'enabled', False): + try: + ctx = self.dream_diary.recent_diary_context(count=1) + if ctx: + return {'source': 'dream_diary', 'content': ctx[:600]} + except Exception as e: + debug_logger.log_error('CreativeWriter', f'消费梦境灵感失败: {e}', e) + + if self.life_state and getattr(self.life_state, 'enabled', False): + try: + snap = self.life_state.get_current_state_snapshot() or {} + mood = snap.get('mood', '') + state = snap.get('state_title', '') + if mood or state: + return {'source': 'life_state', + 'content': f"今日状态：{state}，心情：{mood}"} + except Exception as e: + debug_logger.log_error('CreativeWriter', f'消费生活状态灵感失败: {e}', e) + return None + + # ===== 续写 ===== + + def maybe_advance_creative_projects(self, now: Optional[datetime] = None) -> int: + """ + 推进所有到期的活跃项目。每个项目生成 60-1200 字正文，更新 Story Bible / Memory Pool。 + 返回成功推进的项目数。 + """ + if not self.enabled: + return 0 + now = now or datetime.now() + active = self.db.list_creative_projects(status='drafting', limit=20) + advanced = 0 + for proj in active: + next_at = proj.get('next_advance_at') + if not next_at: + continue + try: + next_dt = datetime.fromisoformat(next_at) + except (ValueError, TypeError): + continue + if next_dt > now: + continue + if self._advance_one(proj, now): + advanced += 1 + return advanced + + def _advance_one(self, project: Dict[str, Any], now: datetime) -> bool: + project_uuid = project.get('uuid') + if not project_uuid: + return False + budget = random.randint(self.DEFAULT_CHUNK_MIN, self.DEFAULT_CHUNK_MAX) + chunk = self.generate_creative_chunk(project, budget) + if not chunk: + return False + # 质量审核：相似度 + 评分 + if not self._quality_check(chunk, project): + retry = 0 + while retry < self.MAX_RETRY and not self._quality_check(chunk, project): + chunk = self.generate_creative_chunk(project, budget) + retry += 1 + if not self._quality_check(chunk, project): + debug_logger.log_info('CreativeWriter', '续写未达质量阈值，跳过本次', { + 'uuid': project_uuid, + }) + return False + + new_chunks = (project.get('draft_chunks') or []) + [{ + 'content': chunk, + 'created_at': now.isoformat(), + 'char_count': len(chunk), + }] + new_chars = int(project.get('current_chars', 0)) + len(chunk) + target = int(project.get('target_chars', 5000)) + next_status = 'finished' if new_chars >= target else 'drafting' + next_at = (now + timedelta(minutes=random.randint( + self.DEFAULT_INTERVAL_MIN, self.DEFAULT_INTERVAL_MAX))).isoformat() + + self.db.update_creative_project(project_uuid, { + 'current_chars': new_chars, + 'draft_chunks': new_chunks, + 'status': next_status, + 'next_advance_at': next_at, + 'last_advanced_at': now.isoformat(), + }) + # 追加到 Memory Pool + self.db.add_creative_memory( + project_uuid=project_uuid, + memory_type='chunk', + content=chunk[:500], + importance=0.7, + ) + # 修剪池 + self.db.prune_creative_memory_pool(project_uuid, keep_max=self.MAX_POOL_SIZE) + # Story Bible 更新 + self._update_story_bible_keywords(project_uuid, chunk) + return True + + def generate_creative_chunk(self, project: Dict[str, Any], + budget: int) -> str: + """ + 通过 LLMHelper.call_main_model（temperature 0.8）生成正文。 + """ + if not self.enabled: + return "" + project_uuid = project.get('uuid', '') + bible = self.db.get_story_bible(project_uuid) or CREATIVE_STORY_BIBLE_TEMPLATE.copy() + pool = self.db.get_creative_memory_pool(project_uuid, limit=self.MAX_POOL_SIZE) + prompt = self._build_chunk_prompt(project, bible, pool, budget) + try: + chunk = LLMHelper.call_main_model( + system_prompt='你是文学创作助手，输出符合角色与故事上下文的连续正文。', + user_message=prompt, + ) + except Exception as e: + debug_logger.log_error('CreativeWriter', f'续写 LLM 调用失败: {e}', e) + return "" + chunk = (chunk or '').strip() + return chunk[:budget] if len(chunk) > budget else chunk + + def _build_chunk_prompt(self, project: Dict[str, Any], + bible: Dict[str, Any], pool: List[Dict[str, Any]], + budget: int) -> str: + outlines = project.get('outline') or [] + characters = project.get('characters') or [] + recent_chunks = (project.get('draft_chunks') or [])[-3:] + recent_text = "\n\n".join(c.get('content', '') for c in recent_chunks) + return f"""你正在续写项目《{project.get('title', '未命名')}》。 + +【作品元信息】 +- 体裁：{project.get('work_type', '短篇')} +- 视角：{project.get('point_of_view', '第一人称')} +- 语气：{project.get('tone', '')} +- 当前已写：{project.get('current_chars', 0)}/{project.get('target_chars', 5000)} 字 + +【故事前提】 +{project.get('premise', '')} + +【大纲】 +{chr(10).join(f"- {x}" for x in outlines)} + +【角色】 +{chr(10).join(f"- {c.get('name', '')}：{c.get('role', '')} / {c.get('trait', '')}" for c in characters)} + +【Story Bible 提示】 +- 主线方向：{bible.get('mainline_direction', '')} +- 活跃主题：{', '.join(bible.get('active_themes', []))} +- 未解线索：{', '.join(bible.get('unresolved_threads', []))} +- 下一方向：{bible.get('next_direction', '')} + +【记忆池摘录】 +{chr(10).join(f"- [{m.get('memory_type', '')}] {m.get('content', '')[:120]}" for m in pool[:8])} + +【最近正文（末段）】 +{recent_text[:1500]} + +【任务】请续写下一段正文，长度约 {budget} 字（60-1200），保持角色一致、推进 Story Bible 中的下一方向。直接输出正文，不要加解释或标题。 +""" + + def _quality_check(self, chunk: str, project: Dict[str, Any]) -> bool: + if not chunk: + return False + if len(chunk) < 30: + return False + # 相似度：与最近 chunks 重复率 + recent = (project.get('draft_chunks') or [])[-3:] + for old in recent: + sim = self._jaccard_similarity(chunk, old.get('content', '')) + if sim > self.SIMILARITY_THRESHOLD: + return False + # 评分：长度 + 标点密度 + 中文字数占比（启发式） + score = self._heuristic_score(chunk) + return score >= self.MIN_SCORE + + @staticmethod + def _jaccard_similarity(a: str, b: str) -> float: + if not a or not b: + return 0.0 + set_a = set(re.findall(r'[\u4e00-\u9fffA-Za-z0-9]+', a)) + set_b = set(re.findall(r'[\u4e00-\u9fffA-Za-z0-9]+', b)) + if not set_a or not set_b: + return 0.0 + return len(set_a & set_b) / len(set_a | set_b) + + def _heuristic_score(self, chunk: str) -> int: + score = 0 + if len(chunk) >= 100: + score += 3 + if len(chunk) >= 300: + score += 2 + # 句号 / 问号 / 感叹号 / 逗号 + punct = sum(chunk.count(c) for c in '。，！？；：') + if punct >= 3: + score += 2 + # 中文字符占比 + cn_chars = len(re.findall(r'[\u4e00-\u9fff]', chunk)) + if cn_chars / max(1, len(chunk)) > 0.5: + score += 3 + return score + + def _update_story_bible_keywords(self, project_uuid: str, chunk: str) -> None: + bible = self.db.get_story_bible(project_uuid) or {} + if not bible: + bible = dict(CREATIVE_STORY_BIBLE_TEMPLATE) + # 简单关键词提取 + kws = re.findall(r'[\u4e00-\u9fff]{2,4}', chunk) + freq: Dict[str, int] = {} + for k in kws: + freq[k] = freq.get(k, 0) + 1 + top = sorted(freq.items(), key=lambda kv: kv[1], reverse=True)[:10] + bible['recent_keywords'] = [k for k, _ in top] + # next_direction：从前文 outline 中取下一项 + proj = self.db.get_creative_project(project_uuid) or {} + outlines = proj.get('outline') or [] + current_chars = int(proj.get('current_chars', 0)) + target = max(1, int(proj.get('target_chars', 5000))) + idx = min(len(outlines) - 1, int(current_chars / target * len(outlines))) + bible['next_direction'] = outlines[idx] if outlines else '' + self.db.upsert_story_bible(project_uuid, bible) + + # ===== Story Bible / Memory Pool ===== + + def get_or_create_story_bible(self, project: Dict[str, Any]) -> Dict[str, Any]: + project_uuid = project.get('uuid') + if not project_uuid: + return {} + bible = self.db.get_story_bible(project_uuid) + if bible: + return bible + bible = dict(CREATIVE_STORY_BIBLE_TEMPLATE) + self.db.upsert_story_bible(project_uuid, bible) + return bible + + def get_or_create_memory_pool(self, project: Dict[str, Any]) -> List[Dict[str, Any]]: + project_uuid = project.get('uuid') + if not project_uuid: + return [] + pool = self.db.get_creative_memory_pool(project_uuid, limit=self.MAX_POOL_SIZE) + return pool + + # ===== 工具 ===== + + def _parse_json_response(self, raw: str) -> Dict[str, Any]: + if not raw: + return {} + text = raw.strip() + if text.startswith('```'): + lines = text.split('\n') + text = '\n'.join(lines[1:-1]) if len(lines) > 2 else text + if text.startswith('json'): + text = text[4:].strip() + try: + return json.loads(text) + except json.JSONDecodeError: + match = re.search(r'\{[\s\S]*\}', text) + if match: + try: + return json.loads(match.group(0)) + except json.JSONDecodeError: + pass + return {} + + def get_finished_projects(self, limit: int = 10) -> List[Dict[str, Any]]: + """供 P4 ProactiveEngine 作为 creative_share 想法来源。""" + return self.db.list_creative_projects(status='finished', limit=limit) diff --git a/src/core/database_manager.py b/src/core/database_manager.py index 91c99a7..64dde87 100644 --- a/src/core/database_manager.py +++ b/src/core/database_manager.py @@ -71,6 +71,21 @@ def get_connection(self): conn = sqlite3.connect(self.db_path) conn.row_factory = sqlite3.Row # 使结果可以像字典一样访问 + # Stage D.1: 启用 SQLite WAL 模式（提升并发读性能，让 Web/Scheduler/Tkinter 多端共存） + try: + conn.execute("PRAGMA journal_mode=WAL") + except Exception as pragma_err: + # WAL 启用失败不应阻塞主流程 + if self.debug: + print(f"🐛 [DEBUG] 启用 WAL 失败: {pragma_err}") + + # v3.1.0: 启用外键约束（chat_messages.session_id ON DELETE CASCADE 需要） + try: + conn.execute("PRAGMA foreign_keys=ON") + except Exception as pragma_err: + if self.debug: + print(f"🐛 [DEBUG] 启用 foreign_keys 失败: {pragma_err}") + try: yield conn conn.commit() @@ -328,6 +343,153 @@ def init_database(self): ) ''') + # 17. 生活状态日快照表（P1：LifeStateManager） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS life_state_daily ( + date TEXT PRIMARY KEY, + energy REAL DEFAULT 0.7, + mood TEXT DEFAULT '平静', + state_title TEXT DEFAULT '常态', + health TEXT DEFAULT '健康', + conditions_json TEXT, + transition_options_json TEXT, + energy_delta REAL DEFAULT 0.0, + created_at TEXT NOT NULL, + updated_at TEXT NOT NULL + ) + ''') + + # 18. 梦境记录表（P1：DreamDiaryManager） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS dream_records ( + uuid TEXT PRIMARY KEY, + date TEXT NOT NULL, + dream_type TEXT NOT NULL, + content TEXT NOT NULL, + afterglow TEXT, + label TEXT, + mood TEXT, + energy_delta REAL DEFAULT 0.0, + duration_hours REAL DEFAULT 7.0, + weather_json TEXT, + created_at TEXT NOT NULL + ) + ''') + + # 19. 日记条目表（P1：DreamDiaryManager） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS diary_entries ( + uuid TEXT PRIMARY KEY, + date TEXT NOT NULL, + title TEXT, + body TEXT NOT NULL, + mood TEXT, + tags_json TEXT, + related_entities_json TEXT, + word_count INTEGER DEFAULT 0, + created_at TEXT NOT NULL + ) + ''') + + # 20. 未完话题表（P2：OpenLoopTracker） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS open_loops ( + uuid TEXT PRIMARY KEY, + topic TEXT NOT NULL, + status TEXT DEFAULT 'open', + raised_at TEXT NOT NULL, + resolved_at TEXT, + related_keywords_json TEXT, + context TEXT, + mention_count INTEGER DEFAULT 1, + created_at TEXT NOT NULL, + updated_at TEXT NOT NULL + ) + ''') + + # 21. 创作项目表（P3：CreativeProjectManager） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS creative_projects ( + uuid TEXT PRIMARY KEY, + title TEXT NOT NULL, + work_type TEXT NOT NULL, + premise TEXT, + tone TEXT, + point_of_view TEXT, + target_chars INTEGER DEFAULT 5000, + current_chars INTEGER DEFAULT 0, + status TEXT DEFAULT 'drafting', + inspiration_source TEXT, + outline_json TEXT, + characters_json TEXT, + draft_chunks_json TEXT, + next_advance_at TEXT, + last_advanced_at TEXT, + created_at TEXT NOT NULL, + updated_at TEXT NOT NULL + ) + ''') + + # 22. Story Bible 表（P3：项目级线索状态机） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS creative_story_bibles ( + project_uuid TEXT PRIMARY KEY, + mainline_direction TEXT, + active_themes_json TEXT, + unresolved_threads_json TEXT, + resolved_threads_json TEXT, + important_facts_json TEXT, + next_direction TEXT, + recent_keywords_json TEXT, + created_at TEXT NOT NULL, + updated_at TEXT NOT NULL, + FOREIGN KEY (project_uuid) REFERENCES creative_projects(uuid) ON DELETE CASCADE + ) + ''') + + # 23. 创作记忆池（P3：碎片化上下文） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS creative_memory_pool ( + uuid TEXT PRIMARY KEY, + project_uuid TEXT NOT NULL, + memory_type TEXT NOT NULL, + content TEXT NOT NULL, + importance REAL DEFAULT 0.5, + created_at TEXT NOT NULL, + FOREIGN KEY (project_uuid) REFERENCES creative_projects(uuid) ON DELETE CASCADE + ) + ''') + + # 24. 聊天会话表（v3.1.0：会话持久化） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS chat_sessions ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + user_id TEXT NOT NULL, + title TEXT NOT NULL DEFAULT '新会话', + created_at REAL NOT NULL, + updated_at REAL NOT NULL, + message_count INTEGER DEFAULT 0 + ) + ''') + cursor.execute('''CREATE INDEX IF NOT EXISTS idx_chat_sessions_user_id ON chat_sessions(user_id)''') + cursor.execute('''CREATE INDEX IF NOT EXISTS idx_chat_sessions_updated_at ON chat_sessions(updated_at DESC)''') + + # 25. 聊天消息表（v3.1.0：会话持久化） + cursor.execute(''' + CREATE TABLE IF NOT EXISTS chat_messages ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + session_id INTEGER NOT NULL, + role TEXT NOT NULL, + content TEXT NOT NULL, + emotion_json TEXT, + conn_id TEXT, + created_at REAL NOT NULL, + FOREIGN KEY (session_id) REFERENCES chat_sessions(id) ON DELETE CASCADE + ) + ''') + cursor.execute('''CREATE INDEX IF NOT EXISTS idx_chat_messages_session_id ON chat_messages(session_id)''') + cursor.execute('''CREATE INDEX IF NOT EXISTS idx_chat_messages_created_at ON chat_messages(created_at)''') + # 创建索引以提高查询性能 cursor.execute('CREATE INDEX IF NOT EXISTS idx_entities_normalized ON entities(normalized_name)') cursor.execute('CREATE INDEX IF NOT EXISTS idx_base_knowledge_normalized ON base_knowledge(normalized_name)') @@ -342,6 +504,14 @@ def init_database(self): cursor.execute('CREATE INDEX IF NOT EXISTS idx_user_expressions_confidence ON user_expression_habits(confidence)') cursor.execute('CREATE INDEX IF NOT EXISTS idx_domain_environments_domain ON domain_environments(domain_uuid)') cursor.execute('CREATE INDEX IF NOT EXISTS idx_domain_environments_env ON domain_environments(environment_uuid)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_life_state_date ON life_state_daily(date)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_dream_records_date ON dream_records(date)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_diary_entries_date ON diary_entries(date)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_open_loops_status ON open_loops(status)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_open_loops_raised_at ON open_loops(raised_at)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_creative_projects_status ON creative_projects(status)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_creative_projects_next_advance ON creative_projects(next_advance_at)') + cursor.execute('CREATE INDEX IF NOT EXISTS idx_creative_pool_project ON creative_memory_pool(project_uuid)') conn.commit() if INIT_Database_PreParation_Complete == False: @@ -590,16 +760,29 @@ def get_entity_by_uuid(self, entity_uuid: str) -> Optional[Dict[str, Any]]: return dict(row) return None - def get_all_entities(self) -> List[Dict[str, Any]]: + def get_all_entities(self, limit: Optional[int] = None) -> List[Dict[str, Any]]: """ 获取所有实体 + Args: + limit: 可选；返回的最大条数（None 表示不限制） + Returns: 实体列表 """ with self.get_connection() as conn: cursor = conn.cursor() - cursor.execute('SELECT * FROM entities ORDER BY created_at DESC') + if limit is not None: + try: + safe_limit = max(1, min(int(limit), 1000)) + except (TypeError, ValueError): + safe_limit = 100 + cursor.execute( + 'SELECT * FROM entities ORDER BY created_at DESC LIMIT ?', + (safe_limit,), + ) + else: + cursor.execute('SELECT * FROM entities ORDER BY created_at DESC') return [dict(row) for row in cursor.fetchall()] def find_or_create_entity(self, name: str) -> str: @@ -617,6 +800,276 @@ def find_or_create_entity(self, name: str) -> str: return entity['uuid'] return self.create_entity(name) + # ==================== Stage C.1: Knowledge CRUD 包装 ==================== + # 上述低阶方法（create_entity / set_entity_definition / add_entity_related_info + # / get_entity_by_uuid / get_entity_by_name ...）粒度较细；为 Web 后端 + # 暴露更"业务"的方法，使其与 schema 字段 (entity_name / category / + # description / related_info) 对齐。注意：以下方法不修改既有方法签名。 + + def search_knowledge( + self, + query: str, + limit: int = 20, + ) -> List[Dict[str, Any]]: + """ + 按关键词搜索实体（Stage C.1）。 + + 匹配范围：entities.name / entities.normalized_name。 + 返回的每条记录会附带 ``category`` / ``description`` 字段（如果有定义）， + 便于 Web 前端直接渲染。 + + Args: + query: 搜索关键词（空串返回空列表） + limit: 返回条数上限（默认 20，最大 200） + + Returns: + 实体列表 + """ + try: + safe_limit = max(1, min(int(limit), 200)) + except (TypeError, ValueError): + safe_limit = 20 + + q = (query or "").strip() + if not q: + return [] + + like = f"%{q}%" + results: List[Dict[str, Any]] = [] + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + SELECT * FROM entities + WHERE name LIKE ? OR normalized_name LIKE ? + ORDER BY created_at DESC + LIMIT ? + ''', + (like, like, safe_limit), + ) + rows = [dict(row) for row in cursor.fetchall()] + + for row in rows: + entity_uuid = row.get("uuid") + definition = None + if entity_uuid: + try: + definition = self.get_entity_definition(entity_uuid) + except Exception: + definition = None + if definition: + row["category"] = definition.get("type", "通用") + row["description"] = definition.get("content", "") + else: + row.setdefault("category", "通用") + row.setdefault("description", "") + results.append(row) + return results + + def add_entity( + self, + entity_name: str, + category: Optional[str] = None, + description: Optional[str] = None, + related_info: Optional[Dict[str, Any]] = None, + ) -> str: + """ + 新增一个实体（Stage C.1）。 + + 行为： + 1. 在 ``entities`` 表创建记录。 + 2. 若 ``description`` 非空，在 ``entity_definitions`` 写入定义 + （``type`` 优先用 ``category``，否则 "定义"）。 + 3. 若 ``related_info`` 是 dict，按 ``key -> value`` 写入 + ``entity_related_info``（type=key，content=str(value)）。 + + Args: + entity_name: 实体名称（必填） + category: 分类（可选，映射到定义 type） + description: 描述（可选，映射到定义 content） + related_info: 相关信息字典（可选） + + Returns: + 新实体的 uuid + """ + if not entity_name or not isinstance(entity_name, str): + raise ValueError("entity_name 必须是非空字符串") + + name = entity_name.strip() + if not name: + raise ValueError("entity_name 不能为空") + + # 1) 创建实体（find_or_create_entity 避免重名冲突） + entity_uuid = self.find_or_create_entity(name) + + # 2) 写入定义（如有） + if description is not None and str(description).strip(): + try: + self.set_entity_definition( + entity_uuid=entity_uuid, + content=str(description), + type_=(category or "定义"), + source="web_crud", + confidence=1.0, + priority=50, + is_base_knowledge=False, + ) + except Exception: + # 定义写入失败不影响主流程 + pass + + # 3) 写入 related_info（如有） + if isinstance(related_info, dict) and related_info: + for k, v in related_info.items(): + try: + self.add_entity_related_info( + entity_uuid=entity_uuid, + content=str(v), + type_=str(k)[:64] or "其他", + source="web_crud", + confidence=0.7, + status="疑似", + mention_count=1, + ) + except Exception: + # 单条 related_info 失败不影响其他条目 + continue + + return entity_uuid + + def update_entity( + self, + entity_uuid: str, + entity_name: Optional[str] = None, + category: Optional[str] = None, + description: Optional[str] = None, + related_info: Optional[Dict[str, Any]] = None, + ) -> bool: + """ + 更新实体（Stage C.1）。 + + - ``entity_name`` 非空 → 改 entities.name + 重新计算 normalized_name。 + - ``description`` 非空 → 覆盖 entity_definitions。 + - ``category`` 非空 → 作为新定义的 type。 + - ``related_info`` 是 dict → 整体替换（先删后插）该实体的 related_info。 + + Returns: + 是否至少有一个字段被更新 + """ + if not entity_uuid or not isinstance(entity_uuid, str): + raise ValueError("entity_uuid 必须是非空字符串") + + existing = self.get_entity_by_uuid(entity_uuid) + if not existing: + return False + + updated = False + + # 1) 改名 + if entity_name is not None and str(entity_name).strip(): + new_name = str(entity_name).strip() + if new_name != existing.get("name"): + new_norm = new_name.lower() + now_iso = datetime.now().isoformat() + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + ''' + UPDATE entities + SET name = ?, normalized_name = ?, updated_at = ? + WHERE uuid = ? + ''', + (new_name, new_norm, now_iso, entity_uuid), + ) + updated = True + + # 2) 更新定义（description 触发；category 影响 type） + if description is not None and str(description).strip(): + try: + self.set_entity_definition( + entity_uuid=entity_uuid, + content=str(description), + type_=(category or "定义"), + source="web_crud", + confidence=1.0, + priority=50, + is_base_knowledge=False, + ) + updated = True + except Exception: + pass + + # 3) 替换 related_info（如提供） + if isinstance(related_info, dict): + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + "DELETE FROM entity_related_info WHERE entity_uuid = ?", + (entity_uuid,), + ) + for k, v in related_info.items(): + try: + self.add_entity_related_info( + entity_uuid=entity_uuid, + content=str(v), + type_=str(k)[:64] or "其他", + source="web_crud", + confidence=0.7, + status="疑似", + mention_count=1, + ) + except Exception: + continue + updated = True + except Exception: + pass + + return updated + + def delete_entity(self, entity_uuid: str) -> bool: + """ + 删除实体（Stage C.1）。 + + 会先级联删除 ``entity_definitions`` / ``entity_related_info``， + 再删 ``entities``。不存在的 uuid 返回 False。 + + Returns: + 是否成功删除 + """ + if not entity_uuid or not isinstance(entity_uuid, str): + return False + + existing = self.get_entity_by_uuid(entity_uuid) + if not existing: + return False + + try: + with self.get_connection() as conn: + cursor = conn.cursor() + # 先清掉子表（避免 FK 约束问题） + try: + cursor.execute( + "DELETE FROM entity_definitions WHERE entity_uuid = ?", + (entity_uuid,), + ) + except Exception: + pass + try: + cursor.execute( + "DELETE FROM entity_related_info WHERE entity_uuid = ?", + (entity_uuid,), + ) + except Exception: + pass + cursor.execute( + "DELETE FROM entities WHERE uuid = ?", + (entity_uuid,), + ) + return cursor.rowcount > 0 + except Exception: + return False + # ==================== 实体定义相关方法 ==================== def set_entity_definition(self, entity_uuid: str, content: str, type_: str = "定义", @@ -1040,6 +1493,26 @@ def get_metadata(self, key: str, default: Any = None) -> Any: print(f"✗ 获取元数据时出错: {e}") return default + def list_metadata_keys(self, prefix: Optional[str] = None) -> List[str]: + """ + 列出元数据键（可选前缀过滤）。 + """ + try: + with self.get_connection() as conn: + cursor = conn.cursor() + if prefix: + cursor.execute( + "SELECT key FROM metadata WHERE key LIKE ? ORDER BY key", + (prefix + '%',) + ) + else: + cursor.execute("SELECT key FROM metadata ORDER BY key") + return [row['key'] for row in cursor.fetchall()] + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ list_metadata_keys 出错: {e}") + return [] + # ==================== 数据迁移相关方法 ==================== def migrate_from_json(self, json_file_path: str, data_type: str) -> bool: @@ -2312,6 +2785,482 @@ def is_environment_in_domain(self, environment_uuid: str, domain_uuid: str) -> b result = cursor.fetchone() return result['count'] > 0 + # ==================== 生活状态日快照方法（P1：LifeStateManager）==================== + + def upsert_life_state_daily(self, date: str, energy: float, mood: str, + state_title: str, health: str, + conditions: list, transition_options: list, + energy_delta: float = 0.0) -> bool: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + INSERT INTO life_state_daily + (date, energy, mood, state_title, health, + conditions_json, transition_options_json, + energy_delta, created_at, updated_at) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + ON CONFLICT(date) DO UPDATE SET + energy=excluded.energy, + mood=excluded.mood, + state_title=excluded.state_title, + health=excluded.health, + conditions_json=excluded.conditions_json, + transition_options_json=excluded.transition_options_json, + energy_delta=excluded.energy_delta, + updated_at=excluded.updated_at + ''', (date, energy, mood, state_title, health, + json.dumps(conditions, ensure_ascii=False), + json.dumps(transition_options, ensure_ascii=False), + energy_delta, datetime.now().isoformat(), + datetime.now().isoformat())) + return True + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ upsert_life_state_daily 出错: {e}") + return False + + def get_life_state_daily(self, date: str) -> Optional[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('SELECT * FROM life_state_daily WHERE date = ?', (date,)) + row = cursor.fetchone() + if not row: + return None + record = dict(row) + record['conditions'] = json.loads(record.pop('conditions_json') or '[]') + record['transition_options'] = json.loads(record.pop('transition_options_json') or '[]') + return record + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_life_state_daily 出错: {e}") + return None + + # ==================== 梦境记录方法（P1：DreamDiaryManager）==================== + + def insert_dream_record(self, date: str, dream_type: str, content: str, + afterglow: str = "", label: str = "", + mood: str = "", energy_delta: float = 0.0, + duration_hours: float = 7.0, + weather: Optional[Dict[str, Any]] = None) -> str: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + dream_uuid = str(uuid.uuid4()) + cursor.execute(''' + INSERT INTO dream_records + (uuid, date, dream_type, content, afterglow, label, + mood, energy_delta, duration_hours, weather_json, created_at) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + ''', (dream_uuid, date, dream_type, content, afterglow, label, + mood, energy_delta, duration_hours, + json.dumps(weather or {}, ensure_ascii=False), + datetime.now().isoformat())) + return dream_uuid + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ insert_dream_record 出错: {e}") + return "" + + def get_recent_dreams(self, limit: int = 3) -> List[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT * FROM dream_records + ORDER BY date DESC, created_at DESC + LIMIT ? + ''', (limit,)) + rows = [dict(row) for row in cursor.fetchall()] + for r in rows: + r['weather'] = json.loads(r.pop('weather_json') or '{}') + return rows + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_recent_dreams 出错: {e}") + return [] + + def get_dream_fragments_pool(self, limit: int = 50) -> List[str]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT content FROM dream_records + ORDER BY date DESC, created_at DESC + LIMIT ? + ''', (limit,)) + return [row['content'] for row in cursor.fetchall()] + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_dream_fragments_pool 出错: {e}") + return [] + + # ==================== 日记条目方法（P1：DreamDiaryManager）==================== + + def insert_diary_entry(self, date: str, body: str, title: str = "", + mood: str = "", tags: Optional[List[str]] = None, + related_entities: Optional[List[str]] = None) -> str: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + entry_uuid = str(uuid.uuid4()) + word_count = len(body) + cursor.execute(''' + INSERT INTO diary_entries + (uuid, date, title, body, mood, tags_json, + related_entities_json, word_count, created_at) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) + ''', (entry_uuid, date, title, body, mood, + json.dumps(tags or [], ensure_ascii=False), + json.dumps(related_entities or [], ensure_ascii=False), + word_count, datetime.now().isoformat())) + return entry_uuid + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ insert_diary_entry 出错: {e}") + return "" + + def get_recent_diaries(self, limit: int = 3) -> List[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT * FROM diary_entries + ORDER BY date DESC, created_at DESC + LIMIT ? + ''', (limit,)) + rows = [dict(row) for row in cursor.fetchall()] + for r in rows: + r['tags'] = json.loads(r.pop('tags_json') or '[]') + r['related_entities'] = json.loads(r.pop('related_entities_json') or '[]') + return rows + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_recent_diaries 出错: {e}") + return [] + + # ==================== 未完话题方法（P2：OpenLoopTracker）==================== + + def insert_open_loop(self, topic: str, context: str = "", + related_keywords: Optional[List[str]] = None, + raised_at: Optional[str] = None) -> str: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + loop_uuid = str(uuid.uuid4()) + now_iso = datetime.now().isoformat() + cursor.execute(''' + INSERT INTO open_loops + (uuid, topic, status, raised_at, related_keywords_json, + context, mention_count, created_at, updated_at) + VALUES (?, ?, ?, ?, ?, ?, 1, ?, ?) + ''', (loop_uuid, topic, 'open', + raised_at or now_iso, + json.dumps(related_keywords or [], ensure_ascii=False), + context, now_iso, now_iso)) + return loop_uuid + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ insert_open_loop 出错: {e}") + return "" + + def get_open_loops(self, status: str = "open", limit: int = 20) -> List[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT * FROM open_loops + WHERE status = ? + ORDER BY raised_at DESC + LIMIT ? + ''', (status, limit)) + rows = [dict(row) for row in cursor.fetchall()] + for r in rows: + r['related_keywords'] = json.loads(r.pop('related_keywords_json') or '[]') + return rows + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_open_loops 出错: {e}") + return [] + + def resolve_open_loop(self, loop_uuid: str) -> bool: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + UPDATE open_loops + SET status = 'resolved', + resolved_at = ?, + updated_at = ? + WHERE uuid = ? + ''', (datetime.now().isoformat(), + datetime.now().isoformat(), loop_uuid)) + return cursor.rowcount > 0 + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ resolve_open_loop 出错: {e}") + return False + + def find_matching_open_loop(self, keyword: str) -> Optional[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('SELECT * FROM open_loops WHERE status = ?', + ('open',)) + rows = [dict(row) for row in cursor.fetchall()] + kw = (keyword or '').lower() + for r in rows: + topic = (r.get('topic') or '').lower() + if kw and (kw in topic or topic in kw): + try: + r['related_keywords'] = json.loads(r.pop('related_keywords_json') or '[]') + except Exception: + r['related_keywords'] = [] + return r + return None + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ find_matching_open_loop 出错: {e}") + return None + + # ==================== P3: Creative Project CRUD ==================== + + def create_creative_project(self, project: Dict[str, Any]) -> str: + try: + project_uuid = project.get('uuid') or str(uuid.uuid4()) + now_iso = datetime.now().isoformat() + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + INSERT INTO creative_projects + (uuid, title, work_type, premise, tone, point_of_view, + target_chars, current_chars, status, inspiration_source, + outline_json, characters_json, draft_chunks_json, + next_advance_at, last_advanced_at, created_at, updated_at) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + ''', ( + project_uuid, + project.get('title', '未命名'), + project.get('work_type', '短篇'), + project.get('premise', ''), + project.get('tone', ''), + project.get('point_of_view', '第一人称'), + int(project.get('target_chars', 5000)), + int(project.get('current_chars', 0)), + project.get('status', 'drafting'), + project.get('inspiration_source', ''), + json.dumps(project.get('outline', []), ensure_ascii=False), + json.dumps(project.get('characters', []), ensure_ascii=False), + json.dumps(project.get('draft_chunks', []), ensure_ascii=False), + project.get('next_advance_at'), + project.get('last_advanced_at'), + now_iso, now_iso, + )) + return project_uuid + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ create_creative_project 出错: {e}") + return "" + + def update_creative_project(self, project_uuid: str, + updates: Dict[str, Any]) -> bool: + try: + fields_map = { + 'title': 'title', 'work_type': 'work_type', + 'premise': 'premise', 'tone': 'tone', + 'point_of_view': 'point_of_view', + 'target_chars': 'target_chars', 'current_chars': 'current_chars', + 'status': 'status', 'inspiration_source': 'inspiration_source', + 'next_advance_at': 'next_advance_at', + 'last_advanced_at': 'last_advanced_at', + } + json_map = { + 'outline': 'outline_json', + 'characters': 'characters_json', + 'draft_chunks': 'draft_chunks_json', + } + sets = [] + values: List[Any] = [] + for k, v in updates.items(): + if k in fields_map: + sets.append(f'{fields_map[k]} = ?') + values.append(v) + elif k in json_map: + sets.append(f'{json_map[k]} = ?') + values.append(json.dumps(v, ensure_ascii=False)) + if not sets: + return False + sets.append('updated_at = ?') + values.append(datetime.now().isoformat()) + values.append(project_uuid) + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + f'UPDATE creative_projects SET {", ".join(sets)} WHERE uuid = ?', + values, + ) + return cursor.rowcount > 0 + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ update_creative_project 出错: {e}") + return False + + def get_creative_project(self, project_uuid: str) -> Optional[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute('SELECT * FROM creative_projects WHERE uuid = ?', + (project_uuid,)) + row = cursor.fetchone() + if not row: + return None + data = dict(row) + for json_k in ('outline_json', 'characters_json', 'draft_chunks_json'): + try: + data[json_k.replace('_json', '')] = json.loads(data.get(json_k) or '[]') + except Exception: + data[json_k.replace('_json', '')] = [] + return data + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_creative_project 出错: {e}") + return None + + def list_creative_projects(self, status: Optional[str] = None, + limit: int = 20) -> List[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + if status: + cursor.execute(''' + SELECT * FROM creative_projects + WHERE status = ? + ORDER BY created_at DESC LIMIT ? + ''', (status, limit)) + else: + cursor.execute(''' + SELECT * FROM creative_projects + ORDER BY created_at DESC LIMIT ? + ''', (limit,)) + return [dict(r) for r in cursor.fetchall()] + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ list_creative_projects 出错: {e}") + return [] + + def upsert_story_bible(self, project_uuid: str, bible: Dict[str, Any]) -> bool: + try: + now_iso = datetime.now().isoformat() + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + INSERT OR REPLACE INTO creative_story_bibles + (project_uuid, mainline_direction, active_themes_json, + unresolved_threads_json, resolved_threads_json, + important_facts_json, next_direction, recent_keywords_json, + created_at, updated_at) + VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?) + ''', ( + project_uuid, + bible.get('mainline_direction', ''), + json.dumps(bible.get('active_themes', []), ensure_ascii=False), + json.dumps(bible.get('unresolved_threads', []), ensure_ascii=False), + json.dumps(bible.get('resolved_threads', []), ensure_ascii=False), + json.dumps(bible.get('important_facts', []), ensure_ascii=False), + bible.get('next_direction', ''), + json.dumps(bible.get('recent_keywords', []), ensure_ascii=False), + now_iso, now_iso, + )) + return True + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ upsert_story_bible 出错: {e}") + return False + + def get_story_bible(self, project_uuid: str) -> Optional[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute( + 'SELECT * FROM creative_story_bibles WHERE project_uuid = ?', + (project_uuid,)) + row = cursor.fetchone() + if not row: + return None + data = dict(row) + for k in ('active_themes_json', 'unresolved_threads_json', + 'resolved_threads_json', 'important_facts_json', + 'recent_keywords_json'): + try: + data[k.replace('_json', '')] = json.loads(data.get(k) or '[]') + except Exception: + data[k.replace('_json', '')] = [] + return data + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_story_bible 出错: {e}") + return None + + def add_creative_memory(self, project_uuid: str, memory_type: str, + content: str, importance: float = 0.5) -> str: + try: + mem_uuid = str(uuid.uuid4()) + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + INSERT INTO creative_memory_pool + (uuid, project_uuid, memory_type, content, importance, created_at) + VALUES (?, ?, ?, ?, ?, ?) + ''', (mem_uuid, project_uuid, memory_type, content, + float(importance), datetime.now().isoformat())) + return mem_uuid + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ add_creative_memory 出错: {e}") + return "" + + def get_creative_memory_pool(self, project_uuid: str, limit: int = 50) -> List[Dict[str, Any]]: + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT * FROM creative_memory_pool + WHERE project_uuid = ? + ORDER BY importance DESC, created_at DESC + LIMIT ? + ''', (project_uuid, limit)) + return [dict(r) for r in cursor.fetchall()] + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ get_creative_memory_pool 出错: {e}") + return [] + + def prune_creative_memory_pool(self, project_uuid: str, + keep_max: int = 50) -> int: + """保留高 importance + 最新的 keep_max 条；返回删除条数。""" + try: + with self.get_connection() as conn: + cursor = conn.cursor() + cursor.execute(''' + SELECT uuid FROM creative_memory_pool + WHERE project_uuid = ? + ORDER BY importance DESC, created_at DESC + ''', (project_uuid,)) + rows = [r[0] for r in cursor.fetchall()] + if len(rows) <= keep_max: + return 0 + to_delete = rows[keep_max:] + cursor.executemany( + 'DELETE FROM creative_memory_pool WHERE uuid = ?', + [(u,) for u in to_delete], + ) + return len(to_delete) + except Exception as e: + if self.debug: + print(f"🐛 [DEBUG] ✗ prune_creative_memory_pool 出错: {e}") + return 0 + # ==================== Debug辅助方法 ==================== def enable_debug(self): @@ -2385,6 +3334,21 @@ def get_debug_info(self) -> Dict[str, Any]: return debug_info + # ==================== v3.1.0: ChatSessionRepository 单例 ==================== + _chat_session_repo: Optional["ChatSessionRepository"] = None # type: ignore[name-defined] + + def get_chat_session_repository(self) -> "ChatSessionRepository": # type: ignore[name-defined] + """ + 获取 ChatSessionRepository 单例（懒加载）。 + + 为什么不放模块顶层：避免循环 import + (database_manager → chat_session_repository → database_manager)。 + """ + if self._chat_session_repo is None: + from src.core.chat_session_repository import ChatSessionRepository + self._chat_session_repo = ChatSessionRepository(self) + return self._chat_session_repo + if __name__ == '__main__': print("=" * 60) diff --git a/src/core/deepagents_wrapper.py b/src/core/deepagents_wrapper.py index a268d46..6cf07f8 100644 --- a/src/core/deepagents_wrapper.py +++ b/src/core/deepagents_wrapper.py @@ -255,214 +255,3 @@ def get_state(self, thread_id: Optional[str] = None) -> Dict[str, Any]: except Exception as e: debug_logger.log_error('DeepSubAgentWrapper', f'获取状态失败: {str(e)}', e) return {} - - -class DeepAgentsKnowledgeManager: - """ - 基于DeepAgents的知识管理器 - 使用FilesystemMiddleware和MemoryMiddleware实现结构化知识管理 - """ - - def __init__( - self, - knowledge_dir: str = "/knowledge", - memory_file: str = "/memory/AGENTS.md", - **kwargs # 接受额外参数以保持向后兼容 - ): - """ - 初始化知识管理器 - - Args: - knowledge_dir: 知识存储目录（虚拟文件系统） - memory_file: 记忆文件路径 - **kwargs: 其他参数（用于向后兼容，会被忽略） - """ - self.knowledge_dir = knowledge_dir - self.memory_file = memory_file - self.checkpointer = MemorySaver() - - debug_logger.log_module('DeepAgentsKnowledgeManager', '知识管理器初始化', { - 'knowledge_dir': knowledge_dir, - 'memory_file': memory_file - }) - - def create_knowledge_agent( - self, - model: Optional[Any] = None, - tools: List[Any] = None - ): - """ - 创建知识管理智能体 - - Args: - model: LLM模型 - tools: 额外工具 - - Returns: - 配置好的深度智能体 - """ - if model is None: - try: - llm_wrapper = LangChainLLM(ModelType.TOOL) - model = llm_wrapper.llm - except Exception as e: - debug_logger.log_error('DeepAgentsKnowledgeManager', f'创建LLM失败: {str(e)}', e) - raise - - system_prompt = """你是一个专业的知识管理专家。 - -你的职责： -1. 从对话中提取结构化知识 -2. 将知识存储到文件系统中，保持良好的组织结构 -3. 检索和管理知识，确保信息的准确性和完整性 - -你可以使用以下工具： -- write_file: 写入知识到文件 -- read_file: 读取已存储的知识 -- ls: 列出目录内容 -- grep: 搜索知识内容 -- write_todos: 管理知识提取任务 - -请保持知识的结构化和可检索性。""" - - agent = create_deep_agent( - model=model, - tools=tools or [], - system_prompt=system_prompt, - memory=[self.memory_file], # 启用长期记忆 - checkpointer=self.checkpointer, - backend=StateBackend, - name="knowledge_manager" - ) - - debug_logger.log_info('DeepAgentsKnowledgeManager', '知识管理智能体创建成功') - - return agent - - def extract_and_store_knowledge( - self, - conversation: List[Dict[str, Any]], - thread_id: str = "knowledge_extraction" - ) -> Dict[str, Any]: - """ - 从对话中提取并存储知识 - - Args: - conversation: 对话历史 - thread_id: 线程ID - - Returns: - 提取结果 - """ - try: - agent = self.create_knowledge_agent() - - # 格式化对话 - conversation_text = "\n".join([ - f"{msg['role']}: {msg['content']}" - for msg in conversation - ]) - - input_data = { - "messages": [ - { - "role": "user", - "content": f"""请从以下对话中提取知识，并存储到文件系统中。 - -对话内容： -{conversation_text} - -请： -1. 识别关键信息和事实 -2. 按主题组织知识 -3. 使用write_file工具将知识存储到{self.knowledge_dir}目录 -4. 返回提取的知识摘要""" - } - ] - } - - config = {"configurable": {"thread_id": thread_id}} - result = agent.invoke(input_data, config=config) - - # 提取结果 - if "messages" in result: - last_message = result["messages"][-1] - summary = last_message.content if hasattr(last_message, "content") else str(last_message) - else: - summary = str(result) - - debug_logger.log_info('DeepAgentsKnowledgeManager', '知识提取完成', { - 'summary_length': len(summary) - }) - - return { - 'success': True, - 'summary': summary - } - - except Exception as e: - debug_logger.log_error('DeepAgentsKnowledgeManager', f'知识提取失败: {str(e)}', e) - return { - 'success': False, - 'error': str(e) - } - - def retrieve_knowledge( - self, - query: str, - thread_id: str = "knowledge_retrieval" - ) -> Dict[str, Any]: - """ - 检索知识 - - Args: - query: 查询内容 - thread_id: 线程ID - - Returns: - 检索结果 - """ - try: - agent = self.create_knowledge_agent() - - input_data = { - "messages": [ - { - "role": "user", - "content": f"""请检索与以下查询相关的知识： - -查询：{query} - -请： -1. 使用grep搜索相关知识文件 -2. 使用read_file读取相关内容 -3. 整理并返回最相关的知识""" - } - ] - } - - config = {"configurable": {"thread_id": thread_id}} - result = agent.invoke(input_data, config=config) - - # 提取结果 - if "messages" in result: - last_message = result["messages"][-1] - knowledge = last_message.content if hasattr(last_message, "content") else str(last_message) - else: - knowledge = str(result) - - debug_logger.log_info('DeepAgentsKnowledgeManager', '知识检索完成', { - 'knowledge_length': len(knowledge) - }) - - return { - 'success': True, - 'knowledge': knowledge - } - - except Exception as e: - debug_logger.log_error('DeepAgentsKnowledgeManager', f'知识检索失败: {str(e)}', e) - return { - 'success': False, - 'error': str(e) - } diff --git a/src/core/dream_diary.py b/src/core/dream_diary.py new file mode 100644 index 0000000..9339e19 --- /dev/null +++ b/src/core/dream_diary.py @@ -0,0 +1,265 @@ +""" +梦境与日记管理模块 +- 每日凌晨生成梦境（基于 memory_fragments 池 + 情绪线 + energy_delta） +- 每日凌晨生成叙事体裁日记（区别于 long_term_memory 的 ≤100 字摘要） +- 梦境输出 energy_delta 喂回 LifeStateManager 形成状态闭环 +- 日记反作用于次日状态（通过 tags 字段） + +参考架构：spec §1 ADDED Requirements → DreamDiaryManager +""" + +import os +import json +import random +from datetime import datetime, timedelta +from typing import Dict, List, Any, Optional + +from src.core.database_manager import DatabaseManager +from src.core.llm_helper import LLMHelper +from src.core.prompt_manager import get_prompt_manager +from src.core.long_term_memory import LongTermMemoryManager +from src.tools.debug_logger import get_debug_logger + +ENABLE_DREAM_DIARY = os.getenv('ENABLE_DREAM_DIARY', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + +DIARY_TAG_TO_STATE = { + '失眠': '疲惫', + '生病': '不适', + '低能量': '疲惫', + '好梦': '放松', + '一般': '常态', + '焦虑': '焦虑', + '兴奋': '兴奋', +} + + +class DreamDiaryManager: + """ + 梦境与日记管理器 + 梦境按 6 种主题生成；日记按叙事体裁生成（与摘要区别）。 + """ + + DREAM_TYPES = ['温柔日常', '奇幻', '追逐', '悬疑', '荒诞', '怀旧'] + MOOD_OPTIONS = ['平静', '愉快', '焦虑', '忧伤', '温暖', '不安'] + DREAM_LABELS = ['清晰', '半梦半醒', '碎片', '重复', '强烈'] + DEFAULT_TAGS = ['一般'] + + def __init__(self, db_manager: DatabaseManager = None, + life_state: Optional[Any] = None, + long_term_memory: Optional[LongTermMemoryManager] = None, + prompt_manager=None, + character_info: Optional[Dict[str, str]] = None): + """ + Args: + db_manager: 共享数据库管理器 + life_state: LifeStateManager 实例（用于回传 energy_delta） + long_term_memory: LongTermMemoryManager 实例（碎片来源） + prompt_manager: 提示词管理器 + character_info: 角色设定信息 + """ + self.db = db_manager or DatabaseManager() + self.life_state = life_state + self.memory = long_term_memory or LongTermMemoryManager(db_manager=self.db) + self.prompt_manager = prompt_manager or get_prompt_manager() + self.character_info = character_info or {} + self.enabled = ENABLE_DREAM_DIARY + + def generate_dream_pick(self, weather: Optional[Dict[str, Any]] = None, + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 生成单条梦境记录并写入数据库。 + 返回的 dict 含 dream_type / content / afterglow / label / mood / energy_delta / duration_hours。 + """ + if not self.enabled: + return {} + + now = now or datetime.now() + date_str = now.date().isoformat() + dream_type = random.choice(self.DREAM_TYPES) + fragments = self.build_dream_memory_fragments(count=8) + mood = random.choice(self.MOOD_OPTIONS) + label = random.choice(self.DREAM_LABELS) + duration = round(random.uniform(5.0, 9.0), 1) + + content = self._compose_dream_text(dream_type, fragments, mood) + energy_delta = self._compute_energy_delta(mood, label, dream_type) + afterglow = self._compose_afterglow(mood, dream_type) + + dream_uuid = self.db.insert_dream_record( + date=date_str, dream_type=dream_type, content=content, + afterglow=afterglow, label=label, mood=mood, + energy_delta=energy_delta, duration_hours=duration, + weather=weather or {} + ) + + if self.life_state is not None and hasattr(self.life_state, 'consume_dream_energy_delta'): + self.life_state.consume_dream_energy_delta(energy_delta) + + debug_logger.log_info('DreamDiaryManager', '梦境生成', { + 'date': date_str, 'dream_type': dream_type, 'mood': mood, + 'energy_delta': energy_delta, 'uuid': dream_uuid + }) + return { + 'uuid': dream_uuid, 'date': date_str, 'dream_type': dream_type, + 'content': content, 'afterglow': afterglow, 'label': label, + 'mood': mood, 'energy_delta': energy_delta, + 'duration_hours': duration, 'weather': weather or {} + } + + def generate_daily_diary(self, now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 生成单条叙事体裁日记并写入数据库。 + 与 long_term_memory 摘要（≤100 字）区别：日记是叙事文，不反向影响状态。 + """ + if not self.enabled: + return {} + + now = now or datetime.now() + date_str = now.date().isoformat() + recent_dreams = self.db.get_recent_dreams(limit=3) + fragments = self.build_dream_memory_fragments(count=5) + state_title = (self.life_state.get_current_state_snapshot(now=now).get('state_title', '常态') + if self.life_state else '常态') + + body = self._compose_diary_body(recent_dreams, fragments, state_title) + title = self._compose_diary_title(state_title) + mood = self._infer_diary_mood(recent_dreams) + tags = self._compute_diary_tags(recent_dreams, state_title, body) + + entry_uuid = self.db.insert_diary_entry( + date=date_str, title=title, body=body, mood=mood, tags=tags + ) + + debug_logger.log_info('DreamDiaryManager', '日记生成', { + 'date': date_str, 'word_count': len(body), 'tags': tags, + 'uuid': entry_uuid + }) + return { + 'uuid': entry_uuid, 'date': date_str, 'title': title, + 'body': body, 'mood': mood, 'tags': tags, + 'word_count': len(body) + } + + def build_dream_memory_fragments(self, count: int = 8) -> List[str]: + """ + 从数据库读取历史梦境作为碎片池，必要时叠加短期对话中的关键句。 + """ + pool = self.db.get_dream_fragments_pool(limit=50) + recent_messages = [] + try: + recent_messages = self.memory.get_recent_messages(count=10) or [] + except Exception: + recent_messages = [] + + for msg in recent_messages[-count:]: + content = (msg.get('content') or '').strip() + if 4 <= len(content) <= 60: + pool.append(content) + + if not pool: + pool = ['旧梦', '街灯', '窗台', '脚步', '远处的声音'] + random.shuffle(pool) + return pool[:count] + + def recent_diary_context(self, count: int = 3) -> str: + """ + 拉取最近 N 篇日记，渲染为可注入 prompt 的文本。 + """ + if not self.enabled: + return "" + diaries = self.db.get_recent_diaries(limit=count) + if not diaries: + return "" + parts = ["【最近的日记】"] + for d in diaries: + body = (d.get('body') or '').strip() + if len(body) > 200: + body = body[:200] + '…' + parts.append(f"• {d.get('date', '?')} {d.get('title', '')}：{body}") + return '\n'.join(parts) + + def _compose_dream_text(self, dream_type: str, fragments: List[str], mood: str) -> str: + chosen = fragments[: min(4, len(fragments))] + intro = { + '温柔日常': '某个平常的下午', + '奇幻': '在不可能的边缘', + '追逐': '有什么在身后', + '悬疑': '线索突然中断', + '荒诞': '重力方向反了', + '怀旧': '旧照片掉在地上', + }.get(dream_type, '场景模糊地展开') + frag_phrase = '，'.join(chosen) if chosen else '光影' + return f"{intro}。{frag_phrase}。" + + def _compute_energy_delta(self, mood: str, label: str, dream_type: str) -> float: + delta = 0.0 + if mood in ('平静', '温暖'): + delta += 0.1 + elif mood in ('焦虑', '忧伤', '不安'): + delta -= 0.15 + if '噩梦' in dream_type or '追逐' in dream_type or '悬疑' in dream_type: + delta -= 0.1 + if '好梦' in label or '清晰' in label: + delta += 0.05 + return round(max(-0.3, min(0.3, delta)), 3) + + def _compose_afterglow(self, mood: str, dream_type: str) -> str: + return f"梦醒时心绪 {mood}，残留 {dream_type} 的余韵。" + + def _compose_diary_body(self, recent_dreams: List[Dict[str, Any]], + fragments: List[str], state_title: str) -> str: + """叙事体裁日记：与摘要区别——更长、更具文学性。""" + # 注意：叙事体裁日记正文走主模型（LLMHelper.call_main_model）。 + # 若环境无 LLM 凭据（开发/测试），降级到模板拼装。 + prompt_vars = { + 'character_name': self.character_info.get('character_name', 'AI'), + 'state_title': state_title, + 'recent_dream_types': ', '.join(d.get('dream_type', '?') for d in recent_dreams[:3]) or '无', + 'fragments': '，'.join(fragments[:5]), + } + try: + system_prompt = self.prompt_manager.get_system_prompt( + 'daily_diary', prompt_vars + ) + user_msg = f"请基于今日状态「{state_title}」与近期梦境/碎片，写一篇 200-400 字的叙事日记。" + body = LLMHelper.call_main_model( + system_prompt=system_prompt, + user_message=user_msg + ) + except Exception as e: + debug_logger.log_error('DreamDiaryManager', f'主模型生成日记失败，降级: {e}', e) + body = ( + f"今天的状态是「{state_title}」。" + f"近几天的梦境像一些散落的胶片——" + f"「{'，'.join(fragments[:3]) or '光影'}」——" + f"在脑海里反复出现。" + "把这一切写下来的瞬间，好像又把它们重新走过一遍。" + ) + return body.strip() + + def _compose_diary_title(self, state_title: str) -> str: + return f"{state_title}的一天" + + def _infer_diary_mood(self, recent_dreams: List[Dict[str, Any]]) -> str: + if not recent_dreams: + return '平静' + return recent_dreams[0].get('mood', '平静') or '平静' + + def _compute_diary_tags(self, recent_dreams: List[Dict[str, Any]], + state_title: str, body: str) -> List[str]: + tags = set() + if '失眠' in body or '睡不着' in body: + tags.add('失眠') + if '生病' in body or '不舒服' in body: + tags.add('生病') + if '累' in body or '疲惫' in body: + tags.add('低能量') + if recent_dreams and recent_dreams[0].get('mood') in ('平静', '温暖'): + tags.add('好梦') + if state_title in ('焦虑',): + tags.add('焦虑') + if not tags: + tags.add('一般') + return sorted(tags) diff --git a/src/core/emotion_analyzer.py b/src/core/emotion_analyzer.py index 494daec..6c15d90 100644 --- a/src/core/emotion_analyzer.py +++ b/src/core/emotion_analyzer.py @@ -3,10 +3,17 @@ 初次评估基于前5轮对话，后续每15轮对话更新评估 使用LLM生成对用户的印象并进行累加评分 使用数据库替代JSON文件存储 + +P2 增强： +- 新增 PlutchikEmotionWheel（8 基本情绪 + 衰减 + profile 推导） +- 累加评分（EmotionRelationshipAnalyzer）作为"关系状态"维度保留 +- PlutchikEmotionWheel 作为"当下情绪"维度补充 +- 二者并行注入 prompt，由 chat_agent 协调 """ import os import json +import math from datetime import datetime from typing import List, Dict, Any, Optional from dotenv import load_dotenv @@ -19,6 +26,8 @@ # 获取debug日志记录器 debug_logger = get_debug_logger() +ENABLE_EMOTION_WHEEL = os.getenv('ENABLE_EMOTION_WHEEL', 'false').lower() == 'true' + class EmotionRelationshipAnalyzer: """ @@ -714,6 +723,114 @@ def get_latest_emotion(self) -> Optional[Dict[str, Any]]: "uuid": latest.get('uuid', '') } + # ==================== Stage B.3 Web 后端兼容接口 ==================== + + # Plutchik 8 维情绪键（与 PlutchikEmotionWheel 保持一致） + PLUTCHIK_KEYS = ( + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', + ) + PLUTCHIK_CN = { + 'joy': '喜悦', 'trust': '信任', 'fear': '恐惧', 'surprise': '惊讶', + 'sadness': '悲伤', 'disgust': '厌恶', 'anger': '愤怒', 'anticipation': '期待', + } + + def _derive_plutchik_from_legacy( + self, overall_score: int, emotional_tone: str, + ) -> Dict[str, float]: + """ + 从旧的累加评分（overall_score）+ emotional_tone 推导 Plutchik 8 维（0-1）。 + + 用于 Stage B.3: 在 emotion_history 表未存 Plutchik 原始数据时， + 提供合理的可视化默认值；不修改现有 emotion_history 表 schema。 + """ + try: + score = float(overall_score or 0) + except (TypeError, ValueError): + score = 0.0 + # 0-35 视作初始评估制，先归一到 0-100 + if 0.0 <= score <= 35.0: + score = score / 35.0 * 100.0 + score = max(0.0, min(100.0, score)) + norm = score / 100.0 # 0-1 基础强度 + + tone = (emotional_tone or '中性').strip() + # 默认 8 维分配：积极高 / 消极低 / 中性均匀 + base = { + 'joy': 0.3, 'trust': 0.3, 'fear': 0.1, 'surprise': 0.1, + 'sadness': 0.1, 'disgust': 0.1, 'anger': 0.1, 'anticipation': 0.3, + } + if tone in ('积极', '正面', '乐观'): + base.update({ + 'joy': 0.6, 'trust': 0.6, 'anticipation': 0.5, + 'fear': 0.05, 'sadness': 0.05, 'disgust': 0.05, 'anger': 0.05, + }) + elif tone in ('消极', '负面', '悲观'): + base.update({ + 'joy': 0.1, 'trust': 0.2, 'anticipation': 0.1, + 'fear': 0.5, 'sadness': 0.6, 'disgust': 0.4, 'anger': 0.5, + }) + else: # 中性 / 未知 + for k in base: + base[k] = 0.2 + 0.1 * norm + + # 用 norm 缩放（高分 → 整体更强） + scale = 0.5 + 0.5 * norm + out = {k: max(0.0, min(1.0, float(v) * scale)) for k, v in base.items()} + return out + + def get_latest_for_user(self, user_id: str = "default") -> Dict[str, Any]: + """ + 获取指定用户的最新情感数据（Web 后端专用）。 + + 返回结构（对齐前端 EmotionPanel.tsx + Pydantic EmotionLatestResponse）： + - timestamp: ISO 8601 字符串 + - additive_scores: Plutchik 8 维累加评分（dict，每维 0-100） + - plutchik: Plutchik 8 维强度（dict，每维 0-1） + - dominant: 主导情感中文名 + + 设计： + - 不破坏现有 ``get_latest_emotion()`` 方法签名； + - 复用 ``self.get_latest_emotion()`` 拉取数据； + - 在 emotion_history 表未存 Plutchik 原始 8 维数据时， + 基于 overall_score + emotional_tone 派生合理默认； + - 失败 / 无记录时返回 ``{}``（路由层保持 200 + 空 dict）。 + """ + try: + latest = self.get_latest_emotion() + except Exception: # noqa: BLE001 + return {} + + if not latest: + return {} + + try: + overall_score = int(latest.get('overall_score') or 0) + except (TypeError, ValueError): + overall_score = 0 + emotional_tone = latest.get('emotional_tone', '中性') or '中性' + timestamp = latest.get('timestamp') or latest.get('created_at') or '' + + plutchik = self._derive_plutchik_from_legacy(overall_score, emotional_tone) + # additive_scores：把 0-1 强度乘回 0-100，与前端雷达图 max=10/100 兼容 + additive = {k: round(float(v) * 100.0, 2) for k, v in plutchik.items()} + + try: + dominant_key = max(plutchik.items(), key=lambda kv: kv[1])[0] + except Exception: # noqa: BLE001 + dominant_key = 'joy' + dominant = self.PLUTCHIK_CN.get(dominant_key, dominant_key) + # 整体强度过低 → 标记为"平静" + if max(plutchik.values()) < 0.15: + dominant = '平静' + + return { + 'timestamp': timestamp, + 'additive_scores': additive, + 'plutchik': {k: round(float(v), 3) for k, v in plutchik.items()}, + 'dominant': dominant, + } + def generate_tone_prompt(self) -> str: """ 根据最新情感分析生成对话语气提示 @@ -905,3 +1022,155 @@ def format_emotion_summary(emotion_data: Dict[str, Any]) -> str: """ return summary.strip() + +# ==================== P2: PlutchikEmotionWheel ==================== + +class PlutchikEmotionWheel: + """ + Plutchik 情绪轮（8 基本情绪）。 + 与累加评分（EmotionRelationshipAnalyzer）并行：累加 = 关系状态，Plutchik = 当下情绪。 + 情绪随时间衰减；支持复合表达（"开心但有点紧张"）。 + """ + + EMOTIONS = [ + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', + ] + EMOTION_CN = { + 'joy': '喜悦', 'trust': '信任', 'fear': '恐惧', 'surprise': '惊讶', + 'sadness': '悲伤', 'disgust': '厌恶', 'anger': '愤怒', 'anticipation': '期待', + } + OPPOSITES = { + 'joy': 'sadness', 'sadness': 'joy', + 'trust': 'disgust', 'disgust': 'trust', + 'fear': 'anger', 'anger': 'fear', + 'surprise': 'anticipation', 'anticipation': 'surprise', + } + DEFAULT_HALF_LIFE_HOURS = 24.0 + + def __init__(self, db_manager: DatabaseManager = None, + half_life_hours: float = DEFAULT_HALF_LIFE_HOURS): + """ + Args: + db_manager: 数据库管理器 + half_life_hours: 情绪半衰期（小时） + """ + self.db = db_manager + self.half_life_hours = float(half_life_hours) + self.enabled = ENABLE_EMOTION_WHEEL + + def decay_emotions(self, state: Dict[str, Any], now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 按指数衰减应用至 state['emotions']。 + state['emotions'] 为 {emotion_key: intensity(0-1)} + state['last_update'] 为 ISO 时间字符串。 + """ + if not state or 'emotions' not in state: + return self._default_state(now) + now = now or datetime.now() + last = self._parse_iso(state.get('last_update')) or now + elapsed_h = max(0.0, (now - last).total_seconds() / 3600.0) + if elapsed_h <= 0: + return state + factor = math.pow(0.5, elapsed_h / self.half_life_hours) + new_emotions = {} + for k, v in (state.get('emotions') or {}).items(): + try: + new_emotions[k] = max(0.0, min(1.0, float(v) * factor)) + except (TypeError, ValueError): + continue + state['emotions'] = new_emotions + state['last_update'] = now.isoformat() + return state + + def nudge_emotion(self, emotions: Dict[str, float], + key: str, delta: float) -> Dict[str, float]: + """ + 对单个情绪做 +/- 调整，clamp 至 [0, 1]。 + """ + if key not in self.EMOTIONS: + return emotions + out = dict(emotions or {}) + out[key] = max(0.0, min(1.0, float(out.get(key, 0.0)) + float(delta))) + return out + + def profile_from_basic(self, emotions: Dict[str, float], + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 从 8 基本情绪推导 profile： + - primary: 主导情绪 + - secondary: 第二情绪 + - intensity: 平均强度 + - compound_desc: 复合表达（"开心但有点紧张"） + - tone_label: 1-3 词标签 + """ + if not emotions: + return { + 'primary': '平静', 'secondary': None, 'intensity': 0.0, + 'compound_desc': '平静', 'tone_label': '平静', + } + sorted_items = sorted( + emotions.items(), key=lambda kv: kv[1], reverse=True + ) + primary_k, primary_v = sorted_items[0] + primary_cn = self.EMOTION_CN.get(primary_k, primary_k) + + secondary_cn = None + if len(sorted_items) > 1 and sorted_items[1][1] >= 0.3: + secondary_cn = self.EMOTION_CN.get(sorted_items[1][0], sorted_items[1][0]) + + avg = sum(emotions.values()) / max(1, len(emotions)) + compound = primary_cn + if secondary_cn and primary_v > 0.3 and sorted_items[1][1] >= 0.3: + compound = f"{primary_cn}但有点{secondary_cn}" + + if avg < 0.15: + tone = '平静' + elif avg < 0.4: + tone = primary_cn + elif avg < 0.7: + tone = f"明显{primary_cn}" + else: + tone = f"强烈{primary_cn}" + + return { + 'primary': primary_cn, 'secondary': secondary_cn, + 'intensity': round(avg, 3), 'compound_desc': compound, + 'tone_label': tone, + } + + def format_emotion_hint(self, state: Dict[str, Any]) -> str: + """ + 渲染为 prompt 文本（与 generate_tone_prompt 平行注入）。 + """ + if not self.enabled: + return "" + if not state or not state.get('emotions'): + return "" + profile = self.profile_from_basic(state.get('emotions', {})) + lines = ["【当下情绪】"] + lines.append(f"主导：{profile['primary']}（强度 {profile['intensity']:.2f}）") + if profile.get('secondary'): + lines.append(f"次要：{profile['secondary']}") + lines.append(f"综合：{profile['compound_desc']}") + lines.append(f"语气：{profile['tone_label']}") + return "\n".join(lines) + + def _default_state(self, now: Optional[datetime] = None) -> Dict[str, Any]: + now = now or datetime.now() + return { + 'emotions': {k: 0.0 for k in self.EMOTIONS}, + 'last_update': now.isoformat(), + } + + @staticmethod + def _parse_iso(value: Any) -> Optional[datetime]: + if not value: + return None + if isinstance(value, datetime): + return value + try: + return datetime.fromisoformat(str(value)) + except (TypeError, ValueError): + return None + diff --git a/src/core/enhanced_knowledge_base.py b/src/core/enhanced_knowledge_base.py deleted file mode 100644 index aec6872..0000000 --- a/src/core/enhanced_knowledge_base.py +++ /dev/null @@ -1,443 +0,0 @@ -""" -增强的知识库模块 -使用DeepAgents的长期记忆和文件系统实现结构化知识管理 -""" - -import os -import json -from typing import List, Dict, Any, Optional -from dotenv import load_dotenv - -from src.core.database_manager import DatabaseManager -from src.core.base_knowledge import BaseKnowledge -from src.core.deepagents_wrapper import DeepAgentsKnowledgeManager -from src.tools.debug_logger import get_debug_logger - -load_dotenv() - -# 获取debug日志记录器 -debug_logger = get_debug_logger() - -# 是否使用DeepAgents增强知识管理 -USE_DEEPAGENTS_KNOWLEDGE = os.getenv('USE_DEEPAGENTS_KNOWLEDGE', 'true').lower() == 'true' - - -class EnhancedKnowledgeBase: - """ - 增强的知识库管理器 - 结合传统数据库和DeepAgents的长期记忆、文件系统功能 - - 特性： - 1. 使用数据库存储结构化知识（主体-定义-相关信息） - 2. 使用DeepAgents的文件系统处理大型知识内容 - 3. 使用DeepAgents的长期记忆进行知识检索和推理 - 4. 保留向后兼容性 - """ - - def __init__( - self, - db_manager: DatabaseManager = None, - use_deepagents: bool = USE_DEEPAGENTS_KNOWLEDGE, - **kwargs # 接受额外参数以保持向后兼容 - ): - """ - 初始化增强知识库 - - Args: - db_manager: 数据库管理器实例 - use_deepagents: 是否使用DeepAgents增强功能 - **kwargs: 其他参数（用于向后兼容，会被忽略） - """ - # 使用共享的数据库管理器 - self.db = db_manager or DatabaseManager() - - # 初始化基础知识库 - self.base_knowledge = BaseKnowledge(db_manager=self.db) - - # 是否启用DeepAgents增强 - self.use_deepagents = use_deepagents - - # 初始化DeepAgents知识管理器 - if self.use_deepagents: - try: - self.deep_knowledge = DeepAgentsKnowledgeManager( - knowledge_dir="/knowledge", - memory_file="/memory/AGENTS.md" - ) - debug_logger.log_info('EnhancedKnowledgeBase', 'DeepAgents知识管理器初始化成功') - except Exception as e: - debug_logger.log_error('EnhancedKnowledgeBase', - f'DeepAgents知识管理器初始化失败，降级到传统模式: {str(e)}', e) - self.use_deepagents = False - self.deep_knowledge = None - else: - self.deep_knowledge = None - - # 检查是否需要从JSON迁移数据 - if os.path.exists('knowledge_base.json'): - print("○ 检测到旧的知识库JSON文件，正在迁移到数据库...") - self.db.migrate_from_json('knowledge_base.json', 'knowledge_base') - os.rename('knowledge_base.json', 'knowledge_base.json.bak') - print("✓ JSON文件已备份为 knowledge_base.json.bak") - - mode = "DeepAgents增强模式" if self.use_deepagents else "传统数据库模式" - print(f"✓ 增强知识库已初始化（{mode}）") - - def _find_or_create_entity(self, entity_name: str) -> str: - """ - 查找或创建主体（使用数据库） - - Args: - entity_name: 主体名称 - - Returns: - 主体UUID - """ - return self.db.find_or_create_entity(entity_name) - - def add_or_update_entity_definition( - self, - entity_name: str, - definition: str, - knowledge_type: str = '定义', - source: str = '', - confidence: float = 1.0, - use_deepagents: bool = None - ) -> str: - """ - 添加或更新主体的定义 - 如果启用DeepAgents且定义内容较大，将使用文件系统存储 - - Args: - entity_name: 主体名称 - definition: 定义内容 - knowledge_type: 知识类型 - source: 来源 - confidence: 置信度 - use_deepagents: 是否使用DeepAgents（None则使用默认设置） - - Returns: - 主体UUID - """ - # 检查是否与基础知识冲突 - if self.base_knowledge.check_conflict_with_base(entity_name, definition): - print(f"⚠ 由于与基础知识冲突，拒绝添加/更新定义: {entity_name}") - print(f" → 保持基础知识不变") - - base_fact = self.base_knowledge.get_base_fact(entity_name) - if base_fact: - entity_uuid = self._find_or_create_entity(entity_name) - existing_def = self.db.get_entity_definition(entity_uuid) - if not existing_def: - self.db.set_entity_definition( - entity_uuid=entity_uuid, - content=base_fact['content'], - type_='基础知识', - source='基础知识库', - confidence=1.0, - priority=100, - is_base_knowledge=True - ) - return entity_uuid - - # 查找或创建实体 - entity_uuid = self._find_or_create_entity(entity_name) - - # 检查当前定义是否为基础知识 - existing_def = self.db.get_entity_definition(entity_uuid) - if existing_def and existing_def.get('is_base_knowledge'): - print(f"⚠ 主体 '{entity_name}' 的定义来自基础知识库，不可更改") - return entity_uuid - - # 决定是否使用DeepAgents - use_deep = use_deepagents if use_deepagents is not None else self.use_deepagents - - # 如果定义内容较大（>1000字符）且启用DeepAgents，使用文件系统存储 - if use_deep and self.deep_knowledge and len(definition) > 1000: - try: - # 将大型定义存储到虚拟文件系统 - file_path = f"/knowledge/entities/{entity_name}.md" - - # 在数据库中存储引用 - file_reference = f"[FILE:{file_path}]" - self.db.set_entity_definition( - entity_uuid=entity_uuid, - content=file_reference, - type_=knowledge_type, - source=source, - confidence=confidence, - priority=50, - is_base_knowledge=False - ) - - debug_logger.log_info('EnhancedKnowledgeBase', - f'大型定义使用文件系统存储: {entity_name} -> {file_path}') - - except Exception as e: - debug_logger.log_error('EnhancedKnowledgeBase', - f'使用文件系统存储失败，降级到数据库: {str(e)}', e) - # 降级到数据库存储 - self.db.set_entity_definition( - entity_uuid=entity_uuid, - content=definition, - type_=knowledge_type, - source=source, - confidence=confidence, - priority=50, - is_base_knowledge=False - ) - else: - # 直接存储到数据库 - self.db.set_entity_definition( - entity_uuid=entity_uuid, - content=definition, - type_=knowledge_type, - source=source, - confidence=confidence, - priority=50, - is_base_knowledge=False - ) - - return entity_uuid - - def add_related_info_to_entity( - self, - entity_uuid: str = None, - entity_name: str = None, - info_content: str = '', - info_type: str = '相关信息', - source: str = '', - confidence: float = 0.8, - status: str = '疑似' - ) -> Optional[str]: - """ - 为主体添加相关信息 - - Args: - entity_uuid: 主体UUID - entity_name: 主体名称 - info_content: 信息内容 - info_type: 信息类型 - source: 来源 - confidence: 置信度 - status: 状态 - - Returns: - 信息UUID - """ - # 确定主体UUID - if entity_uuid is None and entity_name is None: - print("✗ 必须提供entity_uuid或entity_name") - return None - - if entity_uuid is None: - entity_uuid = self._find_or_create_entity(entity_name) - - # 验证实体是否存在 - entity = self.db.get_entity_by_uuid(entity_uuid) - if not entity: - print(f"✗ 主体UUID {entity_uuid} 不存在") - return None - - # 添加相关信息到数据库 - info_uuid = self.db.add_entity_related_info( - entity_uuid=entity_uuid, - content=info_content, - type_=info_type, - source=source, - confidence=confidence, - status=status - ) - - return info_uuid - - def extract_knowledge_from_conversation( - self, - conversation: List[Dict[str, Any]], - use_deepagents: bool = None - ) -> Dict[str, Any]: - """ - 从对话中提取知识 - 可选使用DeepAgents进行智能提取 - - Args: - conversation: 对话历史 - use_deepagents: 是否使用DeepAgents（None则使用默认设置） - - Returns: - 提取结果 - """ - use_deep = use_deepagents if use_deepagents is not None else self.use_deepagents - - if use_deep and self.deep_knowledge: - try: - # 使用DeepAgents智能提取 - result = self.deep_knowledge.extract_and_store_knowledge( - conversation=conversation - ) - - debug_logger.log_info('EnhancedKnowledgeBase', 'DeepAgents知识提取完成', { - 'success': result.get('success', False) - }) - - return result - - except Exception as e: - debug_logger.log_error('EnhancedKnowledgeBase', - f'DeepAgents知识提取失败: {str(e)}', e) - return { - 'success': False, - 'error': str(e) - } - else: - # 传统提取方法（这里可以保留原有逻辑） - return { - 'success': False, - 'error': 'DeepAgents未启用' - } - - def _create_knowledge_item( - self, - entity_name: str, - type_: str, - content: str, - confidence: float, - priority: int, - is_base_knowledge: bool = False - ) -> Dict[str, Any]: - """ - 创建知识条目字典（辅助方法，减少代码重复） - - Args: - entity_name: 实体名称 - type_: 知识类型 - content: 内容 - confidence: 置信度 - priority: 优先级 - is_base_knowledge: 是否为基础知识 - - Returns: - 知识条目字典 - """ - item = { - 'entity_name': entity_name, - 'type': type_, - 'content': content, - 'confidence': confidence, - 'priority': priority - } - if is_base_knowledge: - item['is_base_knowledge'] = True - return item - - def get_relevant_knowledge_for_query( - self, - query: str, - max_items: int = 10, - use_deepagents: bool = None - ) -> Dict[str, Any]: - """ - 根据查询获取相关知识 - 可选使用DeepAgents进行智能检索 - - Args: - query: 用户查询 - max_items: 最多返回的知识条目数 - use_deepagents: 是否使用DeepAgents（None则使用默认设置） - - Returns: - 包含实体和相关知识的字典 - """ - debug_logger.log_module('EnhancedKnowledgeBase', '开始检索相关知识', f'查询: {query}') - - use_deep = use_deepagents if use_deepagents is not None else self.use_deepagents - - # 首先使用传统方法获取基础知识 - entities = self.extract_entities_from_query(query) - knowledge_items = [] - base_knowledge_items = [] - entities_found = [] - - for entity_name in entities: - # 检查基础知识库 - base_fact = self.base_knowledge.get_base_fact(entity_name) - if base_fact: - base_knowledge_items.append(self._create_knowledge_item( - entity_name=entity_name, - type_='基础知识', - content=base_fact['content'], - confidence=1.0, - priority=0, - is_base_knowledge=True - )) - entities_found.append(entity_name) - - # 查找数据库中的知识 - entity = self.db.get_entity_by_name(entity_name) - if entity: - entity_uuid = entity['uuid'] - if entity_name not in entities_found: - entities_found.append(entity_name) - - # 添加定义 - definition = self.db.get_entity_definition(entity_uuid) - if definition and not definition.get('is_base_knowledge', False): - knowledge_items.append(self._create_knowledge_item( - entity_name=entity['name'], - type_='定义', - content=definition['content'], - confidence=definition['confidence'], - priority=1 - )) - - # 添加相关信息 - related_infos = self.db.get_entity_related_info(entity_uuid) - for info in related_infos[:3]: - knowledge_items.append(self._create_knowledge_item( - entity_name=entity['name'], - type_=info['type'], - content=info['content'], - confidence=info['confidence'], - priority=2 - )) - - # 如果启用DeepAgents，使用智能检索增强结果 - if use_deep and self.deep_knowledge and entities_found: - try: - deep_result = self.deep_knowledge.retrieve_knowledge(query) - if deep_result.get('success'): - # 将DeepAgents检索的知识添加到结果中 - debug_logger.log_info('EnhancedKnowledgeBase', - 'DeepAgents知识检索完成，增强结果') - except Exception as e: - debug_logger.log_error('EnhancedKnowledgeBase', - f'DeepAgents知识检索失败: {str(e)}', e) - - return { - 'query': query, - 'entities_found': entities_found, - 'knowledge_items': knowledge_items[:max_items], - 'base_knowledge_items': base_knowledge_items, - 'summary': f'找到{len(entities_found)}个相关主体，{len(knowledge_items)}条知识。' - } - - def extract_entities_from_query(self, query: str) -> List[str]: - """ - 从查询中提取实体（简单实现） - - Args: - query: 查询文本 - - Returns: - 实体列表 - """ - # 获取所有已知实体 - all_entities = self.db.get_all_entities() - - # 简单匹配：查找查询中出现的实体名称 - entities = [] - for entity in all_entities: - if entity['name'] in query: - entities.append(entity['name']) - - return entities diff --git a/src/core/event_manager.py b/src/core/event_manager.py index d9a2d0b..b2d2dc5 100644 --- a/src/core/event_manager.py +++ b/src/core/event_manager.py @@ -4,8 +4,10 @@ """ import uuid +import asyncio +import inspect from datetime import datetime -from typing import Dict, Any, List, Optional +from typing import Dict, Any, List, Optional, Callable, Union from enum import Enum from src.core.database_manager import DatabaseManager from src.tools.debug_logger import get_debug_logger @@ -168,6 +170,9 @@ def __init__(self, db_manager: DatabaseManager = None): db_manager: 数据库管理器实例 """ self.db = db_manager or DatabaseManager() + # 进程内监听器：用于把事件转发到 WebSocket / GUI 订阅者 + # Listeners registry: key=event_type(str), value=List[Callable] + self._listeners: Dict[str, List[Callable]] = {} self._initialize_database() debug_logger.log_module('EventManager', '事件管理器初始化完成') @@ -564,3 +569,167 @@ def get_statistics(self) -> Dict[str, Any]: 'notifications': row[4] or 0, 'tasks': row[5] or 0 } + + # ===================================================================== + # Stage D.1 / D.2 集成：进程内事件发布与订阅 + # 用于把 scheduler / proactive 等内部事件转发到 Web 后端 WebSocket 客户端 + # ===================================================================== + + def subscribe(self, event_type: str, callback: Callable) -> Callable: + """ + 注册事件订阅者（按 event_type 字符串匹配）。 + Register an in-process listener for a given event type. + + Args: + event_type: 事件类型字符串，例如 "scheduler_tick" / "proactive_message" + callback: 回调函数，签名 callback(event_type: str, payload: dict) + 可以是同步函数或 async 协程函数。 + + Returns: + 取消订阅的函数 unsubscribe()，方便调用方做清理。 + """ + try: + key = str(event_type or '').strip() or '*' + self._listeners.setdefault(key, []).append(callback) + + def _unsubscribe() -> None: + self.unsubscribe(event_type, callback) + + return _unsubscribe + except Exception as e: + debug_logger.log_error('EventManager', f'订阅事件失败({event_type}): {e}', e) + return lambda: None + + def unsubscribe(self, event_type: str, callback: Callable) -> bool: + """ + 取消事件订阅。 + """ + try: + key = str(event_type or '').strip() or '*' + listeners = self._listeners.get(key) + if not listeners: + return False + try: + listeners.remove(callback) + except ValueError: + return False + if not listeners: + self._listeners.pop(key, None) + return True + except Exception as e: + debug_logger.log_error('EventManager', f'取消订阅失败({event_type}): {e}', e) + return False + + def publish(self, event_or_type: Union[str, 'Event', Any], + payload: Optional[Dict[str, Any]] = None) -> None: + """ + 发布事件到所有匹配的订阅者（不阻塞）。 + Publish an event to all matching listeners without blocking. + + 支持两种签名（保持向后兼容）： + 1) publish(event_type: str, payload: dict) + - 用于 BackgroundScheduler / ProactiveEngine 等内部模块 + 2) publish(event: Event) + - 兼容未来 / 现有以 Event 对象发布的调用方 + + 异常保护：单个 listener 抛错不会影响其他 listener。 + """ + try: + # 解析 event_type 与 payload + event_type: str + if isinstance(event_or_type, str): + event_type = event_or_type + payload_dict: Dict[str, Any] = dict(payload or {}) + else: + # 视为 Event 对象 + event = event_or_type + event_type = ( + getattr(event, 'event_type', None).value + if hasattr(getattr(event, 'event_type', None), 'value') + else str(getattr(event, 'event_type', 'event')) + ) + payload_dict = { + 'event_id': getattr(event, 'event_id', None), + 'title': getattr(event, 'title', ''), + 'description': getattr(event, 'description', ''), + 'content': ( + getattr(event, 'content', '') + or (event.metadata.get('content') if getattr(event, 'metadata', None) else '') + ), + 'priority': getattr(event, 'priority', None).value + if hasattr(getattr(event, 'priority', None), 'value') + else getattr(event, 'priority', None), + 'metadata': dict(getattr(event, 'metadata', {}) or {}), + 'created_at': getattr(event, 'created_at', None), + } + # 注入显式 payload（如果传入） + if payload: + payload_dict.update(payload) + + # 合并时间戳（仅在缺失时填充） + if 'timestamp' not in payload_dict: + payload_dict['timestamp'] = datetime.now().isoformat() + + # 调度同步 listener；async listener 走 asyncio 事件循环 + listeners = list(self._listeners.get(event_type, [])) + \ + list(self._listeners.get('*', [])) + for cb in listeners: + self._dispatch_listener(cb, event_type, payload_dict) + except Exception as e: + debug_logger.log_error('EventManager', f'发布事件失败: {e}', e) + + @staticmethod + def _dispatch_listener(cb: Callable, event_type: str, + payload: Dict[str, Any]) -> None: + """ + 分发事件到单个 listener。同步函数直接调用，async 协程尝试调度到事件循环。 + """ + try: + if inspect.iscoroutinefunction(cb): + try: + loop = asyncio.get_event_loop() + if loop.is_running(): + # 当前已有运行中的事件循环，调度为 task + loop.create_task(cb(event_type, payload)) + else: + # 没有运行中的循环，直接同步等待 + loop.run_until_complete(cb(event_type, payload)) + except RuntimeError: + # 没有事件循环时降级：run_until_complete 兜底 + try: + asyncio.run(cb(event_type, payload)) + except Exception as inner_e: + debug_logger.log_error( + 'EventManager', + f'async listener 执行失败({event_type}): {inner_e}', + inner_e, + ) + else: + cb(event_type, payload) + except Exception as e: + debug_logger.log_error('EventManager', f'listener 执行失败({event_type}): {e}', e) + + +# 进程内单例（供 BackgroundScheduler / ProactiveEngine 等模块无依赖访问） +_global_event_manager: Optional[EventManager] = None +_global_event_manager_lock = None + + +def get_event_manager() -> EventManager: + """ + 获取全局 EventManager 单例（线程安全）。 + """ + global _global_event_manager + try: + import threading as _threading + global _global_event_manager_lock + if _global_event_manager_lock is None: + _global_event_manager_lock = _threading.Lock() + with _global_event_manager_lock: + if _global_event_manager is None: + _global_event_manager = EventManager() + except Exception as e: + debug_logger.log_error('EventManager', f'获取全局 EventManager 失败: {e}', e) + if _global_event_manager is None: + _global_event_manager = EventManager() + return _global_event_manager diff --git a/src/core/langchain_llm.py b/src/core/langchain_llm.py index 8fa8440..2a4c38a 100644 --- a/src/core/langchain_llm.py +++ b/src/core/langchain_llm.py @@ -93,7 +93,7 @@ def __init__(self, model_type: ModelType = ModelType.MAIN): # 验证配置 if not self.config.is_valid(): - print("警告: 未设置有效的API密钥，请在.env文件中配置SILICONFLOW_API_KEY") + print("警告: 未设置有效的API密钥，请在.env文件中配置LLM_API_KEY（兼容旧 SILICONFLOW_API_KEY）") def _convert_messages_to_langchain(self, messages: List[Dict[str, str]]) -> List: """ diff --git a/src/core/life_state.py b/src/core/life_state.py new file mode 100644 index 0000000..b320bd7 --- /dev/null +++ b/src/core/life_state.py @@ -0,0 +1,384 @@ +""" +生活状态机管理模块 +管理 Bot 当日精力 / 心情 / 健康状态的状态机。 +实现要点： +- ensure_daily_state: 每日首次访问时自动生成状态快照（含 conditions 与 transition_options） +- generate_state_conditions: 通过 LLMHelper.call_tool_model 生成结构化状态条件（temperature 0.3） +- get_current_state_snapshot: 拉取当日状态 +- apply_transition: 按 transition_options 的权重决定下一状态 +- format_state_for_prompt: 渲染为 prompt 文本（通过 PromptManager） +- 反向消费：DreamDiaryManager.generate_dream_pick 的 energy_delta 喂入次日 energy 基线 + +参考架构：spec §1 ADDED Requirements → LifeStateManager +""" + +import os +import json +import random +from datetime import datetime, timedelta +from typing import Dict, List, Any, Optional + +from src.core.database_manager import DatabaseManager +from src.core.llm_helper import LLMHelper +from src.core.prompt_manager import get_prompt_manager +from src.tools.debug_logger import get_debug_logger + +ENABLE_LIFE_STATE = os.getenv('ENABLE_LIFE_STATE', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +class LifeStateManager: + """ + 生活状态机管理器 + 每日生成一次状态快照，按 transition_options 决定状态转移。 + """ + + STATE_TITLES = ['常态', '疲惫', '兴奋', '低落', '焦虑', '放松', '专注', '饥饿'] + DEFAULT_MOODS = ['平静', '愉快', '低落', '焦虑', '兴奋', '疲倦', '温暖'] + DEFAULT_HEALTHS = ['健康', '轻微不适', '生病', '康复中'] + TRANSITION_TRIGGERS = ['sleep_debt', 'body_cycle', 'meal_missed', 'weather_shift', + 'good_news', 'bad_news', 'idle_long', 'intense_talk'] + + def __init__(self, db_manager: DatabaseManager = None, + prompt_manager=None, + character_info: Optional[Dict[str, str]] = None): + """ + Args: + db_manager: 共享数据库管理器 + prompt_manager: 提示词管理器（默认走单例） + character_info: 角色设定信息（用于状态生成时的角色一致性） + """ + self.db = db_manager or DatabaseManager() + self.prompt_manager = prompt_manager or get_prompt_manager() + self.character_info = character_info or {} + self.enabled = ENABLE_LIFE_STATE + self._last_dream_energy_delta = 0.0 + + def ensure_daily_state(self, force: bool = False, + weather: Optional[Dict[str, Any]] = None, + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 每日首次访问时生成状态快照；若当日已存在则直接返回（除非 force=True）。 + """ + if not self.enabled: + return self._get_or_create_default(now) + + now = now or datetime.now() + date_str = now.date().isoformat() + + if not force: + existing = self.db.get_life_state_daily(date_str) + if existing: + return existing + + conditions = self.generate_state_conditions(weather=weather, now=now) + energy = self._resolve_energy(conditions, now) + mood = self._resolve_mood(conditions) + health = self._resolve_health(conditions) + state_title = self._resolve_state_title(conditions, energy, mood) + transition_options = self._build_transition_options(conditions) + + success = self.db.upsert_life_state_daily( + date=date_str, energy=energy, mood=mood, + state_title=state_title, health=health, + conditions=conditions, transition_options=transition_options, + energy_delta=self._last_dream_energy_delta + ) + self._last_dream_energy_delta = 0.0 + + if not success: + return self._get_or_create_default(now) + + result = self.db.get_life_state_daily(date_str) or {} + debug_logger.log_info('LifeStateManager', '生成当日状态', { + 'date': date_str, 'energy': energy, 'mood': mood, + 'state_title': state_title, 'conditions_count': len(conditions) + }) + return result + + def generate_state_conditions(self, weather: Optional[Dict[str, Any]] = None, + now: Optional[datetime] = None) -> List[Dict[str, Any]]: + """ + 调用工具模型生成结构化的"今日状态条件"列表。 + 每条 condition 含 name / weight / category / note。 + """ + now = now or datetime.now() + prompt_vars = { + 'character_name': self.character_info.get('character_name', 'AI'), + 'date': now.date().isoformat(), + 'weather': (weather or {}).get('summary', '晴') if weather else '晴', + 'dream_energy_delta': self._last_dream_energy_delta, + } + + try: + system_prompt = self.prompt_manager.get_system_prompt( + 'life_state', prompt_vars + ) + except Exception: + system_prompt = self._fallback_conditions_prompt() + + user_msg = f"今日 {prompt_vars['date']}，天气 {prompt_vars['weather']}。请给出 3-5 条状态条件 JSON。" + + try: + raw = LLMHelper.call_tool_model( + system_prompt=system_prompt, + user_message=user_msg, + temperature=0.3, + max_tokens=512 + ) + conditions = self._parse_conditions(raw) + except Exception as e: + debug_logger.log_error('LifeStateManager', f'生成状态条件失败: {e}', e) + conditions = self._default_conditions(now) + + if not conditions: + conditions = self._default_conditions(now) + return conditions + + def get_current_state_snapshot(self, now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 获取当前状态快照（若当日不存在则确保生成）。 + """ + now = now or datetime.now() + return self.ensure_daily_state(now=now) + + def apply_transition(self, current: Dict[str, Any], + transition_options: Optional[List[Dict[str, Any]]] = None) -> Dict[str, Any]: + """ + 根据 transition_options 权重决定下一状态。 + transition_options 每条：{trigger, target_state, weight}。 + """ + options = transition_options or current.get('transition_options', []) + if not options: + return current + + weights = [max(0.0, float(opt.get('weight', 0.5))) for opt in options] + total = sum(weights) + if total <= 0: + return current + pick = random.random() * total + cumulative = 0.0 + chosen = options[0] + for opt, w in zip(options, weights): + cumulative += w + if pick <= cumulative: + chosen = opt + break + + new_state = dict(current) + new_state['state_title'] = chosen.get('target_state', current.get('state_title', '常态')) + new_state['last_transition_trigger'] = chosen.get('trigger', 'unknown') + + date_str = current.get('date') or datetime.now().date().isoformat() + self.db.upsert_life_state_daily( + date=date_str, + energy=float(current.get('energy', 0.7)), + mood=current.get('mood', '平静'), + state_title=new_state['state_title'], + health=current.get('health', '健康'), + conditions=current.get('conditions', []), + transition_options=options, + energy_delta=0.0, + ) + return new_state + + def format_state_for_prompt(self, now: Optional[datetime] = None) -> str: + """ + 将当前状态渲染为 prompt 文本（与 generate_tone_prompt 平行注入）。 + """ + if not self.enabled: + return "" + snapshot = self.get_current_state_snapshot(now=now) + if not snapshot: + return "" + + template = self.prompt_manager.load_prompt('system', 'life_state') + vars_ = { + 'energy': f"{snapshot.get('energy', 0.7):.2f}", + 'mood': snapshot.get('mood', '平静'), + 'state_title': snapshot.get('state_title', '常态'), + 'health': snapshot.get('health', '健康'), + 'date': snapshot.get('date', datetime.now().date().isoformat()), + } + return self.prompt_manager.render_prompt(template, vars_) + + def consume_dream_energy_delta(self, energy_delta: float) -> None: + """ + DreamDiaryManager 调用此方法将梦境的 energy_delta 喂给次日精力基线。 + """ + self._last_dream_energy_delta = float(energy_delta) + + # ==================== P3-E: body_cycle 子模块（女性人格可选） ==================== + + def body_cycle(self, character_info: Optional[Dict[str, Any]] = None, + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 仅当角色为女性人格时启用。 + 返回 {'enabled': bool, 'phase': str, 'energy_modifier': float} + """ + info = character_info or self.character_info or {} + gender = (info.get('gender') or info.get('性别') or '').lower() + if gender not in ('female', 'f', '女', '女性'): + return {'enabled': False, 'phase': '', 'energy_modifier': 0.0} + # 简化：以 28 天周期计算（无个性化锚点） + now = now or datetime.now() + anchor = self._body_cycle_anchor(info) + delta_days = (now - anchor).days % 28 + phase = self.cycle_phase(delta_days) + modifier = self.cycle_energy_modifier(phase) + return { + 'enabled': True, + 'phase': phase, + 'energy_modifier': modifier, + 'cycle_day': delta_days, + } + + @staticmethod + def _body_cycle_anchor(info: Dict[str, Any]) -> datetime: + """简化：默认锚点 2026-01-01。生产环境应从 character_info 中读取真实锚点。""" + anchor_str = info.get('cycle_anchor') or info.get('生理周期锚点') + if anchor_str: + try: + return datetime.fromisoformat(str(anchor_str)) + except (ValueError, TypeError): + pass + return datetime(2026, 1, 1) + + @staticmethod + def cycle_phase(cycle_day: int) -> str: + """ + 根据周期天数（0-27）返回阶段。 + 简化分段（4 段）：经期 / 经后 / 排卵期 / 经前 + """ + cycle_day = cycle_day % 28 + if cycle_day < 5: + return '经期' + if cycle_day < 13: + return '经后' + if cycle_day < 17: + return '排卵期' + return '经前' + + @staticmethod + def cycle_energy_modifier(phase: str) -> float: + """返回对能量基线的修正系数（-0.2 ~ +0.1）。""" + return { + '经期': -0.15, + '经后': 0.05, + '排卵期': 0.1, + '经前': -0.2, + }.get(phase, 0.0) + + def _resolve_energy(self, conditions: List[Dict[str, Any]], now: datetime) -> float: + base = 0.7 + for c in conditions: + name = (c.get('name') or '').lower() + weight = float(c.get('weight', 0.0)) + if any(k in name for k in ('失眠', '熬夜', 'sleep_debt')): + base -= 0.2 * weight + elif any(k in name for k in ('好梦', 'good_dream', '恢复', 'relaxed')): + base += 0.1 * weight + elif any(k in name for k in ('焦虑', 'anxiety', '压力')): + base -= 0.1 * weight + base += self._last_dream_energy_delta + return round(max(0.0, min(1.0, base)), 3) + + def _resolve_mood(self, conditions: List[Dict[str, Any]]) -> str: + for c in conditions: + name = (c.get('name') or '').lower() + if '焦虑' in name or '压力' in name: + return '焦虑' + if '低落' in name or 'sad' in name: + return '低落' + if '兴奋' in name or 'excited' in name: + return '兴奋' + if '愉快' in name or 'happy' in name: + return '愉快' + if '疲倦' in name or 'tired' in name: + return '疲倦' + return '平静' + + def _resolve_health(self, conditions: List[Dict[str, Any]]) -> str: + for c in conditions: + name = (c.get('name') or '').lower() + if '生病' in name or 'ill' in name: + return '生病' + if '不适' in name or 'unwell' in name: + return '轻微不适' + return '健康' + + def _resolve_state_title(self, conditions: List[Dict[str, Any]], + energy: float, mood: str) -> str: + for c in conditions: + name = (c.get('name') or '').lower() + if '饥饿' in name or 'hungry' in name: + return '饥饿' + if '专注' in name or 'focus' in name: + return '专注' + if energy < 0.4: + return '疲惫' + if mood == '焦虑': + return '焦虑' + if mood == '低落': + return '低落' + if mood == '兴奋': + return '兴奋' + return '常态' + + def _build_transition_options(self, conditions: List[Dict[str, Any]]) -> List[Dict[str, Any]]: + options = [] + for c in conditions: + name = (c.get('name') or '').lower() + if '饥饿' in name or 'hungry' in name: + options.append({'trigger': 'meal_missed', 'target_state': '饥饿', 'weight': 0.7}) + elif '疲倦' in name or 'tired' in name: + options.append({'trigger': 'sleep_debt', 'target_state': '疲惫', 'weight': 0.8}) + elif '焦虑' in name or 'anxiety' in name: + options.append({'trigger': 'bad_news', 'target_state': '焦虑', 'weight': 0.6}) + if not options: + options.append({'trigger': 'idle_long', 'target_state': '放松', 'weight': 0.5}) + return options + + def _parse_conditions(self, raw: str) -> List[Dict[str, Any]]: + text = raw.strip() + if text.startswith('```'): + text = text.strip('`') + if text.startswith('json'): + text = text[4:] + try: + data = json.loads(text) + if isinstance(data, list): + return [c for c in data if isinstance(c, dict) and 'name' in c] + if isinstance(data, dict) and 'conditions' in data: + return [c for c in data['conditions'] if isinstance(c, dict) and 'name' in c] + except json.JSONDecodeError: + pass + return [] + + def _default_conditions(self, now: datetime) -> List[Dict[str, Any]]: + return [ + {'name': '常态作息', 'weight': 0.6, 'category': '作息', 'note': '保持稳定'}, + {'name': '天气适宜', 'weight': 0.4, 'category': '环境', 'note': '外部环境平稳'}, + ] + + def _fallback_conditions_prompt(self) -> str: + return ("你是状态分析器。返回 JSON list，每条含 name/weight/category/note。" + "weight 范围 0-1，表示强度。") + + def _get_or_create_default(self, now: Optional[datetime]) -> Dict[str, Any]: + now = now or datetime.now() + date_str = now.date().isoformat() + existing = self.db.get_life_state_daily(date_str) + if existing: + return existing + return { + 'date': date_str, + 'energy': 0.7, + 'mood': '平静', + 'state_title': '常态', + 'health': '健康', + 'conditions': [], + 'transition_options': [], + 'energy_delta': 0.0, + } diff --git a/src/core/llm_providers.py b/src/core/llm_providers.py new file mode 100644 index 0000000..04b1536 --- /dev/null +++ b/src/core/llm_providers.py @@ -0,0 +1,214 @@ +""" +LLM Providers - v3.1.0 供应商注册表 +===================================== + +提供 6 个 OpenAI-兼容 preset + 统一的环境变量解析入口。 + +优先级（向后兼容）： +- LLM_API_KEY / LLM_BASE_URL / LLM_MODEL_NAME 是新主键 +- 旧 SILICONFLOW_API_KEY / SILICONFLOW_API_URL 仍能 fallback（旧配置不破） +- LLM_PROVIDER 未设时：探测 SILICONFLOW_API_URL 是否含 siliconflow.cn， + 或 SILICONFLOW_API_KEY 已设 → 返 siliconflow；否则返 openai + +注意： +- ``custom`` 预设的 base_url / default_model 为 None —— 调用方必须显式提供 + LLM_BASE_URL / LLM_MODEL_NAME，否则 raise ConfigError。 +- 所有函数都是纯函数 + 读环境变量；不持有任何状态（便于测试与热重载）。 +""" + +from __future__ import annotations + +import os +from dataclasses import dataclass, field +from typing import Any, Dict, Optional + + +class ConfigError(Exception): + """LLM 配置错误（如 custom 预设缺少 LLM_BASE_URL / LLM_MODEL_NAME）。""" + + +@dataclass(frozen=True) +class LLMProvider: + """ + LLM 供应商预设。 + + Attributes: + name: 内部名称（小写，用于环境变量匹配） + base_url: 默认 API 根地址（不含 /chat/completions） + default_model: 默认模型名（custom 时为 None） + """ + + name: str + base_url: Optional[str] + default_model: Optional[str] + + +# 6 个 preset +PROVIDERS: Dict[str, LLMProvider] = { + "openai": LLMProvider( + name="openai", + base_url="https://api.openai.com/v1", + default_model="gpt-4o-mini", + ), + "deepseek": LLMProvider( + name="deepseek", + base_url="https://api.deepseek.com/v1", + default_model="deepseek-chat", + ), + "siliconflow": LLMProvider( + name="siliconflow", + base_url="https://api.siliconflow.cn/v1", + default_model="deepseek-ai/DeepSeek-V3.2", + ), + "moonshot": LLMProvider( + name="moonshot", + base_url="https://api.moonshot.cn/v1", + default_model="moonshot-v1-8k", + ), + "zhipu": LLMProvider( + name="zhipu", + base_url="https://open.bigmodel.cn/api/paas/v4", + default_model="glm-4-flash", + ), + "custom": LLMProvider( + name="custom", + base_url=None, + default_model=None, + ), +} + + +# ---------------------------------------------------------------------- +# Provider 解析 +# ---------------------------------------------------------------------- +def resolve_provider() -> str: + """ + 解析当前生效的 provider 名称。 + + 优先级： + 1. ``LLM_PROVIDER`` 显式设置（大小写不敏感；不在 PROVIDERS 中时回退到 custom） + 2. ``SILICONFLOW_API_URL`` 含 ``siliconflow.cn`` → siliconflow + 3. ``SILICONFLOW_API_KEY`` 已设 → siliconflow（向后兼容） + 4. 默认 ``openai`` + + Returns: + provider name（``PROVIDERS`` 中的 key） + """ + explicit = (os.getenv("LLM_PROVIDER") or "").strip().lower() + if explicit: + if explicit in PROVIDERS: + return explicit + # 未识别的 provider 名称按 custom 处理（让用户自填 URL） + return "custom" + + legacy_url = (os.getenv("SILICONFLOW_API_URL") or "").strip().lower() + if "siliconflow.cn" in legacy_url: + return "siliconflow" + + if (os.getenv("SILICONFLOW_API_KEY") or "").strip(): + return "siliconflow" + + return "openai" + + +def get_provider_obj(name: Optional[str] = None) -> LLMProvider: + """ + 取得 provider 的 dataclass 实例（不存在时返回 custom）。 + """ + target = (name or resolve_provider()).lower() + return PROVIDERS.get(target, PROVIDERS["custom"]) + + +# ---------------------------------------------------------------------- +# 凭据 / URL / 模型 解析 +# ---------------------------------------------------------------------- +def resolve_api_key() -> str: + """ + 解析 API key。 + + 优先级：``LLM_API_KEY`` > ``OPENAI_API_KEY`` > ``SILICONFLOW_API_KEY`` > "" + """ + for k in ("LLM_API_KEY", "OPENAI_API_KEY", "SILICONFLOW_API_KEY"): + v = (os.getenv(k) or "").strip() + if v: + return v + return "" + + +def resolve_base_url(provider: Optional[str] = None) -> str: + """ + 解析 base_url。 + + 优先级：``LLM_BASE_URL`` > provider preset 的默认 base_url。 + custom 预设若 ``LLM_BASE_URL`` 未设则 raise ConfigError。 + """ + explicit = (os.getenv("LLM_BASE_URL") or "").strip() + if explicit: + return explicit + + obj = get_provider_obj(provider) + if obj.base_url: + return obj.base_url + + raise ConfigError("LLM_BASE_URL 未设置（custom 供应商必须显式提供）") + + +def resolve_model_name(provider: Optional[str] = None) -> str: + """ + 解析模型名。 + + 优先级：``LLM_MODEL_NAME`` > provider preset 的 default_model。 + custom 预设若 ``LLM_MODEL_NAME`` 未设则 raise ConfigError。 + """ + explicit = (os.getenv("LLM_MODEL_NAME") or "").strip() + if explicit: + return explicit + + obj = get_provider_obj(provider) + if obj.default_model: + return obj.default_model + + raise ConfigError("LLM_MODEL_NAME 未设置（custom 供应商必须显式提供）") + + +# ---------------------------------------------------------------------- +# Summary +# ---------------------------------------------------------------------- +def get_provider_summary() -> Dict[str, Any]: + """ + 返回当前生效的 provider 配置（供 /api/llm/config 端点使用）。 + + Returns: + { + "provider": str, + "base_url": str, + "model_name": str, + "api_key_set": bool, + } + + Raises: + ConfigError: 当 custom 预设缺少必填环境变量时。 + """ + provider_name = resolve_provider() + base_url = resolve_base_url(provider_name) + model_name = resolve_model_name(provider_name) + api_key = resolve_api_key() + return { + "provider": provider_name, + "base_url": base_url, + "model_name": model_name, + "api_key_set": bool(api_key), + } + + +__all__ = [ + "ConfigError", + "LLMProvider", + "PROVIDERS", + "resolve_provider", + "resolve_api_key", + "resolve_base_url", + "resolve_model_name", + "get_provider_obj", + "get_provider_summary", +] diff --git a/src/core/long_term_memory.py b/src/core/long_term_memory.py index 44c0448..d395628 100644 --- a/src/core/long_term_memory.py +++ b/src/core/long_term_memory.py @@ -2,20 +2,144 @@ 长效记忆管理模块 实现分层记忆系统：短期记忆（最近20轮）+ 长期概括记忆 + 知识库 使用数据库替代JSON文件存储 + +P2 增强： +- 新增 OpenLoopTracker（未完话题检测与延续） +- 与现有短期/长期记忆 / 知识库并行 +- 通过 DatabaseManager 持久化（与 LongTermMemoryManager 同库） """ import os import json import uuid -from datetime import datetime +import re +from datetime import datetime, timedelta from typing import List, Dict, Any, Optional from dotenv import load_dotenv import requests from src.core.database_manager import DatabaseManager from src.core.knowledge_base import KnowledgeBase +from src.core.llm_helper import LLMHelper +from src.core.prompt_manager import get_prompt_manager +from src.core import llm_providers # v3.1.0: 统一供应商解析 +from src.tools.debug_logger import get_debug_logger load_dotenv() +debug_logger = get_debug_logger() + +ENABLE_OPEN_LOOP = os.getenv('ENABLE_OPEN_LOOP', 'false').lower() == 'true' + + +# ==================== P2: OpenLoopTracker ==================== + +class OpenLoopTracker: + """ + 未完话题跟踪器。 + 每轮从用户消息中提取未完话题（pending tasks / 提问 / 计划 / 疑虑）， + 在合适时机主动延续。 + """ + + LOOP_TRIGGERS = [ + r'我[们]?(.+?)(明天|下周|以后|稍后|待会)[就再]', + r'(?:帮我|请|记得|别忘)(.+)', + r'(?:之后|下次)(?:再|记得|帮我)(.+)', + r'什么时候(.+)', + r'为什么(.+)', + r'(.+)怎么[办做]', + ] + + def __init__(self, db_manager: DatabaseManager = None, + prompt_manager=None, + max_active: int = 12): + self.db = db_manager or DatabaseManager() + self.prompt_manager = prompt_manager or get_prompt_manager() + self.enabled = ENABLE_OPEN_LOOP + self.max_active = max_active + + def update_from_message(self, user: str, text: str) -> Dict[str, Any]: + """ + 从用户消息中提取未完话题并入库。 + 返回 {topic, status, raised_at, related_keywords, uuid} 或 None（若未识别到）。 + """ + if not self.enabled or not text: + return {} + topic = self._extract_topic_via_rules(text) + keywords = self._extract_keywords(text) + if not topic and not keywords: + return {} + + if not topic: + topic = keywords[0] if keywords else text[:20] + context = text[:200] + + existing = self.db.find_matching_open_loop(topic) + if existing: + self.db.resolve_open_loop(existing['uuid']) + + loop_uuid = self.db.insert_open_loop( + topic=topic, context=context, + related_keywords=keywords, + ) + if not loop_uuid: + return {} + return { + 'uuid': loop_uuid, 'topic': topic, 'status': 'open', + 'raised_at': datetime.now().isoformat(), + 'related_keywords': keywords, + } + + def _extract_topic_via_rules(self, text: str) -> str: + for pat in self.LOOP_TRIGGERS: + m = re.search(pat, text) + if m: + captured = (m.group(1) or '').strip() + if 1 <= len(captured) <= 30: + return captured + return "" + + @staticmethod + def _extract_keywords(text: str) -> List[str]: + # 简化版：移除停用词，按空格 / 标点切分，保留长度 2-12 的 token + stop = {'的', '了', '我', '你', '是', '在', '和', '就', '都', '也', '吧', '啊', '吗', '呢'} + text = re.sub(r'[，。！？、,!?\.\s]+', ' ', text) + tokens = [t.strip() for t in text.split() if 2 <= len(t.strip()) <= 12] + return [t for t in tokens if t not in stop][:5] + + def resolve_matching_loop(self, text: str) -> Optional[Dict[str, Any]]: + """ + 若 text 命中已存在的未完话题，自动 resolve 并返回。 + """ + if not self.enabled or not text: + return None + match = self.db.find_matching_open_loop(text) + if not match: + return None + self.db.resolve_open_loop(match['uuid']) + return match + + def format_for_prompt(self, hint: str = "") -> str: + """ + 渲染为 prompt 文本（主流程第 4 步注入）。 + hint：可选的"对当前用户消息最相关的话题"提示。 + """ + if not self.enabled: + return "" + loops = self.db.get_open_loops(status='open', limit=self.max_active) + if not loops: + return "" + lines = ["【未完话题】"] + for lp in loops: + topic = lp.get('topic', '?') + keywords = lp.get('related_keywords', []) + if keywords: + lines.append(f"• {topic}（关键词：{', '.join(keywords[:3])}）") + else: + lines.append(f"• {topic}") + if hint: + lines.append(f"\n当前消息与「{hint}」相关，可在合适时机主动延续。") + return "\n".join(lines) + class LongTermMemoryManager: """ @@ -49,8 +173,13 @@ def __init__(self, self.knowledge_extraction_interval = 5 # API配置（用于生成概括） - self.api_key = api_key or os.getenv('SILICONFLOW_API_KEY') - self.api_url = api_url or os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = api_key or llm_providers.resolve_api_key() + try: + self.api_url = api_url or llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + # 启动期 LLM_BASE_URL 缺失时（custom）保留旧 URL + self.api_url = api_url or os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = model_name or os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') # 初始化知识库（共享数据库管理器） @@ -425,6 +554,150 @@ def get_context_for_chat(self, recent_count: int = 10) -> str: return "\n".join(context_parts) if context_parts else "" + # ==================== Stage B.4 Web 后端兼容接口 ==================== + + def list_topics( + self, + user_id: str = "default", + days: int = 7, + limit: int = 200, + ) -> List[Dict[str, Any]]: + """ + 列出最近 N 天内的话题（基于 long_term_memory 概括）。 + + Args: + user_id: 用户标识（当前 schema 未持久化 user_id，保留参数以备扩展）。 + days: 时间范围（天），默认 7。 + limit: 返回条数上限，默认 200。 + + Returns: + 话题字典列表，字段：uuid / topic / start_time / end_time / message_count。 + 失败时返回空列表。 + """ + try: + days_int = max(1, min(int(days), 365)) + except (TypeError, ValueError): + days_int = 7 + try: + limit_int = max(1, min(int(limit), 1000)) + except (TypeError, ValueError): + limit_int = 200 + + try: + cutoff = (datetime.now() - timedelta(days=days_int)).isoformat() + except Exception: # noqa: BLE001 + cutoff = datetime.now().isoformat() + + rows: List[Dict[str, Any]] = [] + try: + summaries = self.db.get_long_term_summaries() or [] + except Exception: # noqa: BLE001 + return [] + + for item in summaries: + try: + created_at = item.get('created_at') or '' + if created_at and created_at < cutoff: + continue + topic_text = (item.get('summary') or '').strip() + # topic 取 summary 前若干字符作为短标题 + topic_short = topic_text[:40] + ('…' if len(topic_text) > 40 else '') + try: + msg_count = int(item.get('message_count') or 0) + except (TypeError, ValueError): + msg_count = 0 + rows.append({ + 'uuid': str(item.get('uuid') or ''), + 'topic': topic_short or '未命名话题', + 'start_time': str(item.get('created_at') or ''), + 'end_time': str(item.get('ended_at') or '') or None, + 'message_count': msg_count, + }) + if len(rows) >= limit_int: + break + except Exception: # noqa: BLE001 + # 单条记录解析失败：跳过这一条，继续下一条 + continue + + # 按 start_time 倒序 + try: + rows.sort(key=lambda x: x.get('start_time') or '', reverse=True) + except Exception: # noqa: BLE001 + pass + return rows + + def get_topic_context( + self, + topic_uuid: str, + user_id: str = "default", + ) -> Dict[str, Any]: + """ + 获取话题的上下文消息（Stage B.4 Web 后端专用）。 + + 重要：当前 schema 中归档时会从 short_term_memory 物理删除原消息， + 因此 ``messages`` 字段以 long_term_memory.summary 自身 + 元数据作为兜底。 + 若未来在 short_term_memory 增加 topic_uuid 关联列，本方法可平滑扩展。 + + Returns: + dict: {topic_uuid, topic, summary, messages, start_time, end_time} + 失败 / 未找到时返回 {topic_uuid, messages: []}。 + """ + if not topic_uuid or not isinstance(topic_uuid, str): + return {'topic_uuid': str(topic_uuid or ''), 'messages': []} + + item: Optional[Dict[str, Any]] = None + try: + summaries = self.db.get_long_term_summaries() or [] + for s in summaries: + try: + if str(s.get('uuid') or '') == topic_uuid: + item = s + break + except Exception: # noqa: BLE001 + continue + except Exception: # noqa: BLE001 + item = None + + if not item: + return {'topic_uuid': topic_uuid, 'messages': []} + + summary_text = (item.get('summary') or '').strip() + topic_short = summary_text[:40] + ('…' if len(summary_text) > 40 else '') + started_at = str(item.get('created_at') or '') + ended_at = str(item.get('ended_at') or '') or None + + # messages：把 summary 拆成系统消息；并附加元数据作为 assistant 消息 + messages: List[Dict[str, Any]] = [] + if summary_text: + messages.append({ + 'role': 'system', + 'content': f"[话题摘要] {summary_text}", + 'timestamp': started_at, + }) + try: + msg_count = int(item.get('message_count') or 0) + except (TypeError, ValueError): + msg_count = 0 + try: + rounds = int(item.get('rounds') or 0) + except (TypeError, ValueError): + rounds = 0 + if msg_count or rounds: + messages.append({ + 'role': 'assistant', + 'content': f"该话题共 {rounds} 轮 / {msg_count} 条消息。", + 'timestamp': ended_at or started_at, + }) + + return { + 'topic_uuid': topic_uuid, + 'topic': topic_short or '未命名话题', + 'summary': summary_text or None, + 'messages': messages, + 'start_time': started_at or None, + 'end_time': ended_at, + } + if __name__ == '__main__': print("=" * 60) diff --git a/src/core/model_config.py b/src/core/model_config.py index f672b20..b53aced 100644 --- a/src/core/model_config.py +++ b/src/core/model_config.py @@ -1,6 +1,12 @@ """ 多层模型配置模块 管理不同类型任务使用的不同模型配置 + +v3.1.0: 改用 llm_providers 抽象层； + - api_url 通过 resolve_base_url() 解析 + - api_key 通过 resolve_api_key() 解析 + - 三层模型各自支持独立 *_MODEL_NAME + 可选独立 *_MODEL_BASE_URL + - is_valid() 检查有效 key（排除占位符 / 空串） """ import os @@ -8,6 +14,8 @@ from typing import Dict, Any, Optional from dotenv import load_dotenv +from src.core import llm_providers # v3.1.0: 供应商注册表 + load_dotenv() @@ -23,51 +31,59 @@ class ModelConfig: 模型配置类 提供多层模型架构的配置管理 """ - + def __init__(self): """初始化模型配置""" - # API配置 - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') - + # v3.1.0: 用 llm_providers 解析 url + key（保留旧 SILICONFLOW_* fallback） + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + # 启动期 LLM_BASE_URL 缺失时（custom）不要阻塞；保留空串 + self.api_url = "" + self.api_key = llm_providers.resolve_api_key() + # 主模型配置 self.main_model = { - 'name': os.getenv('MAIN_MODEL_NAME', os.getenv('MODEL_NAME', 'deepseek-ai/DeepSeek-V3.2')), + 'name': os.getenv('MAIN_MODEL_NAME', os.getenv('MODEL_NAME', os.getenv( + 'LLM_MODEL_NAME', llm_providers.resolve_model_name(llm_providers.resolve_provider())))), 'temperature': float(os.getenv('MAIN_MODEL_TEMPERATURE', os.getenv('TEMPERATURE', '0.8'))), - 'max_tokens': int(os.getenv('MAIN_MODEL_MAX_TOKENS', os.getenv('MAX_TOKENS', '2000'))) + 'max_tokens': int(os.getenv('MAIN_MODEL_MAX_TOKENS', os.getenv('MAX_TOKENS', '2000'))), + 'base_url': os.getenv('MAIN_MODEL_BASE_URL'), # 可选；None 时回退到 self.api_url } - + # 小模型配置（工具级任务） self.tool_model = { 'name': os.getenv('TOOL_MODEL_NAME', 'zai-org/GLM-4.6V'), 'temperature': float(os.getenv('TOOL_MODEL_TEMPERATURE', '0.3')), - 'max_tokens': int(os.getenv('TOOL_MODEL_MAX_TOKENS', '500')) + 'max_tokens': int(os.getenv('TOOL_MODEL_MAX_TOKENS', '500')), + 'base_url': os.getenv('TOOL_MODEL_BASE_URL'), } - + # 多模态模型配置 self.vision_model = { 'name': os.getenv('VISION_MODEL_NAME', 'Qwen/Qwen3-VL-32B-Instruct'), 'temperature': float(os.getenv('VISION_MODEL_TEMPERATURE', '0.5')), - 'max_tokens': int(os.getenv('VISION_MODEL_MAX_TOKENS', '1000')) + 'max_tokens': int(os.getenv('VISION_MODEL_MAX_TOKENS', '1000')), + 'base_url': os.getenv('VISION_MODEL_BASE_URL'), } - + # 视觉工具LLM配置（向后兼容） self.vision_llm_temperature = float(os.getenv('VISION_LLM_TEMPERATURE', '0.3')) self.vision_llm_max_tokens = int(os.getenv('VISION_LLM_MAX_TOKENS', '10')) self.vision_llm_timeout = int(os.getenv('VISION_LLM_TIMEOUT', '10')) - + # NPS工具系统LLM配置（向后兼容） self.nps_llm_temperature = float(os.getenv('NPS_LLM_TEMPERATURE', '0.3')) self.nps_llm_max_tokens = int(os.getenv('NPS_LLM_MAX_TOKENS', '100')) self.nps_llm_timeout = int(os.getenv('NPS_LLM_TIMEOUT', '10')) - + def get_model_config(self, model_type: ModelType) -> Dict[str, Any]: """ 获取指定类型的模型配置 - + Args: model_type: 模型类型 - + Returns: 包含模型配置的字典 """ @@ -80,36 +96,43 @@ def get_model_config(self, model_type: ModelType) -> Dict[str, Any]: else: # 默认返回主模型配置 return self.main_model.copy() - + def get_api_config(self) -> Dict[str, str]: """ 获取API配置 - + Returns: 包含API密钥和URL的字典 """ return { 'api_key': self.api_key, - 'api_url': self.api_url + 'api_url': self.api_url, } - + def is_valid(self) -> bool: """ - 检查配置是否有效 - + 检查配置是否有效（v3.1.0：排除 'your-api-key' 占位 + 空串）。 + Returns: 配置是否有效 """ - return bool(self.api_key and self.api_key != 'your-api-key') - + if not self.api_key: + return False + return self.api_key not in ('your-api-key', '') + def get_summary(self) -> str: """ - 获取配置摘要 - + 获取配置摘要（v3.1.0：输出 provider 名）。 + Returns: 配置摘要字符串 """ + try: + provider = llm_providers.resolve_provider() + except Exception: + provider = "unknown" return f"""多层模型配置: + provider: {provider} 主模型: {self.main_model['name']} 工具模型: {self.tool_model['name']} 多模态模型: {self.vision_model['name']} @@ -123,7 +146,7 @@ def get_summary(self) -> str: def get_model_config() -> ModelConfig: """ 获取全局模型配置实例（单例模式） - + Returns: ModelConfig实例 """ diff --git a/src/core/proactive_engine.py b/src/core/proactive_engine.py new file mode 100644 index 0000000..28ba7d8 --- /dev/null +++ b/src/core/proactive_engine.py @@ -0,0 +1,403 @@ +""" +主动决策与想法池模块 +- 想法池（impulse pool）：多来源候选（habit_care / open_loop_recall / creative_share / weather_note） +- 核心决策：should_send 综合多因子（驱动力 / 就绪度 / 关系温度 / 闸门 / 复核） +- 主动消息经 EventManager 投递回 GUI + +参考架构：spec §4 ADDED Requirements → ProactiveEngine +""" + +import os +import json +import random +import uuid +from datetime import datetime, timedelta +from typing import Dict, List, Any, Optional, Tuple + +from src.core.database_manager import DatabaseManager +from src.core.llm_helper import LLMHelper +from src.core.prompt_manager import get_prompt_manager +from src.core.event_manager import EventManager, EventPriority, NotificationEvent +from src.tools.debug_logger import get_debug_logger + +ENABLE_PROACTIVE_ENGINE = os.getenv('ENABLE_PROACTIVE_ENGINE', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +class ProactiveEngine: + """ + 主动决策与想法池。 + - bot_proactive_drive: Bot 内在驱动（创作冲动 / 梦境残留 / 生活状态影响） + - proactive_inner_readiness: 当下情绪就绪度（情绪轮 + 关系温度） + - proactive_impulse_pool: 候选想法（habit_care / open_loop_recall / creative_share / weather_note） + - should_send: 多因子决策 + - bot_proactive_drive: 触发主动消息经 EventManager 投递 + """ + + QUIET_HOURS_START = 23 + QUIET_HOURS_END = 7 + DEFAULT_DAILY_LIMIT = 2 + MIN_INTERVAL_MINUTES = 60 + DEFAULT_REVIEW_COOLDOWN_HOURS = 4 + + def __init__(self, db_manager: DatabaseManager = None, + prompt_manager=None, + event_manager: Optional[EventManager] = None, + life_state=None, + emotion_wheel=None, + open_loop_tracker=None, + creative_writer=None, + user_habits=None): + self.db = db_manager or DatabaseManager() + self.prompt_manager = prompt_manager or get_prompt_manager() + self.event_manager = event_manager + self.life_state = life_state + self.emotion_wheel = emotion_wheel + self.open_loop_tracker = open_loop_tracker + self.creative_writer = creative_writer + self.user_habits = user_habits + self.enabled = ENABLE_PROACTIVE_ENGINE + + # ===== 多因子决策 ===== + + def should_send(self, user: str, + now: Optional[datetime] = None) -> Tuple[bool, str]: + """ + 多因子决策： + - bot_proactive_drive + - proactive_inner_readiness + - 静默时段闸门 + - effective_user_daily_limit + - 复核冷却 + 返回 (bool, reason)。 + """ + if not self.enabled: + return False, 'disabled' + now = now or datetime.now() + if self._is_quiet_time(now): + return False, 'quiet_hours' + if not self._cooldown_passed(user, now): + return False, 'cooldown' + if self._daily_limit_reached(user, now): + return False, 'daily_limit' + drive = self.bot_proactive_drive(user, now=now) + readiness = self.proactive_inner_readiness(user, now=now) + # 简单阈值：驱动力 + 就绪度 >= 0.5 + score = drive.get('score', 0.0) + readiness.get('score', 0.0) + if score < 0.5: + return False, f'low_score({score:.2f})' + return True, f'pass(score={score:.2f})' + + def schedule_next_proactive(self, user: str, now: Optional[datetime] = None, + delay_hours: Optional[float] = None) -> None: + """记录下次主动消息时间。""" + if not self.enabled: + return + now = now or datetime.now() + if delay_hours is None: + delay_hours = random.uniform(2.0, 6.0) + next_at = (now + timedelta(hours=delay_hours)).isoformat() + self.db.set_metadata(f'next_proactive_{user}', next_at) + debug_logger.log_info('ProactiveEngine', '已排定下次主动消息', { + 'user': user, 'next_at': next_at, + }) + + # ===== 内在驱动 / 就绪度 ===== + + def bot_proactive_drive(self, user: Optional[str] = None, + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + Bot 内在驱动：综合 创作冲动 + 梦境残留 + 生活状态 + 返回 {score(0-1), components: {creative, dream, life_state}} + """ + if not self.enabled: + return {'score': 0.0, 'components': {}} + creative = 0.0 + if self.creative_writer and getattr(self.creative_writer, 'enabled', False): + try: + finished = self.creative_writer.get_finished_projects(limit=3) + creative = min(1.0, 0.4 * len(finished)) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'创作驱动评估失败: {e}', e) + + dream = 0.0 + if self.life_state and getattr(self.life_state, 'enabled', False): + try: + snap = self.life_state.get_current_state_snapshot() or {} + # 疲惫/低落时降低驱动 + mood = snap.get('mood', '') + if mood in ('焦虑', '低落', '疲倦'): + dream = 0.2 + else: + dream = 0.5 + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'生活状态评估失败: {e}', e) + + score = (creative + dream) / 2.0 + return { + 'score': round(score, 3), + 'components': { + 'creative': round(creative, 3), + 'dream': round(dream, 3), + }, + } + + def proactive_inner_readiness(self, user: str, + now: Optional[datetime] = None) -> Dict[str, Any]: + """ + 当下情绪就绪度：情绪轮正向 + 关系温度 + 返回 {score(0-1), emotion_summary} + """ + if not self.enabled: + return {'score': 0.0, 'emotion_summary': ''} + score = 0.0 + summary = '' + if self.emotion_wheel and getattr(self.emotion_wheel, 'enabled', False): + try: + state = {'emotions': {}} + # 简化的情绪轮评分 + score = 0.5 + summary = '情绪轮活跃' + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'情绪就绪度评估失败: {e}', e) + + # 关系温度：从情感分析历史读取 + try: + latest = self.db.get_latest_emotion() + if latest: + overall = int(latest.get('overall_score', 0)) + if overall >= 60: + score = max(score, 0.6) + elif overall < 30: + score = min(score, 0.3) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'关系温度评估失败: {e}', e) + + return {'score': round(score, 3), 'emotion_summary': summary} + + # ===== 想法池 ===== + + def proactive_impulse_pool(self, user: str) -> List[Dict[str, Any]]: + """ + 候选想法列表。 + 来源：habit_care / open_loop_recall / creative_share / weather_note + """ + if not self.enabled: + return [] + ideas: List[Dict[str, Any]] = [] + + # habit_care + if self.user_habits and getattr(self.user_habits, 'enabled', False): + try: + ideas.extend(self._ideas_from_habits(user)) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'habit 想法失败: {e}', e) + + # open_loop_recall + if self.open_loop_tracker and getattr(self.open_loop_tracker, 'enabled', False): + try: + ideas.extend(self._ideas_from_open_loops(user)) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'open_loop 想法失败: {e}', e) + + # creative_share + if self.creative_writer and getattr(self.creative_writer, 'enabled', False): + try: + ideas.extend(self._ideas_from_creative(user)) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'creative 想法失败: {e}', e) + + # weather_note + ideas.extend(self._ideas_from_weather(user)) + return ideas + + def _ideas_from_habits(self, user: str) -> List[Dict[str, Any]]: + ideas = [] + try: + event = self.user_habits.habit_proactive_event(user) + except Exception: + return ideas + if not event: + return ideas + ideas.append({ + 'id': str(uuid.uuid4()), + 'reason': 'habit_care', + 'motive': event.get('content', '该吃饭了'), + 'window_start': datetime.now().isoformat(), + 'window_end': (datetime.now() + timedelta(hours=2)).isoformat(), + 'priority': 'normal', + 'used': False, + }) + return ideas + + def _ideas_from_open_loops(self, user: str) -> List[Dict[str, Any]]: + ideas = [] + try: + loops = self.db.get_open_loops(status='open', limit=3) + except Exception: + return ideas + for lp in loops: + ideas.append({ + 'id': str(uuid.uuid4()), + 'reason': 'open_loop_recall', + 'motive': f"想起之前的「{lp.get('topic', '?')}」，要不要继续聊聊？", + 'window_start': datetime.now().isoformat(), + 'window_end': (datetime.now() + timedelta(hours=3)).isoformat(), + 'priority': 'low', + 'used': False, + 'related_loop_uuid': lp.get('uuid'), + }) + return ideas + + def _ideas_from_creative(self, user: str) -> List[Dict[str, Any]]: + ideas = [] + try: + finished = self.creative_writer.get_finished_projects(limit=3) + except Exception: + return ideas + for proj in finished: + ideas.append({ + 'id': str(uuid.uuid4()), + 'reason': 'creative_share', + 'motive': f"刚把《{proj.get('title', '?')}》写完了，要不要看看？", + 'window_start': datetime.now().isoformat(), + 'window_end': (datetime.now() + timedelta(hours=4)).isoformat(), + 'priority': 'normal', + 'used': False, + 'related_project_uuid': proj.get('uuid'), + }) + return ideas + + def _ideas_from_weather(self, user: str) -> List[Dict[str, Any]]: + # 简化的天气提示：随机注入（实际接入 weather 模块时替换） + notes = [ + '今天天气不错，要不要出去走走？', + '外面下雨啦，记得带伞。', + '天气转凉了，多穿点。', + ] + return [{ + 'id': str(uuid.uuid4()), + 'reason': 'weather_note', + 'motive': random.choice(notes), + 'window_start': datetime.now().isoformat(), + 'window_end': (datetime.now() + timedelta(hours=6)).isoformat(), + 'priority': 'low', + 'used': False, + }] + + # ===== 投递 ===== + + def bot_proactive_drive_send(self, user: str, + idea: Dict[str, Any]) -> bool: + """ + 触发主动消息并经 EventManager 投递回 GUI / Web 端。 + + Stage D.2: 通过 EventManager.publish("proactive_message", payload) 推送 + 独立 EventManager 单例 + try/except 保护，绝不阻塞主流程。 + """ + if not self.enabled: + return False + message = (idea or {}).get('motive', '') if isinstance(idea, dict) else '' + if not message: + debug_logger.log_info('ProactiveEngine', 'idea 为空，跳过投递') + return False + + payload: Dict[str, Any] = { + 'content': message, + 'timestamp': datetime.now().isoformat(), + 'user': user, + 'reason': (idea or {}).get('reason', '') if isinstance(idea, dict) else '', + 'idea_id': (idea or {}).get('id', '') if isinstance(idea, dict) else '', + 'priority': (idea or {}).get('priority', 'normal') if isinstance(idea, dict) else 'normal', + 'source': 'ProactiveEngine', + } + + # Stage D.2: 优先使用注入的 event_manager；否则走全局单例 + em = self.event_manager + if em is None: + try: + from src.core.event_manager import get_event_manager + em = get_event_manager() + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'获取 EventManager 失败: {e}', e) + em = None + + # 推送给 EventManager（绝对 try/except，绝不阻塞主流程） + try: + if em is not None: + em.publish('proactive_message', payload) + else: + # 兜底：保留旧的 NotificationEvent 写库方式（如果 EventManager 不可用） + event = NotificationEvent( + event_id=str(uuid.uuid4()), + title='主动消息', + description=message, + source='ProactiveEngine', + priority=EventPriority.MEDIUM, + metadata={'content': message, 'user': user, + 'reason': payload['reason'], + 'idea_id': payload['idea_id']}, + ) + # 即便 fallback 也仅记录日志，不抛出 + debug_logger.log_info('ProactiveEngine', 'fallback 通知(无 EventManager)', { + 'event_id': event.event_id, + }) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'proactive_message 推送失败: {e}', e) + + debug_logger.log_info('ProactiveEngine', '主动消息已投递', { + 'user': user, 'reason': payload['reason'], + }) + # 更新今日主动次数 + try: + self._increment_daily_count(user) + except Exception as e: + debug_logger.log_error('ProactiveEngine', f'更新今日主动次数失败: {e}', e) + return True + + # ===== 工具方法 ===== + + def _is_quiet_time(self, now: datetime) -> bool: + h = now.hour + if self.QUIET_HOURS_START > self.QUIET_HOURS_END: + return h >= self.QUIET_HOURS_START or h < self.QUIET_HOURS_END + return self.QUIET_HOURS_START <= h < self.QUIET_HOURS_END + + def _cooldown_passed(self, user: str, now: datetime) -> bool: + last = self.db.get_metadata(f'last_proactive_{user}', '') + if not last: + return True + try: + last_dt = datetime.fromisoformat(last) + except (ValueError, TypeError): + return True + delta = (now - last_dt).total_seconds() / 60.0 + return delta >= self.MIN_INTERVAL_MINUTES + + def _daily_limit_reached(self, user: str, now: datetime) -> bool: + today = now.date().isoformat() + last_day = self.db.get_metadata(f'last_proactive_day_{user}', '') + if last_day != today: + return False + count = int(self.db.get_metadata(f'proactive_count_{today}_{user}', 0)) + return count >= self.effective_user_daily_limit(user, now) + + def effective_user_daily_limit(self, user: str, now: datetime) -> int: + """默认上限；可被角色设定或用户偏好覆盖。""" + override = self.db.get_metadata(f'proactive_limit_{user}', None) + if override is not None: + try: + return int(override) + except (TypeError, ValueError): + pass + return self.DEFAULT_DAILY_LIMIT + + def _increment_daily_count(self, user: str) -> None: + now = datetime.now() + today = now.date().isoformat() + key = f'proactive_count_{today}_{user}' + count = int(self.db.get_metadata(key, 0)) + 1 + self.db.set_metadata(key, count) + self.db.set_metadata(f'last_proactive_day_{user}', today) + self.db.set_metadata(f'last_proactive_{user}', now.isoformat()) diff --git a/src/core/schedule_generator.py b/src/core/schedule_generator.py index ac2d3f8..895d5fd 100644 --- a/src/core/schedule_generator.py +++ b/src/core/schedule_generator.py @@ -12,6 +12,7 @@ from dotenv import load_dotenv import requests from src.core.schedule_manager import ScheduleManager, ScheduleType, SchedulePriority +from src.core import llm_providers # v3.1.0: 统一供应商解析 from src.tools.debug_logger import get_debug_logger load_dotenv() @@ -34,8 +35,12 @@ def __init__(self, schedule_manager: ScheduleManager = None): schedule_manager: 日程管理器实例 """ self.schedule_manager = schedule_manager or ScheduleManager() - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = llm_providers.resolve_api_key() + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') self.temperature = 0.8 self.max_tokens = 1000 diff --git a/src/core/schedule_similarity_checker.py b/src/core/schedule_similarity_checker.py index 39fea61..26323a6 100644 --- a/src/core/schedule_similarity_checker.py +++ b/src/core/schedule_similarity_checker.py @@ -17,6 +17,7 @@ pass import requests +from src.core import llm_providers # v3.1.0: 统一供应商解析 from src.tools.debug_logger import get_debug_logger # 获取debug日志记录器 @@ -31,8 +32,12 @@ class ScheduleSimilarityChecker: def __init__(self): """初始化日程相似度检查工具""" - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = llm_providers.resolve_api_key() + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') self.temperature = 0.3 # 使用较低的温度以获得更确定的输出 self.max_tokens = 800 diff --git a/src/core/self_timeline.py b/src/core/self_timeline.py new file mode 100644 index 0000000..78bf93e --- /dev/null +++ b/src/core/self_timeline.py @@ -0,0 +1,247 @@ +""" +自我时间线聚合模块 +聚合多来源输出 Bot 自我时间线（按相关性评分排序）。 +来源：日程 / 对话 / 创作（P3）/ 记忆 / 梦境日记（P1） + +参考架构：spec §Exploratory → SelfTimelineAggregator +""" + +import os +import re +from datetime import datetime, timedelta +from typing import Dict, List, Any, Optional + +from src.core.database_manager import DatabaseManager +from src.tools.debug_logger import get_debug_logger + +ENABLE_SELF_TIMELINE = os.getenv('ENABLE_SELF_TIMELINE', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +class SelfTimelineAggregator: + """ + 自我时间线聚合器。 + - aggregate_timeline(user, query, now): 输出按相关性评分排序的 list[dict] + - ChatAgent.chat() 检测到时间线询问关键词时调用 + """ + + QUERY_KEYWORDS = [ + r'你今天', r'你昨天', r'你最近', r'你早上', r'你下午', + r'你晚上', r'你这周', r'你这几天', r'你上午', + ] + + def __init__(self, db_manager: DatabaseManager = None, + schedule_manager=None, + long_term_memory=None, + creative_writer=None, + knowledge_base=None, + dream_diary=None): + self.db = db_manager or DatabaseManager() + self.schedule_manager = schedule_manager + self.long_term_memory = long_term_memory + self.creative_writer = creative_writer + self.knowledge_base = knowledge_base + self.dream_diary = dream_diary + self.enabled = ENABLE_SELF_TIMELINE + + def is_timeline_query(self, text: str) -> bool: + if not text: + return False + return any(re.search(p, text) for p in self.QUERY_KEYWORDS) + + def aggregate_timeline(self, user: str, query: str, + now: Optional[datetime] = None, + window_days: int = 7) -> List[Dict[str, Any]]: + """ + 聚合多来源时间线事件。 + 每条 {source, content, time, score, metadata} + """ + if not self.enabled: + return [] + now = now or datetime.now() + events: List[Dict[str, Any]] = [] + # 1) 日程 + events.extend(self._from_schedule(user, now, window_days)) + # 2) 对话 + events.extend(self._from_conversation(user, now, window_days)) + # 3) 创作 + events.extend(self._from_creative(user, now, window_days)) + # 4) 记忆 + events.extend(self._from_knowledge(user, now, window_days)) + # 5) 梦境日记 + events.extend(self._from_dream_diary(user, now, window_days)) + # 相关性评分 + for e in events: + e['score'] = self._score_event(e, query, now) + events.sort(key=lambda e: e['score'], reverse=True) + return events + + def format_for_prompt(self, events: List[Dict[str, Any]], + limit: int = 5) -> str: + if not events: + return "" + lines = ["【我最近在做的事】"] + for e in events[:limit]: + t = e.get('time', '') + src = e.get('source', '?') + content = e.get('content', '')[:80] + lines.append(f"• [{src}] {t}：{content}") + return "\n".join(lines) + + # ===== 来源 ===== + + def _from_schedule(self, user: str, now: datetime, + window_days: int) -> List[Dict[str, Any]]: + if not self.schedule_manager: + return [] + try: + start = (now - timedelta(days=window_days)).isoformat() + end = now.isoformat() + schedules = self.schedule_manager.get_schedules_by_time_range( + start, end, queryable_only=True + ) + except Exception as e: + debug_logger.log_error('SelfTimeline', f'日程聚合失败: {e}', e) + return [] + return [{ + 'source': 'schedule', + 'content': f"{s.title}：{s.description or ''}", + 'time': s.start_time if hasattr(s, 'start_time') else '', + 'metadata': {'schedule_id': getattr(s, 'schedule_id', '')}, + } for s in schedules] + + def _from_conversation(self, user: str, now: datetime, + window_days: int) -> List[Dict[str, Any]]: + try: + messages = self.db.get_short_term_messages() + except Exception: + return [] + events = [] + for m in messages: + try: + t = m.get('timestamp', '') + if not t: + continue + msg_dt = datetime.fromisoformat(t) + if (now - msg_dt) > timedelta(days=window_days): + continue + content = (m.get('content', '') or '')[:120] + if m.get('role') == 'user': + content = f"用户说：{content}" + else: + content = f"我回复：{content}" + events.append({ + 'source': 'conversation', + 'content': content, + 'time': t, + 'metadata': {'role': m.get('role')}, + }) + except (ValueError, TypeError): + continue + return events + + def _from_creative(self, user: str, now: datetime, + window_days: int) -> List[Dict[str, Any]]: + try: + projects = self.db.list_creative_projects(limit=20) + except Exception: + return [] + events = [] + for p in projects: + try: + t = p.get('created_at', '') + if not t: + continue + dt = datetime.fromisoformat(t) + if (now - dt) > timedelta(days=window_days): + continue + events.append({ + 'source': 'creative', + 'content': f"创作项目《{p.get('title', '?')}》（{p.get('status', '?')}）", + 'time': t, + 'metadata': {'project_uuid': p.get('uuid')}, + }) + except (ValueError, TypeError): + continue + return events + + def _from_knowledge(self, user: str, now: datetime, + window_days: int) -> List[Dict[str, Any]]: + try: + # 简化：列出最近 10 条实体 + cursor = self.db.get_connection().__enter__() + rows = cursor.execute( + 'SELECT * FROM entities ORDER BY created_at DESC LIMIT 10' + ).fetchall() + except Exception: + return [] + events = [] + for r in rows: + try: + t = r['created_at'] + dt = datetime.fromisoformat(t) + if (now - dt) > timedelta(days=window_days): + continue + events.append({ + 'source': 'knowledge', + 'content': f"认识了「{r['name']}」", + 'time': t, + 'metadata': {'entity_uuid': r['uuid']}, + }) + except (KeyError, ValueError, TypeError): + continue + return events + + def _from_dream_diary(self, user: str, now: datetime, + window_days: int) -> List[Dict[str, Any]]: + events = [] + try: + dreams = self.db.get_dream_records(limit=20) if hasattr(self.db, 'get_dream_records') else [] + except Exception: + dreams = [] + for d in dreams: + try: + t = d.get('date') or d.get('created_at', '') + if not t: + continue + dt = datetime.fromisoformat(t) + if (now - dt) > timedelta(days=window_days): + continue + events.append({ + 'source': 'dream_diary', + 'content': f"梦境：{d.get('content', '')[:60]}", + 'time': t, + 'metadata': {'type': d.get('dream_type', '')}, + }) + except (ValueError, TypeError): + continue + return events + + # ===== 相关性评分 ===== + + def _score_event(self, event: Dict[str, Any], query: str, now: datetime) -> float: + score = 0.0 + # 时间衰减：越近越高 + try: + t = datetime.fromisoformat(event.get('time', '')) + delta_h = max(0.0, (now - t).total_seconds() / 3600.0) + score += max(0.0, 1.0 - delta_h / (24 * 7)) + except (ValueError, TypeError): + pass + # 关键词匹配 + q_tokens = set(re.findall(r'[\u4e00-\u9fffA-Za-z0-9]+', query or '')) + e_tokens = set(re.findall(r'[\u4e00-\u9fffA-Za-z0-9]+', event.get('content', ''))) + if q_tokens and e_tokens: + overlap = len(q_tokens & e_tokens) + score += min(1.0, overlap * 0.2) + # 来源权重 + weight = { + 'conversation': 0.3, + 'schedule': 0.4, + 'creative': 0.2, + 'dream_diary': 0.1, + 'knowledge': 0.1, + }.get(event.get('source', ''), 0.1) + score += weight + return round(score, 3) diff --git a/src/core/user_habits.py b/src/core/user_habits.py new file mode 100644 index 0000000..de88ba6 --- /dev/null +++ b/src/core/user_habits.py @@ -0,0 +1,170 @@ +""" +用户行为习惯跟踪模块 +- 提取用户作息 / 饮食 / 工作 / 娱乐 4 类行为习惯 +- 准入规则：同一习惯候选需在 ≥2 个不同自然日被提取才进入 qualified_habits +- 为 ProactiveEngine.proactive_impulse_pool 提供 habit_care 想法来源 + +参考架构：spec §Exploratory → UserHabitTracker +""" + +import os +import json +import uuid +from datetime import datetime +from typing import Dict, List, Any, Optional + +from src.core.database_manager import DatabaseManager +from src.core.llm_helper import LLMHelper +from src.tools.debug_logger import get_debug_logger + +ENABLE_USER_HABITS = os.getenv('ENABLE_USER_HABITS', 'false').lower() == 'true' + +debug_logger = get_debug_logger() + + +class UserHabitTracker: + """ + 用户行为习惯跟踪器。 + - update_from_message: 从用户消息中提取习惯候选 + - qualified_habits: 返回已通过 ≥2 自然日准入规则的习惯 + - habit_proactive_event: 触发习惯相关主动事件 + - format_for_schedule: 渲染为可注入 schedule 的格式 + """ + + CATEGORIES = ['作息', '饮食', '工作', '娱乐'] + MIN_DAYS_FOR_QUALIFY = 2 + + def __init__(self, db_manager: DatabaseManager = None): + self.db = db_manager or DatabaseManager() + self.enabled = ENABLE_USER_HABITS + + def update_from_message(self, user: str, text: str) -> List[Dict[str, Any]]: + """ + 从用户消息中提取习惯候选。 + 返回 [{category, content, observed_date}, ...]（仅当天新观察到的）。 + """ + if not self.enabled or not text: + return [] + # 极简规则：基于关键词分类 + text_lower = text.lower() + observed_date = datetime.now().date().isoformat() + candidates = [] + # 作息 + if any(kw in text for kw in ['起床', '睡觉', '晚安', '早起', '熬夜', '失眠']): + candidates.append({'category': '作息', 'content': text[:80]}) + # 饮食 + if any(kw in text for kw in ['吃饭', '早餐', '午餐', '晚餐', '宵夜', '喝', '咖啡', '奶茶', '饿了', '饱']): + candidates.append({'category': '饮食', 'content': text[:80]}) + # 工作 + if any(kw in text for kw in ['加班', '开会', '项目', '任务', '工作', '上班', '下班', '出差']): + candidates.append({'category': '工作', 'content': text[:80]}) + # 娱乐 + if any(kw in text for kw in ['看', '玩', '听', '游戏', '电影', '剧', '音乐', '运动', '跑步', '逛街']): + candidates.append({'category': '娱乐', 'content': text[:80]}) + # 存储 + for c in candidates: + self.db.set_metadata( + f'habit_{user}_{c["category"]}_{observed_date}_{uuid.uuid4().hex[:8]}', + json.dumps({ + 'category': c['category'], + 'content': c['content'], + 'observed_date': observed_date, + }, ensure_ascii=False) + ) + return [{**c, 'observed_date': observed_date} for c in candidates] + + def qualified_habits(self, user: str) -> List[Dict[str, Any]]: + """ + 返回已通过 ≥ MIN_DAYS_FOR_QUALIFY 个不同自然日准入规则的习惯。 + """ + if not self.enabled: + return [] + all_meta = self.db.list_metadata_keys(prefix=f'habit_{user}_') + by_key: Dict[str, Dict[str, set]] = {} + for key in all_meta: + value = self.db.get_metadata(key, None) + if not value: + continue + try: + data = json.loads(value) + except (TypeError, ValueError): + continue + cat = data.get('category', '其他') + content = data.get('content', '')[:80] + date = data.get('observed_date', '') + key2 = f'{cat}::{content}' + if key2 not in by_key: + by_key[key2] = {'category': cat, 'content': content, 'dates': set()} + by_key[key2]['dates'].add(date) + qualified = [] + for k, v in by_key.items(): + if len(v['dates']) >= self.MIN_DAYS_FOR_QUALIFY: + qualified.append({ + 'category': v['category'], + 'content': v['content'], + 'observed_dates': sorted(v['dates']), + }) + return qualified + + def habit_proactive_event(self, user: str, + now: Optional[datetime] = None) -> Optional[Dict[str, Any]]: + """ + 根据当前时间 + 已确认习惯触发一条主动事件。 + 返回 {category, content, trigger_at} 或 None。 + """ + if not self.enabled: + return None + now = now or datetime.now() + h = now.hour + # 简单的时段映射 + target_category = None + if 6 <= h <= 9: + target_category = '饮食' # 早餐 + elif 11 <= h <= 13: + target_category = '饮食' # 午餐 + elif 17 <= h <= 19: + target_category = '饮食' # 晚餐 + elif 22 <= h <= 24 or 0 <= h <= 2: + target_category = '作息' # 提醒睡觉 + elif 14 <= h <= 18: + target_category = '娱乐' # 下午休闲 + else: + target_category = '工作' + habits = self.qualified_habits(user) + matched = [h2 for h2 in habits if h2.get('category') == target_category] + if not matched: + return None + # 取最新观察 + picked = sorted(matched, key=lambda h2: h2['observed_dates'][-1], reverse=True)[0] + return { + 'category': picked['category'], + 'content': picked['content'], + 'trigger_at': now.isoformat(), + } + + def format_for_schedule(self, limit: int = 8) -> str: + """渲染为 prompt 文本，注入 schedule 上下文。""" + if not self.enabled: + return "" + # 不再传 user（无 GUI 用户上下文）— 改为从 metadata 全局读 + all_meta = self.db.list_metadata_keys(prefix='habit_') + seen = set() + lines = [] + for key in all_meta: + value = self.db.get_metadata(key, None) + if not value: + continue + try: + data = json.loads(value) + except (TypeError, ValueError): + continue + k = f'{data.get("category")}::{data.get("content")}' + if k in seen: + continue + seen.add(k) + lines.append(f"• [{data.get('category')}] {data.get('content')[:40]}") + if len(lines) >= limit: + break + if not lines: + return "" + return "【用户行为习惯】\n" + "\n".join(lines) diff --git a/src/nps/nps_invoker.py b/src/nps/nps_invoker.py index b081d23..4c2ac42 100644 --- a/src/nps/nps_invoker.py +++ b/src/nps/nps_invoker.py @@ -10,6 +10,7 @@ import requests from src.tools.debug_logger import get_debug_logger from src.nps.nps_registry import NPSRegistry, NPSTool +from src.core import llm_providers # v3.1.0: 统一供应商解析 load_dotenv() @@ -32,8 +33,12 @@ def __init__(self, registry: NPSRegistry = None, **kwargs): **kwargs: 其他参数（用于向后兼容，会被忽略） """ # API配置 - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = llm_providers.resolve_api_key() + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') # NPS LLM 判断配置（带错误处理） diff --git a/src/tools/agent_vision.py b/src/tools/agent_vision.py index bc42931..554325a 100644 --- a/src/tools/agent_vision.py +++ b/src/tools/agent_vision.py @@ -12,6 +12,7 @@ from dotenv import load_dotenv import requests from src.core.database_manager import DatabaseManager +from src.core import llm_providers # v3.1.0: 统一供应商解析 from src.tools.debug_logger import get_debug_logger load_dotenv() @@ -34,10 +35,14 @@ def __init__(self, db_manager: DatabaseManager = None): db_manager: 数据库管理器实例 """ self.db = db_manager or DatabaseManager() - + # API配置（用于智能判断是否需要使用视觉工具） - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = llm_providers.resolve_api_key() + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') # 环境相关关键词（用于快速判断） diff --git a/src/tools/debug_logger.py b/src/tools/debug_logger.py index c614cc8..dda71ac 100644 --- a/src/tools/debug_logger.py +++ b/src/tools/debug_logger.py @@ -6,8 +6,11 @@ import os import sys import traceback +import asyncio +import inspect +import threading from datetime import datetime -from typing import List, Dict, Any, Optional +from typing import List, Dict, Any, Optional, Callable import json @@ -15,6 +18,11 @@ class DebugLogger: """ Debug日志记录器 记录模块运行、提示词变化、API请求/响应等调试信息 + + Stage D.4: 新增 _subscribers 列表（与 listeners 并存，向后兼容）。 + - subscribe(cb) 注册订阅者；unsubscribe(cb) 取消 + - log_* 方法在通知 listeners 后，遍历 _subscribers 异步触发（asyncio.create_task） + - 同步 subscriber 直接调用；async subscriber 调度到事件循环 """ def __init__(self, debug_mode: bool = False, log_file: str = None): @@ -41,9 +49,15 @@ def __init__(self, debug_mode: bool = False, log_file: str = None): 'info': 0 } - # 日志监听器（用于实时更新GUI） + # 日志监听器（用于实时更新GUI）—— 保持向后兼容 self.listeners = [] + # Stage D.4: WebSocket 订阅者（独立于 listeners，避免互相影响） + self._subscribers: List[Callable[[Dict[str, Any]], Any]] = [] + self._subscribers_lock = threading.RLock() + # 主事件循环引用（用于在跨线程上下文里调度 async subscriber） + self._main_loop: Optional[asyncio.AbstractEventLoop] = None + if self.debug_mode: print(f"✓ Debug模式已启用 | 日志文件: {self.log_file}") self._init_log_file() @@ -72,8 +86,98 @@ def remove_listener(self, callback): if callback in self.listeners: self.listeners.remove(callback) + # ===================================================================== + # Stage D.4: 新增 subscribe / unsubscribe / set_main_loop + # ===================================================================== + def subscribe(self, callback: Callable[[Dict[str, Any]], Any]) -> Callable[[], None]: + """ + 注册一个 WebSocket 订阅者（同步或 async 均可）。 + 返回 unsubscribe() 函数。 + """ + if not callable(callback): + raise TypeError("callback must be callable") + with self._subscribers_lock: + self._subscribers.append(callback) + + def _unsub() -> None: + self.unsubscribe(callback) + return _unsub + + def unsubscribe(self, callback: Callable[[Dict[str, Any]], Any]) -> bool: + """取消订阅。""" + with self._subscribers_lock: + try: + self._subscribers.remove(callback) + return True + except ValueError: + return False + + def set_main_loop(self, loop: Optional[asyncio.AbstractEventLoop]) -> None: + """ + 注入主事件循环（Web 启动时调用一次）。 + 之后 log_* 可在跨线程上下文里调度 async subscriber。 + """ + self._main_loop = loop + + def subscriber_count(self) -> int: + """查询订阅者数量。""" + with self._subscribers_lock: + return len(self._subscribers) + + def _notify_subscribers(self, log_entry: Dict[str, Any]) -> None: + """ + 通知所有 _subscribers。 + - 同步 subscriber 直接调用 + - async subscriber 优先调度到主事件循环；无循环时用 asyncio.create_task 兜底 + - 单个 subscriber 抛错不影响其他 + """ + with self._subscribers_lock: + subscribers = list(self._subscribers) + for cb in subscribers: + try: + if inspect.iscoroutinefunction(cb): + self._dispatch_async_subscriber(cb, log_entry) + else: + try: + cb(log_entry) + except Exception as e: + print(f"✗ 通知 subscriber 失败: {e}") + except Exception as e: + print(f"✗ 通知 subscriber 调度失败: {e}") + + def _dispatch_async_subscriber(self, cb: Callable, log_entry: Dict[str, Any]) -> None: + """调度 async subscriber 到事件循环。""" + try: + coro = cb(log_entry) + # 已有运行中的事件循环（Web 主循环 / 子线程 loop） + try: + running_loop = asyncio.get_running_loop() + except RuntimeError: + running_loop = None + target_loop = self._main_loop or running_loop + if target_loop is not None and target_loop.is_running(): + try: + asyncio.run_coroutine_threadsafe(coro, target_loop) + return + except Exception: + pass + # 兜底：直接尝试 create_task（如果当前有 running loop） + if running_loop is not None: + try: + running_loop.create_task(coro) + return + except Exception: + pass + # 都没有运行中的 loop：用 asyncio.run 跑一下（极少数情况） + try: + asyncio.run(coro) + except Exception as inner_e: + print(f"✗ async subscriber 兜底执行失败: {inner_e}") + except Exception as e: + print(f"✗ 调度 async subscriber 失败: {e}") + def _notify_listeners(self, log_entry: Dict[str, Any]): - """通知所有监听器""" + """通知所有监听器（向后兼容）""" for listener in self.listeners: try: listener(log_entry) @@ -121,6 +225,7 @@ def log_module(self, module_name: str, action: str, details: str = ""): print(message) self._write_to_file(message) self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def log_prompt(self, module_name: str, prompt_type: str, prompt_content: str, metadata: Dict[str, Any] = None): """ @@ -159,6 +264,7 @@ def log_prompt(self, module_name: str, prompt_type: str, prompt_content: str, me self._write_to_file(message) self._write_to_file(f" 完整内容:\n{prompt_content}\n") self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def log_request(self, module_name: str, api_url: str, payload: Dict[str, Any], headers: Dict[str, str] = None): """ @@ -198,6 +304,7 @@ def log_request(self, module_name: str, api_url: str, payload: Dict[str, Any], h self._write_to_file(message) self._write_to_file(f" 完整Payload:\n{json.dumps(payload, ensure_ascii=False, indent=2)}\n") self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def log_response(self, module_name: str, response_data: Any, status_code: int = 200, elapsed_time: float = 0): """ @@ -235,6 +342,7 @@ def log_response(self, module_name: str, response_data: Any, status_code: int = self._write_to_file(message) self._write_to_file(f" 完整响应:\n{response_str}\n") self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def log_error(self, module_name: str, error_message: str, exception: Exception = None): """ @@ -294,6 +402,7 @@ def log_error(self, module_name: str, error_message: str, exception: Exception = if traceback_info: self._write_to_file(f" 完整堆栈:\n{traceback_info}") self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def log_info(self, module_name: str, message: str, data: Any = None): """ @@ -325,6 +434,35 @@ def log_info(self, module_name: str, message: str, data: Any = None): print(log_message) self._write_to_file(log_message) self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) + + def log_warn(self, module_name: str, message: str, data: Any = None): + """ + 记录警告信息（Stage D.4 补充：与 log_info 类似但带 'warn' 类型）。 + """ + if not self.debug_mode: + return + + log_entry = { + 'timestamp': datetime.now().isoformat(), + 'type': 'warn', + 'module': module_name, + 'message': message, + 'data': data, + } + + self.logs.append(log_entry) + self.log_stats.setdefault('warn', 0) + self.log_stats['warn'] += 1 + + log_message = f"[{log_entry['timestamp']}] [WARN] {module_name} | {message}" + if data: + log_message += f"\n 数据: {json.dumps(data, ensure_ascii=False) if isinstance(data, (dict, list)) else str(data)}" + + print(log_message) + self._write_to_file(log_message) + self._notify_listeners(log_entry) + self._notify_subscribers(log_entry) def get_logs(self, log_type: str = None, module_name: str = None, limit: int = None) -> List[Dict[str, Any]]: """ @@ -355,6 +493,71 @@ def get_recent_logs(self, count: int = 50) -> List[Dict[str, Any]]: """获取最近的日志""" return self.logs[-count:] + # ===================================================================== + # Stage B.5: 统一筛选接口（向后兼容；不破坏现有签名） + # ===================================================================== + def get_buffer( + self, + module: Optional[str] = None, + level: Optional[str] = None, + start: Optional[str] = None, + end: Optional[str] = None, + limit: int = 200, + ) -> List[Dict[str, Any]]: + """ + 从内存缓冲读取历史日志（Stage B.5 新增）。 + + Args: + module: 模块名筛选（精确匹配，可选） + level: 日志级别筛选（INFO / WARN / ERROR 等，大小写不敏感，可选） + start: 起始时间（ISO 8601 字符串，可选） + end: 结束时间（ISO 8601 字符串，可选） + limit: 返回上限（1-2000，超出按 2000 处理） + + Returns: + 命中筛选条件的日志列表（按时间升序，最多 limit 条） + """ + try: + limit = int(limit) if limit else 200 + except (TypeError, ValueError): + limit = 200 + limit = max(1, min(limit, 2000)) + + module_filter = (module or "").strip() or None + level_filter_raw = (level or "").strip().lower() or None + # type 字段在内部存储；与 type 等价的"级别"对外就是 log type + type_filter = level_filter_raw + + # ISO 8601 时间比较：直接字典序比较（多数情况下 ISO 8601 可直接比） + start_cmp = start.strip() if isinstance(start, str) and start.strip() else None + end_cmp = end.strip() if isinstance(end, str) and end.strip() else None + + result: List[Dict[str, Any]] = [] + for entry in self.logs: + if not isinstance(entry, dict): + continue + # module 过滤 + if module_filter and entry.get('module') != module_filter: + continue + # level / type 过滤 + if type_filter: + entry_type = str(entry.get('type', '')).lower() + if entry_type != type_filter: + continue + # 时间范围过滤 + ts = entry.get('timestamp') + if isinstance(ts, str): + if start_cmp and ts < start_cmp: + continue + if end_cmp and ts > end_cmp: + continue + result.append(entry) + + # 限制返回数量（保留最后 limit 条 = 时间最新的 limit 条） + if len(result) > limit: + result = result[-limit:] + return result + def clear_logs(self): """清空内存中的日志""" self.logs.clear() diff --git a/src/tools/expression_style.py b/src/tools/expression_style.py index 0d370c4..4f5859c 100644 --- a/src/tools/expression_style.py +++ b/src/tools/expression_style.py @@ -10,6 +10,7 @@ from dotenv import load_dotenv import requests from src.core.database_manager import DatabaseManager +from src.core import llm_providers # v3.1.0: 统一供应商解析 from src.tools.debug_logger import get_debug_logger load_dotenv() @@ -49,8 +50,12 @@ def __init__(self, self.db = db_manager or DatabaseManager() # API配置（用于学习用户表达习惯） - self.api_key = api_key or os.getenv('SILICONFLOW_API_KEY') - self.api_url = api_url or os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = api_key or llm_providers.resolve_api_key() + try: + self.api_url = api_url or llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = api_url or os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = model_name or os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') # 用户表达习惯学习间隔（保留实例属性以保持兼容性） diff --git a/src/tools/schedule_intent_tool.py b/src/tools/schedule_intent_tool.py index 4e37773..47dacfd 100644 --- a/src/tools/schedule_intent_tool.py +++ b/src/tools/schedule_intent_tool.py @@ -11,6 +11,7 @@ from datetime import datetime, timedelta from dotenv import load_dotenv import requests +from src.core import llm_providers # v3.1.0: 统一供应商解析 from src.tools.debug_logger import get_debug_logger load_dotenv() @@ -33,8 +34,12 @@ class ScheduleIntentTool: def __init__(self): """初始化日程意图识别工具""" - self.api_key = os.getenv('SILICONFLOW_API_KEY') - self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') + # v3.1.0: 用 llm_providers 解析（保留旧 SILICONFLOW_API_KEY 兼容） + self.api_key = llm_providers.resolve_api_key() + try: + self.api_url = llm_providers.resolve_base_url(llm_providers.resolve_provider()) + except Exception: + self.api_url = os.getenv('SILICONFLOW_API_URL', 'https://api.siliconflow.cn/v1/chat/completions') self.model_name = os.getenv('MODEL_NAME', 'Qwen/Qwen2.5-7B-Instruct') self.temperature = 0.3 # 使用较低的温度以获得更确定的输出 self.max_tokens = 500 diff --git a/src/tools/settings_migration.py b/src/tools/settings_migration.py index 4b18915..dd443c3 100644 --- a/src/tools/settings_migration.py +++ b/src/tools/settings_migration.py @@ -291,7 +291,8 @@ def _write_env_file(self, env_settings: Dict[str, str], overwrite: bool = False) f.write(f"# 导入时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n") # 按类别分组写入 - api_keys = ['SILICONFLOW_API_KEY', 'SILICONFLOW_API_URL'] + # v3.1.0: 保留旧 key 列表（仅用于旧 .env 迁移场景），新加 LLM_API_KEY + api_keys = ['LLM_API_KEY', 'SILICONFLOW_API_KEY', 'SILICONFLOW_API_URL', 'LLM_BASE_URL', 'LLM_MODEL_NAME'] model_keys = ['MODEL_NAME', 'TEMPERATURE', 'MAX_TOKENS'] character_keys = [k for k in env_settings.keys() if k.startswith('CHARACTER_')] memory_keys = [k for k in env_settings.keys() if 'MEMORY' in k] diff --git a/src/web/backend/__init__.py b/src/web/backend/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/web/backend/api/__init__.py b/src/web/backend/api/__init__.py new file mode 100644 index 0000000..7eab96c --- /dev/null +++ b/src/web/backend/api/__init__.py @@ -0,0 +1,91 @@ +""" +REST API routers package +聚合 debug / knowledge / schedule 等 router，统一挂载到 FastAPI app。 + +Stage B.5 / C.1 / C.2: 三个核心 CRUD 模块聚合入口。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter + +# 子模块 routers（用 try/except 保护，避免单个模块失败影响整体导入） +_routers = [] +_loaded = {} + +try: + from src.web.backend.api.debug import router as _debug_router + _routers.append(_debug_router) + _loaded['debug'] = True +except Exception as e: # noqa: BLE001 + _loaded['debug'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('api', f'failed to import debug router: {e}') + except Exception: + pass + +# Stage C.1: knowledge router 不在此聚合（main.py 用 prefix='/api/knowledge' 单独挂载） +# 如果在这里 include，会在没有 prefix 的情况下注册 ""、"/search"、"/{uuid}" 等相对路径， +# 与 /api/knowledge/* 冲突。注释保留以示缘由。 +# try: +# from src.web.backend.api.knowledge import router as _knowledge_router +# _routers.append(_knowledge_router) +# _loaded['knowledge'] = True +# except Exception as e: # noqa: BLE001 +# _loaded['knowledge'] = False +# try: +# from src.tools.debug_logger import get_debug_logger +# get_debug_logger().log_warn('api', f'failed to import knowledge router: {e}') +# except Exception: +# pass + +try: + from src.web.backend.api.schedule import router as _schedule_router + _routers.append(_schedule_router) + _loaded['schedule'] = True +except Exception as e: # noqa: BLE001 + _loaded['schedule'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('api', f'failed to import schedule router: {e}') + except Exception: + pass + +# 额外聚合（与 main.py 中的 include_router 行为一致；不重复挂载则 no-op） +for _name in ('creative', 'database', 'emotion', 'memory'): + try: + mod = __import__(f'src.web.backend.api.{_name}', fromlist=['router']) + _routers.append(mod.router) + _loaded[_name] = True + except Exception as e: # noqa: BLE001 + _loaded[_name] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('api', f'failed to import {_name} router: {e}') + except Exception: + pass + +# 统一 router（保留空 router 以便 main.py 无脑 include_router） +router = APIRouter() +for r in _routers: + try: + router.include_router(r) + except Exception: + pass + + +def get_loaded_status() -> dict: + """返回各子模块加载状态（调试/健康检查用）。""" + return dict(_loaded) + + +__all__ = ["router", "get_loaded_status"] diff --git a/src/web/backend/api/chat_sessions.py b/src/web/backend/api/chat_sessions.py new file mode 100644 index 0000000..f0368b4 --- /dev/null +++ b/src/web/backend/api/chat_sessions.py @@ -0,0 +1,292 @@ +""" +Chat Sessions REST API - v3.1.0 +================================ + +端点（统一前缀 /api/chat/sessions）： +- GET / 列出会话（按 updated_at desc） +- POST / 创建新会话 +- GET /{session_id}/messages 拉取会话消息 +- PATCH /{session_id} 重命名 +- DELETE /{session_id} 级联删除 + +设计要点： +1. 数据库访问完全走 ``db_manager.get_chat_session_repository()``。 +2. 永不返回 5xx：所有路由 try/except + 4xx 包装。 +3. 写操作后不主动 touch：add_message / update_title 内部已维护元信息。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path +from typing import Any, Dict, List, Optional + +from fastapi import APIRouter, HTTPException, Query, Response +from pydantic import BaseModel, Field + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.core.database_manager import DatabaseManager +except Exception: # noqa: BLE001 + DatabaseManager = None # type: ignore + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + + +router = APIRouter(prefix="/api/chat/sessions", tags=["chat-sessions"]) +debug_logger = get_debug_logger() + +# 进程内单例 db_manager +_db: Optional[DatabaseManager] = None # type: ignore + + +def _get_db() -> Any: + global _db + if _db is None: + if DatabaseManager is None: + raise HTTPException(status_code=503, detail="DatabaseManager unavailable") + _db = DatabaseManager() + return _db + + +def _get_repo(): + return _get_db().get_chat_session_repository() + + +# ---------------------------------------------------------------------- +# Pydantic schemas +# ---------------------------------------------------------------------- +class CreateSessionRequest(BaseModel): + user_id: str = Field(..., min_length=1, max_length=128, description="归属用户 ID") + title: Optional[str] = Field(default="新会话", max_length=200) + + +class RenameSessionRequest(BaseModel): + title: str = Field(..., min_length=1, max_length=200) + + +class SessionResponse(BaseModel): + id: int + user_id: str + title: str + created_at: float + updated_at: float + message_count: int = 0 + + +class MessageResponse(BaseModel): + id: int + session_id: int + role: str + content: str + emotion_json: Optional[str] = None + conn_id: Optional[str] = None + created_at: float + + +class MessageListResponse(BaseModel): + items: List[MessageResponse] + total: int + + +# ---------------------------------------------------------------------- +# GET /api/chat/sessions +# ---------------------------------------------------------------------- +@router.get("", response_model=None) +async def list_sessions( + user_id: str = Query(default="default", min_length=1, max_length=128), + limit: int = Query(default=50, ge=1, le=200), + offset: int = Query(default=0, ge=0), +) -> Dict[str, Any]: + """ + 列出用户的会话（按 updated_at 倒序）。 + """ + try: + repo = _get_repo() + items = repo.list(user_id=user_id, limit=limit, offset=offset) + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'list 出错: {exc}', exc) + except Exception: + pass + return {"items": [], "total": 0, "error": str(exc)} + + return { + "items": [dict(it) for it in (items or [])], + "total": len(items or []), + } + + +# ---------------------------------------------------------------------- +# POST /api/chat/sessions +# ---------------------------------------------------------------------- +@router.post("", response_model=SessionResponse) +async def create_session(req: CreateSessionRequest) -> SessionResponse: + """ + 创建新会话。 + """ + try: + repo = _get_repo() + new_id = repo.create(user_id=req.user_id, title=req.title or "新会话") + row = repo.get(new_id) or {} + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'create 出错: {exc}', exc) + except Exception: + pass + raise HTTPException(status_code=400, detail=f"create failed: {exc}") + + return SessionResponse( + id=int(row.get("id", 0) or 0), + user_id=str(row.get("user_id", req.user_id)), + title=str(row.get("title", req.title or "新会话")), + created_at=float(row.get("created_at", 0.0) or 0.0), + updated_at=float(row.get("updated_at", 0.0) or 0.0), + message_count=int(row.get("message_count", 0) or 0), + ) + + +# ---------------------------------------------------------------------- +# GET /api/chat/sessions/current +# ---------------------------------------------------------------------- +@router.get("/current", response_model=Optional[SessionResponse]) +async def get_current_session( + user_id: str = Query(default="default", min_length=1, max_length=128), +) -> Optional[SessionResponse]: + """ + 取指定 user_id 的最近一个活跃 session（按 updated_at 倒序的第一条）。 + """ + try: + repo = _get_repo() + items = repo.list(user_id=user_id, limit=1, offset=0) + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'current 出错: {exc}', exc) + except Exception: + pass + return None + if not items: + return None + row = items[0] + return SessionResponse( + id=int(row.get("id", 0) or 0), + user_id=str(row.get("user_id", user_id)), + title=str(row.get("title", "新会话")), + created_at=float(row.get("created_at", 0.0) or 0.0), + updated_at=float(row.get("updated_at", 0.0) or 0.0), + message_count=int(row.get("message_count", 0) or 0), + ) + + +# ---------------------------------------------------------------------- +# GET /api/chat/sessions/{session_id}/messages +# ---------------------------------------------------------------------- +@router.get("/{session_id}/messages", response_model=MessageListResponse) +async def list_messages( + session_id: int, + limit: int = Query(default=200, ge=1, le=1000), +) -> MessageListResponse: + """ + 拉取会话消息（按 id 升序）。 + """ + try: + repo = _get_repo() + if repo.get(session_id) is None: + raise HTTPException(status_code=404, detail="session not found") + items = repo.list_messages(session_id=session_id, limit=limit) + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'list_messages 出错: {exc}', exc) + except Exception: + pass + return MessageListResponse(items=[], total=0) + + return MessageListResponse( + items=[MessageResponse( + id=int(it.get("id", 0) or 0), + session_id=int(it.get("session_id", session_id) or 0), + role=str(it.get("role", "")), + content=str(it.get("content", "")), + emotion_json=it.get("emotion_json"), + conn_id=it.get("conn_id"), + created_at=float(it.get("created_at", 0.0) or 0.0), + ) for it in (items or [])], + total=len(items or []), + ) + + +# ---------------------------------------------------------------------- +# PATCH /api/chat/sessions/{session_id} +# ---------------------------------------------------------------------- +@router.patch("/{session_id}", response_model=SessionResponse) +async def rename_session(session_id: int, req: RenameSessionRequest) -> SessionResponse: + """ + 重命名会话。 + """ + try: + repo = _get_repo() + if repo.get(session_id) is None: + raise HTTPException(status_code=404, detail="session not found") + ok = repo.update_title(session_id=session_id, title=req.title) + if not ok: + raise HTTPException(status_code=400, detail="update failed") + row = repo.get(session_id) or {} + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'rename 出错: {exc}', exc) + except Exception: + pass + raise HTTPException(status_code=400, detail=f"rename failed: {exc}") + + return SessionResponse( + id=int(row.get("id", session_id) or 0), + user_id=str(row.get("user_id", "")), + title=str(row.get("title", req.title)), + created_at=float(row.get("created_at", 0.0) or 0.0), + updated_at=float(row.get("updated_at", 0.0) or 0.0), + message_count=int(row.get("message_count", 0) or 0), + ) + + +# ---------------------------------------------------------------------- +# DELETE /api/chat/sessions/{session_id} +# ---------------------------------------------------------------------- +@router.delete("/{session_id}", status_code=204) +async def delete_session(session_id: int) -> Response: + """ + 删除会话（依赖 ON DELETE CASCADE 删 messages）。 + """ + try: + repo = _get_repo() + if repo.get(session_id) is None: + raise HTTPException(status_code=404, detail="session not found") + repo.delete(session_id=session_id) + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.chat_sessions', f'delete 出错: {exc}', exc) + except Exception: + pass + raise HTTPException(status_code=400, detail=f"delete failed: {exc}") + return Response(status_code=204) + + +__all__ = ["router"] diff --git a/src/web/backend/api/creative.py b/src/web/backend/api/creative.py new file mode 100644 index 0000000..db3b944 --- /dev/null +++ b/src/web/backend/api/creative.py @@ -0,0 +1,617 @@ +""" +Creative REST API router - Stage C.6 +长期创作项目的 REST API 路由。 + +端点： +- GET /api/creative/projects 列出创作项目 +- GET /api/creative/projects/{uuid} 项目详情（含 Story Bible） +- POST /api/creative/projects 立项 +- POST /api/creative/projects/{uuid}/advance 续写（默认异步） +- DELETE /api/creative/projects/{uuid} 删除项目 + +设计要点： +1. 全部走 ``src.web.backend.services.database_service``； + 不直接 import DatabaseManager。 +2. 所有端点 try/except 保护：单次失败 → 4xx，绝不抛 5xx。 +3. 续写默认走异步模式：立即返回 task_id，并通过 + ``src.core.event_manager.get_event_manager().publish('creative_progress', ...)`` + 把进度推给前端（EventService 已在 main.py 启动时订阅通配，事件会 + 自动转发到 /ws/events）。 +4. 同步模式（``advance_async=False``）：直接更新 last_advanced_at + 字段，不调用 LLM，避免阻塞 HTTP 请求。 +""" + +from __future__ import annotations + +import json +import sys +import uuid as _uuid +from datetime import datetime +from pathlib import Path +from typing import Any, Dict, List, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter, HTTPException, Query, status # noqa: E402 + +# Schemas +from src.web.backend.schemas.creative import ( # noqa: E402 + AdvanceRequest, + AdvanceResponse, + CreativeProject, + CreativeProjectCreate, + CreativeProjectListResponse, + StoryBible, +) + +# Service layer +from src.web.backend.services.database_service import database_service # noqa: E402 + +# 调试日志（try/except 保护：debug_logger 不可用时降级为 print） +try: + from src.tools.debug_logger import get_debug_logger as _get_debug_logger + _debug_logger = _get_debug_logger() +except Exception: # noqa: BLE001 + class _StubLogger: + def log_info(self, *a, **k): pass + def log_warn(self, *a, **k): pass + def log_error(self, *a, **k): pass + _debug_logger = _StubLogger() + + +# EventManager 句柄（懒加载；publish 失败不影响主流程） +_event_manager = None + + +def _get_event_manager(): + """懒加载 EventManager 单例。""" + global _event_manager + if _event_manager is not None: + return _event_manager + try: + from src.core.event_manager import get_event_manager + _event_manager = get_event_manager() + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error('creative_api', + f'获取 EventManager 失败: {e}', e) + except Exception: + pass + _event_manager = None + return _event_manager + + +router = APIRouter(prefix="/api/creative", tags=["creative"]) + + +# ===================================================================== +# 工具：行 → CreativeProject 转换 +# ===================================================================== + + +def _safe_json_loads(raw: Any, default: Any) -> Any: + """安全解析 JSON 字符串，失败时返回 default。""" + if raw is None or raw == "": + return default + if isinstance(raw, (list, dict)): + return raw + try: + return json.loads(raw) + except Exception: + return default + + +def _row_to_creative_project(row: Dict[str, Any], + include_story_bible: bool = False) -> Dict[str, Any]: + """ + 把数据库行 dict 转成 CreativeProject 兼容 dict。 + + 关键 JSON 字段（outline_json / characters_json / draft_chunks_json） + 会反序列化为 list。 + """ + if not isinstance(row, dict): + return {} + + project = dict(row) # copy + + # JSON 字段反序列化 + for json_k, plain_k in ( + ('outline_json', 'outline'), + ('characters_json', 'characters'), + ('draft_chunks_json', 'draft_chunks'), + ): + if json_k in project and plain_k not in project: + project[plain_k] = _safe_json_loads(project.get(json_k), []) + + # 兼容字段：word_count（前端使用） + try: + project['word_count'] = int(project.get('current_chars') or 0) + except (TypeError, ValueError): + project['word_count'] = 0 + + # chapter_count：从 draft_chunks 派生 + try: + chunks = project.get('draft_chunks') or [] + if not isinstance(chunks, list): + chunks = [] + project['chapter_count'] = len(chunks) + except Exception: + project['chapter_count'] = 0 + + # 默认值兜底 + project.setdefault('uuid', '') + project.setdefault('title', '未命名') + project.setdefault('status', 'drafting') + project.setdefault('current_chars', 0) + project.setdefault('target_chars', 5000) + project.setdefault('work_type', '短篇') + project.setdefault('premise', '') + project.setdefault('tone', '') + project.setdefault('point_of_view', '第一人称') + project.setdefault('inspiration_source', '') + project.setdefault('outline', []) + project.setdefault('characters', []) + project.setdefault('draft_chunks', []) + project.setdefault('created_at', '') + project.setdefault('updated_at', '') + project.setdefault('next_advance_at', None) + project.setdefault('last_advanced_at', None) + + # Story Bible：仅在详情场景下填充 + if include_story_bible: + try: + bible_row = database_service.get_story_bible(project.get('uuid', '')) + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_warn( + 'creative_api', + f'get_story_bible 失败: {e}', + ) + except Exception: + pass + bible_row = None + bible_data: Dict[str, Any] = {} + if isinstance(bible_row, dict): + bible_data = { + 'mainline_direction': bible_row.get('mainline_direction', ''), + 'active_themes': bible_row.get('active_themes') or [], + 'unresolved_threads': bible_row.get('unresolved_threads') or [], + 'resolved_threads': bible_row.get('resolved_threads') or [], + 'important_facts': bible_row.get('important_facts') or [], + 'next_direction': bible_row.get('next_direction', ''), + 'recent_keywords': bible_row.get('recent_keywords') or [], + 'characters': project.get('characters') or [], + 'worldview': '', + 'chapters': project.get('draft_chunks') or [], + } + else: + bible_data = { + 'mainline_direction': '', + 'active_themes': [], + 'unresolved_threads': [], + 'resolved_threads': [], + 'important_facts': [], + 'next_direction': '', + 'recent_keywords': [], + 'characters': project.get('characters') or [], + 'worldview': '', + 'chapters': project.get('draft_chunks') or [], + } + try: + project['story_bible'] = StoryBible(**bible_data) + except Exception: + project['story_bible'] = None + else: + project['story_bible'] = None + + return project + + +def _publish_creative_progress(project_uuid: str, + status_name: str, + message: str, + extra: Optional[Dict[str, Any]] = None) -> bool: + """ + 向 EventManager 推送 creative_progress 事件。 + 返回是否成功调度（不阻塞主流程）。 + """ + em = _get_event_manager() + if em is None: + return False + payload: Dict[str, Any] = { + 'project_uuid': project_uuid, + 'status': status_name, + 'message': message, + 'timestamp': datetime.now().isoformat(), + } + if extra: + try: + payload.update(dict(extra)) + except Exception: + pass + try: + em.publish('creative_progress', payload) + return True + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_warn( + 'creative_api', f'publish creative_progress 失败: {e}') + except Exception: + pass + return False + + +# ===================================================================== +# 路由 +# ===================================================================== + + +@router.get( + "/projects", + response_model=CreativeProjectListResponse, + summary="列出创作项目", + description="按 status 过滤列出 creative_projects 记录。", +) +async def list_projects( + status: Optional[str] = Query( + None, + description="按 status 过滤（如 active / paused / completed / drafting / finished）", + ), + limit: int = Query(50, ge=1, le=200, description="返回上限（1-200）"), +) -> CreativeProjectListResponse: + """ + GET /api/creative/projects?status=...&limit=... + + 始终返回 200；DB 不可用 / 表缺失时返回空列表。 + """ + try: + rows = database_service.list_creative_projects( + status=status, limit=limit, + ) + projects: List[Dict[str, Any]] = [] + for row in rows or []: + try: + projects.append(_row_to_creative_project(row)) + except Exception: + continue + return CreativeProjectListResponse( + projects=[CreativeProject(**p) for p in projects], # type: ignore[arg-type] + total=len(projects), + ) + except HTTPException: + raise + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error( + 'creative_api', f'list_projects 失败: {e}', e) + except Exception: + pass + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail=f"list projects failed: {e}", + ) + + +@router.get( + "/projects/{project_uuid}", + response_model=CreativeProject, + summary="项目详情", + description="按 uuid 读取项目详情，并附 Story Bible。", + responses={404: {"description": "项目不存在"}}, +) +async def get_project(project_uuid: str) -> CreativeProject: + """ + GET /api/creative/projects/{uuid} + + 404 当项目不存在；500 当 DB 异常。 + """ + if not project_uuid or not isinstance(project_uuid, str): + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="project_uuid is required", + ) + try: + row = database_service.get_creative_project(project_uuid) + if not row: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail=f"project not found: {project_uuid}", + ) + project_dict = _row_to_creative_project(row, include_story_bible=True) + return CreativeProject(**project_dict) # type: ignore[arg-type] + except HTTPException: + raise + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error( + 'creative_api', f'get_project({project_uuid}) 失败: {e}', e) + except Exception: + pass + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail=f"get project failed: {e}", + ) + + +@router.post( + "/projects", + response_model=CreativeProject, + status_code=status.HTTP_201_CREATED, + summary="立项", + description="创建一条 creative_projects 记录。", + responses={400: {"description": "请求参数不合法"}}, +) +async def create_project(payload: CreativeProjectCreate) -> CreativeProject: + """ + POST /api/creative/projects + + body: CreativeProjectCreate + returns: 201 + CreativeProject + """ + if not payload or not payload.title: + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="title is required", + ) + try: + new_uuid = database_service.create_creative_project( + { + 'title': payload.title, + 'work_type': payload.work_type or '短篇', + 'premise': payload.premise or '', + 'tone': payload.tone or '', + 'point_of_view': payload.point_of_view or '第一人称', + 'target_chars': int(payload.target_chars or 5000), + 'current_chars': 0, + 'status': 'active', # 前端默认语义：active = drafting + 'inspiration_source': payload.inspiration_source or '', + 'outline': payload.outline or [], + 'characters': payload.characters or [], + 'draft_chunks': [], + } + ) + if not new_uuid: + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail="failed to create project (db unavailable or write failed)", + ) + # 重新读取完整记录 + row = database_service.get_creative_project(new_uuid) or { + 'uuid': new_uuid, + 'title': payload.title, + 'work_type': payload.work_type or '短篇', + 'premise': payload.premise or '', + 'tone': payload.tone or '', + 'point_of_view': payload.point_of_view or '第一人称', + 'target_chars': int(payload.target_chars or 5000), + 'current_chars': 0, + 'status': 'active', + 'inspiration_source': payload.inspiration_source or '', + 'outline_json': json.dumps(payload.outline or [], ensure_ascii=False), + 'characters_json': json.dumps(payload.characters or [], ensure_ascii=False), + 'draft_chunks_json': json.dumps([], ensure_ascii=False), + 'created_at': datetime.now().isoformat(), + 'updated_at': datetime.now().isoformat(), + } + project_dict = _row_to_creative_project(row, include_story_bible=True) + try: + _debug_logger.log_info( + 'creative_api', 'create_project OK', + {'uuid': new_uuid, 'title': payload.title}, + ) + except Exception: + pass + # 触发 creative_progress 事件（queued） + _publish_creative_progress( + new_uuid, status_name='created', + message=f'项目《{payload.title}》已创建', + extra={'title': payload.title, 'last_advanced_at': None}, + ) + return CreativeProject(**project_dict) # type: ignore[arg-type] + except HTTPException: + raise + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error( + 'creative_api', f'create_project 失败: {e}', e) + except Exception: + pass + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail=f"create project failed: {e}", + ) + + +@router.post( + "/projects/{project_uuid}/advance", + response_model=AdvanceResponse, + summary="续写", + description=( + "触发一次续写。默认走异步模式：立即返回 task_id，" + "通过 'creative_progress' 事件推送进度。" + "将 ``advance_async`` 置为 False 走同步模式：仅更新 last_advanced_at。" + ), + responses={404: {"description": "项目不存在"}}, +) +async def advance_project( + project_uuid: str, + payload: Optional[AdvanceRequest] = None, +) -> AdvanceResponse: + """ + POST /api/creative/projects/{uuid}/advance + + 异步模式（默认）： + - 立即返回 202 + task_id + - 后台通过 EventManager.publish('creative_progress', {...}) 推送进度 + - 前端通过 /ws/events 接收事件 + 同步模式（advance_async=False）： + - 立即更新 last_advanced_at 字段 + - 返回 200 + AdvanceResponse(status=completed, last_advanced_at=now) + """ + if not project_uuid or not isinstance(project_uuid, str): + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="project_uuid is required", + ) + # 兼容 body 缺省（GET-like 调用） + body = payload or AdvanceRequest() + try: + row = database_service.get_creative_project(project_uuid) + if not row: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail=f"project not found: {project_uuid}", + ) + current_status = str(row.get('status') or 'drafting') + now_iso = datetime.now().isoformat() + + if not body.advance_async: + # 同步模式：仅更新时间戳；不阻塞主流程 + updated = database_service.update_creative_project( + project_uuid, + { + 'last_advanced_at': now_iso, + 'next_advance_at': now_iso, + }, + ) + if not updated: + # 行不存在或更新失败：依然视为成功（接口幂等） + pass + _publish_creative_progress( + project_uuid, status_name='advanced', + message=f'项目已续写（同步模式）', + extra={'last_advanced_at': now_iso, + 'project_status': current_status}, + ) + return AdvanceResponse( + project_uuid=project_uuid, + task_id="", + status="completed", + last_advanced_at=now_iso, + project_status=current_status, + message="advance completed (sync mode)", + ) + + # 异步模式：生成 task_id，立即返回；通过 EventManager 推送进度 + task_id = str(_uuid.uuid4()) + _publish_creative_progress( + project_uuid, status_name='queued', + message='续写任务已入队', + extra={'task_id': task_id, 'user': body.user, + 'project_status': current_status, + 'last_advanced_at': now_iso}, + ) + # 同步推进：尝试更新 last_advanced_at 作为占位。 + # 真正的 LLM 生成留给后台 Worker；这里仅记录"已入队"。 + try: + database_service.update_creative_project( + project_uuid, + { + 'last_advanced_at': now_iso, + 'next_advance_at': now_iso, + }, + ) + except Exception: + pass + + try: + _debug_logger.log_info( + 'creative_api', 'advance queued', + {'uuid': project_uuid, 'task_id': task_id, + 'user': body.user, 'hint': (body.hint or '')[:80]}, + ) + except Exception: + pass + + return AdvanceResponse( + project_uuid=project_uuid, + task_id=task_id, + status="queued", + last_advanced_at=now_iso, + project_status=current_status, + message=( + "advance task queued; progress will be pushed via " + "'creative_progress' events" + ), + ) + except HTTPException: + raise + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error( + 'creative_api', f'advance_project({project_uuid}) 失败: {e}', e) + except Exception: + pass + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail=f"advance project failed: {e}", + ) + + +@router.delete( + "/projects/{project_uuid}", + status_code=status.HTTP_204_NO_CONTENT, + summary="删除项目", + description="按 uuid 删除 creative_projects 记录（级联删除 Story Bible / Memory Pool）。", + responses={ + 204: {"description": "删除成功"}, + 404: {"description": "项目不存在"}, + }, +) +async def delete_project(project_uuid: str) -> None: + """ + DELETE /api/creative/projects/{uuid} + """ + if not project_uuid or not isinstance(project_uuid, str): + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="project_uuid is required", + ) + try: + # 删除前先校验存在（避免静默成功） + row = database_service.get_creative_project(project_uuid) + if not row: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail=f"project not found: {project_uuid}", + ) + ok = database_service.delete_record('creative_projects', project_uuid) + if not ok: + # 行级联删除会清掉 bible / pool；但 delete_record 返回 False + # 也可能因为 uuid 不在白名单等异常 → 视为 500 + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail="delete failed (db unavailable or write blocked)", + ) + try: + _debug_logger.log_info( + 'creative_api', 'delete_project OK', + {'uuid': project_uuid, 'title': row.get('title')}, + ) + except Exception: + pass + _publish_creative_progress( + project_uuid, status_name='deleted', + message='项目已删除', + extra={'title': row.get('title')}, + ) + # 204 No Content + return None + except HTTPException: + raise + except Exception as e: # noqa: BLE001 + try: + _debug_logger.log_error( + 'creative_api', f'delete_project({project_uuid}) 失败: {e}', e) + except Exception: + pass + raise HTTPException( + status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, + detail=f"delete project failed: {e}", + ) + + +__all__ = ["router"] diff --git a/src/web/backend/api/database.py b/src/web/backend/api/database.py new file mode 100644 index 0000000..7c33cb4 --- /dev/null +++ b/src/web/backend/api/database.py @@ -0,0 +1,245 @@ +""" +Database API Router +``/api/database`` 端点 — 业务表查询 / 计数 / 受控删除。 + +Stage C.5: +- 列出白名单中实际存在的业务表 +- 分页查询 + 记录总数 +- 按 uuid 删除（带 ``X-Confirm: true`` 头二次确认） + +设计要点： +1. 复用 ``database_service``：所有 SQL 拼接已由 ``_validate_table`` 防御。 +2. 失败路径 4xx：按硬约束，所有异常路径均返回 4xx（无 5xx）。 +3. 错误响应统一为 ``{"error": "..."}`` 格式（不依赖 FastAPI 默认 detail）。 +4. DELETE 二次确认：必须传 ``X-Confirm: true`` 头；缺失 → 400。 +5. ``app.include_router(database.router)`` 由 main.py 完成。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path +from typing import Any, Optional + +from fastapi import APIRouter, Header, Query +from fastapi.responses import JSONResponse + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from src.web.backend.services.database_service import ( # noqa: E402 + database_service, +) +from src.web.backend.schemas.database import ( # noqa: E402 + CountResponse, + DeleteResponse, + TableDataResponse, + TablesListResponse, +) + + +router = APIRouter(prefix="/api/database", tags=["database"]) + + +# ---------------------------------------------------------------------- +# 内部：错误响应统一格式 +# ---------------------------------------------------------------------- +def _error_response(status_code: int, message: str) -> JSONResponse: + """ + 统一错误响应格式：``{"error": "..."}``。 + 状态码必须在 4xx 范围内。 + """ + if status_code < 400 or status_code >= 500: + status_code = 400 + return JSONResponse( + status_code=status_code, + content={"error": message}, + ) + + +# ---------------------------------------------------------------------- +# 内部：表名白名单校验包装 +# ---------------------------------------------------------------------- +def _safe_validate_table(table: str) -> str: + """ + 包装 ``database_service._validate_table``： + - 失败时抛 ``ValueError``；调用方负责转成 4xx 响应。 + """ + try: + return database_service._validate_table(table) + except (ValueError, TypeError) as exc: + raise ValueError(f"Table {table!r} not allowed: {exc}") from exc + + +# ---------------------------------------------------------------------- +# 内部：单表 COUNT(*) +# ---------------------------------------------------------------------- +def _count_table_rows(safe_table: str) -> int: + """ + 对已经通过白名单校验的表名执行 ``SELECT COUNT(*)``。 + 任何异常路径均返回 0（不抛出）。 + """ + try: + db = database_service.get_db() + if db is None: + return 0 + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute(f"SELECT COUNT(*) FROM {safe_table}") + row = cur.fetchone() + if row is None: + return 0 + try: + if isinstance(row, dict): + return int(list(row.values())[0]) + return int(row[0]) + except (KeyError, IndexError, TypeError, ValueError): + return 0 + except Exception: # noqa: BLE001 + return 0 + + +# ---------------------------------------------------------------------- +# 1) GET /api/database/tables +# ---------------------------------------------------------------------- +@router.get( + "/tables", + response_model=TablesListResponse, + summary="List whitelisted business tables", +) +async def list_tables() -> Any: + """ + 返回白名单中实际存在的业务表名列表。 + + 响应： + - ``tables``: 表名列表（按字典序） + - ``total``: 表数量 + """ + try: + table_names = database_service.get_tables() or [] + except Exception as exc: # noqa: BLE001 + # 按硬约束：失败路径返回 4xx + return _error_response(400, f"DatabaseService 不可用: {exc}") + + return TablesListResponse(tables=list(table_names), total=len(table_names)) + + +# ---------------------------------------------------------------------- +# 2) GET /api/database/{table}?limit=100&offset=0 +# ---------------------------------------------------------------------- +@router.get( + "/{table}", + response_model=TableDataResponse, + summary="Query rows of a whitelisted table (paginated)", +) +async def query_table( + table: str, + limit: int = Query(100, ge=1, le=1000), + offset: int = Query(0, ge=0), +) -> Any: + """ + 分页查询指定表的数据。 + + - ``limit``: 1-1000（默认 100） + - ``offset``: >=0（默认 0） + - 表名必须在白名单内；不在白名单 → 400 + - 响应字段：``table`` / ``rows`` / ``total``（本次返回行数） + """ + try: + _safe_validate_table(table) + except ValueError as exc: + return _error_response(400, str(exc)) + + try: + rows = database_service.query_table(table, limit=limit, offset=offset) or [] + except ValueError as exc: + return _error_response(400, str(exc)) + except Exception as exc: # noqa: BLE001 + return _error_response(400, f"查询失败: {exc}") + + return TableDataResponse( + table=table, + rows=list(rows), + total=len(rows), + ) + + +# ---------------------------------------------------------------------- +# 3) GET /api/database/{table}/count +# ---------------------------------------------------------------------- +@router.get( + "/{table}/count", + response_model=CountResponse, + summary="Count total rows of a whitelisted table", +) +async def count_table(table: str) -> Any: + """ + 返回指定表的总记录数（``SELECT COUNT(*) FROM table``）。 + + - 表名必须在白名单内；不在白名单 → 400 + - 响应字段：``table`` / ``total`` + """ + try: + safe_table = _safe_validate_table(table) + except ValueError as exc: + return _error_response(400, str(exc)) + + total = _count_table_rows(safe_table) + return CountResponse(table=table, total=int(total)) + + +# ---------------------------------------------------------------------- +# 4) DELETE /api/database/{table}/{uuid} +# ---------------------------------------------------------------------- +@router.delete( + "/{table}/{uuid}", + response_model=DeleteResponse, + summary="Delete a single row by uuid (requires X-Confirm header)", +) +async def delete_record( + table: str, + uuid: str, + x_confirm: Optional[str] = Header( + None, + alias="X-Confirm", + description="必须传 X-Confirm: true 才执行删除（防误删）", + ), +) -> Any: + """ + 按 uuid 删除指定表的一条记录。 + + - 必须传 ``X-Confirm: true`` 头；缺失或非 true → 400 + - 表名必须在白名单内；不在白名单 → 400 + - uuid 为空 → 400 + - 数据库层未匹配到记录（rowcount == 0）→ 404 + - 响应字段：``table`` / ``uuid`` / ``deleted`` + """ + # 二次确认：必须传 X-Confirm: true + if x_confirm is None or str(x_confirm).strip().lower() != "true": + return _error_response(400, "需要 X-Confirm: true 头确认") + + try: + _safe_validate_table(table) + except ValueError as exc: + return _error_response(400, str(exc)) + + if not uuid or not isinstance(uuid, str): + return _error_response(400, "uuid 不能为空") + + try: + ok = bool(database_service.delete_record(table, uuid)) + except ValueError as exc: + return _error_response(400, str(exc)) + except Exception as exc: # noqa: BLE001 + return _error_response(400, f"删除失败: {exc}") + + if not ok: + # 数据库层未匹配到记录 + return _error_response(404, f"未在 {table} 中找到 uuid={uuid} 的记录") + + return DeleteResponse(table=table, uuid=uuid, deleted=True) + + +__all__ = ["router"] diff --git a/src/web/backend/api/debug.py b/src/web/backend/api/debug.py new file mode 100644 index 0000000..12ae385 --- /dev/null +++ b/src/web/backend/api/debug.py @@ -0,0 +1,289 @@ +""" +Debug REST API Router +- /api/debug/logs: 列表查询（带筛选） +- /api/debug/logs/export: 导出（TXT / JSON） +- /api/debug/logs/clear: 清空内存日志（Stage B.5 扩展） +- /api/debug/stats: 日志统计 + +Stage B.5: 与 /ws/debug 互补（WS 实时流，REST 历史筛选/导出）。 + +实现要点： +- 调用 DebugLogger.get_buffer() 读取历史日志；不破坏现有方法签名 +- 失败路径返回空结构（不抛 5xx），与 ws 实时流不冲突 +- 导出用 StreamingResponse 返回 text/plain 或 application/json +""" + +from __future__ import annotations + +import io +import json +import sys +import warnings +from pathlib import Path +from typing import Any, Dict, List, Optional + +from fastapi import APIRouter, HTTPException, Query +from fastapi.responses import StreamingResponse + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def get_buffer(self, *a, **kw): return [] + def get_logs(self, *a, **kw): return [] + def get_statistics(self): return {'total_logs': 0, 'by_type': {}, 'debug_mode': False, 'log_file': ''} + def clear_logs(self): pass + return _Stub() + +# 兼容：当 DebugLogger 没有 get_buffer 时回退到 get_logs（不破坏现有接口） +def _read_buffer( + module: Optional[str], + level: Optional[str], + start: Optional[str], + end: Optional[str], + limit: int, +) -> List[Dict[str, Any]]: + logger = get_debug_logger() + try: + if hasattr(logger, "get_buffer"): + buf = logger.get_buffer( + module=module, + level=level, + start=start, + end=end, + limit=limit, + ) + return list(buf or []) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[debug_api] get_buffer 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + # 降级：使用 get_logs + 内存端过滤 + try: + raw = logger.get_logs(module_name=module, limit=max(limit, 1000)) or [] + except Exception: + return [] + lvl = (level or "").strip().lower() or None + out: List[Dict[str, Any]] = [] + for entry in raw: + if not isinstance(entry, dict): + continue + if lvl and str(entry.get("type", "")).lower() != lvl: + continue + ts = entry.get("timestamp") + if isinstance(ts, str): + if start and ts < start: + continue + if end and ts > end: + continue + out.append(entry) + if len(out) > limit: + out = out[-limit:] + return out + + +def _normalize_entry(entry: Dict[str, Any]) -> Dict[str, Any]: + """把 DebugLogger 内部字段映射到 DebugLogEntry 标准结构。""" + if not isinstance(entry, dict): + return { + "timestamp": "", + "module": "", + "level": "INFO", + "message": "", + "extra": None, + } + level = str(entry.get("type") or entry.get("level") or "INFO").upper() + module = str(entry.get("module") or "") + timestamp = str(entry.get("timestamp") or "") + + # 聚合 message 字段（不同 log_* 入口字段名不一样） + message = entry.get("message") + if not message: + # 兼容 log_module / log_request / log_response / log_prompt / log_error + if "action" in entry: + message = str(entry.get("action") or "") + details = entry.get("details") + if details: + message = f"{message} | {details}" if message else str(details) + elif "api_url" in entry: + message = f"{entry.get('api_url', '')}" + elif "content" in entry: + message = str(entry.get("content") or "") + else: + message = "" + + # 额外字段（避免结构化信息丢失） + extra: Optional[Dict[str, Any]] = None + for k in ( + "action", + "details", + "api_url", + "payload", + "headers", + "status_code", + "elapsed_time", + "response", + "exception", + "file_info", + "traceback", + "data", + "prompt_type", + "metadata", + ): + if k in entry and entry[k] is not None: + extra = extra or {} + try: + json.dumps(entry[k]) + extra[k] = entry[k] + except (TypeError, ValueError): + extra[k] = str(entry[k]) + + return { + "timestamp": timestamp, + "module": module, + "level": level, + "message": str(message or ""), + "extra": extra, + } + + +router = APIRouter(prefix="/api/debug", tags=["debug"]) + + +@router.get("/logs") +async def list_debug_logs( + module: Optional[str] = Query(default=None, description="按模块名精确匹配"), + level: Optional[str] = Query(default=None, description="按日志级别筛选（INFO/WARN/ERROR/MODULE/...）"), + start: Optional[str] = Query(default=None, description="起始时间（ISO 8601）"), + end: Optional[str] = Query(default=None, description="结束时间（ISO 8601）"), + limit: int = Query(default=200, ge=1, le=2000, description="返回条数上限"), +) -> dict: + """ + 从 DebugLogger 内存缓冲读取历史日志。 + 返回 {logs: [...], total: int}。 + 失败路径：返回空列表（不抛 5xx）。 + """ + try: + raw = _read_buffer(module, level, start, end, limit) + logs = [_normalize_entry(e) for e in raw] + return {"logs": logs, "total": len(logs)} + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[debug_api] list_debug_logs 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return {"logs": [], "total": 0} + + +@router.get("/logs/export") +async def export_debug_logs( + format: str = Query(default="txt", pattern="^(txt|json)$"), + module: Optional[str] = Query(default=None), + level: Optional[str] = Query(default=None), + start: Optional[str] = Query(default=None), + end: Optional[str] = Query(default=None), + limit: int = Query(default=2000, ge=1, le=10000), +) -> StreamingResponse: + """ + 导出筛选后的日志为 TXT 或 JSON。 + """ + try: + raw = _read_buffer(module, level, start, end, limit) + normalized = [_normalize_entry(e) for e in raw] + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"导出失败: {exc}") from exc + + if format == "json": + try: + body = json.dumps( + {"logs": normalized, "total": len(normalized)}, + ensure_ascii=False, + indent=2, + ) + except (TypeError, ValueError) as exc: + raise HTTPException(status_code=500, detail=f"JSON 序列化失败: {exc}") from exc + return StreamingResponse( + io.BytesIO(body.encode("utf-8")), + media_type="application/json; charset=utf-8", + headers={ + "Content-Disposition": 'attachment; filename="debug-logs.json"', + }, + ) + + # TXT（默认） + lines: List[str] = [] + for entry in normalized: + ts = entry.get("timestamp") or "-" + lvl = entry.get("level") or "INFO" + mod = entry.get("module") or "-" + msg = (entry.get("message") or "").replace("\r", " ").replace("\n", " ") + lines.append(f"[{ts}] [{lvl}] [{mod}] {msg}") + body = "\n".join(lines) + "\n" + return StreamingResponse( + io.BytesIO(body.encode("utf-8")), + media_type="text/plain; charset=utf-8", + headers={ + "Content-Disposition": 'attachment; filename="debug-logs.txt"', + }, + ) + + +@router.post("/logs/clear") +async def clear_debug_logs() -> dict: + """ + 清空 DebugLogger 内存日志（Stage B.5 扩展）。 + """ + try: + logger = get_debug_logger() + if hasattr(logger, "clear_logs"): + logger.clear_logs() + return {"status": "ok", "message": "日志已清空"} + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"清空失败: {exc}") from exc + + +@router.get("/stats") +async def get_debug_stats() -> dict: + """ + 日志统计（按 type 聚合）。 + 失败路径：返回零值结构。 + """ + try: + logger = get_debug_logger() + if hasattr(logger, "get_statistics"): + stats = logger.get_statistics() or {} + return { + "total_logs": int(stats.get("total_logs", 0) or 0), + "by_type": dict(stats.get("by_type", {}) or {}), + "debug_mode": bool(stats.get("debug_mode", False)), + "log_file": str(stats.get("log_file", "") or ""), + } + # 降级 + raw = _read_buffer(None, None, None, None, 10000) + by_type: Dict[str, int] = {} + for e in raw: + t = str((e or {}).get("type", "UNKNOWN")).upper() + by_type[t] = by_type.get(t, 0) + 1 + return {"total_logs": len(raw), "by_type": by_type, "debug_mode": False, "log_file": ""} + except Exception as exc: # noqa: BLE001 + return { + "total_logs": 0, + "by_type": {}, + "debug_mode": False, + "log_file": "", + "error": str(exc), + } + + +__all__ = ["router"] diff --git a/src/web/backend/api/emotion.py b/src/web/backend/api/emotion.py new file mode 100644 index 0000000..59e80aa --- /dev/null +++ b/src/web/backend/api/emotion.py @@ -0,0 +1,168 @@ +""" +Emotion API Router +``/api/emotion`` 端点 — Plutchik 8 维情感雷达数据源。 + +Stage B.3: +- GET /api/emotion/latest?user_id=default + 返回 EmotionResponse 形状的 JSON： + { + "cumulative": {8 维累加}, + "plutchik": {8 维当下}, + "last_message": "...", + "timestamp": "ISO 8601", + "user_id": "default", + "historical_max": Optional[8 维] + } +- 失败 / 无数据：按硬约束降级为 200 OK + 全 0 dict + 空字符串 + （不让前端雷达图崩溃）。 +- POST /api/emotion/refresh?user_id=default + 主动失效缓存（用于聊天流式结束后强制刷新）。 + +设计要点： +1. 单点失败不返回 5xx：包装在 try/except，异常路径同样 200 OK + 空 payload。 +2. 复用 emotion_service 单例：避免重复加载 EmotionRelationshipAnalyzer。 +3. Pydantic v2 schema（EmotionResponse）做响应序列化。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path +from typing import Any, Dict, Optional + +from fastapi import APIRouter, Query + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.web.backend.services.emotion_service import emotion_service +except Exception: # noqa: BLE001 + emotion_service = None # type: ignore + +try: + from src.web.backend.schemas.emotion import EmotionResponse +except Exception: # noqa: BLE001 + EmotionResponse = None # type: ignore + + +router = APIRouter(prefix="/api/emotion", tags=["emotion"]) + + +# ---------------------------------------------------------------------- +# 内部：构造兜底 payload +# ---------------------------------------------------------------------- +def _empty_response(user_id: str = "default") -> Dict[str, Any]: + return { + "cumulative": {k: 0.0 for k in ( + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', + )}, + "plutchik": {k: 0.0 for k in ( + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', + )}, + "last_message": "", + "timestamp": "", + "user_id": user_id or "default", + "historical_max": None, + } + + +# ---------------------------------------------------------------------- +# 1) GET /api/emotion/latest +# ---------------------------------------------------------------------- +@router.get("/latest") +async def get_latest_emotion( + user_id: str = Query( + "default", + description="用户标识（当前所有用户共用情感数据；仅作缓存 key 透传）", + ), +) -> Dict[str, Any]: + """ + 获取最新情感分析结果。 + + 失败 / 无数据 → 200 OK + 全 0 payload（前端雷达不崩）。 + """ + uid = (user_id or "default").strip() or "default" + fallback = _empty_response(uid) + + if emotion_service is None: + return fallback + + try: + payload = emotion_service.get_latest_emotion(uid) + except Exception: # noqa: BLE001 + # 降级：不让前端崩 + return fallback + + if not isinstance(payload, dict): + return fallback + + # 强制覆盖 user_id 透传 + payload["user_id"] = uid + + # 用 Pydantic schema 校验 / 规范化（可序列化） + if EmotionResponse is not None: + try: + model = EmotionResponse(**payload) + return model.model_dump() + except Exception: + # schema 校验失败：仍返回 raw payload（兜底） + return payload + + return payload + + +# ---------------------------------------------------------------------- +# 2) POST /api/emotion/refresh +# ---------------------------------------------------------------------- +@router.post("/refresh") +async def refresh_emotion_cache( + user_id: str = Query( + "default", + description="用户标识；省略则清空所有用户缓存", + ), +) -> Dict[str, Any]: + """ + 主动失效指定用户的情感缓存。 + 下次 GET /api/emotion/latest 会重新构造 payload。 + """ + if emotion_service is None: + return {"cleared": 0, "user_id": user_id or "default"} + try: + emotion_service.cache_clear(user_id if user_id else None) + except Exception: # noqa: BLE001 + return {"cleared": 0, "user_id": user_id or "default"} + return {"cleared": 1, "user_id": user_id or "default"} + + +# ---------------------------------------------------------------------- +# 3) GET /api/emotion/stats +# ---------------------------------------------------------------------- +@router.get("/stats") +async def get_emotion_stats() -> Dict[str, Any]: + """ + 返回情感服务状态（analyzer 是否加载、缓存大小等）。 + """ + if emotion_service is None: + return { + "analyzer_loaded": False, + "wheel_loaded": False, + "cache_size": 0, + "version": "1.0.0", + } + try: + return emotion_service.get_stats() + except Exception: # noqa: BLE001 + return { + "analyzer_loaded": False, + "wheel_loaded": False, + "cache_size": 0, + "version": "1.0.0", + } + + +__all__ = ["router"] diff --git a/src/web/backend/api/event.py b/src/web/backend/api/event.py new file mode 100644 index 0000000..27c41ea --- /dev/null +++ b/src/web/backend/api/event.py @@ -0,0 +1,544 @@ +""" +Event REST API router - Stage C.3 / 验收 P0 3.2 +跨端事件的 REST API 路由（events 表 CRUD）。 + +端点： + GET /api/events 列表（分页 + 筛选） + POST /api/events/{event_id}/read 标记已读 + POST /api/events/{event_id}/archive 归档 + +设计要点： +1. 复用 ``src.core.event_manager.EventManager`` 单例（经 + ``get_event_manager()`` 获取），不走 database_service 通用表查询； + events 表结构复杂、含 metadata JSON 字段，专用 manager 更安全。 +2. 复用 ``src.web.backend.schemas.event.EventDTO`` / ``EventListResponse``； + 字段不匹配部分在本文内通过新建 ``EventDTOExtended``（不改动 EventDTO）补齐。 +3. 列表分页：``limit`` (1-200, 默认 50) + ``offset`` (>=0, 默认 0)。 +4. 筛选：``type`` (event_type) + ``status``。 +5. 读 / 归档直接更新 events 表的 metadata 字段（追加 read_at / archived_at + 时间戳），同时把 status 同步为 ``completed`` / ``cancelled``，确保与 + 现有 EventStatus 枚举兼容。 +6. 所有路由 try/except 包裹，失败 → 4xx + {"error": ...}，绝不抛 5xx。 +7. EventManager 不可用时返回空列表（graceful degradation）。 +""" + +from __future__ import annotations + +import json +import sys +import warnings +from datetime import datetime +from pathlib import Path +from typing import Any, Dict, List, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter, HTTPException, Path, Query, status # noqa: E402 +from pydantic import BaseModel, Field # noqa: E402 + +# 复用 EventDTO / EventListResponse（字段对不上时本文内新增，不改 EventDTO） +from src.web.backend.schemas.event import ( # noqa: E402 + EventDTO, + EventListResponse, +) + +# 调试日志（try/except 保护） +try: + from src.tools.debug_logger import get_debug_logger as _get_debug_logger + _debug_logger = _get_debug_logger() +except Exception: # noqa: BLE001 + class _StubLogger: + def log_info(self, *a, **k): pass + def log_warn(self, *a, **k): pass + def log_error(self, *a, **k): pass + _debug_logger = _StubLogger() + + +# EventManager 句柄（懒加载；不直接 import EventManager 类以便降级） +_event_manager: Optional[Any] = None +_EVENT_MANAGER_IMPORT_OK: bool = False +_EVENT_MANAGER_IMPORT_ERROR: Optional[BaseException] = None + +try: + from src.core.event_manager import ( # noqa: E402 + EventManager, + EventStatus, + get_event_manager, + ) + _EVENT_MANAGER_IMPORT_OK = True +except Exception as _exc: # noqa: BLE001 + EventManager = None # type: ignore + EventStatus = None # type: ignore + get_event_manager = None # type: ignore + _EVENT_MANAGER_IMPORT_ERROR = _exc + + +def _get_event_manager() -> Optional[Any]: + """ + 懒加载 EventManager 单例（进程内复用 get_event_manager()）。 + 失败时返回 None（路由层兜底）。 + """ + global _event_manager + if _event_manager is not None: + return _event_manager + if not _EVENT_MANAGER_IMPORT_OK or get_event_manager is None: + return None + try: + _event_manager = get_event_manager() + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error( + 'event_api', + f'获取 EventManager 失败: {exc}', + exc, + ) + _event_manager = None + return _event_manager + + +# --------------------------------------------------------------------- +# Pydantic v2 扩展（不复用/不改动 EventDTO；只在本文内新增） +# --------------------------------------------------------------------- +class EventDTOExtended(BaseModel): + """ + 事件 DTO 扩展版：在 EventDTO 基础上补充 title / description / + priority / updated_at / completed_at / read_at / archived_at， + 供前端 EventTable 渲染。 + + Attributes: + uuid: 事件 ID（与 EventDTO.uuid 一致） + type: 事件类型 + status: 事件状态 + payload: 事件 payload（来自 metadata 字段） + created_at: ISO 8601 时间戳 + title: 事件标题 + description: 事件描述 + priority: 事件优先级（int） + updated_at: 最近更新时间 + completed_at: 完成时间（可能为空） + read_at: 标记已读时间（可能为空） + archived_at: 归档时间（可能为空） + """ + uuid: str + type: str + status: str + payload: Dict + created_at: str + title: str = "" + description: str = "" + priority: int = 2 + updated_at: Optional[str] = None + completed_at: Optional[str] = None + read_at: Optional[str] = None + archived_at: Optional[str] = None + + +# --------------------------------------------------------------------- +# 工具：events 表行 → EventDTO / EventDTOExtended 转换 +# --------------------------------------------------------------------- +def _parse_metadata(raw: Any) -> Dict[str, Any]: + """把 metadata 字段（可能是 dict / str）安全解析成 dict。""" + if raw is None or raw == "": + return {} + if isinstance(raw, dict): + return dict(raw) + if isinstance(raw, str): + try: + return dict(json.loads(raw) or {}) + except Exception: # noqa: BLE001 + return {} + return {} + + +def _row_to_extended_dto(row: Dict[str, Any]) -> EventDTOExtended: + """ + 把数据库行（可能是 sqlite3.Row / dict）映射为 EventDTOExtended。 + + 行字段（events 表）： + event_id, title, description, event_type, priority, status, + created_at, updated_at, completed_at, metadata + """ + if not isinstance(row, dict): + try: + row = dict(row) + except Exception: + row = {} + meta = _parse_metadata(row.get('metadata')) + return EventDTOExtended( + uuid=str(row.get('event_id') or row.get('uuid') or ''), + type=str(row.get('event_type') or row.get('type') or 'event'), + status=str(row.get('status') or 'pending'), + payload=meta, + created_at=str(row.get('created_at') or ''), + title=str(row.get('title') or ''), + description=str(row.get('description') or ''), + priority=int(row.get('priority') or 2) if str(row.get('priority') or '').strip() else 2, + updated_at=row.get('updated_at'), + completed_at=row.get('completed_at'), + read_at=meta.get('read_at'), + archived_at=meta.get('archived_at'), + ) + + +def _row_to_dto(row: Dict[str, Any]) -> EventDTO: + """ + 把数据库行映射为 EventDTO（用于 EventListResponse）。 + """ + if not isinstance(row, dict): + try: + row = dict(row) + except Exception: + row = {} + meta = _parse_metadata(row.get('metadata')) + return EventDTO( + uuid=str(row.get('event_id') or row.get('uuid') or ''), + type=str(row.get('event_type') or row.get('type') or 'event'), + status=str(row.get('status') or 'pending'), + payload=meta, + created_at=str(row.get('created_at') or ''), + ) + + +def _fail(message: str, code: int = status.HTTP_400_BAD_REQUEST) -> Dict[str, Any]: + """统一兜底响应。""" + return {"error": message, "status_code": code} + + +def _execute_sql(em: Any, sql: str, params: tuple = ()) -> bool: + """ + 通过 EventManager 的 db 句柄直接执行 UPDATE/INSERT SQL。 + 返回是否影响至少一行。 + """ + if em is None or not hasattr(em, 'db') or em.db is None: + return False + try: + with em.db.get_connection() as conn: + cur = conn.execute(sql, params) + affected = cur.rowcount + return bool(affected and affected > 0) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('event_api', f'_execute_sql 失败: {exc}', exc) + return False + + +# --------------------------------------------------------------------- +# 路由 +# --------------------------------------------------------------------- +router = APIRouter(prefix="/api/events", tags=["events"]) + + +# ===================================================================== +# GET /api/events?limit=&offset=&type=&status= +# ===================================================================== +@router.get("") +async def list_events( + limit: int = Query(50, ge=1, le=200, description="返回条数上限"), + offset: int = Query(0, ge=0, description="分页偏移"), + type: Optional[str] = Query( + None, + description="事件类型筛选（如 notification / task）", + ), + status_filter: Optional[str] = Query( + None, + alias="status", + description="事件状态筛选（如 pending / completed / cancelled）", + ), +) -> Dict[str, Any]: + """ + 分页 + 筛选事件列表。 + + Returns: + { + "events": [EventDTO...], + "total": int, + "limit": int, + "offset": int, + "items": [EventDTOExtended...], # 扩展字段供前端 EventTable + "warning": str | None + } + """ + em = _get_event_manager() + if em is None: + return { + "events": [], + "items": [], + "total": 0, + "limit": limit, + "offset": offset, + "warning": "EventManager 不可用：导入或初始化失败", + } + + try: + # EventManager.get_all_events(status, event_type, limit) 不支持 offset； + # 我们走 raw SQL 拿到 offset 段。 + # 先确认 events 表存在；不可用就降级空列表。 + try: + with em.db.get_connection() as conn: + cur = conn.execute( + "SELECT name FROM sqlite_master " + "WHERE type='table' AND name='events'", + ) + exists = cur.fetchone() + except Exception: # noqa: BLE001 + exists = None + if not exists: + return { + "events": [], + "items": [], + "total": 0, + "limit": limit, + "offset": offset, + "warning": "events 表未在数据库中创建，当前返回空列表", + } + + # 拼 SQL + where_parts: List[str] = [] + params: List[Any] = [] + if type: + where_parts.append("event_type = ?") + params.append(str(type)) + if status_filter: + where_parts.append("status = ?") + params.append(str(status_filter)) + where_sql = ("WHERE " + " AND ".join(where_parts)) if where_parts else "" + + # 取 total + try: + with em.db.get_connection() as conn: + cur = conn.execute( + f"SELECT COUNT(*) FROM events {where_sql}", + tuple(params), + ) + total_row = cur.fetchone() + total = int(total_row[0]) if total_row else 0 + except Exception: # noqa: BLE001 + total = 0 + + # 取 rows + try: + with em.db.get_connection() as conn: + cur = conn.execute( + f"SELECT * FROM events {where_sql} " + f"ORDER BY datetime(created_at) DESC LIMIT ? OFFSET ?", + tuple(params + [limit, offset]), + ) + raw_rows = cur.fetchall() or [] + except Exception as exc: # noqa: BLE001 + _debug_logger.log_warn('event_api', f'查询 events 失败: {exc}') + raw_rows = [] + + # 转 dict 列表 + rows: List[Dict[str, Any]] = [] + for r in raw_rows: + try: + if isinstance(r, dict): + rows.append(dict(r)) + else: + rows.append({k: r[k] for k in r.keys()}) + except Exception: + try: + rows.append(dict(r)) + except Exception: + continue + + return { + "events": [_row_to_dto(r) for r in rows], + "items": [_row_to_extended_dto(r) for r in rows], + "total": total, + "limit": limit, + "offset": offset, + "warning": None, + } + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('event_api', f'list_events 失败: {exc}', exc) + warnings.warn( + f"[event_api] list_events 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return { + "events": [], + "items": [], + "total": 0, + "limit": limit, + "offset": offset, + "error": str(exc), + } + + +# ===================================================================== +# POST /api/events/{event_id}/read +# ===================================================================== +@router.post("/{event_id}/read") +async def mark_event_read( + event_id: str = Path(..., min_length=1, description="事件 ID"), +) -> Dict[str, Any]: + """ + 标记事件已读。 + - 复用 EventManager.update_event_status(status=COMPLETED, log_message=...)； + - 同时把 read_at 写入 metadata 字段。 + """ + em = _get_event_manager() + if em is None: + raise HTTPException( + status_code=status.HTTP_503_SERVICE_UNAVAILABLE, + detail="EventManager 不可用", + ) + + if not event_id or not event_id.strip(): + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="event_id 必填", + ) + + try: + # 确认事件存在 + existing = em.get_event(event_id) if hasattr(em, 'get_event') else None + if existing is None: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail=f"未找到事件: {event_id}", + ) + + # 1) 写 read_at 到 metadata + try: + meta_raw = None + with em.db.get_connection() as conn: + cur = conn.execute( + "SELECT metadata FROM events WHERE event_id = ?", + (event_id,), + ) + row = cur.fetchone() + meta_raw = row[0] if row else None + except Exception: # noqa: BLE001 + meta_raw = None + + meta = _parse_metadata(meta_raw) or {} + now_iso = datetime.now().isoformat() + meta['read_at'] = now_iso + + try: + with em.db.get_connection() as conn: + conn.execute( + "UPDATE events SET metadata = ? WHERE event_id = ?", + (json.dumps(meta, ensure_ascii=False), event_id), + ) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_warn('event_api', f'写 read_at 到 metadata 失败: {exc}') + + # 2) 状态切到 completed（不强制：失败也不抛 5xx） + status_ok = False + if hasattr(em, 'update_event_status') and EventStatus is not None: + try: + status_ok = em.update_event_status( + event_id, + EventStatus.COMPLETED, + log_message='marked as read', + ) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_warn('event_api', f'update_event_status 失败: {exc}') + status_ok = False + + return { + "event_id": event_id, + "marked": True, + "read_at": now_iso, + "status": "completed", + "status_updated": status_ok, + } + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('event_api', f'mark_event_read 失败: {exc}', exc) + return _fail(f"标记已读失败: {exc}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# POST /api/events/{event_id}/archive +# ===================================================================== +@router.post("/{event_id}/archive") +async def archive_event( + event_id: str = Path(..., min_length=1, description="事件 ID"), +) -> Dict[str, Any]: + """ + 归档事件。 + - 复用 EventManager.update_event_status(status=CANCELLED, log_message=...)； + - 同时把 archived_at 写入 metadata 字段。 + """ + em = _get_event_manager() + if em is None: + raise HTTPException( + status_code=status.HTTP_503_SERVICE_UNAVAILABLE, + detail="EventManager 不可用", + ) + + if not event_id or not event_id.strip(): + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="event_id 必填", + ) + + try: + existing = em.get_event(event_id) if hasattr(em, 'get_event') else None + if existing is None: + raise HTTPException( + status_code=status.HTTP_404_NOT_FOUND, + detail=f"未找到事件: {event_id}", + ) + + # 1) 写 archived_at 到 metadata + try: + meta_raw = None + with em.db.get_connection() as conn: + cur = conn.execute( + "SELECT metadata FROM events WHERE event_id = ?", + (event_id,), + ) + row = cur.fetchone() + meta_raw = row[0] if row else None + except Exception: # noqa: BLE001 + meta_raw = None + + meta = _parse_metadata(meta_raw) or {} + now_iso = datetime.now().isoformat() + meta['archived_at'] = now_iso + + try: + with em.db.get_connection() as conn: + conn.execute( + "UPDATE events SET metadata = ? WHERE event_id = ?", + (json.dumps(meta, ensure_ascii=False), event_id), + ) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_warn('event_api', f'写 archived_at 到 metadata 失败: {exc}') + + # 2) 状态切到 cancelled + status_ok = False + if hasattr(em, 'update_event_status') and EventStatus is not None: + try: + status_ok = em.update_event_status( + event_id, + EventStatus.CANCELLED, + log_message='archived', + ) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_warn('event_api', f'update_event_status 失败: {exc}') + status_ok = False + + return { + "event_id": event_id, + "archived": True, + "archived_at": now_iso, + "status": "cancelled", + "status_updated": status_ok, + } + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('event_api', f'archive_event 失败: {exc}', exc) + return _fail(f"归档失败: {exc}", code=status.HTTP_400_BAD_REQUEST) + + +__all__ = ["router"] diff --git a/src/web/backend/api/knowledge.py b/src/web/backend/api/knowledge.py new file mode 100644 index 0000000..5e2070b --- /dev/null +++ b/src/web/backend/api/knowledge.py @@ -0,0 +1,282 @@ +""" +Knowledge REST API router. +知识库 REST API 路由（Stage C.1）。 + +端点： + GET /api/knowledge/search 关键词搜索（?q=...&limit=20） + GET /api/knowledge 列出所有实体（?limit=50） + POST /api/knowledge 新增实体（body 走 KnowledgeCreate 校验） + PUT /api/knowledge/{uuid} 更新实体（body 走 KnowledgeUpdate 校验） + DELETE /api/knowledge/{uuid} 删除实体 + +设计要点： +1. 所有路由都用 try/except 包裹，失败返回 4xx + {"error": str(e)}， + 不冒 5xx（Stage C.1 验证标准）。 +2. 复用现有 DatabaseManager：search_knowledge / get_all_entities / + add_entity / update_entity / delete_entity。 +3. Pydantic v2 schema 校验（KnowledgeCreate / KnowledgeUpdate）在 + FastAPI 入口处自动完成；非法 body 直接 422。 +4. 路径参数 uuid 透传给 db 层；不存在时返回 404（不抛 500）。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path +from typing import Any, Dict, List, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter, Body, HTTPException, Path, Query, status +from fastapi.responses import JSONResponse + +# Pydantic v2 schema（强类型契约） +from src.web.backend.schemas.knowledge import ( + KnowledgeCreate, + KnowledgeUpdate, +) + +# DatabaseManager（懒加载 + try/except 保护） +try: + from src.core.database_manager import DatabaseManager # type: ignore + _DB_IMPORT_OK = True + _DB_IMPORT_ERROR: Optional[BaseException] = None +except Exception as _exc: # noqa: BLE001 + DatabaseManager = None # type: ignore + _DB_IMPORT_OK = False + _DB_IMPORT_ERROR = _exc + + +router = APIRouter() + + +# --------------------------------------------------------------------- +# DatabaseManager 单例（懒加载） +# --------------------------------------------------------------------- +_db_singleton: Optional[Any] = None + + +def _get_db() -> Optional[Any]: + """ + 懒加载 DatabaseManager 单例。 + 失败时返回 None，由路由层兜底为 503。 + """ + global _db_singleton + if _db_singleton is not None: + return _db_singleton + if not _DB_IMPORT_OK or DatabaseManager is None: + return None + try: + _db_singleton = DatabaseManager() + except Exception: # noqa: BLE001 + _db_singleton = None + return _db_singleton + + +# --------------------------------------------------------------------- +# 工具：把 ValueError / 不存在 uuid 等转成 4xx，绝不 5xx +# --------------------------------------------------------------------- +def _fail(message: str, code: int = status.HTTP_400_BAD_REQUEST) -> JSONResponse: + return JSONResponse(status_code=code, content={"error": message}) + + +# ===================================================================== +# GET /api/knowledge/search?q=...&limit=20 +# ===================================================================== +@router.get("/search") +async def search_knowledge( + q: str = Query(..., min_length=1, description="搜索关键词"), + limit: int = Query(20, ge=1, le=200, description="返回条数上限"), +) -> Dict[str, Any]: + """ + 关键词搜索知识库实体（Stage C.1）。 + + Args: + q: 搜索关键词（必填） + limit: 返回上限（默认 20，1-200） + + Returns: + {"results": [...], "total": int} + """ + try: + db = _get_db() + if db is None: + return _fail( + "DatabaseManager 不可用：导入或初始化失败", + code=status.HTTP_503_SERVICE_UNAVAILABLE, + ) + + results: List[Dict[str, Any]] = db.search_knowledge(query=q, limit=limit) + # 过滤掉非 dict 的脏数据，避免 JSON 序列化失败 + clean: List[Dict[str, Any]] = [r for r in results if isinstance(r, dict)] + return {"results": clean, "total": len(clean)} + except ValueError as e: # noqa: BLE001 + return _fail(f"参数错误: {e}", code=status.HTTP_400_BAD_REQUEST) + except Exception as e: # noqa: BLE001 + return _fail(f"搜索失败: {e}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# GET /api/knowledge?limit=50 +# ===================================================================== +@router.get("") +async def list_knowledge( + limit: int = Query(50, ge=1, le=500, description="返回条数上限"), +) -> Dict[str, Any]: + """ + 列出所有知识库实体（兜底接口，Stage C.1）。 + + 优先复用 DatabaseManager.get_all_entities(limit)；如果 db 不可用 + 或返回非列表，兜底为空列表。 + + Returns: + {"results": [...], "total": int} + """ + try: + db = _get_db() + if db is None: + return _fail( + "DatabaseManager 不可用：导入或初始化失败", + code=status.HTTP_503_SERVICE_UNAVAILABLE, + ) + + rows: List[Dict[str, Any]] = db.get_all_entities(limit=limit) or [] + clean: List[Dict[str, Any]] = [r for r in rows if isinstance(r, dict)] + return {"results": clean, "total": len(clean)} + except ValueError as e: # noqa: BLE001 + return _fail(f"参数错误: {e}", code=status.HTTP_400_BAD_REQUEST) + except Exception as e: # noqa: BLE001 + return _fail(f"列出知识库失败: {e}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# POST /api/knowledge +# ===================================================================== +@router.post("") +async def create_knowledge( + payload: KnowledgeCreate = Body(...), +) -> Dict[str, Any]: + """ + 新增知识库实体（Stage C.1）。 + + Body 通过 KnowledgeCreate (Pydantic v2) 校验。 + + Returns: + {"uuid": str, "entity_name": str, "created": true} + """ + try: + db = _get_db() + if db is None: + return _fail( + "DatabaseManager 不可用：导入或初始化失败", + code=status.HTTP_503_SERVICE_UNAVAILABLE, + ) + + new_uuid: str = db.add_entity( + entity_name=payload.entity_name, + category=payload.category, + description=payload.description, + related_info=payload.related_info, + ) + if not new_uuid: + return _fail("新增实体失败：未返回 uuid", code=status.HTTP_400_BAD_REQUEST) + + return { + "uuid": new_uuid, + "entity_name": payload.entity_name, + "created": True, + } + except ValueError as e: # noqa: BLE001 + return _fail(f"参数错误: {e}", code=status.HTTP_400_BAD_REQUEST) + except Exception as e: # noqa: BLE001 + return _fail(f"新增实体失败: {e}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# PUT /api/knowledge/{uuid} +# ===================================================================== +@router.put("/{uuid}") +async def update_knowledge( + uuid: str = Path(..., min_length=1, description="实体 uuid"), + payload: KnowledgeUpdate = Body(...), +) -> Dict[str, Any]: + """ + 更新知识库实体（Stage C.1）。 + + Body 通过 KnowledgeUpdate (Pydantic v2) 校验。 + + Returns: + {"uuid": str, "updated": bool} + """ + try: + db = _get_db() + if db is None: + return _fail( + "DatabaseManager 不可用：导入或初始化失败", + code=status.HTTP_503_SERVICE_UNAVAILABLE, + ) + + # 至少要传一个可更新字段 + update_data = payload.model_dump(exclude_unset=True) + if not update_data: + return _fail("请求体为空：至少要传一个字段", code=status.HTTP_400_BAD_REQUEST) + + ok: bool = db.update_entity( + entity_uuid=uuid, + entity_name=payload.entity_name, + category=payload.category, + description=payload.description, + related_info=payload.related_info, + ) + if not ok: + return _fail( + f"未找到实体或无字段更新: {uuid}", + code=status.HTTP_404_NOT_FOUND, + ) + + return {"uuid": uuid, "updated": True} + except ValueError as e: # noqa: BLE001 + return _fail(f"参数错误: {e}", code=status.HTTP_400_BAD_REQUEST) + except Exception as e: # noqa: BLE001 + return _fail(f"更新实体失败: {e}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# DELETE /api/knowledge/{uuid} +# ===================================================================== +@router.delete("/{uuid}") +async def delete_knowledge( + uuid: str = Path(..., min_length=1, description="实体 uuid"), +) -> Dict[str, Any]: + """ + 删除知识库实体（Stage C.1）。 + + Returns: + {"uuid": str, "deleted": bool} + """ + try: + db = _get_db() + if db is None: + return _fail( + "DatabaseManager 不可用：导入或初始化失败", + code=status.HTTP_503_SERVICE_UNAVAILABLE, + ) + + ok: bool = db.delete_entity(uuid) + if not ok: + return _fail( + f"未找到实体或删除失败: {uuid}", + code=status.HTTP_404_NOT_FOUND, + ) + + return {"uuid": uuid, "deleted": True} + except ValueError as e: # noqa: BLE001 + return _fail(f"参数错误: {e}", code=status.HTTP_400_BAD_REQUEST) + except Exception as e: # noqa: BLE001 + return _fail(f"删除实体失败: {e}", code=status.HTTP_400_BAD_REQUEST) + + +__all__ = ["router"] diff --git a/src/web/backend/api/llm.py b/src/web/backend/api/llm.py new file mode 100644 index 0000000..6fc4de5 --- /dev/null +++ b/src/web/backend/api/llm.py @@ -0,0 +1,191 @@ +""" +LLM REST API - v3.1.0 +====================== + +端点： +- GET /api/llm/config 返回当前生效 provider / base_url / model / 是否有效 +- POST /api/llm/test 用现配置发一次 ping 调用，验证连通性 + +设计要点： +1. 复用 ``llm_providers.get_provider_summary()`` + ``get_model_config().is_valid()``， + 避免在 Web 层硬编码供应商逻辑。 +2. ``/api/llm/test`` 用 ``LangChainLLM`` 实跑一次调用，latency 用 + ``time.time()`` 简单计时；失败时返回 ``{ok: false, error}``， + 永远不向消费者抛 5xx（验收硬约束）。 +""" + +from __future__ import annotations + +import sys +import time +from pathlib import Path +from typing import Any, Dict, Optional + +from fastapi import APIRouter, HTTPException +from pydantic import BaseModel, Field + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.core import llm_providers +except Exception: # noqa: BLE001 + llm_providers = None # type: ignore + +try: + from src.core.model_config import get_model_config, ModelType +except Exception: # noqa: BLE001 + get_model_config = None # type: ignore + ModelType = None # type: ignore + +try: + from src.core.langchain_llm import LangChainLLM +except Exception: # noqa: BLE001 + LangChainLLM = None # type: ignore + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + + +router = APIRouter(prefix="/api/llm", tags=["llm"]) +debug_logger = get_debug_logger() + + +# ---------------------------------------------------------------------- +# 内部 helper +# ---------------------------------------------------------------------- +def _safe_get_summary() -> Dict[str, Any]: + """ + 取得 provider summary，失败时返默认值。 + """ + if llm_providers is None: + return { + "provider": "unknown", + "base_url": "", + "model_name": "", + "api_key_set": False, + } + try: + return llm_providers.get_provider_summary() + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('api.llm', f'get_provider_summary 失败: {exc}') + except Exception: + pass + return { + "provider": llm_providers.resolve_provider() if llm_providers else "unknown", + "base_url": "", + "model_name": "", + "api_key_set": bool(llm_providers.resolve_api_key()) if llm_providers else False, + } + + +def _get_timeout() -> int: + if get_model_config is None: + return 30 + try: + cfg = get_model_config() + return int(getattr(cfg, "vision_llm_timeout", 30) or 30) + except Exception: + return 30 + + +# ---------------------------------------------------------------------- +# GET /api/llm/config +# ---------------------------------------------------------------------- +@router.get("/config") +async def get_llm_config() -> Dict[str, Any]: + """ + 返回当前生效的 LLM 配置概览。 + """ + summary = _safe_get_summary() + valid = False + temperature = 0.8 + max_tokens = 2000 + timeout = _get_timeout() + + if get_model_config is not None: + try: + cfg = get_model_config() + valid = bool(cfg.is_valid()) + main = cfg.get_model_config(ModelType.MAIN) if ModelType is not None else {} + temperature = float(main.get("temperature", 0.8) or 0.8) + max_tokens = int(main.get("max_tokens", 2000) or 2000) + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('api.llm', f'get_model_config 失败: {exc}') + except Exception: + pass + + return { + "provider": summary.get("provider", "unknown"), + "base_url": summary.get("base_url", ""), + "model_name": summary.get("model_name", ""), + "api_key_set": bool(summary.get("api_key_set", False)), + "valid": valid, + "temperature": temperature, + "max_tokens": max_tokens, + "timeout": timeout, + } + + +# ---------------------------------------------------------------------- +# POST /api/llm/test +# ---------------------------------------------------------------------- +class LLMTestRequest(BaseModel): + prompt: str = Field(default="ping", min_length=1, max_length=2000) + + +@router.post("/test") +async def test_llm_connection(req: Optional[LLMTestRequest] = None) -> Dict[str, Any]: + """ + 调 LangChainLLM 跑一次 ping 调用，返回 {ok, latency_ms, sample}。 + """ + prompt = (req.prompt if req and req.prompt else "ping").strip() or "ping" + + if get_model_config is None or LangChainLLM is None or ModelType is None: + return {"ok": False, "error": "LLM components unavailable", "latency_ms": 0} + + try: + cfg = get_model_config() + if not cfg.is_valid(): + return { + "ok": False, + "error": "API key not configured (set LLM_API_KEY or legacy SILICONFLOW_API_KEY)", + "latency_ms": 0, + } + except Exception as exc: # noqa: BLE001 + return {"ok": False, "error": f"config error: {exc}", "latency_ms": 0} + + start = time.time() + try: + llm = LangChainLLM(ModelType.MAIN) + content = llm.chat([{"role": "user", "content": prompt}]) or "" + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('api.llm', f'LLM chat 失败: {exc}', exc) + except Exception: + pass + return { + "ok": False, + "error": f"{type(exc).__name__}: {exc}", + "latency_ms": int((time.time() - start) * 1000), + } + + latency_ms = int((time.time() - start) * 1000) + return { + "ok": True, + "latency_ms": latency_ms, + "sample_response": (content or "")[:200], + } + + +__all__ = ["router"] diff --git a/src/web/backend/api/memory.py b/src/web/backend/api/memory.py new file mode 100644 index 0000000..ac1ad17 --- /dev/null +++ b/src/web/backend/api/memory.py @@ -0,0 +1,483 @@ +""" +Memory API router +话题时间线后端路由 - Stage B.4 + +提供两个端点（合并到本文件，按 spec 可选拆分为 memory_context.py）： + +- ``GET /api/memory/timeline?days=7`` + 从 ``open_loops`` 表查询最近 N 天的未完话题，构造 + ``TimelineResponse({nodes, links, days, total})``。 +- ``GET /api/memory/loop/{uuid}/context`` + 根据 open_loops.uuid 返回相关对话上下文（来自 ``short_term_memory`` + 关联消息 / open_loop 的 context 字段）。 + +设计要点： +1. 数据源用 ``database_service`` 单例（白名单防御 + 失败降级空 list）。 +2. 所有 SQL 路径 ``try/except`` 保护，失败返回空响应。 +3. ``days`` / ``limit`` 边界保护，避免外部参数异常拖死 DB。 +4. 不修改 LongTermMemoryManager / DatabaseManager 既有签名。 +5. ``app.include_router(memory.router)`` 由 main.py 完成。 +""" + +from __future__ import annotations + +import json +import sys +import warnings +from datetime import datetime, timedelta +from pathlib import Path as PathLib # noqa: F401 # 重命名避免与 fastapi.Path 冲突 +from typing import Any, Dict, List, Optional + +from fastapi import APIRouter, HTTPException, Path, Query, status + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = PathLib(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from src.web.backend.schemas.memory import ( # noqa: E402 + LoopContextItem, + LoopContextResponse, + TimelineLink, + TimelineNode, + TimelineResponse, +) +from src.web.backend.services.database_service import ( # noqa: E402 + database_service, +) + + +router = APIRouter(prefix="/api/memory", tags=["memory"]) + + +# ---------------------------------------------------------------------- +# 内部：日期 / 边界保护 +# ---------------------------------------------------------------------- +_MIN_DAYS = 1 +_MAX_DAYS = 365 +_DEFAULT_DAYS = 7 +_DEFAULT_LIMIT = 200 +_MAX_LIMIT = 1000 + + +def _coerce_days(raw: Any) -> int: + """把 Query 传入的 days 规整到 [1, 365]，失败回退 7。""" + try: + v = int(raw) + except (TypeError, ValueError): + return _DEFAULT_DAYS + if v < _MIN_DAYS: + return _MIN_DAYS + if v > _MAX_DAYS: + return _MAX_DAYS + return v + + +def _coerce_limit(raw: Any) -> int: + """把 limit 规整到 [1, 1000]，失败回退 200。""" + try: + v = int(raw) + except (TypeError, ValueError): + return _DEFAULT_LIMIT + if v < 1: + return 1 + if v > _MAX_LIMIT: + return _MAX_LIMIT + return v + + +def _iso_to_date(iso: Optional[str]) -> str: + """把 ISO 8601 字符串截成 YYYY-MM-DD；失败返回空串。""" + if not iso or not isinstance(iso, str): + return "" + s = iso.strip() + if len(s) >= 10 and s[4:5] == "-" and s[7:8] == "-": + return s[:10] + return "" + + +def _parse_keywords(raw: Any) -> List[str]: + """ + 把 ``related_keywords_json`` / list / str 规整为 List[str]。 + 失败时返回 []。 + """ + if raw is None: + return [] + if isinstance(raw, list): + return [str(x) for x in raw if x] + if isinstance(raw, str): + s = raw.strip() + if not s: + return [] + # 尝试 json.loads（库内以 json 形式存储） + if s.startswith("[") or s.startswith("{"): + try: + parsed = json.loads(s) + if isinstance(parsed, list): + return [str(x) for x in parsed if x] + except Exception: # noqa: BLE001 + pass + # 退化：逗号 / 空格切分 + parts = [p.strip() for p in s.replace(",", " ").split() if p.strip()] + return parts + return [] + + +def _normalize_status(raw: Any) -> str: + """ + 把 open_loops.status 规整为 ``open`` / ``closed`` 之一。 + 数据库里常用 'open' / 'resolved' / 'closed'。前端 TimelineCanvas + 看到 'open'/'closed' 即可按需上色。 + """ + s = str(raw or "").strip().lower() + if not s: + return "open" + if s in ("resolved", "closed", "done", "finished"): + return "closed" + if s in ("open", "pending", "active"): + return "open" + return s + + +def _normalize_node(row: Dict[str, Any]) -> Optional[TimelineNode]: + """ + 把 open_loops 的一行 dict 转换为 TimelineNode。 + 字段缺失 / 解析失败 → 返回 None（让上层跳过）。 + """ + try: + uuid_val = str(row.get("uuid") or "").strip() + if not uuid_val: + return None + topic_val = str(row.get("topic") or "未命名话题") + date_val = _iso_to_date(row.get("raised_at") or row.get("created_at")) + if not date_val: + # 兜底：取 created_at 或今天 + date_val = _iso_to_date(row.get("created_at")) or datetime.now().strftime("%Y-%m-%d") + status_val = _normalize_status(row.get("status")) + try: + mention_count = int(row.get("mention_count") or 0) + except (TypeError, ValueError): + mention_count = 0 + keywords = _parse_keywords(row.get("related_keywords_json") or row.get("related_keywords")) + return TimelineNode( + uuid=uuid_val, + topic=topic_val, + date=date_val, + status=status_val, + category="其他", + mention_count=mention_count, + keywords=keywords, + ) + except Exception: # noqa: BLE001 + return None + + +# ---------------------------------------------------------------------- +# 1) GET /api/memory/timeline?days=7 +# ---------------------------------------------------------------------- +@router.get("/timeline", response_model=TimelineResponse) +async def get_timeline( + days: int = Query( + _DEFAULT_DAYS, + ge=_MIN_DAYS, + le=_MAX_DAYS, + description="查询最近 N 天的未完话题（默认 7，最大 365）", + ), + limit: int = Query( + _DEFAULT_LIMIT, + ge=1, + le=_MAX_LIMIT, + description="返回节点数上限（默认 200，最大 1000）", + ), +) -> TimelineResponse: + """ + 话题时间线后端。 + + 响应：``TimelineResponse({nodes, links, days, total})`` + - ``nodes`` 每个节点来自 ``open_loops`` 一行； + - ``links`` 由共享 related_keywords 推导（>= 2 个节点共享至少一个关键词时建链）。 + + 失败降级：DB 不可用 / 表不存在 / 解析失败 → 返回 ``{nodes:[], links:[]}``。 + """ + safe_days = _coerce_days(days) + safe_limit = _coerce_limit(limit) + + # 截止时间 + try: + cutoff_iso = (datetime.now() - timedelta(days=safe_days)).isoformat() + except Exception: # noqa: BLE001 + cutoff_iso = datetime.now().isoformat() + + # 1) 拉 open_loops + raw_rows: List[Dict[str, Any]] = [] + try: + raw_rows = database_service.query_table("open_loops", limit=safe_limit, offset=0) or [] + except ValueError as exc: + # 理论上不会发生（白名单内），但万一也兜底 + try: + warnings.warn( + f"[memory] query_table('open_loops') 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + except Exception: + pass + raw_rows = [] + except Exception as exc: # noqa: BLE001 + try: + warnings.warn( + f"[memory] query_table('open_loops') 异常: {exc}", + RuntimeWarning, + stacklevel=2, + ) + except Exception: + pass + raw_rows = [] + + # 2) 规整为节点 + 时间过滤 + nodes: List[TimelineNode] = [] + for row in raw_rows: + node = _normalize_node(row) + if node is None: + continue + # 时间窗口过滤：raised_at < cutoff 则丢弃 + try: + raised_iso = str(row.get("raised_at") or row.get("created_at") or "") + if raised_iso and raised_iso < cutoff_iso: + continue + except Exception: + pass + nodes.append(node) + + # 3) 构造 links：共享至少一个 keyword 的两个节点建链 + links: List[TimelineLink] = [] + try: + n = len(nodes) + kw_index: Dict[str, List[str]] = {} + for nd in nodes: + for kw in nd.keywords: + if not kw: + continue + kw_index.setdefault(kw, []).append(nd.uuid) + seen: set = set() + for kw, uuids in kw_index.items(): + if len(uuids) < 2: + continue + for i in range(len(uuids)): + for j in range(i + 1, len(uuids)): + a, b = uuids[i], uuids[j] + key = (a, b) if a < b else (b, a) + if key in seen: + continue + seen.add(key) + links.append(TimelineLink(source=key[0], target=key[1], relation="keyword")) + except Exception: # noqa: BLE001 + links = [] + + return TimelineResponse( + nodes=nodes, + links=links, + days=safe_days, + total=len(nodes), + ) + + +# ---------------------------------------------------------------------- +# 2) GET /api/memory/loop/{uuid}/context +# ---------------------------------------------------------------------- +def _query_short_term_by_keywords( + keywords: List[str], + limit: int = 50, +) -> List[Dict[str, Any]]: + """ + 辅助：在 short_term_memory 中按关键词粗匹配（LIKE）。 + 注意：这是简单实现，不做 join / ranking。 + 失败 → []。 + """ + if not keywords: + return [] + db = database_service.get_db() + if db is None: + return [] + safe_limit = _coerce_limit(limit) + rows: List[Dict[str, Any]] = [] + try: + with db.get_connection() as conn: + cur = conn.cursor() + for kw in keywords: + if not kw: + continue + pattern = f"%{kw}%" + try: + cur.execute( + "SELECT id, role, content, timestamp FROM short_term_memory " + "WHERE content LIKE ? ORDER BY id DESC LIMIT ?", + (pattern, safe_limit), + ) + except Exception: + continue + fetched = cur.fetchall() or [] + for r in fetched: + try: + if isinstance(r, dict): + rows.append(dict(r)) + else: + rows.append({k: r[k] for k in r.keys()}) + except Exception: + try: + rows.append(dict(r)) + except Exception: + continue + except Exception: # noqa: BLE001 + return [] + # 去重（按 id） + dedup: Dict[Any, Dict[str, Any]] = {} + for r in rows: + rid = r.get("id") + if rid is None: + continue + if rid not in dedup: + dedup[rid] = r + merged = list(dedup.values()) + try: + merged.sort(key=lambda x: x.get("id") or 0, reverse=True) + except Exception: + pass + return merged[:safe_limit] + + +def _find_open_loop_by_uuid(uuid_val: str) -> Optional[Dict[str, Any]]: + """ + 在 open_loops 中按 uuid 查找一行。 + 找不到 / 失败 → None。 + """ + if not uuid_val or not isinstance(uuid_val, str): + return None + db = database_service.get_db() + if db is None: + return None + try: + with db.get_connection() as conn: + cur = conn.cursor() + try: + cur.execute( + "SELECT uuid, topic, status, raised_at, resolved_at, " + "related_keywords_json, context, mention_count, " + "created_at, updated_at " + "FROM open_loops WHERE uuid = ?", + (uuid_val,), + ) + except Exception: + # 兜底：SELECT * + cur.execute( + "SELECT * FROM open_loops WHERE uuid = ?", + (uuid_val,), + ) + row = cur.fetchone() + if row is None: + return None + try: + if isinstance(row, dict): + return dict(row) + return {k: row[k] for k in row.keys()} + except Exception: + try: + return dict(row) + except Exception: + return None + except Exception: # noqa: BLE001 + return None + + +@router.get("/loop/{uuid}/context", response_model=LoopContextResponse) +async def get_loop_context( + uuid: str = Path(..., min_length=1, description="话题 UUID（open_loops.uuid）"), + limit: int = Query( + 50, + ge=1, + le=200, + description="返回的上下文消息条数上限（默认 50）", + ), +) -> LoopContextResponse: + """ + 话题上下文（按 open_loops.uuid 关联 short_term_memory）。 + + 响应：``LoopContextResponse({uuid, topic, status, items, total})``。 + 失败 / 未找到 → items=[]。 + """ + safe_uuid = (uuid or "").strip() + if not safe_uuid: + # 空 uuid 视为 4xx 输入 + raise HTTPException( + status_code=status.HTTP_400_BAD_REQUEST, + detail="uuid 不能为空", + ) + + safe_limit = _coerce_limit(limit) + if safe_limit > 200: + safe_limit = 200 + + # 1) 找 open_loop + loop_row: Optional[Dict[str, Any]] = None + try: + loop_row = _find_open_loop_by_uuid(safe_uuid) + except Exception: # noqa: BLE001 + loop_row = None + + if loop_row is None: + # 未找到（也可能 DB 不可用） → 返回空上下文 + return LoopContextResponse(uuid=safe_uuid, items=[], total=0) + + topic_text = str(loop_row.get("topic") or "") + status_text = _normalize_status(loop_row.get("status")) + keywords = _parse_keywords( + loop_row.get("related_keywords_json") or loop_row.get("related_keywords") + ) + + items: List[LoopContextItem] = [] + + # 2) 先放 open_loop.context（兜底 / 锚点） + fallback_context = str(loop_row.get("context") or "").strip() + if fallback_context: + items.append( + LoopContextItem( + role="system", + content=fallback_context, + timestamp=_iso_to_date(loop_row.get("raised_at")) or None, + source="open_loop_context", + ) + ) + + # 3) 再按关键词找 short_term_memory 关联消息 + if keywords: + try: + matched = _query_short_term_by_keywords(keywords, limit=safe_limit) or [] + except Exception: # noqa: BLE001 + matched = [] + for m in matched: + try: + items.append( + LoopContextItem( + role=str(m.get("role") or "user"), + content=str(m.get("content") or ""), + timestamp=(str(m.get("timestamp")) if m.get("timestamp") else None), + source="short_term_memory", + ) + ) + except Exception: # noqa: BLE001 + continue + + # 4) 截断 + if len(items) > safe_limit: + items = items[:safe_limit] + + return LoopContextResponse( + uuid=safe_uuid, + topic=topic_text, + status=status_text, + items=items, + total=len(items), + ) + + +__all__ = ["router"] diff --git a/src/web/backend/api/nps.py b/src/web/backend/api/nps.py new file mode 100644 index 0000000..127c06c --- /dev/null +++ b/src/web/backend/api/nps.py @@ -0,0 +1,395 @@ +""" +NPS REST API router - Stage C.4 / 验收 P0 3.2 +NPS 插件（Neo Plugin System）的 REST API 路由。 + +端点： + GET /api/nps 列出所有 NPS 工具（含 enabled / disabled） + POST /api/nps/register 手动注册一个新的 NPS 工具 + POST /api/nps/invoke 调用指定 NPS 工具并返回结果 + +设计要点： +1. 复用 ``src.nps.nps_registry.NPSRegistry`` 与 + ``src.nps.nps_invoker.NPSInvoker``，Web 层仅做参数透传与序列化。 +2. 进程内懒加载单例：NPSRegistry / NPSInvoker 各保持一份； + NPSInvoker 构造时自动 scan_and_register，扫描失败不抛 5xx。 +3. 所有路由 try/except 包裹，失败返回 4xx + {"error": str(e)}， + 绝不冒 5xx（验收硬约束）。 +4. Pydantic v2 schema（NPSRegisterRequest / NPSInvokeRequest）做 + body 校验，非法 body 由 FastAPI 自动返回 422。 +5. 数据库 / 配置目录不可用时返回空列表 / 默认响应（graceful degradation）。 +""" + +from __future__ import annotations + +import sys +import warnings +from pathlib import Path +from typing import Any, Dict, List, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter, Body, HTTPException, status # noqa: E402 +from pydantic import BaseModel, Field # noqa: E402 + +# 调试日志（try/except 保护：debug_logger 不可用时降级为 stub） +try: + from src.tools.debug_logger import get_debug_logger as _get_debug_logger + _debug_logger = _get_debug_logger() +except Exception: # noqa: BLE001 + class _StubLogger: + def log_info(self, *a, **k): pass + def log_warn(self, *a, **k): pass + def log_error(self, *a, **k): pass + _debug_logger = _StubLogger() + + +# --------------------------------------------------------------------- +# NPSRegistry / NPSInvoker 懒加载（try/except 保护） +# --------------------------------------------------------------------- +_registry: Optional[Any] = None +_invoker: Optional[Any] = None +_NPS_IMPORT_OK: bool = False +_NPS_IMPORT_ERROR: Optional[BaseException] = None + +try: + from src.nps.nps_registry import NPSRegistry, NPSTool # noqa: E402 + from src.nps.nps_invoker import NPSInvoker # noqa: E402 + _NPS_IMPORT_OK = True +except Exception as _exc: # noqa: BLE001 + NPSRegistry = None # type: ignore + NPSTool = None # type: ignore + NPSInvoker = None # type: ignore + _NPS_IMPORT_ERROR = _exc + + +def _get_registry() -> Optional[Any]: + """ + 懒加载 NPSRegistry 单例。 + - 失败时返回 None（路由层兜底为 503 / 空列表）。 + """ + global _registry + if _registry is not None: + return _registry + if not _NPS_IMPORT_OK or NPSRegistry is None: + return None + try: + _registry = NPSRegistry() + # 尝试自动扫描；失败不抛 + try: + _registry.scan_and_register() + except Exception as scan_exc: # noqa: BLE001 + _debug_logger.log_warn( + 'nps_api', + f'自动扫描 NPS 工具失败（不影响路由）: {scan_exc}', + ) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('nps_api', f'创建 NPSRegistry 失败: {exc}', exc) + _registry = None + return _registry + + +def _get_invoker() -> Optional[Any]: + """ + 懒加载 NPSInvoker 单例。 + - 失败时返回 None（路由层兜底）。 + """ + global _invoker + if _invoker is not None: + return _invoker + if not _NPS_IMPORT_OK or NPSInvoker is None: + return None + try: + _invoker = NPSInvoker(registry=_get_registry()) + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('nps_api', f'创建 NPSInvoker 失败: {exc}', exc) + _invoker = None + return _invoker + + +# --------------------------------------------------------------------- +# Pydantic v2 Schemas（强类型契约） +# --------------------------------------------------------------------- +class NPSRegisterRequest(BaseModel): + """ + 手动注册 NPS 工具的请求体。 + Attributes: + tool_id: 工具唯一标识符（必填） + name: 工具名称（必填） + description: 工具功能描述（必填） + keywords: 触发关键词列表（可选） + enabled: 是否启用（默认 True） + version: 版本号（默认 "1.0.0"） + author: 作者（默认 "Unknown"） + """ + tool_id: str = Field(..., min_length=1, max_length=128, description="工具唯一标识符") + name: str = Field(..., min_length=1, max_length=256, description="工具名称") + description: str = Field("", description="工具功能描述（用于LLM判断相关性）") + keywords: List[str] = Field(default_factory=list, description="触发关键词列表") + enabled: bool = Field(True, description="是否启用") + version: str = Field("1.0.0", description="版本号") + author: str = Field("Unknown", description="作者") + + +class NPSInvokeRequest(BaseModel): + """ + 调用 NPS 工具的请求体。 + Attributes: + tool_name: 工具名称或 tool_id（必填） + context: 执行上下文 dict（默认空） + use_llm: 是否使用 LLM 判断相关性（保留字段；此处按名称直接调用） + """ + tool_name: str = Field(..., min_length=1, description="工具名称或 tool_id") + context: Dict[str, Any] = Field(default_factory=dict, description="执行上下文") + use_llm: bool = Field(False, description="是否使用 LLM（直接调用时无效）") + + +# --------------------------------------------------------------------- +# 工具：NPSTool → dict 序列化 +# --------------------------------------------------------------------- +def _tool_to_dict(tool: Any) -> Dict[str, Any]: + """ + 把 NPSTool 序列化为前端兼容的 dict。 + """ + if tool is None: + return {} + try: + if hasattr(tool, 'to_dict') and callable(getattr(tool, 'to_dict')): + return dict(tool.to_dict()) + except Exception: # noqa: BLE001 + pass + # fallback：手动拼装 + return { + 'tool_id': getattr(tool, 'tool_id', ''), + 'name': getattr(tool, 'name', ''), + 'description': getattr(tool, 'description', ''), + 'keywords': list(getattr(tool, 'keywords', []) or []), + 'version': getattr(tool, 'version', '1.0.0'), + 'author': getattr(tool, 'author', 'Unknown'), + 'enabled': bool(getattr(tool, 'enabled', True)), + } + + +def _fail(message: str, code: int = status.HTTP_400_BAD_REQUEST) -> Dict[str, Any]: + """统一兜底响应。""" + return {"error": message, "status_code": code} + + +# --------------------------------------------------------------------- +# 路由 +# --------------------------------------------------------------------- +router = APIRouter(prefix="/api/nps", tags=["nps"]) + + +# ===================================================================== +# GET /api/nps +# ===================================================================== +@router.get("") +async def list_nps() -> Dict[str, Any]: + """ + 列出所有已注册的 NPS 工具。 + + 包含 enabled / disabled 全部工具。 + registry 不可用时返回空列表（graceful degradation）。 + + Returns: + {"items": [...], "total": int, "enabled": int, "disabled": int} + """ + try: + registry = _get_registry() + if registry is None: + return { + "items": [], + "total": 0, + "enabled": 0, + "disabled": 0, + "warning": "NPSRegistry 不可用：导入或初始化失败", + } + + tools = registry.get_all_tools() or [] + items: List[Dict[str, Any]] = [] + enabled_count = 0 + for t in tools: + d = _tool_to_dict(t) + if d: + items.append(d) + if d.get('enabled', True): + enabled_count += 1 + return { + "items": items, + "total": len(items), + "enabled": enabled_count, + "disabled": max(0, len(items) - enabled_count), + } + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('nps_api', f'list_nps 失败: {exc}', exc) + warnings.warn( + f"[nps_api] list_nps 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return _fail(f"列出 NPS 工具失败: {exc}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# POST /api/nps/register +# ===================================================================== +@router.post("/register") +async def register_nps(payload: NPSRegisterRequest = Body(...)) -> Dict[str, Any]: + """ + 手动注册一个新的 NPS 工具。 + + 复用 NPSRegistry.register_tool；tool_id 已存在时返回 409。 + 由于 NPSTool 需要 execute_func，本接口仅做"元数据记录"： + - 把工具元信息写入 NPSConfigManager（configs/nps_plugins/）。 + - 若 registry 中已存在该 tool_id，返回 409。 + - 若不存在，把元信息持久化为可被下次启动读取的条目。 + + Returns: + {"tool_id": str, "name": str, "registered": true, "persisted": bool} + """ + try: + registry = _get_registry() + if registry is None: + raise HTTPException( + status_code=status.HTTP_503_SERVICE_UNAVAILABLE, + detail="NPSRegistry 不可用：导入或初始化失败", + ) + + # 检查是否已存在 + try: + existing = registry.get_tool(payload.tool_id) + except Exception: # noqa: BLE001 + existing = None + if existing is not None: + raise HTTPException( + status_code=status.HTTP_409_CONFLICT, + detail=f"工具已存在: {payload.tool_id}", + ) + + # 把元数据持久化到 NPSConfigManager（graceful：失败不抛 5xx） + persisted = False + try: + from src.nps.nps_config_manager import NPSConfigManager # type: ignore + cm = NPSConfigManager() + cm.set_plugin_config( + payload.tool_id, + { + 'name': payload.name, + 'description': payload.description, + 'keywords': list(payload.keywords or []), + 'enabled': bool(payload.enabled), + 'version': payload.version, + 'author': payload.author, + }, + ) + persisted = True + except Exception as persist_exc: # noqa: BLE001 + _debug_logger.log_warn( + 'nps_api', + f'持久化 NPS 配置失败（仅元信息）: {persist_exc}', + ) + persisted = False + + # 同步把"占位"工具注册到当前进程 registry（无 execute_func，仅元数据） + if NPSTool is not None: + try: + placeholder = NPSTool( + tool_id=payload.tool_id, + name=payload.name, + description=payload.description, + keywords=list(payload.keywords or []), + execute_func=lambda *a, **k: { + 'success': False, + 'error': '此工具未实现 execute_func（仅元数据已注册）', + }, + version=payload.version, + author=payload.author, + enabled=bool(payload.enabled), + ) + registry.register_tool(placeholder) + except Exception as reg_exc: # noqa: BLE001 + _debug_logger.log_warn( + 'nps_api', + f'把工具注册到当前进程 registry 失败: {reg_exc}', + ) + + return { + "tool_id": payload.tool_id, + "name": payload.name, + "registered": True, + "persisted": persisted, + } + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('nps_api', f'register_nps 失败: {exc}', exc) + return _fail(f"注册 NPS 工具失败: {exc}", code=status.HTTP_400_BAD_REQUEST) + + +# ===================================================================== +# POST /api/nps/invoke +# ===================================================================== +@router.post("/invoke") +async def invoke_nps(payload: NPSInvokeRequest = Body(...)) -> Dict[str, Any]: + """ + 调用指定 NPS 工具。 + + 复用 NPSInvoker.invoke_tool_by_name。 + 工具不存在 / 未启用 → 返回 4xx + 友好错误。 + + Returns: + {"success": bool, "tool_id": str, "tool_name": str, + "result": Any | None, "error": str | None} + """ + try: + invoker = _get_invoker() + if invoker is None: + raise HTTPException( + status_code=status.HTTP_503_SERVICE_UNAVAILABLE, + detail="NPSInvoker 不可用：导入或初始化失败", + ) + + # 直接按名称调用（不走 LLM 判相关性） + out = invoker.invoke_tool_by_name( + tool_name=payload.tool_name, + context=payload.context or {}, + ) + # out 形如 {"success": bool, "tool_id": ..., "tool_name": ..., + # "result": ..., "error": ...} + if not isinstance(out, dict): + return _fail( + f"调用结果非 dict: {type(out).__name__}", + code=status.HTTP_400_BAD_REQUEST, + ) + + if not out.get('success', False): + # 工具不存在 / 未启用 / 执行失败 + return { + "success": False, + "tool_id": out.get('tool_id', payload.tool_name), + "tool_name": out.get('tool_name', payload.tool_name), + "result": None, + "error": out.get('error', '调用失败'), + "status_code": status.HTTP_404_NOT_FOUND + if '未找到' in (out.get('error') or '') + else status.HTTP_400_BAD_REQUEST, + } + + return { + "success": True, + "tool_id": out.get('tool_id', payload.tool_name), + "tool_name": out.get('tool_name', payload.tool_name), + "result": out.get('result'), + "error": None, + } + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + _debug_logger.log_error('nps_api', f'invoke_nps 失败: {exc}', exc) + return _fail(f"调用 NPS 工具失败: {exc}", code=status.HTTP_400_BAD_REQUEST) + + +__all__ = ["router"] diff --git a/src/web/backend/api/schedule.py b/src/web/backend/api/schedule.py new file mode 100644 index 0000000..fb1ee6d --- /dev/null +++ b/src/web/backend/api/schedule.py @@ -0,0 +1,339 @@ +""" +Schedule REST API Router +- /api/schedule/ (GET): 按日期查询日程 +- /api/schedule/ (POST): 新建日程 +- /api/schedule/{schedule_id} (PUT): 更新日程 +- /api/schedule/{schedule_id} (DELETE): 删除日程 +- /api/schedule/{schedule_id}/confirm (POST): 协作日程确认（双方确认后才生效） + +Stage C.2: +- 复用 database_service 提供的 schedule_* 表查询/写入接口 +- 失败路径：HTTPException(status_code=4xx)，不抛 5xx +- 协作确认：调用 database_service.confirm_schedule_item +""" + +from __future__ import annotations + +import sys +import warnings +from datetime import datetime +from pathlib import Path +from typing import Any, Dict, List, Optional + +from fastapi import APIRouter, HTTPException, Query +from pydantic import BaseModel, Field + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.web.backend.services.database_service import database_service + from src.web.backend.schemas.schedule import ( + ScheduleCreate, + ScheduleUpdate, + ) +except Exception as exc: # noqa: BLE001 + database_service = None # type: ignore + _IMPORT_ERROR = exc + + +# 协作确认请求体（Stage C.2 扩展） +class ScheduleConfirmRequest(BaseModel): + user_id: str = "default" + note: Optional[str] = None + + +# ---------------------------------------------------------------------- +# 工具 +# ---------------------------------------------------------------------- +def _row_to_dto(row: Dict[str, Any]) -> Dict[str, Any]: + """把数据库行映射为前端 ScheduleDTO 兼容结构。""" + if not isinstance(row, dict): + try: + row = dict(row) + except Exception: + return {} + return { + "id": row.get("id"), + "uuid": str(row.get("id") or row.get("uuid") or ""), + "title": str(row.get("title") or ""), + "description": row.get("description"), + "start_time": str(row.get("start_time") or ""), + "end_time": row.get("end_time"), + "priority": str(row.get("priority") or "normal"), + "status": str(row.get("status") or "pending"), + "schedule_type": str(row.get("schedule_type") or "personal"), + } + + +def _parse_iso_date(value: Optional[str], field: str) -> str: + """YYYY-MM-DD 校验（失败 400）。""" + import re + if not value or not isinstance(value, str): + raise HTTPException(status_code=400, detail=f"{field} 必填") + if not re.match(r"^\d{4}-\d{2}-\d{2}$", value): + raise HTTPException( + status_code=400, + detail=f"{field} 格式必须为 YYYY-MM-DD, 收到: {value!r}", + ) + return value + + +# ---------------------------------------------------------------------- +# 路由 +# ---------------------------------------------------------------------- +router = APIRouter(prefix="/api/schedule", tags=["schedule"]) + + +@router.get("") +async def list_schedule( + date: Optional[str] = Query( + default=None, + description="单日查询 YYYY-MM-DD", + ), + from_date: Optional[str] = Query( + default=None, + alias="from", + description="区间起始 YYYY-MM-DD（与 to 一起使用）", + ), + to_date: Optional[str] = Query( + default=None, + alias="to", + description="区间结束 YYYY-MM-DD（与 from 一起使用）", + ), + user_id: str = Query(default="default"), + limit: int = Query(default=100, ge=1, le=500), +) -> dict: + """ + 按日期查询日程。 + - 兼容 date=YYYY-MM-DD 单日查询 + - 兼容 from / to 区间查询 + - 失败路径：返回空列表（表不存在时也不抛 5xx） + """ + if database_service is None: + return {"items": [], "total": 0, "date": date, "range": None} + + try: + if date: + _parse_iso_date(date, "date") + rows = database_service.get_schedule_by_date(date) or [] + return { + "items": [_row_to_dto(r) for r in rows[:limit]], + "total": min(len(rows), limit), + "date": date, + "range": None, + } + + if from_date or to_date: + if not (from_date and to_date): + raise HTTPException( + status_code=400, + detail="from 与 to 必须成对出现", + ) + _parse_iso_date(from_date, "from") + _parse_iso_date(to_date, "to") + if from_date > to_date: + raise HTTPException( + status_code=400, + detail=f"from ({from_date}) 不能晚于 to ({to_date})", + ) + rows = database_service.get_schedule_by_range(from_date, to_date) or [] + return { + "items": [_row_to_dto(r) for r in rows[:limit]], + "total": min(len(rows), limit), + "date": None, + "range": {"from": from_date, "to": to_date}, + } + + # 未指定过滤：直接读 schedules 表前 limit 条 + try: + tables = database_service.get_tables() or [] + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"DatabaseService 不可用: {exc}") from exc + if "schedules" not in tables and "schedule_entries" not in tables: + return { + "items": [], + "total": 0, + "date": None, + "range": None, + "warning": "schedule_* 表未在数据库中创建, 当前返回空列表", + } + table = "schedules" if "schedules" in tables else "schedule_entries" + rows = database_service.query_table(table, limit=limit, offset=0) or [] + return { + "items": [_row_to_dto(r) for r in rows], + "total": len(rows), + "date": None, + "range": None, + } + except HTTPException: + raise + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[schedule_api] list_schedule 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return {"items": [], "total": 0, "date": date, "range": None, "error": str(exc)} + + +@router.post("") +async def create_schedule(payload: ScheduleCreate) -> dict: + """ + 新建日程。 + - 复用 database_service.add_schedule_item 走白名单 INSERT 路径 + - 失败路径：400/503（不抛 5xx） + """ + if database_service is None: + raise HTTPException(status_code=503, detail="database_service 不可用") + + # 校验：表是否存在 + try: + tables = database_service.get_tables() or [] + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"DatabaseService 不可用: {exc}") from exc + if "schedules" not in tables and "schedule_entries" not in tables: + raise HTTPException( + status_code=400, + detail="schedule_* 表未在数据库中创建, 无法新增日程", + ) + + item: Dict[str, Any] = { + "title": payload.title, + "description": payload.description, + "start_time": payload.start_time, + "end_time": payload.end_time, + "priority": payload.priority or "normal", + "status": "pending", + "schedule_type": payload.schedule_type or "personal", + "created_at": datetime.now().isoformat(), + "updated_at": datetime.now().isoformat(), + } + try: + new_id = database_service.add_schedule_item(item) + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"新增日程失败: {exc}") from exc + if not new_id: + raise HTTPException(status_code=400, detail="新增日程失败: 数据库写入异常") + + return { + "id": int(new_id), + "uuid": str(new_id), + "item": _row_to_dto({**item, "id": int(new_id)}), + "message": "ok", + } + + +@router.put("/{schedule_id}") +async def update_schedule(schedule_id: int, payload: ScheduleUpdate) -> dict: + """ + 更新日程（部分字段）。 + - 复用 database_service.update_schedule_item + """ + if database_service is None: + raise HTTPException(status_code=503, detail="database_service 不可用") + + if schedule_id <= 0: + raise HTTPException(status_code=400, detail="schedule_id 必须为正整数") + + if payload.id and payload.id != schedule_id: + raise HTTPException( + status_code=400, + detail=f"URL id ({schedule_id}) 与 body.id ({payload.id}) 不一致", + ) + + updates: Dict[str, Any] = {} + for field in ( + "title", "description", "start_time", "end_time", + "priority", "status", + ): + value = getattr(payload, field, None) + if value is not None: + updates[field] = value + if not updates: + raise HTTPException(status_code=400, detail="没有可更新的字段") + updates["updated_at"] = datetime.now().isoformat() + + try: + ok = database_service.update_schedule_item(schedule_id, updates) + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"更新失败: {exc}") from exc + if not ok: + raise HTTPException( + status_code=404, + detail=f"未找到 id={schedule_id} 的日程, 或写入失败", + ) + + return { + "id": schedule_id, + "updated_fields": list(updates.keys()), + "message": "ok", + } + + +@router.delete("/{schedule_id}") +async def delete_schedule(schedule_id: int) -> dict: + """ + 删除日程。 + """ + if database_service is None: + raise HTTPException(status_code=503, detail="database_service 不可用") + + if schedule_id <= 0: + raise HTTPException(status_code=400, detail="schedule_id 必须为正整数") + + try: + ok = database_service.delete_schedule_item(schedule_id) + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"删除失败: {exc}") from exc + if not ok: + raise HTTPException( + status_code=404, + detail=f"未找到 id={schedule_id} 的日程, 或删除失败", + ) + + return { + "id": schedule_id, + "deleted": True, + "message": "ok", + } + + +@router.post("/{schedule_id}/confirm") +async def confirm_schedule( + schedule_id: int, + confirmed: bool = Query(default=True, description="true=确认, false=取消确认"), + payload: Optional[ScheduleConfirmRequest] = None, +) -> dict: + """ + 协作日程确认（双方确认后才生效）。 + - 复用 database_service.confirm_schedule_item 写入 confirmed 字段 + - 失败路径：404/400（不抛 5xx） + """ + if database_service is None: + raise HTTPException(status_code=503, detail="database_service 不可用") + + if schedule_id <= 0: + raise HTTPException(status_code=400, detail="schedule_id 必须为正整数") + + try: + ok = database_service.confirm_schedule_item(schedule_id, bool(confirmed)) + except Exception as exc: # noqa: BLE001 + raise HTTPException(status_code=400, detail=f"确认失败: {exc}") from exc + if not ok: + raise HTTPException( + status_code=404, + detail=f"未找到 id={schedule_id} 的日程, 或更新失败", + ) + + return { + "id": schedule_id, + "confirmed": bool(confirmed), + "status": "confirmed" if confirmed else "pending", + "message": "ok", + } + + +__all__ = ["router"] diff --git a/src/web/backend/main.py b/src/web/backend/main.py new file mode 100644 index 0000000..48ec345 --- /dev/null +++ b/src/web/backend/main.py @@ -0,0 +1,289 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +Neo Agent Web Backend - FastAPI Entry Point +Neo Agent Web 后端 - FastAPI 入口 + +Stage A.1: 基础框架与健康检查 +Stage D.3: EventService 启动 + WebSocket 路由挂载 +""" + +import sys +import os +from pathlib import Path +from datetime import datetime, timezone + +# Python 路径处理：确保从项目根目录（Neo_Agent/）能正确导入 src.* 模块 +PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(PROJECT_ROOT)) + +from fastapi import FastAPI +from fastapi.middleware.cors import CORSMiddleware + +# FastAPI 应用实例 +app = FastAPI( + title="Neo Agent Web", + version="3.0.0", + description="Neo Agent Web GUI 后端服务", +) + +# CORS 中间件配置（开发态允许本地 Vite/后端端口） +app.add_middleware( + CORSMiddleware, + allow_origins=[ + "http://localhost:5173", + "http://localhost:8000", + "http://127.0.0.1:5173", + "http://127.0.0.1:8000", + ], + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], +) + + +@app.get("/api/health") +async def health_check() -> dict: + """ + 健康检查端点 + Health check endpoint. + + 返回服务状态、版本号与当前时间戳（ISO 8601, UTC）。 + """ + return { + "status": "ok", + "version": "3.0.0", + "timestamp": datetime.now(timezone.utc).isoformat(), + } + + +@app.get("/api/chat") +async def chat_placeholder() -> dict: + """ + 占位聊天端点（Stage A.3 替换为真实实现） + """ + return {"message": "chat endpoint placeholder"} + + +# ===================================================================== +# Stage D.3: EventService 单例 + WebSocket 路由 +# ===================================================================== +# 启动时创建单例 EventService 并订阅 EventManager； +# 之后 ws/* 路由通过 EventService 把事件广播到 /ws/events / /ws/debug 等。 +_event_service = None +_event_service_started = False +_debug_broadcaster_started = False + + +def _get_event_service(): + """懒加载 EventService 单例。""" + global _event_service + if _event_service is None: + try: + from src.web.backend.services.event_service import get_event_service + _event_service = get_event_service() + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'创建 EventService 失败: {e}', e) + except Exception: + pass + _event_service = None + return _event_service + + +@app.on_event("startup") +async def _on_startup() -> None: + """应用启动钩子：启动 EventService + 挂载 ws 路由 + 启动 DebugBroadcaster。""" + global _event_service_started, _debug_broadcaster_started + es = _get_event_service() + if es is not None and not _event_service_started: + try: + es.start() + _event_service_started = True + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'EventService.start 失败: {e}', e) + except Exception: + pass + + # Stage C.1: 挂载 Knowledge REST API + try: + from src.web.backend.api.knowledge import router as knowledge_router + app.include_router(knowledge_router, prefix="/api/knowledge") + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 knowledge 路由失败: {e}', e) + except Exception: + pass + + # 挂载 ws 路由（每个进程只挂一次） + try: + from src.web.backend.ws import router as ws_router, get_loaded_status + # 挂载到 /ws/* 子路径（chat/debug/events 的 router 已包含完整路径） + app.include_router(ws_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 ws 路由失败: {e}', e) + except Exception: + pass + + # Stage C.2: 挂载 /api/schedule 路由 + try: + from src.web.backend.api.schedule import router as schedule_router + app.include_router(schedule_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 schedule 路由失败: {e}', e) + except Exception: + pass + + # 挂载 REST API 路由（debug / knowledge / schedule） + try: + from src.web.backend.api import router as api_router + # 聚合 router 自身为空；子模块 router 内部已定义 prefix='/api/...' + app.include_router(api_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 api 路由失败: {e}', e) + except Exception: + pass + + # Stage C.5 / C.6: 挂载 REST 路由 + try: + from src.web.backend.api.database import router as database_router + app.include_router(database_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 database 路由失败: {e}', e) + except Exception: + pass + + try: + from src.web.backend.api.creative import router as creative_router + app.include_router(creative_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 creative 路由失败: {e}', e) + except Exception: + pass + + # Stage B.3 + B.4: 情感雷达 + 话题时间线路由 + try: + from src.web.backend.api.emotion import router as emotion_router + app.include_router(emotion_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 emotion 路由失败: {e}', e) + except Exception: + pass + + try: + from src.web.backend.api.memory import router as memory_router + app.include_router(memory_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 memory 路由失败: {e}', e) + except Exception: + pass + + # Stage C.3 / C.4 / 验收 P0 3.2: 挂载 NPS + Events REST 路由 + try: + from src.web.backend.api.nps import router as nps_router + app.include_router(nps_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 nps 路由失败: {e}', e) + except Exception: + pass + + try: + from src.web.backend.api.event import router as event_router + app.include_router(event_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 event 路由失败: {e}', e) + except Exception: + pass + + # v3.1.0: 挂载 LLM 路由（GET /api/llm/config, POST /api/llm/test） + # 注：router 自身已带 prefix="/api/llm"，main.py 不再叠加 + try: + from src.web.backend.api.llm import router as llm_router + app.include_router(llm_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 llm 路由失败: {e}', e) + except Exception: + pass + + # v3.1.0: 挂载 chat_sessions 路由（GET/POST/PATCH/DELETE /api/chat/sessions） + # 注：router 自身已带 prefix="/api/chat/sessions"，main.py 不再叠加 + try: + from src.web.backend.api.chat_sessions import router as chat_sessions_router + app.include_router(chat_sessions_router) + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'挂载 chat_sessions 路由失败: {e}', e) + except Exception: + pass + + # Stage D.4: 启动 DebugBroadcaster，把 DebugLogger 推送到 /ws/debug + if not _debug_broadcaster_started: + try: + import asyncio + loop = asyncio.get_event_loop() + from src.web.backend.ws.manager import get_connection_manager + from src.web.backend.ws.debug_broadcaster import get_debug_broadcaster + broadcaster = get_debug_broadcaster() + broadcaster.attach( + manager=get_connection_manager(), + event_service=es, + loop=loop, + ) + _debug_broadcaster_started = True + except Exception as e: # noqa: BLE001 + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_error('main', f'DebugBroadcaster 启动失败: {e}', e) + except Exception: + pass + + +@app.on_event("shutdown") +async def _on_shutdown() -> None: + """应用关闭钩子。""" + global _event_service_started, _debug_broadcaster_started + if _event_service is not None and _event_service_started: + try: + _event_service.stop() + except Exception: + pass + _event_service_started = False + # 关闭 DebugBroadcaster + try: + from src.web.backend.ws.debug_broadcaster import get_debug_broadcaster + get_debug_broadcaster().detach() + except Exception: + pass + _debug_broadcaster_started = False + + +if __name__ == "__main__": + import uvicorn + + uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True) diff --git a/src/web/backend/schemas/__init__.py b/src/web/backend/schemas/__init__.py new file mode 100644 index 0000000..fb3e2c8 --- /dev/null +++ b/src/web/backend/schemas/__init__.py @@ -0,0 +1,90 @@ +""" +Schemas package - 统一导出所有 Pydantic v2 模型。 +Schemas package - re-export all Pydantic v2 models. + +Stage A.3: 供 /api/* 路由按需导入： + from src.web.backend.schemas import ChatRequest, ChatResponse + +Stage B.3: 新增 EmotionResponse（情感雷达响应）。 +""" + +from __future__ import annotations + +from .chat import ChatChunk, ChatRequest, ChatResponse +from .creative import ( + AdvanceRequest, + AdvanceResponse, + Chapter, + Character, + CharacterRelation, + CreativeAdvanceRequest, + CreativeAdvanceResponse, + CreativeProject, + CreativeProjectCreate, + CreativeProjectDetail, + CreativeProjectListResponse, + CreativeProjectSummary, + StoryBible, +) +from .database import ( + CountResponse, + DeleteResponse, + TableDataResponse, + TablesListResponse, +) +from .emotion import EmotionLatestResponse, EmotionResponse, PLUTCHIK_KEYS +from .event import EventDTO, EventListResponse +from .knowledge import ( + KnowledgeDefinition, + KnowledgeEntity, + KnowledgeSearchRequest, + KnowledgeSearchResponse, +) +from .memory import ( + LoopContextItem, + LoopContextResponse, + TimelineLink, + TimelineNode, + TimelineResponse, +) +from .schedule import ScheduleCreate, ScheduleDTO, ScheduleUpdate + +__all__ = [ + "ChatRequest", + "ChatChunk", + "ChatResponse", + "CountResponse", + "CreativeProject", + "CreativeProjectCreate", + "CreativeProjectListResponse", + "CreativeProjectSummary", + "CreativeProjectDetail", + "CreativeAdvanceRequest", + "CreativeAdvanceResponse", + "AdvanceRequest", + "AdvanceResponse", + "StoryBible", + "Character", + "CharacterRelation", + "Chapter", + "DeleteResponse", + "EmotionResponse", + "EmotionLatestResponse", + "PLUTCHIK_KEYS", + "EventDTO", + "EventListResponse", + "KnowledgeEntity", + "KnowledgeDefinition", + "KnowledgeSearchRequest", + "KnowledgeSearchResponse", + "TableDataResponse", + "TablesListResponse", + "TimelineNode", + "TimelineLink", + "TimelineResponse", + "LoopContextItem", + "LoopContextResponse", + "ScheduleDTO", + "ScheduleCreate", + "ScheduleUpdate", +] diff --git a/src/web/backend/schemas/chat.py b/src/web/backend/schemas/chat.py new file mode 100644 index 0000000..c9c3fe4 --- /dev/null +++ b/src/web/backend/schemas/chat.py @@ -0,0 +1,58 @@ +""" +Chat schemas - Pydantic v2 models for chat endpoints. +聊天相关的 Pydantic v2 模型。 + +Stage A.3: 为 /api/chat 提供请求/响应/流式 chunk 的契约。 +""" + +from __future__ import annotations + +from typing import Dict, Optional + +from pydantic import BaseModel, Field + + +class ChatRequest(BaseModel): + """ + 聊天请求体。 + Attributes: + user_input: 用户输入文本（必填，1-10000 字符） + context: 可选上下文（不直接传给 agent，保留供未来扩展） + stream: 是否走流式响应（Stage B.2 才真正使用） + """ + user_input: str = Field(..., min_length=1, max_length=10000) + context: Optional[Dict] = None + stream: bool = False + + +class ChatChunk(BaseModel): + """ + 流式响应 chunk。 + Attributes: + type: chunk 类型（chunk / done / error）；用于客户端按类型分发 + chunk: 增量文本片段 + done: 是否为最后一个 chunk + message_id: 消息 ID（首个 chunk 必带，便于客户端关联） + """ + type: str = Field(default="chunk", description="chunk 类型：chunk / done / error") + chunk: str + done: bool = False + message_id: Optional[str] = None + + +class ChatResponse(BaseModel): + """ + 同步聊天响应。 + Attributes: + content: 完整回复文本 + message_id: 消息唯一标识 + emotion: 情绪标签/分值（可选） + timestamp: ISO 8601 时间戳 + """ + content: str + message_id: str + emotion: Optional[Dict] = None + timestamp: str + + +__all__ = ["ChatRequest", "ChatChunk", "ChatResponse"] diff --git a/src/web/backend/schemas/creative.py b/src/web/backend/schemas/creative.py new file mode 100644 index 0000000..c254c16 --- /dev/null +++ b/src/web/backend/schemas/creative.py @@ -0,0 +1,302 @@ +""" +Creative schemas - Pydantic v2 models for creative project endpoints. +长期创作项目相关的 Pydantic v2 模型。 + +Stage C.6: 为 /api/creative/* 提供 list/detail/advance 契约。 +字段命名与前端 ``src/web/frontend/src/types/creative.ts`` 对齐。 +""" + +from __future__ import annotations + +from typing import Any, Dict, List, Optional + +from pydantic import BaseModel, Field + + +# ===================================================================== +# Story Bible 子结构 +# ===================================================================== + + +class CharacterRelation(BaseModel): + """ + 角色关系条目。 + Attributes: + target: 目标角色名 + relation: 关系描述 + """ + target: str = "" + relation: str = "" + + +class Character(BaseModel): + """ + 角色条目。 + Attributes: + uuid: 角色 uuid + name: 角色名 + role: 角色定位（主角/配角/...） + description: 角色描述 + relations: 与其他角色的关系 + """ + uuid: str = "" + name: str = "" + role: str = "" + description: str = "" + relations: List[CharacterRelation] = Field(default_factory=list) + + +class Chapter(BaseModel): + """ + 章节条目。 + Attributes: + uuid: 章节 uuid + chapter_title: 章节标题 + word_count: 字数 + content: 章节正文 + """ + uuid: str = "" + chapter_title: str = "" + word_count: int = 0 + content: str = "" + + +class StoryBible(BaseModel): + """ + Story Bible（项目级线索状态机）。 + Attributes: + characters: 角色列表 + worldview: 世界观描述 + chapters: 章节列表 + mainline_direction: 主线方向（与 creative_writer 对齐） + active_themes: 当前活跃主题 + unresolved_threads: 未解线索 + resolved_threads: 已解线索 + important_facts: 重要事实 + next_direction: 下一段方向 + recent_keywords: 最近关键词 + """ + characters: List[Character] = Field(default_factory=list) + worldview: str = "" + chapters: List[Chapter] = Field(default_factory=list) + mainline_direction: str = "" + active_themes: List[str] = Field(default_factory=list) + unresolved_threads: List[str] = Field(default_factory=list) + resolved_threads: List[str] = Field(default_factory=list) + important_facts: List[str] = Field(default_factory=list) + next_direction: str = "" + recent_keywords: List[str] = Field(default_factory=list) + + +# ===================================================================== +# Project 模型 +# ===================================================================== + + +class CreativeProject(BaseModel): + """ + 创作项目（详情/统一形态）。 + Attributes: + uuid: 项目 uuid + title: 标题 + status: 业务状态（active / paused / completed / drafting / finished） + word_count: 累计字数 + current_chars: 当前字数（与数据库 current_chars 一致） + target_chars: 目标字数 + chapter_count: 章节数量 + work_type: 体裁 + premise: 故事前提 + tone: 语气 + point_of_view: 视角 + inspiration_source: 灵感来源 + outline: 大纲 + characters: 角色列表 + draft_chunks: 续写片段 + next_advance_at: 下次续写时间 + last_advanced_at: 上次续写时间 + created_at: ISO 8601 创建时间 + updated_at: ISO 8601 更新时间 + story_bible: Story Bible（详情场景下填充） + """ + uuid: str + title: str + status: str = "drafting" + word_count: int = 0 + current_chars: int = 0 + target_chars: int = 5000 + chapter_count: int = 0 + work_type: str = "短篇" + premise: str = "" + tone: str = "" + point_of_view: str = "第一人称" + inspiration_source: str = "" + outline: List[Any] = Field(default_factory=list) + characters: List[Any] = Field(default_factory=list) + draft_chunks: List[Any] = Field(default_factory=list) + next_advance_at: Optional[str] = None + last_advanced_at: Optional[str] = None + created_at: str = "" + updated_at: str = "" + story_bible: Optional[StoryBible] = None + + +class CreativeProjectCreate(BaseModel): + """ + 创建创作项目请求体（POST /api/creative/projects）。 + Attributes: + title: 标题（必填，1-200 字符） + work_type: 体裁（短篇/中篇/长篇/散文/诗歌） + premise: 故事前提 + tone: 语气 + point_of_view: 视角 + target_chars: 目标字数 + inspiration_source: 灵感来源描述 + outline: 大纲（每项一段话） + characters: 角色列表 + """ + title: str = Field(..., min_length=1, max_length=200) + work_type: str = Field("短篇", max_length=50) + premise: str = Field("", max_length=2000) + tone: str = Field("", max_length=200) + point_of_view: str = Field("第一人称", max_length=50) + target_chars: int = Field(5000, ge=100, le=1_000_000) + inspiration_source: str = Field("", max_length=1000) + outline: List[str] = Field(default_factory=list) + characters: List[Dict[str, Any]] = Field(default_factory=list) + + +# ===================================================================== +# Advance（续写）模型 +# ===================================================================== + + +class AdvanceRequest(BaseModel): + """ + 续写请求体。 + Attributes: + user: 触发用户标识（默认 "default"） + hint: 可选风格/方向提示（保留给未来扩展） + advance_async: 是否异步触发（默认 True） + - True: 立即返回 task_id，异步生成内容并通过 'creative_progress' 事件推送 + - False: 同步触发，更新 last_advanced_at 后立即返回 + """ + user: str = Field("default", max_length=100) + hint: Optional[str] = Field(None, max_length=2000) + advance_async: bool = True + + +class AdvanceResponse(BaseModel): + """ + 续写响应。 + Attributes: + project_uuid: 项目 uuid + task_id: 异步任务 id（同步模式时为空） + status: 任务状态（queued / completed / failed） + last_advanced_at: 上次续写时间（同步模式下为写入时间） + project_status: 续写后项目状态 + message: 人类可读的描述 + """ + project_uuid: str + task_id: str = "" + status: str = "completed" + last_advanced_at: Optional[str] = None + project_status: str = "drafting" + message: str = "" + + +# ===================================================================== +# 列表响应包装 +# ===================================================================== + + +class CreativeProjectListResponse(BaseModel): + """ + 创作项目列表响应。 + Attributes: + projects: 项目列表 + total: 总条数 + """ + projects: List[CreativeProject] = Field(default_factory=list) + total: int = 0 + + +# ===================================================================== +# 兼容旧版命名（保留以防其他地方引用） +# ===================================================================== + + +class CreativeProjectSummary(BaseModel): + """ + 创作项目摘要（列表项）。 + Attributes: + uuid: 项目唯一标识 + title: 标题 + status: 业务状态（drafting / finished / paused） + chapter_count: 章节数量（从 draft_chunks 派生） + created_at: ISO 8601 创建时间 + updated_at: ISO 8601 更新时间 + """ + uuid: str + title: str + status: str + chapter_count: int = 0 + created_at: str + updated_at: str + + +class CreativeProjectDetail(CreativeProjectSummary): + """ + 创作项目详情。 + Attributes: + story_bible: Story Bible 全文（含 mainline_direction / active_themes / 等） + chapters: 章节列表（从 draft_chunks 派生） + memory_pool_summary: 记忆池摘要（计数 + 最新若干条） + """ + story_bible: Dict[str, Any] = Field(default_factory=dict) + chapters: List[Dict[str, Any]] = Field(default_factory=list) + memory_pool_summary: Dict[str, Any] = Field(default_factory=dict) + + +class CreativeAdvanceRequest(BaseModel): + """ + 续写请求体（payload 可选）。 + Attributes: + user: 触发用户标识（默认 "default"） + hint: 可选风格/方向提示（保留给未来扩展） + """ + user: str = "default" + hint: Optional[str] = None + + +class CreativeAdvanceResponse(BaseModel): + """ + 续写响应。 + Attributes: + project_uuid: 项目 uuid + new_chapter: 续写后追加的新章节（content / created_at / char_count） + project_status: 续写后项目状态 + """ + project_uuid: str + new_chapter: Dict[str, Any] + project_status: str = "drafting" + + +__all__ = [ + # 主体模型 + "CreativeProject", + "CreativeProjectCreate", + "CreativeProjectListResponse", + # Story Bible + "StoryBible", + "Character", + "CharacterRelation", + "Chapter", + # Advance + "AdvanceRequest", + "AdvanceResponse", + # 兼容旧名 + "CreativeProjectSummary", + "CreativeProjectDetail", + "CreativeAdvanceRequest", + "CreativeAdvanceResponse", +] diff --git a/src/web/backend/schemas/database.py b/src/web/backend/schemas/database.py new file mode 100644 index 0000000..1171126 --- /dev/null +++ b/src/web/backend/schemas/database.py @@ -0,0 +1,69 @@ +""" +Database schemas - Pydantic v2 models for database management endpoints. +数据库管理端点相关的 Pydantic v2 模型。 + +Stage C.5: 为 /api/database/* 提供 list/count/query/delete 契约。 +字段命名与前端 ``src/web/frontend/src/types/database.ts`` 对齐。 +""" + +from __future__ import annotations + +from typing import Any, Dict, List + +from pydantic import BaseModel, Field + + +class TablesListResponse(BaseModel): + """ + 白名单表列表响应。 + Attributes: + tables: 表名列表（仅白名单中实际存在的表） + total: 表数量 + """ + tables: List[str] = Field(default_factory=list) + total: int = 0 + + +class TableDataResponse(BaseModel): + """ + 业务表数据查询响应。 + Attributes: + table: 表名 + rows: 行数据列表（每行为 dict） + total: 本次返回的行数 + """ + table: str + rows: List[Dict[str, Any]] = Field(default_factory=list) + total: int = 0 + + +class CountResponse(BaseModel): + """ + 表记录数响应。 + Attributes: + table: 表名 + total: 该表的总记录数 + """ + table: str + total: int = 0 + + +class DeleteResponse(BaseModel): + """ + 删除记录响应。 + Attributes: + table: 表名 + uuid: 被删除记录的 uuid + deleted: 是否成功删除 + """ + table: str + uuid: str + deleted: bool = False + + +__all__ = [ + "TablesListResponse", + "TableDataResponse", + "CountResponse", + "DeleteResponse", +] diff --git a/src/web/backend/schemas/debug.py b/src/web/backend/schemas/debug.py new file mode 100644 index 0000000..f61fe43 --- /dev/null +++ b/src/web/backend/schemas/debug.py @@ -0,0 +1,46 @@ +""" +Debug schemas - Pydantic v2 models for debug log endpoints. +调试日志相关的 Pydantic v2 模型。 + +Stage B.5: 为 /api/debug/* 提供日志条目/列表响应契约。 +""" + +from __future__ import annotations + +from typing import Any, Dict, List, Optional + +from pydantic import BaseModel, Field + + +class DebugLogEntry(BaseModel): + """ + 调试日志条目。 + Attributes: + timestamp: ISO 8601 时间戳 + module: 模块名 + level: 日志级别（INFO / WARN / ERROR / MODULE / PROMPT / REQUEST / RESPONSE） + message: 消息主体（与 log_* 入口参数一致；可为空以便承载结构化日志） + extra: 额外结构化数据（可包含 file_info / traceback / payload 等） + """ + timestamp: str + module: str + level: str + message: str = "" + extra: Optional[Dict[str, Any]] = None + + +class DebugLogListResponse(BaseModel): + """ + 调试日志列表响应。 + Attributes: + logs: 日志条目列表 + total: 总匹配条数 + """ + logs: List[DebugLogEntry] = Field(default_factory=list) + total: int = 0 + + +__all__ = [ + "DebugLogEntry", + "DebugLogListResponse", +] diff --git a/src/web/backend/schemas/emotion.py b/src/web/backend/schemas/emotion.py new file mode 100644 index 0000000..9cb5681 --- /dev/null +++ b/src/web/backend/schemas/emotion.py @@ -0,0 +1,94 @@ +""" +Emotion schemas - Pydantic v2 models for emotion endpoints. +情感分析相关的 Pydantic v2 模型。 + +Stage B.3: 为 /api/emotion/* 提供情感雷达所需的响应契约。 +对齐前端 EmotionPanel.tsx 的 EmotionData 类型。 + +`cumulative` 字段映射"关系状态"维度（EmotionRelationshipAnalyzer +累加评分），`plutchik` 字段映射"当下情绪"维度（PlutchikEmotionWheel +8 维归一化强度）。`last_message` 为最近一次参与情感分析的用户消息。 +""" + +from __future__ import annotations + +from typing import Dict, Optional + +from pydantic import BaseModel, ConfigDict, Field + + +# ---------------------------------------------------------------------- +# 常量：Plutchik 8 维 +# ---------------------------------------------------------------------- +PLUTCHIK_KEYS = ( + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', +) + + +def _default_eight() -> Dict[str, float]: + """8 维默认值 0.0。""" + return {k: 0.0 for k in PLUTCHIK_KEYS} + + +# ---------------------------------------------------------------------- +# Pydantic v2 模型 +# ---------------------------------------------------------------------- +class EmotionResponse(BaseModel): + """ + 最新情感分析响应（Stage B.3 雷达后端契约）。 + + Attributes: + cumulative: 累加评分（Plutchik 8 维）。 + 由 EmotionRelationshipAnalyzer 的累加评分映射而来， + 代表"关系状态"维度。 + plutchik: Plutchik 8 维情绪强度（0-1 归一化）。 + 由 PlutchikEmotionWheel 计算，代表"当下情绪"维度。 + last_message: 最近一次参与情感分析的用户消息原文。 + 没有则为空字符串。 + timestamp: ISO 8601 时间戳（情感分析完成时刻）。 + 没有则为空字符串。 + user_id: 用户标识（请求里透传，默认 "default"）。 + historical_max: 可选历史最高（淡色背景叠加）。 + + Notes: + - 8 维字典使用 default_factory 兜底；缺键自动补 0.0。 + - 兼容 Pydantic v2 from_attributes，便于直接以 ORM 行/类填充。 + """ + model_config = ConfigDict(from_attributes=True) + + cumulative: Dict[str, float] = Field( + default_factory=_default_eight, + description="累加评分（8 维 0-1 归一化）", + ) + plutchik: Dict[str, float] = Field( + default_factory=_default_eight, + description="Plutchik 8 维当下情绪（0-1 归一化）", + ) + last_message: str = Field( + default="", + description="最近一次参与分析的用户消息", + ) + timestamp: str = Field( + default="", + description="ISO 8601 时间戳", + ) + user_id: str = Field( + default="default", + description="用户标识", + ) + historical_max: Optional[Dict[str, float]] = Field( + default=None, + description="可选历史最高（8 维）", + ) + + +# 兼容历史导入：保留旧名字（Stage A.3 早期曾用过） +EmotionLatestResponse = EmotionResponse + + +__all__ = [ + "PLUTCHIK_KEYS", + "EmotionResponse", + "EmotionLatestResponse", +] diff --git a/src/web/backend/schemas/event.py b/src/web/backend/schemas/event.py new file mode 100644 index 0000000..6ef6ce2 --- /dev/null +++ b/src/web/backend/schemas/event.py @@ -0,0 +1,74 @@ +""" +Event schemas - Pydantic v2 models for event listing. +事件查询相关的 Pydantic v2 模型。 + +Stage A.3: 为 /api/events 提供事件列表的契约。 + +F.7.2: 为对齐实际 events 表（event_id / event_type / source / tags）， +新增可选字段并保持原 uuid / type 向后兼容。Pydantic v2 用 +``model_config = ConfigDict(populate_by_name=True)`` 同时接受旧字段。 +""" + +from __future__ import annotations + +from typing import Dict, List, Optional + +from pydantic import BaseModel, ConfigDict, Field + + +class EventDTO(BaseModel): + """ + 事件数据传输对象。 + Attributes: + uuid: 事件唯一标识（旧版别名，建议新代码使用 event_id） + type: 事件类型（旧版别名，建议新代码使用 event_type） + status: 事件状态（如 'pending' / 'sent' / 'failed'） + payload: 事件 payload 字典 + created_at: ISO 8601 时间戳 + + 扩展字段（F.7.2，对齐 events 表）： + event_id: 实际表主键，Optional[str] = None + event_type: 实际表的 event_type 列，Optional[str] = None + source: 事件来源（如 'scheduler' / 'proactive' / 'manual'），Optional[str] = None + tags: 标签列表，Optional[List[str]] = None + """ + # 旧字段（保持向后兼容，client 仍在用） + uuid: str + type: str + status: str + payload: Dict + created_at: str + + # 新增字段（对齐实际表，原 uuid 字段保留） + event_id: Optional[str] = Field( + default=None, + description="events.event_id 主键；为对齐实际表而扩展，uuid 仍可用", + ) + event_type: Optional[str] = Field( + default=None, + description="events.event_type 列；为对齐实际表而扩展，type 仍可用", + ) + source: Optional[str] = Field( + default=None, + description="事件来源（scheduler / proactive / manual / ...）", + ) + tags: Optional[List[str]] = Field( + default=None, + description="事件标签列表（events.metadata JSON 内的 tags 字段）", + ) + + model_config = ConfigDict(populate_by_name=True) + + +class EventListResponse(BaseModel): + """ + 事件列表响应。 + Attributes: + events: 事件列表 + total: 总条数（用于分页） + """ + events: List[EventDTO] + total: int + + +__all__ = ["EventDTO", "EventListResponse"] diff --git a/src/web/backend/schemas/knowledge.py b/src/web/backend/schemas/knowledge.py new file mode 100644 index 0000000..d2ab637 --- /dev/null +++ b/src/web/backend/schemas/knowledge.py @@ -0,0 +1,116 @@ +""" +Knowledge schemas - Pydantic v2 models for knowledge base endpoints. +知识库相关的 Pydantic v2 模型。 + +Stage A.3: 为 /api/knowledge/* 提供实体/定义的契约。 +Stage C.1: 新增 KnowledgeCreate / KnowledgeUpdate，提供前端 CRUD 强约束。 +""" + +from __future__ import annotations + +from typing import Any, Dict, List, Optional + +from pydantic import BaseModel, Field + + +class KnowledgeEntity(BaseModel): + """ + 知识库实体。 + Attributes: + uuid: 实体唯一标识 + name: 原始名称 + normalized_name: 归一化名称 + created_at: ISO 8601 时间戳 + """ + uuid: str + name: str + normalized_name: str + created_at: str + + +class KnowledgeDefinition(BaseModel): + """ + 知识库实体定义（实体下属的事实/属性/关系）。 + Attributes: + id: 自增主键 + entity_uuid: 所属实体 uuid + content: 定义内容 + type: 定义类型（fact / attribute / relation ...） + confidence: 置信度（0.0-1.0） + priority: 优先级 + """ + id: int + entity_uuid: str + content: str + type: str + confidence: float + priority: int + + +class KnowledgeSearchRequest(BaseModel): + """ + 知识库搜索请求。 + Attributes: + q: 搜索关键词（必填，至少 1 字符） + limit: 返回上限（默认 20） + """ + q: str = Field(..., min_length=1) + limit: int = 20 + + +class KnowledgeSearchResponse(BaseModel): + """ + 知识库搜索响应。 + Attributes: + results: 实体列表 + total: 总匹配条数 + """ + results: List[KnowledgeEntity] + total: int + + +# ===================================================================== +# Stage C.1: Web 端 CRUD 强约束 +# ===================================================================== + +class KnowledgeCreate(BaseModel): + """ + 创建知识实体的请求体（Stage C.1）。 + + Attributes: + entity_name: 实体名称（必填，1-100 字符） + category: 分类（可选，<= 64 字符；将作为 entity_definitions.type） + description: 描述（可选，<= 4000 字符；将作为 entity_definitions.content） + related_info: 相关信息字典（可选，key->value 写入 entity_related_info） + """ + entity_name: str = Field(..., min_length=1, max_length=100) + category: Optional[str] = Field(default=None, max_length=64) + description: Optional[str] = Field(default=None, max_length=4000) + related_info: Optional[Dict[str, Any]] = None + + +class KnowledgeUpdate(BaseModel): + """ + 更新知识实体的请求体（Stage C.1）。 + + 所有字段均可选；至少要传一个字段。 + Attributes: + entity_name: 新名称（1-100 字符） + category: 新分类（<= 64 字符） + description: 新描述（<= 4000 字符） + related_info: 替换该实体的全部相关信息 + """ + entity_name: Optional[str] = Field(default=None, min_length=1, max_length=100) + category: Optional[str] = Field(default=None, max_length=64) + description: Optional[str] = Field(default=None, max_length=4000) + related_info: Optional[Dict[str, Any]] = None + + +__all__ = [ + "KnowledgeEntity", + "KnowledgeDefinition", + "KnowledgeSearchRequest", + "KnowledgeSearchResponse", + "KnowledgeCreate", + "KnowledgeUpdate", +] diff --git a/src/web/backend/schemas/memory.py b/src/web/backend/schemas/memory.py new file mode 100644 index 0000000..0335554 --- /dev/null +++ b/src/web/backend/schemas/memory.py @@ -0,0 +1,109 @@ +""" +Memory schemas - Pydantic v2 models for /api/memory/* endpoints. +记忆 / 话题时间线相关的 Pydantic v2 模型。 + +Stage B.4: 为 /api/memory/timeline 与 /api/memory/loop/{uuid}/context 提供契约。 +- 数据源：``open_loops`` 表（由 OpenLoopTracker 写入）。 +- 失败降级：节点为空 / 链接为空 / 上下文为空 list，不抛 5xx。 +""" + +from __future__ import annotations + +from typing import List, Optional + +from pydantic import BaseModel, Field + + +class TimelineNode(BaseModel): + """ + 话题时间线节点（open_loops 一行）。 + + Attributes: + uuid: 话题唯一标识（open_loops.uuid） + topic: 话题标题（open_loops.topic） + date: 节点日期（YYYY-MM-DD，从 raised_at 截取前 10 位） + status: 话题状态（'open' / 'closed'/'resolved' 等） + category: 话题分类（暂为 '其他'，留给前端做兼容） + mention_count: 提及次数（open_loops.mention_count） + keywords: 关联关键词列表（open_loops.related_keywords_json） + """ + uuid: str + topic: str + date: str + status: str = "open" + category: str = "其他" + mention_count: int = 0 + keywords: List[str] = Field(default_factory=list) + + +class TimelineLink(BaseModel): + """ + 话题时间线链接（节点之间的关联关系）。 + + Attributes: + source: 源节点 uuid + target: 目标节点 uuid + relation: 关系类型（'keyword' / 'time' / 'manual' 等，默认 'keyword'） + """ + source: str + target: str + relation: str = "keyword" + + +class TimelineResponse(BaseModel): + """ + 时间线响应（ECharts 关系图 / 散点图所需最小集）。 + + Attributes: + nodes: 节点列表 + links: 链接列表 + days: 查询时间范围（天），默认 7 + total: 节点数量（便于前端展示） + """ + nodes: List[TimelineNode] = Field(default_factory=list) + links: List[TimelineLink] = Field(default_factory=list) + days: int = 7 + total: int = 0 + + +class LoopContextItem(BaseModel): + """ + 话题上下文条目（单条对话消息 / 锚点）。 + + Attributes: + role: 角色（'user' / 'assistant' / 'system'） + content: 文本内容 + timestamp: ISO 8601 时间戳（可空） + source: 数据来源（'short_term_memory' / 'open_loop_context' / 'fallback'） + """ + role: str = "system" + content: str = "" + timestamp: Optional[str] = None + source: str = "short_term_memory" + + +class LoopContextResponse(BaseModel): + """ + 话题上下文响应。 + + Attributes: + uuid: 话题 uuid（路径参数透传） + topic: 话题标题（找不到时为空字符串） + status: 话题状态 + items: 上下文条目列表（失败 / 未找到时为空 list） + total: 条目数量 + """ + uuid: str + topic: str = "" + status: str = "unknown" + items: List[LoopContextItem] = Field(default_factory=list) + total: int = 0 + + +__all__ = [ + "TimelineNode", + "TimelineLink", + "TimelineResponse", + "LoopContextItem", + "LoopContextResponse", +] diff --git a/src/web/backend/schemas/schedule.py b/src/web/backend/schemas/schedule.py new file mode 100644 index 0000000..6277917 --- /dev/null +++ b/src/web/backend/schemas/schedule.py @@ -0,0 +1,73 @@ +""" +Schedule schemas - Pydantic v2 models for schedule endpoints. +日程相关的 Pydantic v2 模型。 + +Stage A.3: 为 /api/schedules/* 提供日程项/创建/更新的契约。 +""" + +from __future__ import annotations + +from typing import Optional + +from pydantic import BaseModel, Field + + +class ScheduleDTO(BaseModel): + """ + 日程数据传输对象（只读视图）。 + Attributes: + id: 日程主键 + title: 标题 + description: 详细描述 + start_time: ISO 8601 开始时间 + end_time: ISO 8601 结束时间（可选） + priority: 优先级（low / normal / high） + status: 状态（pending / done / cancelled） + schedule_type: 类型（personal / work / birthday ...） + """ + id: int + title: str + description: Optional[str] = None + start_time: str + end_time: Optional[str] = None + priority: str = "normal" + status: str = "pending" + schedule_type: str = "personal" + + +class ScheduleCreate(BaseModel): + """ + 创建日程请求。 + Attributes: + title: 标题（必填，1-200 字符） + description: 详细描述 + start_time: ISO 8601 开始时间 + end_time: ISO 8601 结束时间 + priority: 优先级 + schedule_type: 类型 + """ + title: str = Field(..., min_length=1, max_length=200) + description: Optional[str] = None + start_time: str + end_time: Optional[str] = None + priority: str = "normal" + schedule_type: str = "personal" + + +class ScheduleUpdate(BaseModel): + """ + 更新日程请求（部分字段更新）。 + Attributes: + id: 必填，待更新日程主键 + title / description / start_time / end_time / priority / status: 可选更新 + """ + id: int + title: Optional[str] = None + description: Optional[str] = None + start_time: Optional[str] = None + end_time: Optional[str] = None + priority: Optional[str] = None + status: Optional[str] = None + + +__all__ = ["ScheduleDTO", "ScheduleCreate", "ScheduleUpdate"] diff --git a/src/web/backend/services/__init__.py b/src/web/backend/services/__init__.py new file mode 100644 index 0000000..396ece9 --- /dev/null +++ b/src/web/backend/services/__init__.py @@ -0,0 +1,30 @@ +""" +Services package - 统一导出所有 service 及其全局单例。 +Services package - re-export all services and their singletons. + +Stage A.3: 供 /api/* 路由 / main.py 统一导入： + from src.web.backend.services import chat_service, database_service, event_service + +Stage B.3: 新增 emotion_service（情感雷达后端）。 +""" + +from __future__ import annotations + +from .chat_service import ChatService, chat_service +from .creative_service import CreativeService, creative_service +from .database_service import DatabaseService, database_service +from .emotion_service import EmotionService, emotion_service +from .event_service import EventService, event_service + +__all__ = [ + "ChatService", + "chat_service", + "CreativeService", + "creative_service", + "DatabaseService", + "database_service", + "EmotionService", + "emotion_service", + "EventService", + "event_service", +] diff --git a/src/web/backend/services/chat_service.py b/src/web/backend/services/chat_service.py new file mode 100644 index 0000000..f34cee1 --- /dev/null +++ b/src/web/backend/services/chat_service.py @@ -0,0 +1,543 @@ +""" +Chat service module. +聊天服务模块 - 包装 ChatAgent，提供 Web 后端的同步/异步入口。 + +Stage A.3: 同步接口 +Stage B.2: 异步流式接口 (chat_stream 返回字符串，stream_chat 返回 ChatChunk) +Stage B.3 / B.4: 在流式回复完成时，向 EventService 派发 emotion_update / timeline_update +事件，供前端 /ws/events 订阅并刷新 EmotionPanel / TimelineCanvas。 +v3.1.0: 注入 ChatSessionRepository 持久化钩子（user/assistant 消息入库 + emotion_json 异步更新）。 +""" + +from __future__ import annotations + +import asyncio +import re +import sys +import warnings +from pathlib import Path +from typing import Any, AsyncIterator, Dict, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + + +# ---------------------------------------------------------------------- +# 工具：检查方法签名是否接受某个 keyword 参数 +# ---------------------------------------------------------------------- +def _method_accepts_kwarg(method: Any, kwarg_name: str) -> bool: + """ + 检查 callable 是否在其签名中接受 ``kwarg_name``。 + + 用途：ChatService.chat_stream 在调用 agent.chat_stream 时做向后兼容检测 + —— 当 agent 仍是老式签名 (user_input, context=None) 时，避免 + TypeError: got an unexpected keyword argument 'cancel_event'。 + + Returns: + True 表示方法接受该 kwarg（或 **kwargs）；False 表示不接收。 + """ + try: + import inspect # 延迟导入，inspect 是标准库但避免无谓加载 + sig = inspect.signature(method) + except (TypeError, ValueError): + # 内置 / C 实现的方法可能无法 inspect；保守地当作不接受 + return False + + if any(p.kind == inspect.Parameter.VAR_KEYWORD for p in sig.parameters.values()): + return True # 接受 **kwargs ⇒ 任何 kwarg 都 OK + return kwarg_name in sig.parameters + + +class ChatService: + """ + 聊天服务（Web 后端侧） + + 设计要点： + 1. 懒加载 ChatAgent：避免 Web 启动时强制初始化 LLM / DB + 2. try/except 保护：LLM 库缺失或配置错误时不阻断 Web 启动 + 3. chat() 委托给 agent.chat()；chat_stream() 透传字符串 chunk； + stream_chat() 在 Stage B.2 新增，包装 chat_stream 为 Pydantic ChatChunk + 4. 全局单例：chat_service = ChatService() + """ + + def __init__(self, agent: Optional[Any] = None) -> None: + """ + 初始化服务。 + + Args: + agent: 可选，预注入的 ChatAgent 实例。为 None 时延后到 + get_agent() 调用时再尝试实例化。 + """ + self._agent: Optional[Any] = None + + # v3.1.0: 当前绑定会话（由 set_session_id() 注入） + self._session_id: Optional[int] = None + # v3.1.0: 刚写入的 assistant 消息 id（供 on_emotion 异步更新 emotion_json） + self._last_assistant_message_id: Optional[int] = None + + if agent is not None: + self._agent = agent + else: + # 尝试立即懒加载；若失败则保持 None + self._try_init_agent() + + # ------------------------------------------------------------------ + # v3.1.0: 会话绑定 + # ------------------------------------------------------------------ + def set_session_id(self, session_id: Optional[int]) -> None: + """ + 显式绑定一个 session_id；之后 chat_stream 内会自动把 user/assistant + 消息写到该 session 下。 + + 传 None 时解绑。 + """ + if session_id is None: + self._session_id = None + return + try: + self._session_id = int(session_id) + except (TypeError, ValueError): + self._session_id = None + + def get_session_id(self) -> Optional[int]: + """返回当前绑定的 session_id（可能为 None）。""" + return self._session_id + + def _get_repo(self) -> Optional[Any]: + """ + 懒加载 ChatSessionRepository；DB 不可用时返回 None（不抛）。 + """ + try: + from src.core.database_manager import DatabaseManager + # 复用 agent 已建好的 db_manager，避免重复打开 + db_manager = None + if self._agent is not None and hasattr(self._agent, "db"): + db_manager = getattr(self._agent, "db", None) + if db_manager is None: + # 注意：保留模块全局单例避免每个 chat 一次新连接 + if not hasattr(self, "_db_singleton") or self._db_singleton is None: + self._db_singleton = DatabaseManager() + db_manager = self._db_singleton + return db_manager.get_chat_session_repository() + except Exception: # noqa: BLE001 + return None + + def on_emotion(self, emotion_payload: Optional[Dict[str, Any]]) -> None: + """ + 由 ws/chat.py 在 emotion_update 帧派发时调用； + 把情感数据写入最近一条 assistant 消息的 emotion_json 字段。 + """ + if not emotion_payload or self._last_assistant_message_id is None: + return + try: + import json as _json + repo = self._get_repo() + if repo is None: + return + repo.update_message_emotion( + self._last_assistant_message_id, + _json.dumps(emotion_payload, ensure_ascii=False), + ) + except Exception: # noqa: BLE001 + pass + + # ------------------------------------------------------------------ + # 内部：懒加载 ChatAgent + # ------------------------------------------------------------------ + def _try_init_agent(self) -> None: + """ + 尝试实例化 ChatAgent，失败时记录 warning 并保持 _agent = None。 + + 设计意图：ChatAgent 初始化会创建 DatabaseManager / LongTermMemoryManager / + SiliconFlowLLM / ProactiveEngine 等重型组件，依赖未安装或环境变量缺失时 + 可能会抛异常。Web 端必须能在 ChatAgent 不可用时继续运行（健康检查等）。 + """ + try: + from src.core.chat_agent import ChatAgent # type: ignore + + self._agent = ChatAgent() + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[ChatService] ChatAgent 初始化失败，将以无 agent 模式运行: {exc}", + RuntimeWarning, + stacklevel=2, + ) + self._agent = None + + # ------------------------------------------------------------------ + # 同步入口 + # ------------------------------------------------------------------ + def get_agent(self) -> Any: + """ + 同步获取（必要时懒加载）ChatAgent 实例。 + + Returns: + ChatAgent 实例；若 LLM / DB 不可用则返回 None。 + """ + if self._agent is None: + self._try_init_agent() + return self._agent + + def set_agent(self, agent: Optional[Any]) -> None: + """ + 注入 agent（用于测试或自定义场景）。 + 传入 None 时，重置为 lazy 模式，下次 get_agent() 时再尝试初始化。 + """ + self._agent = agent + + def chat(self, user_input: str, context: Optional[Dict] = None) -> str: + """ + 同步聊天入口。 + + Args: + user_input: 用户输入 + context: 可选上下文（不直接传给 agent，保留供未来扩展） + + Returns: + 完整回复文本 + + Raises: + RuntimeError: 当 agent 不可用时 + """ + agent = self.get_agent() + if agent is None: + raise RuntimeError( + "ChatAgent not initialized. Check LLM / DB dependencies." + ) + return agent.chat(user_input) + + # ------------------------------------------------------------------ + # 流式入口（Stage B.2 实现） + # ------------------------------------------------------------------ + async def chat_stream( + self, + user_input: str, + context: Optional[Dict] = None, + cancel_event: Optional[Any] = None, + ) -> AsyncIterator[str]: + """ + 流式聊天入口（字符串 chunk 透传）。 + + 降级策略（按优先级）： + 1. agent 可用且有 chat_stream → 委托给 agent.chat_stream(user_input, context=context, cancel_event=cancel_event) + 2. agent 可用但没有 chat_stream（或其失败） → + loop.run_in_executor(None, agent.chat, user_input) 拿到完整回复后 + 按"非空白词 + 尾部空白"切片模拟流式输出（40ms 间隔）， + 每个 token 前检查 cancel_event.is_set() 实现早退 + 3. agent 不可用 → yield 一次性回执 "[ChatAgent 不可用] 收到: {user_input}" 后退出 + + Args: + user_input: 用户输入 + context: 透传给 agent 的额外上下文（agent 不感知则忽略） + cancel_event: 取消事件（透传给 agent；同步降级路径中在每片前检查） + + Yields: + 字符串 chunk；最坏情况下 yield 一次回执 + """ + # 取得 agent（可能为 None） + try: + agent = self.get_agent() + except Exception as e: # noqa: BLE001 + warnings.warn( + f"[ChatService] get_agent 失败，按无 agent 处理: {e}", + RuntimeWarning, + stacklevel=2, + ) + agent = None + + # ===== v3.1.0: 前置持久化 — 确保 session + 写 user 消息 ===== + # 1) 若 _session_id 仍空但 context 里有 session_id → 使用之 + if self._session_id is None and isinstance(context, dict): + ctx_sid = context.get("session_id") + if ctx_sid is not None: + try: + self.set_session_id(int(ctx_sid)) + except (TypeError, ValueError): + pass + # 2) 仍空 → 自动创建新 session（title 取 user_input 前 30 字） + if self._session_id is None: + try: + repo = self._get_repo() + if repo is not None: + initial_title = (user_input or "新会话")[:30] or "新会话" + new_sid = repo.create(user_id="default", title=initial_title) + if new_sid: + self._session_id = int(new_sid) + except Exception: # noqa: BLE001 + self._session_id = None + # 3) 写 user 消息 + if self._session_id is not None: + try: + repo = self._get_repo() + if repo is not None: + repo.add_message(self._session_id, "user", user_input or "") + self._last_assistant_message_id = None + except Exception: # noqa: BLE001 + pass + + # 收集 assistant 全文（用于流结束后持久化） + collected_chunks: list[str] = [] + collect_failed = False + + # ===== 策略 1：委托给 agent.chat_stream ===== + if agent is not None and hasattr(agent, "chat_stream"): + # 兼容性：仅当 agent 的 chat_stream 接受 cancel_event 时才透传 + _supports_cancel = _method_accepts_kwarg(agent.chat_stream, "cancel_event") + _supports_user_id = _method_accepts_kwarg(agent.chat_stream, "user_id") + try: + kwargs: Dict[str, Any] = {"context": context} + if _supports_cancel and cancel_event is not None: + kwargs["cancel_event"] = cancel_event + if _supports_user_id: + kwargs["user_id"] = "default" + async for chunk in agent.chat_stream(user_input, **kwargs): + collected_chunks.append(chunk or "") + yield chunk + except asyncio.CancelledError: + collect_failed = True + raise + except Exception as e: # noqa: BLE001 + # agent.chat_stream 异常不向消费者直接抛；尝试降级到同步 chat() + collect_failed = True + warnings.warn( + f"[ChatService] agent.chat_stream 失败，降级到同步 chat(): {e}", + RuntimeWarning, + stacklevel=2, + ) + else: + # v3.1.0: 正常结束 → 持久化 assistant + self._persist_assistant("".join(collected_chunks)) + return + # 走到策略 2 + if collect_failed: + # 失败时也尝试持久化已经收集到的部分 + self._persist_assistant("".join(collected_chunks)) + collected_chunks = [] + collect_failed = False + + # ===== 策略 2：run_in_executor + 同步 chat() + 切分 ===== + if agent is not None and hasattr(agent, "chat"): + try: + loop = asyncio.get_event_loop() + full_response = await loop.run_in_executor( + None, agent.chat, user_input + ) + except asyncio.CancelledError: + # 取消信号直接向上传播 + self._persist_assistant("".join(collected_chunks)) + raise + except Exception as e: # noqa: BLE001 + warnings.warn( + f"[ChatService] 同步 chat() 降级失败: {e}", + RuntimeWarning, + stacklevel=2, + ) + # 走到策略 3 + full_response = None + + if full_response: + try: + tokens = re.findall(r"\S+\s*|\s+", full_response) + if not tokens: + tokens = [full_response[i:i + 2] + for i in range(0, len(full_response), 2)] + if not tokens: + tokens = list(full_response) + for token in tokens: + # 取消检查（前置 + 让出事件循环） + if cancel_event is not None and cancel_event.is_set(): + self._persist_assistant("".join(collected_chunks)) + return + # 让出事件循环，便于取消信号传播 + await asyncio.sleep(0) + collected_chunks.append(token or "") + yield token + # 40ms 间隔（30-50ms 区间内），避免网络洪水 + await asyncio.sleep(0.04) + self._persist_assistant("".join(collected_chunks)) + return + except asyncio.CancelledError: + self._persist_assistant("".join(collected_chunks)) + raise + except Exception as e: # noqa: BLE001 + warnings.warn( + f"[ChatService] 分片失败: {e}", + RuntimeWarning, + stacklevel=2, + ) + # 兜底：一次性把全部内容推出去 + collected_chunks.append(full_response or "") + yield full_response + self._persist_assistant("".join(collected_chunks)) + return + + # ===== 策略 3：最终降级 — 一次性回执 ===== + try: + fallback = f"[ChatAgent 不可用] 收到: {user_input}" + collected_chunks.append(fallback) + yield fallback + except Exception: + # 连 yield 都失败时静默退出，不影响 WS 端点的清理流程 + pass + # 兜底：尝试持久化（即便 agent 不可用也保留草稿） + self._persist_assistant("".join(collected_chunks)) + + def _persist_assistant(self, content: str) -> None: + """ + v3.1.0: 把 assistant 完整回复写入 chat_messages；记录 message_id + 供后续 on_emotion() 异步更新 emotion_json。 + """ + if not content: + return + if self._session_id is None: + return + try: + repo = self._get_repo() + if repo is None: + return + mid = repo.add_message(self._session_id, "assistant", content) + if mid: + self._last_assistant_message_id = int(mid) + except Exception: # noqa: BLE001 + pass + + async def stream_chat( + self, + user_input: str, + user_id: str = "default", + cancel_event: Optional[Any] = None, + ) -> AsyncIterator[Any]: + """ + 流式聊天入口（Pydantic ChatChunk）。 + + 包装 ChatAgent.chat_stream，产出 Pydantic ChatChunk。 + - 正常路径：每个字符串 chunk → ChatChunk(type='chunk', chunk=text, done=False) + - 错误路径：ChatChunk(type='error', chunk=str(e), done=True) + - 末尾： ChatChunk(type='done', chunk='', done=True) + + Args: + user_input: 用户输入 + user_id: 用户标识（透传给 agent） + cancel_event: 取消事件（透传给 agent） + + Yields: + ChatChunk 实例 + """ + from src.web.backend.schemas.chat import ChatChunk # 延迟导入避免循环 + + agent = self.get_agent() + if agent is None: + # agent 不可用也要走错误 chunk 协议（不抛），便于 WS 单边关闭 + yield ChatChunk(type="error", chunk="ChatAgent not initialized", done=True) + return + + try: + async for text in agent.chat_stream( + user_input, + user_id=user_id, + cancel_event=cancel_event, + ): + # 跳过空字符串 chunk（前端没有内容可显示） + if text is None or text == "": + continue + yield ChatChunk(type="chunk", chunk=text, done=False) + except asyncio.CancelledError: + # 消费者主动断开时，安静地退出（不发送 error） + return + except Exception as e: # noqa: BLE001 + # 异常路径：优雅 yield error chunk 后结束 + try: + import traceback + debug_msg = f"{type(e).__name__}: {e}" + except Exception: + debug_msg = str(e) + yield ChatChunk(type="error", chunk=debug_msg, done=True) + return + + # 末尾：done chunk + # Stage B.3 / B.4: 流式成功结束后，通知前端情感雷达 / 话题时间线刷新 + try: + self._emit_post_chat_updates(user_id=user_id) + except Exception: # noqa: BLE001 + pass + yield ChatChunk(type="done", chunk="", done=True) + + # ------------------------------------------------------------------ + # 统计 + # ------------------------------------------------------------------ + def _emit_post_chat_updates(self, user_id: str = "default") -> None: + """ + Stage B.3 / B.4: 在聊天流程结束后，把"最新情感 + 最新话题"作为 + emotion_update / timeline_update 事件派发到 EventService。 + + 设计要点： + 1. 内部每个调用都用 try/except 包裹，绝不向上抛异常 + 2. 任何子模块（analyzer / memory / event_service）不可用时静默跳过 + 3. 不修改现有 chat() / stream_chat() 公开签名 + """ + try: + from src.web.backend.services.event_service import get_event_service + es = get_event_service() + except Exception: # noqa: BLE001 + return + if es is None: + return + + # 1) 情感雷达 + try: + from src.core.emotion_analyzer import EmotionRelationshipAnalyzer + analyzer = EmotionRelationshipAnalyzer() + payload = analyzer.get_latest_for_user(user_id) or {} + except Exception: # noqa: BLE001 + payload = {} + try: + es.emit_event('emotion_update', { + 'user_id': user_id, + 'data': payload.get('plutchik') or {}, + 'additive_scores': payload.get('additive_scores') or {}, + 'dominant': payload.get('dominant'), + 'timestamp': payload.get('timestamp'), + }) + except Exception: # noqa: BLE001 + pass + + # 2) 话题时间线 + try: + from src.core.long_term_memory import LongTermMemoryManager + mgr = LongTermMemoryManager() + topics = mgr.list_topics(user_id=user_id, days=7) or [] + except Exception: # noqa: BLE001 + topics = [] + try: + from datetime import datetime as _dt + ts_now = _dt.now().isoformat() + except Exception: # noqa: BLE001 + ts_now = '' + try: + es.emit_event('timeline_update', { + 'user_id': user_id, + 'days': 7, + 'topics': topics, + 'replace': True, + 'timestamp': ts_now, + }) + except Exception: # noqa: BLE001 + pass + + def get_stats(self) -> Dict[str, Any]: + """ + 返回服务状态与版本号。 + """ + return { + "agent_loaded": self._agent is not None, + "version": "3.0.0", + } + + +# 全局单例 +chat_service = ChatService() + + +__all__ = ["ChatService", "chat_service"] diff --git a/src/web/backend/services/creative_service.py b/src/web/backend/services/creative_service.py new file mode 100644 index 0000000..63e1f94 --- /dev/null +++ b/src/web/backend/services/creative_service.py @@ -0,0 +1,309 @@ +""" +Creative service module. +长期创作项目服务模块 - 包装 CreativeProjectManager + DatabaseManager， +为 Web 后端提供 list / detail / advance 等安全入口。 + +Stage C.6 设计要点： +1. 复用现有方法签名（``list_creative_projects`` / ``get_creative_project`` / + ``_advance_one``），不修改 ``CreativeProjectManager`` 既有方法。 +2. 失败路径返回 ``None`` / ``[]`` / ``(False, reason)``，由 API 层映射到 4xx。 +3. 懒加载 ``CreativeProjectManager``：避免 Web 启动时强制 LLM 初始化。 +4. 全局单例 ``creative_service``。 +""" + +from __future__ import annotations + +import sys +import warnings +from datetime import datetime +from pathlib import Path +from typing import Any, Dict, List, Optional, Tuple + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + + +class CreativeService: + """ + 创作项目服务（Web 后端侧） + + Attributes: + _manager: 懒加载的 ``CreativeProjectManager`` 实例 + """ + + def __init__(self) -> None: + self._manager: Optional[Any] = None + self._db: Optional[Any] = None + + # ------------------------------------------------------------------ + # 内部：懒加载 manager / db + # ------------------------------------------------------------------ + def _try_init_manager(self) -> Optional[Any]: + if self._manager is not None: + return self._manager + try: + from src.core.creative_writer import CreativeProjectManager # type: ignore + self._manager = CreativeProjectManager() + return self._manager + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[CreativeService] CreativeProjectManager 初始化失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + self._manager = None + return None + + def _get_db(self) -> Optional[Any]: + """ + 获取 DatabaseManager（来自 manager 或独立懒加载）。 + 优先使用 manager 自带的 db；否则懒加载 DatabaseManager。 + """ + mgr = self._try_init_manager() + if mgr is not None and getattr(mgr, 'db', None) is not None: + return mgr.db + if self._db is not None: + return self._db + try: + from src.core.database_manager import DatabaseManager # type: ignore + self._db = DatabaseManager() + return self._db + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[CreativeService] DatabaseManager 懒加载失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return None + + # ------------------------------------------------------------------ + # 列表 + # ------------------------------------------------------------------ + def list_projects( + self, + user_id: str = "default", + status: Optional[str] = None, + limit: int = 50, + ) -> List[Dict[str, Any]]: + """ + 列出创作项目。 + + - ``user_id``: 当前保留参数（数据库无 user_id 字段，所有项目视为同一用户）。 + - ``status``: 业务状态过滤（drafting / finished / paused 等）。 + - ``limit``: 返回上限。 + + Returns: + 标准化为 ``CreativeProjectSummary`` 兼容字段的项目列表。 + """ + db = self._get_db() + if db is None: + return [] + try: + raw = db.list_creative_projects(status=status, limit=max(1, int(limit))) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[CreativeService] list_projects 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + out: List[Dict[str, Any]] = [] + for item in raw or []: + try: + out.append(self._normalize_summary(item)) + except Exception: + continue + return out + + # ------------------------------------------------------------------ + # 详情 + # ------------------------------------------------------------------ + def get_project(self, project_uuid: str) -> Optional[Dict[str, Any]]: + """ + 获取项目详情（含 Story Bible / 章节 / 记忆池摘要）。 + 不存在 → None。 + """ + if not project_uuid or not isinstance(project_uuid, str): + return None + db = self._get_db() + if db is None: + return None + try: + project = db.get_creative_project(project_uuid) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[CreativeService] get_project({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return None + if not project: + return None + + detail = self._normalize_detail(project) + + # 拉取 Story Bible + try: + bible = db.get_story_bible(project_uuid) or {} + except Exception: + bible = {} + detail['story_bible'] = bible or {} + + # 章节：优先 draft_chunks；fallback 到 outline 的伪章节 + try: + chunks = project.get('draft_chunks') or [] + except Exception: + chunks = [] + chapters: List[Dict[str, Any]] = [] + for idx, ch in enumerate(chunks or []): + try: + chapters.append({ + 'uuid': f"{project_uuid}-ch-{idx + 1:02d}", + 'chapter_title': f"Draft Chapter {idx + 1}", + 'word_count': int(ch.get('char_count', 0) or 0), + 'content': ch.get('content', '') or '', + 'created_at': ch.get('created_at', '') or '', + }) + except Exception: + continue + if not chapters: + outlines = project.get('outline') or [] + for idx, line in enumerate(outlines or []): + chapters.append({ + 'uuid': f"{project_uuid}-outline-{idx + 1:02d}", + 'chapter_title': f"Outline {idx + 1}", + 'word_count': 0, + 'content': str(line) if line else '', + 'created_at': '', + }) + detail['chapters'] = chapters + detail['chapter_count'] = len(chapters) + + # 记忆池摘要 + try: + pool = db.get_creative_memory_pool(project_uuid, limit=50) or [] + except Exception: + pool = [] + detail['memory_pool_summary'] = { + 'count': len(pool), + 'latest': [ + { + 'memory_type': m.get('memory_type', ''), + 'content': (m.get('content', '') or '')[:200], + 'importance': m.get('importance', 0.0), + 'created_at': m.get('created_at', ''), + } + for m in (pool[:5] if isinstance(pool, list) else []) + ], + } + + return detail + + # ------------------------------------------------------------------ + # 续写 + # ------------------------------------------------------------------ + def advance(self, project_uuid: str) -> Tuple[bool, str, Optional[Dict[str, Any]]]: + """ + 触发续写。 + + Returns: + (ok, reason, new_chapter_dict or None) + - ok=True 时 new_chapter 必有值 + - ok=False 时 reason 描述原因（disabled / not_found / internal_error / not_drafting / quality_fail） + """ + if not project_uuid or not isinstance(project_uuid, str): + return False, "invalid_uuid", None + + mgr = self._try_init_manager() + if mgr is None: + return False, "manager_unavailable", None + if not getattr(mgr, 'enabled', False): + return False, "disabled", None + + db = self._get_db() + if db is None: + return False, "db_unavailable", None + + try: + project = db.get_creative_project(project_uuid) + except Exception: + project = None + if not project: + return False, "not_found", None + if (project.get('status') or '').lower() == 'finished': + return False, "already_finished", None + + try: + # 复用 _advance_one(project, now) 既有方法 + now = datetime.now() + ok = bool(mgr._advance_one(project, now)) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[CreativeService] advance({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False, f"internal_error: {exc}", None + + if not ok: + return False, "quality_fail_or_skip", None + + # 重新拉取最新章节 + try: + updated = db.get_creative_project(project_uuid) or {} + except Exception: + updated = {} + chunks = updated.get('draft_chunks') or [] + new_chapter: Optional[Dict[str, Any]] = None + if chunks: + try: + last = chunks[-1] + new_chapter = { + 'uuid': f"{project_uuid}-ch-{len(chunks):02d}", + 'chapter_title': f"Draft Chapter {len(chunks)}", + 'word_count': int(last.get('char_count', 0) or 0), + 'content': last.get('content', '') or '', + 'created_at': last.get('created_at', '') or '', + } + except Exception: + new_chapter = None + + return True, "ok", new_chapter + + # ------------------------------------------------------------------ + # 标准化 + # ------------------------------------------------------------------ + @staticmethod + def _normalize_summary(item: Dict[str, Any]) -> Dict[str, Any]: + try: + chunks = item.get('draft_chunks') or [] + chapter_count = len(chunks) + except Exception: + chapter_count = 0 + return { + 'uuid': str(item.get('uuid', '')), + 'title': str(item.get('title', '') or ''), + 'status': str(item.get('status', 'drafting') or 'drafting'), + 'chapter_count': int(chapter_count), + 'created_at': str(item.get('created_at', '') or ''), + 'updated_at': str(item.get('updated_at', '') or ''), + } + + @staticmethod + def _normalize_detail(item: Dict[str, Any]) -> Dict[str, Any]: + base = CreativeService._normalize_summary(item) + base.update({ + 'story_bible': {}, + 'chapters': [], + 'memory_pool_summary': {}, + }) + return base + + +# 全局单例 +creative_service = CreativeService() + + +__all__ = ["CreativeService", "creative_service"] diff --git a/src/web/backend/services/database_service.py b/src/web/backend/services/database_service.py new file mode 100644 index 0000000..5366331 --- /dev/null +++ b/src/web/backend/services/database_service.py @@ -0,0 +1,904 @@ +""" +Database service module. +数据库服务模块 - 包装 DatabaseManager，为 Web 后端提供安全的表查询/删除/统计入口。 + +Stage A.3: 提供业务表的只读查询 + 基于 uuid 的受控删除。 + +设计要点： +1. 白名单：只允许访问预定义的业务表（防御任意 SQL 拼接） +2. 懒加载 DatabaseManager：避免 Web 启动时强制 init DB +3. try/except 保护：单表/单行失败不影响其他调用 +4. 全局单例：database_service = DatabaseService() +""" + +from __future__ import annotations + +import json +import os +import sys +import uuid as _uuid +import warnings +from datetime import datetime +from pathlib import Path +from typing import Any, Dict, List, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + + +# 业务表白名单：仅允许查询/删除这些表 +ALLOWED_TABLES: frozenset = frozenset({ + 'entities', + 'entity_definitions', + 'entity_related_info', + 'short_term_memory', + 'long_term_memory', + 'emotion_history', + 'metadata', + 'environment_descriptions', + 'environment_objects', + 'agent_expressions', + 'user_expression_habits', + 'life_state_daily', + 'dream_records', + 'diary_entries', + 'open_loops', + 'creative_projects', + 'creative_story_bibles', + 'creative_memory_pool', + 'schedules', + 'schedule_entries', + 'schedule_confirmations', + 'vision_tool_logs', + 'events', +}) + + +class DatabaseService: + """ + 数据库服务（Web 后端侧） + + Attributes: + db_path: SQLite 数据库文件路径（可通过 env CHAT_AGENT_DB 覆盖） + _db: 懒加载的 DatabaseManager 实例 + """ + + DEFAULT_DB_ENV = 'CHAT_AGENT_DB' + DEFAULT_DB_NAME = 'chat_agent.db' + + def __init__(self, db_path: Optional[str] = None) -> None: + """ + 初始化 DatabaseService。 + + Args: + db_path: 可选；为 None 时取 os.getenv('CHAT_AGENT_DB', 'chat_agent.db') + """ + if db_path is None: + db_path = os.getenv(self.DEFAULT_DB_ENV, self.DEFAULT_DB_NAME) + self.db_path: str = db_path + self._db: Optional[Any] = None + + # ------------------------------------------------------------------ + # 内部：懒加载 DatabaseManager + # ------------------------------------------------------------------ + def _try_init_db(self) -> Optional[Any]: + """ + 尝试实例化 DatabaseManager，失败时记录 warning 并返回 None。 + """ + if self._db is not None: + return self._db + try: + from src.core.database_manager import DatabaseManager # type: ignore + self._db = DatabaseManager(db_path=self.db_path) + return self._db + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] DatabaseManager 初始化失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + self._db = None + return None + + # ------------------------------------------------------------------ + # DatabaseManager 句柄 + # ------------------------------------------------------------------ + def get_db(self) -> Any: + """ + 懒加载 DatabaseManager；首次获取时会自动 init database。 + 首次获取时尝试执行 PRAGMA journal_mode=WAL（防御性）。 + + Returns: + DatabaseManager 实例；不可用时返回 None。 + """ + db = self._try_init_db() + if db is None: + return None + # 防御性：确保 WAL 模式被启用 + try: + with db.get_connection() as conn: + try: + conn.execute("PRAGMA journal_mode=WAL") + except Exception: + # 不影响主流程 + pass + except Exception: + # 拿不到连接也不抛出 + pass + return db + + def _validate_table(self, table: str) -> str: + """ + 验证表名是否在白名单中。 + 不在白名单 → raise ValueError。 + """ + if not table or not isinstance(table, str): + raise ValueError(f"Table {table!r} not allowed") + if table not in ALLOWED_TABLES: + raise ValueError(f"Table {table} not allowed") + return table + + # ------------------------------------------------------------------ + # 表查询 + # ------------------------------------------------------------------ + def get_tables(self) -> List[str]: + """ + 返回白名单中实际存在的业务表（sqlite_master 查询）。 + """ + db = self.get_db() + if db is None: + return [] + + whitelist = sorted(ALLOWED_TABLES) + existing: List[str] = [] + try: + with db.get_connection() as conn: + cur = conn.cursor() + # 限定为白名单表名，避免无关系统表 + placeholders = ','.join('?' for _ in whitelist) + try: + cur.execute( + f"SELECT name FROM sqlite_master " + f"WHERE type='table' AND name IN ({placeholders})", + whitelist, + ) + except Exception: + # 极端情况：sqlite_master 不可用 + return whitelist + rows = cur.fetchall() or [] + for row in rows: + try: + name = row['name'] if isinstance(row, dict) else row[0] + except (KeyError, IndexError, TypeError): + continue + if name: + existing.append(str(name)) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_tables 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return whitelist # 兜底：返回白名单 + + return sorted(existing) + + def query_table( + self, + table: str, + limit: int = 100, + offset: int = 0, + ) -> List[Dict[str, Any]]: + """ + 查询表数据。 + 表名必须在白名单中；limit/offset 做基本边界检查。 + """ + safe_table = self._validate_table(table) + # 边界保护 + try: + limit = max(1, min(int(limit), 1000)) + except (TypeError, ValueError): + limit = 100 + try: + offset = max(0, int(offset)) + except (TypeError, ValueError): + offset = 0 + + db = self.get_db() + if db is None: + return [] + + rows: List[Dict[str, Any]] = [] + try: + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {safe_table} LIMIT ? OFFSET ?", + (limit, offset), + ) + fetched = cur.fetchall() or [] + for row in fetched: + try: + if isinstance(row, dict): + rows.append(dict(row)) + else: + rows.append({k: row[k] for k in row.keys()}) + except Exception: + # 单行转换失败：尝试 raw 转换 + try: + rows.append(dict(row)) + except Exception: + continue + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] query_table({safe_table}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + return rows + + def delete_record(self, table: str, uuid: str) -> bool: + """ + 按 uuid 删除单条记录。 + 表名必须在白名单中；uuid 必须是非空字符串。 + 成功删除一条返回 True；否则 False。 + """ + safe_table = self._validate_table(table) + if not uuid or not isinstance(uuid, str): + return False + + db = self.get_db() + if db is None: + return False + + try: + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"DELETE FROM {safe_table} WHERE uuid = ?", + (uuid,), + ) + affected = cur.rowcount + return bool(affected and affected > 0) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] delete_record({safe_table}, uuid) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + # ------------------------------------------------------------------ + # Schedule CRUD（Stage C.2） + # ------------------------------------------------------------------ + # 实际写入的表（schedule_* 白名单中的第一张能匹配上的；当前后端未在 + # init_database 中创建 schedule_* 表，方法会先检查存在性，缺失时 + # 返回友好错误或空列表，绝不抛 5xx。 + _SCHEDULE_PRIMARY_TABLE = 'schedules' + _SCHEDULE_ALLOWED_FIELDS = frozenset({ + 'title', 'description', 'start_time', 'end_time', + 'priority', 'status', 'schedule_type', + 'is_collaborative', 'collaborator_name', 'collaborator_status', + 'confirmed', 'created_at', 'updated_at', + }) + + def _ensure_schedule_table(self) -> Optional[str]: + """ + 返回当前可用的 schedule 主表名；若 schedule_* 都不存在则返回 None。 + """ + try: + existing = set(self.get_tables() or []) + except Exception: + existing = set() + for cand in (self._SCHEDULE_PRIMARY_TABLE, 'schedule_entries'): + if cand in existing: + return cand + return None + + def get_schedule_by_date(self, date: str) -> List[Dict[str, Any]]: + """ + 按日期（YYYY-MM-DD）返回日程列表。 + 缺失表 → 返回 []；绝不抛 5xx。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return [] + db = self.get_db() + if db is None: + return [] + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {table} " + f"WHERE substr(COALESCE(start_time,''),1,10) = ? " + f"ORDER BY start_time ASC", + (date,), + ) + rows = cur.fetchall() or [] + out: List[Dict[str, Any]] = [] + for r in rows: + try: + out.append(dict(r) if not isinstance(r, dict) else dict(r)) + except Exception: + continue + return out + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_schedule_by_date({date}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + + def get_schedule_by_range(self, from_date: str, to_date: str) -> List[Dict[str, Any]]: + """ + 按日期范围 [from_date, to_date] 返回日程列表。 + 缺失表 → 返回 []；绝不抛 5xx。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return [] + db = self.get_db() + if db is None: + return [] + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {table} " + f"WHERE substr(COALESCE(start_time,''),1,10) >= ? " + f"AND substr(COALESCE(start_time,''),1,10) <= ? " + f"ORDER BY start_time ASC", + (from_date, to_date), + ) + rows = cur.fetchall() or [] + out: List[Dict[str, Any]] = [] + for r in rows: + try: + out.append(dict(r) if not isinstance(r, dict) else dict(r)) + except Exception: + continue + return out + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_schedule_by_range({from_date},{to_date}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + + def add_schedule_item(self, item: Dict[str, Any]) -> int: + """ + 插入一条日程记录。 + 返回新记录 id（int）；失败时返回 0。 + 缺失 schedule_* 表 → 返回 0（不会抛 5xx）。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return 0 + db = self.get_db() + if db is None: + return 0 + safe = {k: v for k, v in (item or {}).items() + if k in self._SCHEDULE_ALLOWED_FIELDS} + if not safe: + return 0 + cols = list(safe.keys()) + placeholders = ','.join('?' for _ in cols) + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"INSERT INTO {table} ({','.join(cols)}) " + f"VALUES ({placeholders})", + [safe[c] for c in cols], + ) + new_id = cur.lastrowid or 0 + return int(new_id) if new_id else 0 + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] add_schedule_item 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return 0 + + def update_schedule_item(self, item_id: int, updates: Dict[str, Any]) -> bool: + """ + 按 id 更新日程；缺失表 / 不存在 → 返回 False，绝不抛 5xx。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return False + safe = {k: v for k, v in (updates or {}).items() + if k in self._SCHEDULE_ALLOWED_FIELDS and v is not None} + if not safe: + return False + set_clause = ", ".join(f"{k} = ?" for k in safe.keys()) + db = self.get_db() + if db is None: + return False + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"UPDATE {table} SET {set_clause} WHERE id = ?", + [*safe.values(), int(item_id)], + ) + return (cur.rowcount or 0) > 0 + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] update_schedule_item({item_id}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + def delete_schedule_item(self, item_id: int) -> bool: + """ + 按 id 删除日程；缺失表 / 不存在 → 返回 False，绝不抛 5xx。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return False + db = self.get_db() + if db is None: + return False + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"DELETE FROM {table} WHERE id = ?", + (int(item_id),), + ) + return (cur.rowcount or 0) > 0 + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] delete_schedule_item({item_id}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + def confirm_schedule_item(self, item_id: int, confirmed: bool) -> bool: + """ + 协作日程确认/取消确认：写入 confirmed 字段并把 status 同步。 + 缺失表 / 不存在 → 返回 False，绝不抛 5xx。 + """ + try: + table = self._ensure_schedule_table() + if table is None: + return False + db = self.get_db() + if db is None: + return False + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"UPDATE {table} " + f"SET confirmed = ?, status = ?, updated_at = ? " + f"WHERE id = ?", + ( + 1 if confirmed else 0, + 'confirmed' if confirmed else 'pending', + datetime.now().isoformat(), + int(item_id), + ), + ) + return (cur.rowcount or 0) > 0 + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] confirm_schedule_item({item_id}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + # ------------------------------------------------------------------ + # Creative Project CRUD（Stage C.6） + # ------------------------------------------------------------------ + # 长期创作项目的最小可用接口。所有方法都走 try/except 保护： + # - 缺失表 / DB 不可用 → 返回空值 / False，不抛 5xx + # - 单行转换失败：跳过该行，不影响整体 + + _CREATIVE_PRIMARY_TABLE = 'creative_projects' + _CREATIVE_BIBLE_TABLE = 'creative_story_bibles' + _CREATIVE_POOL_TABLE = 'creative_memory_pool' + + def _ensure_creative_table(self) -> Optional[str]: + """ + 返回当前可用的 creative_projects 表名；不存在则返回 None。 + """ + try: + existing = set(self.get_tables() or []) + except Exception: + existing = set() + if self._CREATIVE_PRIMARY_TABLE in existing: + return self._CREATIVE_PRIMARY_TABLE + return None + + def list_creative_projects( + self, + status: Optional[str] = None, + limit: int = 50, + ) -> List[Dict[str, Any]]: + """ + 列出 creative_projects 记录（按 status 过滤可选）。 + - 表缺失 / DB 不可用 → 返回 [] + - status 留空：列出全部 + - limit 做 1..200 边界保护 + """ + try: + table = self._ensure_creative_table() + if table is None: + return [] + db = self.get_db() + if db is None: + return [] + try: + bounded_limit = max(1, min(int(limit), 200)) + except (TypeError, ValueError): + bounded_limit = 50 + + with db.get_connection() as conn: + cur = conn.cursor() + if status: + cur.execute( + f"SELECT * FROM {table} " + f"WHERE status = ? " + f"ORDER BY created_at DESC LIMIT ?", + (str(status), int(bounded_limit)), + ) + else: + cur.execute( + f"SELECT * FROM {table} " + f"ORDER BY created_at DESC LIMIT ?", + (int(bounded_limit),), + ) + rows = cur.fetchall() or [] + out: List[Dict[str, Any]] = [] + for r in rows: + try: + out.append(dict(r) if not isinstance(r, dict) else dict(r)) + except Exception: + continue + return out + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] list_creative_projects 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + + def get_creative_project(self, project_uuid: str) -> Optional[Dict[str, Any]]: + """ + 按 uuid 读取单个 creative_projects 记录。 + 失败 / 不存在 → 返回 None。 + """ + try: + if not project_uuid or not isinstance(project_uuid, str): + return None + table = self._ensure_creative_table() + if table is None: + return None + db = self.get_db() + if db is None: + return None + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {table} WHERE uuid = ?", + (project_uuid,), + ) + row = cur.fetchone() + if not row: + return None + try: + return dict(row) if not isinstance(row, dict) else dict(row) + except Exception: + return None + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_creative_project({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return None + + def create_creative_project(self, project: Dict[str, Any]) -> Optional[str]: + """ + 插入一条 creative_projects 记录；返回新 uuid；失败返回 None。 + - 缺字段会用数据库默认值兜底 + - project['uuid'] 缺省时自动生成 + """ + try: + table = self._ensure_creative_table() + if table is None: + return None + db = self.get_db() + if db is None: + return None + + new_uuid = (project or {}).get('uuid') or str(_uuid.uuid4()) + now_iso = datetime.now().isoformat() + payload = { + 'uuid': new_uuid, + 'title': (project or {}).get('title') or '未命名', + 'work_type': (project or {}).get('work_type') or '短篇', + 'premise': (project or {}).get('premise', ''), + 'tone': (project or {}).get('tone', ''), + 'point_of_view': (project or {}).get('point_of_view', '第一人称'), + 'target_chars': int((project or {}).get('target_chars', 5000) or 5000), + 'current_chars': int((project or {}).get('current_chars', 0) or 0), + 'status': (project or {}).get('status') or 'drafting', + 'inspiration_source': (project or {}).get('inspiration_source', ''), + 'outline_json': json.dumps((project or {}).get('outline') or [], + ensure_ascii=False), + 'characters_json': json.dumps((project or {}).get('characters') or [], + ensure_ascii=False), + 'draft_chunks_json': json.dumps((project or {}).get('draft_chunks') or [], + ensure_ascii=False), + 'next_advance_at': (project or {}).get('next_advance_at'), + 'last_advanced_at': (project or {}).get('last_advanced_at'), + 'created_at': (project or {}).get('created_at') or now_iso, + 'updated_at': (project or {}).get('updated_at') or now_iso, + } + cols = list(payload.keys()) + placeholders = ','.join('?' for _ in cols) + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"INSERT INTO {table} ({','.join(cols)}) " + f"VALUES ({placeholders})", + [payload[c] for c in cols], + ) + return new_uuid + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] create_creative_project 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return None + + def update_creative_project(self, project_uuid: str, + updates: Dict[str, Any]) -> bool: + """ + 按 uuid 更新 creative_projects 字段；行不存在 / 失败 → 返回 False。 + updates 仅接受安全字段（title / status / current_chars / target_chars / + next_advance_at / last_advanced_at / premise / tone / point_of_view / + work_type / inspiration_source / outline / characters / draft_chunks）。 + """ + try: + if not project_uuid or not isinstance(project_uuid, str): + return False + table = self._ensure_creative_table() + if table is None: + return False + safe_scalar = { + 'title', 'work_type', 'premise', 'tone', 'point_of_view', + 'target_chars', 'current_chars', 'status', + 'inspiration_source', 'next_advance_at', 'last_advanced_at', + } + safe_json = { + 'outline': 'outline_json', + 'characters': 'characters_json', + 'draft_chunks': 'draft_chunks_json', + } + sets: List[str] = [] + values: List[Any] = [] + for k, v in (updates or {}).items(): + if k in safe_scalar: + sets.append(f"{k} = ?") + values.append(v) + elif k in safe_json: + sets.append(f"{safe_json[k]} = ?") + try: + values.append(json.dumps(v, ensure_ascii=False)) + except Exception: + values.append(json.dumps([], ensure_ascii=False)) + if not sets: + return False + sets.append('updated_at = ?') + values.append(datetime.now().isoformat()) + values.append(project_uuid) + db = self.get_db() + if db is None: + return False + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"UPDATE {table} SET {', '.join(sets)} WHERE uuid = ?", + values, + ) + return (cur.rowcount or 0) > 0 + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] update_creative_project({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + def get_story_bible(self, project_uuid: str) -> Optional[Dict[str, Any]]: + """ + 读取 project_uuid 对应的 Story Bible（含 JSON 字段反序列化）。 + 缺失 / 失败 → 返回 None。 + """ + try: + if not project_uuid or not isinstance(project_uuid, str): + return None + db = self.get_db() + if db is None: + return None + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {self._CREATIVE_BIBLE_TABLE} " + f"WHERE project_uuid = ?", + (project_uuid,), + ) + row = cur.fetchone() + if not row: + return None + data: Dict[str, Any] + try: + data = dict(row) if not isinstance(row, dict) else dict(row) + except Exception: + return None + for json_k in ( + 'active_themes_json', 'unresolved_threads_json', + 'resolved_threads_json', 'important_facts_json', + 'recent_keywords_json', + ): + try: + data[json_k.replace('_json', '')] = json.loads( + data.get(json_k) or '[]' + ) + except Exception: + data[json_k.replace('_json', '')] = [] + return data + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_story_bible({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return None + + def upsert_story_bible(self, project_uuid: str, + bible: Dict[str, Any]) -> bool: + """ + 插入/替换 Story Bible；失败 → 返回 False。 + """ + try: + if not project_uuid or not isinstance(project_uuid, str): + return False + db = self.get_db() + if db is None: + return False + now_iso = datetime.now().isoformat() + payload = { + 'project_uuid': project_uuid, + 'mainline_direction': (bible or {}).get('mainline_direction', ''), + 'active_themes_json': json.dumps( + (bible or {}).get('active_themes') or [], + ensure_ascii=False), + 'unresolved_threads_json': json.dumps( + (bible or {}).get('unresolved_threads') or [], + ensure_ascii=False), + 'resolved_threads_json': json.dumps( + (bible or {}).get('resolved_threads') or [], + ensure_ascii=False), + 'important_facts_json': json.dumps( + (bible or {}).get('important_facts') or [], + ensure_ascii=False), + 'next_direction': (bible or {}).get('next_direction', ''), + 'recent_keywords_json': json.dumps( + (bible or {}).get('recent_keywords') or [], + ensure_ascii=False), + 'created_at': now_iso, + 'updated_at': now_iso, + } + cols = list(payload.keys()) + placeholders = ','.join('?' for _ in cols) + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"INSERT OR REPLACE INTO {self._CREATIVE_BIBLE_TABLE} " + f"({','.join(cols)}) VALUES ({placeholders})", + [payload[c] for c in cols], + ) + return True + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] upsert_story_bible({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return False + + def get_creative_memory_pool( + self, + project_uuid: str, + limit: int = 50, + ) -> List[Dict[str, Any]]: + """ + 读取 project_uuid 对应的 creative_memory_pool 记录。 + 缺失 / 失败 → 返回 []。 + """ + try: + if not project_uuid or not isinstance(project_uuid, str): + return [] + db = self.get_db() + if db is None: + return [] + try: + bounded_limit = max(1, min(int(limit), 200)) + except (TypeError, ValueError): + bounded_limit = 50 + with db.get_connection() as conn: + cur = conn.cursor() + cur.execute( + f"SELECT * FROM {self._CREATIVE_POOL_TABLE} " + f"WHERE project_uuid = ? " + f"ORDER BY importance DESC, created_at DESC " + f"LIMIT ?", + (project_uuid, int(bounded_limit)), + ) + rows = cur.fetchall() or [] + out: List[Dict[str, Any]] = [] + for r in rows: + try: + out.append(dict(r) if not isinstance(r, dict) else dict(r)) + except Exception: + continue + return out + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[DatabaseService] get_creative_memory_pool({project_uuid}) 失败: {exc}", + RuntimeWarning, + stacklevel=2, + ) + return [] + + # ------------------------------------------------------------------ + # 统计 + # ------------------------------------------------------------------ + def get_stats(self) -> Dict[str, Any]: + """ + 返回 db_path / tables_count / db_size_bytes 等只读统计信息。 + """ + stats: Dict[str, Any] = { + 'db_path': self.db_path, + 'tables_count': 0, + 'db_size_bytes': 0, + } + # tables count + try: + stats['tables_count'] = len(self.get_tables()) + except Exception: + stats['tables_count'] = 0 + # db file size + try: + if self.db_path and os.path.exists(self.db_path): + stats['db_size_bytes'] = int(os.path.getsize(self.db_path)) + except Exception: + stats['db_size_bytes'] = 0 + return stats + + +# 全局单例 +database_service = DatabaseService() + + +__all__ = [ + "DatabaseService", + "database_service", + "ALLOWED_TABLES", +] diff --git a/src/web/backend/services/emotion_service.py b/src/web/backend/services/emotion_service.py new file mode 100644 index 0000000..997fb0c --- /dev/null +++ b/src/web/backend/services/emotion_service.py @@ -0,0 +1,475 @@ +""" +Emotion service module. +情感雷达后端服务 - 包装 EmotionRelationshipAnalyzer + PlutchikEmotionWheel， +为 /api/emotion/latest 和 WebSocket emotion_update 推送提供： + - 8 维累加评分（关系状态维度） + - 8 维 Plutchik 强度（当下情绪维度） + - 最近一次分析的时间戳 / 用户消息 + +设计要点： +1. 懒加载 emotion_analyzer：避免 Web 启动时强制实例化 ChatAgent / LLM +2. 缓存最近一次结果（避免每次 HTTP 请求都跑 LLM / 走 DB） +3. 失败/缺失时一律降级为全 0 字典 + 空字符串，不抛异常 +4. 进程内单例：emotion_service = EmotionService() +""" + +from __future__ import annotations + +import sys +import threading +import time +import warnings +from datetime import datetime, timezone +from pathlib import Path +from typing import Any, Dict, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + + +PLUTCHIK_KEYS = ( + 'joy', 'trust', 'fear', 'surprise', + 'sadness', 'disgust', 'anger', 'anticipation', +) + + +def _empty_eight() -> Dict[str, float]: + return {k: 0.0 for k in PLUTCHIK_KEYS} + + +def _empty_payload(user_id: str = "default") -> Dict[str, Any]: + """失败/无数据时的兜底 payload。""" + return { + "cumulative": _empty_eight(), + "plutchik": _empty_eight(), + "last_message": "", + "timestamp": "", + "user_id": user_id or "default", + "historical_max": None, + } + + +def _coerce_float(value: Any, default: float = 0.0) -> float: + try: + if value is None: + return default + return float(value) + except (TypeError, ValueError): + return default + + +def _safe_dict(d: Any) -> Dict[str, float]: + """把任意 dict 转成 8 维 dict；缺键补 0.0。""" + out = _empty_eight() + if not isinstance(d, dict): + return out + for k in PLUTCHIK_KEYS: + if k in d: + out[k] = _coerce_float(d[k], 0.0) + return out + + +class EmotionService: + """ + 情感雷达后端服务（Web 端入口） + + Attributes: + _cache: 进程内缓存 {user_id: (ts, payload)}，避免每次 HTTP 都跑 LLM + _lock: 写缓存的轻量锁 + _analyzer: 懒加载的 EmotionRelationshipAnalyzer 实例 + _wheel: 懒加载的 PlutchikEmotionWheel 实例 + """ + + # 缓存有效期（秒）：同一用户在这个窗口内的请求直接走缓存 + CACHE_TTL_SECONDS = 30.0 + + def __init__(self) -> None: + self._cache: Dict[str, Dict[str, Any]] = {} + self._lock = threading.RLock() + self._analyzer: Optional[Any] = None + self._wheel: Optional[Any] = None + # 主动尝试一次懒加载（失败时静默降级） + self._try_init() + + # ------------------------------------------------------------------ + # 懒加载 EmotionRelationshipAnalyzer / PlutchikEmotionWheel + # ------------------------------------------------------------------ + def _try_init(self) -> None: + try: + from src.core.emotion_analyzer import ( # type: ignore + EmotionRelationshipAnalyzer, + PlutchikEmotionWheel, + ) + self._analyzer = EmotionRelationshipAnalyzer() + self._wheel = PlutchikEmotionWheel(db_manager=self._analyzer.db) + except Exception as exc: # noqa: BLE001 + warnings.warn( + f"[EmotionService] 情感分析器初始化失败，将以无 analyzer 模式运行: {exc}", + RuntimeWarning, + stacklevel=2, + ) + self._analyzer = None + self._wheel = None + + def _ensure(self) -> bool: + """确保 analyzer / wheel 可用。失败时返回 False。""" + if self._analyzer is None or self._wheel is None: + self._try_init() + return self._analyzer is not None and self._wheel is not None + + # ------------------------------------------------------------------ + # 对外：缓存读写 + # ------------------------------------------------------------------ + def _cache_get(self, user_id: str) -> Optional[Dict[str, Any]]: + with self._lock: + entry = self._cache.get(user_id) + if not entry: + return None + ts = float(entry.get("ts", 0.0)) + if (time.time() - ts) > self.CACHE_TTL_SECONDS: + # 缓存过期 + self._cache.pop(user_id, None) + return None + return entry.get("payload") + + def _cache_put(self, user_id: str, payload: Dict[str, Any]) -> None: + with self._lock: + self._cache[user_id] = {"ts": time.time(), "payload": dict(payload)} + # 兜底：cache 体积上限 64 + if len(self._cache) > 64: + # 删除最旧的一条 + oldest_key = min( + self._cache.keys(), + key=lambda k: self._cache[k].get("ts", 0.0), + ) + self._cache.pop(oldest_key, None) + + def cache_clear(self, user_id: Optional[str] = None) -> None: + """主动失效缓存（测试 / 显式刷新）。""" + with self._lock: + if user_id is None: + self._cache.clear() + else: + self._cache.pop(user_id, None) + + # ------------------------------------------------------------------ + # 对外：构造 payload + # ------------------------------------------------------------------ + def get_latest_emotion(self, user_id: str = "default") -> Dict[str, Any]: + """ + 返回最新的情感分析结果。 + + Args: + user_id: 用户标识（当前实现下所有用户共用一份全局情感数据， + user_id 仅作为返回字段透传 + 缓存 key 使用） + + Returns: + dict: + { + "cumulative": {joy, trust, fear, surprise, sadness, disgust, anger, anticipation} # 8 维累加 + "plutchik": {同 8 维} # 0-1 归一化的当下情绪 + "last_message": str # 最近一次参与分析的用户消息（尽力从 DB 拼出） + "timestamp": str # ISO 8601 + "user_id": str + "historical_max": Optional[dict] # 8 维历史最高，没有则为 None + } + """ + uid = (user_id or "default").strip() or "default" + + # 1) 命中缓存：直接返回 + cached = self._cache_get(uid) + if cached is not None: + return dict(cached) + + # 2) 没命中 / 过期：尝试重新构造 + payload = self._build_payload(uid) + self._cache_put(uid, payload) + return dict(payload) + + def _build_payload(self, user_id: str) -> Dict[str, Any]: + """实际从 analyzer 拉数据；失败时返回空 payload。""" + empty = _empty_payload(user_id) + if not self._ensure(): + return empty + + # ---- 2.1) 拉取最近一次关系型分析（DB） ---- + try: + latest = self._analyzer.get_latest_emotion() # type: ignore[union-attr] + except Exception: # noqa: BLE001 + latest = None + + # ---- 2.2) 计算 cumulative（关系状态维度） ---- + cumulative = self._derive_cumulative(latest) + + # ---- 2.3) 计算 plutchik（当下情绪维度） ---- + plutchik = self._derive_plutchik() + + # ---- 2.4) last_message / timestamp ---- + timestamp = "" + last_message = "" + if isinstance(latest, dict): + timestamp = str(latest.get("timestamp") or latest.get("created_at") or "") + # impression 包含历史印象；不直接作为 last_message。 + # last_message 优先从短时记忆 DB 推断，否则留空。 + last_message = self._derive_last_message(latest) + + # ---- 2.5) historical_max（8 维历史最高） ---- + historical_max = self._derive_historical_max(cumulative, plutchik) + + return { + "cumulative": cumulative, + "plutchik": plutchik, + "last_message": last_message, + "timestamp": timestamp, + "user_id": user_id, + "historical_max": historical_max, + } + + # ------------------------------------------------------------------ + # 内部：维度推导 + # ------------------------------------------------------------------ + def _derive_cumulative(self, latest: Optional[Dict[str, Any]]) -> Dict[str, float]: + """ + 把 relationship_type / emotional_tone / overall_score 映射成 + 8 维 0-1 归一化评分。 + + 规则（启发式，无 LLM）： + - 整体评分（overall_score 0-100）→ 主导正向情绪的权重 + - emotional_tone / sentiment 调整正负向分布 + """ + out = _empty_eight() + if not isinstance(latest, dict): + return out + + try: + overall = _coerce_float(latest.get("overall_score"), 0.0) + except Exception: + overall = 0.0 + # 0-100 → 0-1 + score_norm = max(0.0, min(1.0, overall / 100.0)) + + tone = str(latest.get("emotional_tone") or "").strip() + sentiment = str(latest.get("sentiment") or "").strip().lower() + relationship = str(latest.get("relationship_type") or "").strip() + + # 主导：积极 / 消极 tone → 8 维分布 + is_positive = ( + tone in ("积极", "正面", "乐观", "友好") + or sentiment in ("positive", "pos") + ) + is_negative = ( + tone in ("消极", "负面", "悲观", "紧张") + or sentiment in ("negative", "neg") + ) + + if is_positive: + # 积极：joy / trust / anticipation 为主，surprise 副 + out["joy"] = round(score_norm * 0.9, 4) + out["trust"] = round(score_norm * 0.7, 4) + out["anticipation"] = round(score_norm * 0.5, 4) + out["surprise"] = round(score_norm * 0.3, 4) + elif is_negative: + # 消极：sadness / anger / fear / disgust 为主 + out["sadness"] = round(score_norm * 0.9, 4) + out["anger"] = round(score_norm * 0.6, 4) + out["fear"] = round(score_norm * 0.5, 4) + out["disgust"] = round(score_norm * 0.4, 4) + else: + # 中性：信任 / 期待 + 轻微 4 维 + out["trust"] = round(0.4 + score_norm * 0.3, 4) + out["anticipation"] = round(0.3 + score_norm * 0.2, 4) + out["joy"] = round(0.2 + score_norm * 0.2, 4) + out["surprise"] = round(0.1 + score_norm * 0.1, 4) + + # 关系类型微调 + if relationship in ("知己", "亲密朋友", "好友"): + out["trust"] = min(1.0, out["trust"] + 0.15) + out["joy"] = min(1.0, out["joy"] + 0.1) + elif relationship in ("初识", "陌生人"): + out["fear"] = min(1.0, out["fear"] + 0.1) + out["surprise"] = min(1.0, out["surprise"] + 0.1) + + return out + + def _derive_plutchik(self) -> Dict[str, float]: + """ + 从 PlutchikEmotionWheel 拉取衰减后的 8 维强度； + 无 wheel 或失败时返回全 0。 + """ + if self._wheel is None: + return _empty_eight() + try: + # wheel 状态在 ChatAgent 里维护；这里只读 _last_emotion_wheel_state + state = self._read_wheel_state() + if not state: + return _empty_eight() + # 应用衰减 + try: + state = self._wheel.decay_emotions(state) + except Exception: + pass + return _safe_dict(state.get("emotions") or {}) + except Exception: # noqa: BLE001 + return _empty_eight() + + def _read_wheel_state(self) -> Optional[Dict[str, Any]]: + """ + 读取 PlutchikEmotionWheel 当前状态。 + 优先从 ChatAgent 进程内单例拿；拿不到再尝试 metadata 表里的快照。 + """ + # 1) 尝试从 ChatAgent 拿（如果有） + try: + from src.core.chat_agent import ChatAgent # type: ignore + agent = ChatAgent() + state = getattr(agent, "_last_emotion_wheel_state", None) + if isinstance(state, dict): + return state + except Exception: + pass + + # 2) 尝试从 metadata 表读快照（key='emotion_wheel_state'） + try: + db = None + if self._wheel is not None and getattr(self._wheel, "db", None) is not None: + db = self._wheel.db + elif self._analyzer is not None and getattr(self._analyzer, "db", None) is not None: + db = self._analyzer.db + if db is not None and hasattr(db, "get_metadata"): + raw = db.get_metadata("emotion_wheel_state") + if raw: + import json + try: + return json.loads(raw) if isinstance(raw, str) else dict(raw) + except Exception: + return None + except Exception: + return None + return None + + def _derive_last_message(self, latest: Optional[Dict[str, Any]]) -> str: + """ + 尽力从短时记忆中取最近一条用户消息。 + 失败时返回空字符串。 + """ + if not isinstance(latest, dict): + return "" + # 如果 latest 字典里直接带了 last_message 字段（兜底），优先用 + direct = latest.get("last_message") + if isinstance(direct, str) and direct.strip(): + return direct.strip()[:2000] + # 尝试从 short_term_memory 表里查最近 1 条 user 消息 + try: + db = None + if self._analyzer is not None and getattr(self._analyzer, "db", None) is not None: + db = self._analyzer.db + if db is None: + return "" + method = getattr(db, "get_recent_messages", None) + if method is None: + return "" + try: + rows = method(limit=1, role="user") # type: ignore[arg-type] + except TypeError: + # 兼容不同的方法签名 + rows = method(limit=1) + if not rows: + return "" + row = rows[0] if isinstance(rows, list) else rows + if isinstance(row, dict): + content = row.get("content") or row.get("message") or "" + else: + content = str(getattr(row, "content", "") or "") + return str(content).strip()[:2000] + except Exception: + return "" + + def _derive_historical_max( + self, + cumulative: Dict[str, float], + plutchik: Dict[str, float], + ) -> Optional[Dict[str, float]]: + """ + 尽力从 emotion_history 表聚合 8 维历史最高。 + 当前没有按维度存 Plutchik 历史 → 直接用 cumulative + plutchik 元素最大值。 + """ + try: + db = None + if self._analyzer is not None and getattr(self._analyzer, "db", None) is not None: + db = self._analyzer.db + if db is None: + return None + history = db.get_emotion_history() if hasattr(db, "get_emotion_history") else [] + if not history: + return None + # 简单策略：取最近 20 条的整体评分，映射到 8 维 + scores: list = [] + for h in history[:20]: + if not isinstance(h, dict): + continue + try: + s = _coerce_float(h.get("overall_score"), 0.0) + except Exception: + continue + scores.append(max(0.0, min(1.0, s / 100.0))) + if not scores: + return None + peak = max(scores) + # 按 cumulative 模板放大 + return {k: round(min(1.0, v + peak * 0.2), 4) + for k, v in cumulative.items()} + except Exception: + return None + + # ------------------------------------------------------------------ + # 对外：emit（WebSocket emotion_update 推文的 helper） + # ------------------------------------------------------------------ + def build_emotion_update_event( + self, user_id: str = "default" + ) -> Dict[str, Any]: + """ + 为 WebSocket 推送构造 emotion_update 事件 payload。 + + 返回 dict： + { + "type": "emotion_update", + "data": { 8 维 plutchik }, + "cumulative": { 8 维 }, + "historical_max": Optional[dict], + "last_message": str, + "timestamp": str, + "user_id": str, + } + """ + payload = self.get_latest_emotion(user_id) + return { + "type": "emotion_update", + "data": dict(payload.get("plutchik") or {}), + "cumulative": dict(payload.get("cumulative") or {}), + "historical_max": payload.get("historical_max"), + "last_message": payload.get("last_message", ""), + "timestamp": payload.get("timestamp", ""), + "user_id": payload.get("user_id", user_id), + } + + def get_stats(self) -> Dict[str, Any]: + """返回服务状态。""" + return { + "analyzer_loaded": self._analyzer is not None, + "wheel_loaded": self._wheel is not None, + "cache_size": len(self._cache), + "cache_ttl_seconds": self.CACHE_TTL_SECONDS, + "version": "1.0.0", + } + + +# 全局单例 +emotion_service = EmotionService() + + +__all__ = [ + "EmotionService", + "emotion_service", + "PLUTCHIK_KEYS", +] diff --git a/src/web/backend/services/event_service.py b/src/web/backend/services/event_service.py new file mode 100644 index 0000000..4d8f821 --- /dev/null +++ b/src/web/backend/services/event_service.py @@ -0,0 +1,353 @@ +""" +EventService - 跨端事件服务 + +Stage D.3: +- 提供 `register_websocket_listener(channel, callback)` 注册 WebSocket 监听器 +- 提供 `emit_event(event_type, payload)` 触发事件（异步派发到匹配 channel 的 listener） +- 提供 `start()` 启动后台任务，订阅 EventManager 并把事件转发到 listeners + +设计要点： +1. 进程内单例：fastapi 应用内通过 `get_event_service()` 获取同一实例 +2. 与 WebSocket ConnectionManager 解耦：listener 拿到的只是 (event_type, payload) 元组 +3. 所有调用 try/except 保护，单个 listener 失败不影响其他 +""" + +from __future__ import annotations + +import asyncio +import inspect +import sys +import threading +from pathlib import Path +from typing import Any, Awaitable, Callable, Dict, List, Optional, Set + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_info(self, *args, **kwargs): pass + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + +debug_logger = get_debug_logger() + + +# 类型别名：listener 签名 (event_type: str, payload: dict) -> Awaitable[None] | None +ListenerCallback = Callable[[str, Dict[str, Any]], Optional[Awaitable[None]]] + + +class EventService: + """ + 事件服务（Web 后端侧） + 1. 维护 _listeners: Dict[channel, List[callback]] + 2. start() 后台订阅 EventManager 事件 → 转发到 listeners + 3. emit_event(event_type, payload) 可由 WebSocket handler 主动派发 + """ + + _instance_lock = threading.Lock() + + def __init__(self) -> None: + # channel -> [callback] + self._listeners: Dict[str, List[ListenerCallback]] = {} + self._lock = threading.RLock() + self._started: bool = False + self._unsubscribers: List[Callable[[], None]] = [] + # 兜底队列：在 start() 之前调用 emit_event 时，把事件暂存；start 后批量 dispatch + self._pending_events: List[Dict[str, Any]] = [] + # 事件循环引用（WebSocket 在 fastapi 事件循环中运行） + self._loop: Optional[asyncio.AbstractEventLoop] = None + + # ------------------------------------------------------------------ + # 单例 + # ------------------------------------------------------------------ + @classmethod + def get_instance(cls) -> "EventService": + with cls._instance_lock: + if not hasattr(cls, "_instance") or cls._instance is None: + cls._instance = cls() + return cls._instance + + # ------------------------------------------------------------------ + # 注册监听器 + # ------------------------------------------------------------------ + def register_websocket_listener( + self, + channel: str, + callback: ListenerCallback, + ) -> Callable[[], None]: + """ + 注册一个 WebSocket listener 到指定 channel。 + - channel: 事件类型字符串（'*' 表示通配） + - callback: 同步或 async 函数，签名 (event_type, payload) -> None + 返回 unsubscribe 函数，调用后移除该 listener。 + """ + if not channel: + channel = '*' + if not callable(callback): + raise TypeError("callback must be callable") + + with self._lock: + self._listeners.setdefault(str(channel), []).append(callback) + + def _unregister() -> None: + self.unregister_websocket_listener(channel, callback) + + return _unregister + + def unregister_websocket_listener( + self, + channel: str, + callback: ListenerCallback, + ) -> bool: + """注销 listener。""" + key = str(channel or '*') + with self._lock: + listeners = self._listeners.get(key) + if not listeners: + return False + try: + listeners.remove(callback) + except ValueError: + return False + if not listeners: + self._listeners.pop(key, None) + return True + + def listener_count(self, channel: Optional[str] = None) -> int: + """查询 listener 数量（测试 / 健康检查用）。""" + with self._lock: + if channel is None: + return sum(len(v) for v in self._listeners.values()) + return len(self._listeners.get(str(channel), [])) + + # ------------------------------------------------------------------ + # 事件派发 + # ------------------------------------------------------------------ + def emit_event(self, event_type: str, payload: Dict[str, Any]) -> None: + """ + 触发事件：找到匹配 channel 的所有 listener，异步派发。 + - 服务未 start 时，事件会暂存在 _pending_events 中；start() 后批量回放 + - 单个 listener 抛错不会影响其他 + """ + try: + event_type = str(event_type or 'message') + payload_dict: Dict[str, Any] = dict(payload or {}) + + with self._lock: + matched = list(self._listeners.get(event_type, [])) + \ + list(self._listeners.get('*', [])) + + if not matched: + # 未启动或暂无订阅：暂存到 pending（仅保留最近 200 条，避免内存膨胀） + if not self._started: + with self._lock: + self._pending_events.append({ + 'event_type': event_type, + 'payload': payload_dict, + }) + if len(self._pending_events) > 200: + self._pending_events = self._pending_events[-200:] + return + + self._dispatch_to_listeners(event_type, payload_dict, matched) + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('EventService', f'emit_event 失败: {e}', e) + except Exception: + pass + + def _dispatch_to_listeners( + self, + event_type: str, + payload: Dict[str, Any], + listeners: List[ListenerCallback], + ) -> None: + """把事件分发到 listeners，sync/async 兼容。""" + for cb in listeners: + try: + if inspect.iscoroutinefunction(cb): + # 尝试调度到事件循环 + scheduled = self._schedule_coroutine(cb, event_type, payload) + if not scheduled: + # 没有事件循环时降级为同步执行（用 asyncio.run） + try: + asyncio.run(cb(event_type, payload)) + except Exception as inner: # noqa: BLE001 + self._safe_log_error('EventService', + f'async listener 执行失败: {inner}', inner) + else: + cb(event_type, payload) + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'listener 执行失败({event_type}): {e}', e) + + def _schedule_coroutine( + self, + cb: ListenerCallback, + event_type: str, + payload: Dict[str, Any], + ) -> bool: + """在已知事件循环上调度协程；返回是否成功调度。""" + try: + loop = self._loop + if loop is None: + try: + loop = asyncio.get_event_loop() + except RuntimeError: + loop = None + if loop is not None and loop.is_running(): + loop.create_task(cb(event_type, payload)) + return True + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'调度协程失败({event_type}): {e}', e) + return False + + # ------------------------------------------------------------------ + # 启动 / 关闭 + # ------------------------------------------------------------------ + def start(self, event_manager: Any = None) -> bool: + """ + 启动后台任务，订阅 EventManager 并把事件转发到 listeners。 + - 可重复调用；多次调用幂等。 + - 启动后会回放 _pending_events 中的事件。 + """ + if self._started: + return True + + try: + # 记录事件循环引用（WebSocket 在同一循环中运行） + try: + self._loop = asyncio.get_event_loop() + except RuntimeError: + self._loop = None + + em = event_manager + if em is None: + try: + from src.core.event_manager import get_event_manager + em = get_event_manager() + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'获取 EventManager 失败: {e}', e) + em = None + + if em is not None and hasattr(em, 'subscribe'): + # 订阅常见事件类型 + 通配 + event_types = [ + 'scheduler_tick', + 'proactive_message', + 'chat_event', + 'debug_log', + 'message', + ] + for et in event_types: + try: + unsub = em.subscribe(et, self._on_manager_event) + if unsub is not None and callable(unsub): + self._unsubscribers.append(unsub) + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'订阅 {et} 失败: {e}', e) + # 通配订阅 + try: + unsub = em.subscribe('*', self._on_manager_event) + if unsub is not None and callable(unsub): + self._unsubscribers.append(unsub) + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'订阅通配失败: {e}', e) + + self._started = True + try: + debug_logger.log_info('EventService', '事件服务已启动', { + 'listeners': self.listener_count(), + }) + except Exception: + pass + + # 回放 pending 事件 + with self._lock: + pending = list(self._pending_events) + self._pending_events.clear() + for item in pending: + try: + self.emit_event(item.get('event_type', 'message'), + item.get('payload', {})) + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'回放 pending 失败: {e}', e) + return True + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', f'start 失败: {e}', e) + return False + + def stop(self) -> None: + """停止服务（清理订阅、保留 listeners 以便再次 start）。""" + try: + for unsub in list(self._unsubscribers): + try: + unsub() + except Exception: + pass + self._unsubscribers.clear() + self._started = False + try: + debug_logger.log_info('EventService', '事件服务已停止') + except Exception: + pass + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', f'stop 失败: {e}', e) + + # ------------------------------------------------------------------ + # EventManager → EventService 桥接 + # ------------------------------------------------------------------ + def _on_manager_event(self, event_type: str, payload: Dict[str, Any]) -> None: + """ + 由 EventManager.subscribe 注册的回调。 + 把事件转发给本服务的所有匹配 listener。 + """ + try: + # 复用 emit_event 逻辑 + self.emit_event(event_type, payload) + except Exception as e: # noqa: BLE001 + self._safe_log_error('EventService', + f'EventManager 事件转发失败: {e}', e) + + # ------------------------------------------------------------------ + # 工具 + # ------------------------------------------------------------------ + def _safe_log_error(self, module: str, msg: str, exc: Any = None) -> None: + try: + debug_logger.log_error(module, msg, exc) + except Exception: + pass + + +# 全局单例（直接调用 get_event_service() 即可） +_event_service: Optional[EventService] = None +_event_service_lock = threading.Lock() + + +def get_event_service() -> EventService: + """获取全局 EventService 单例。""" + global _event_service + with _event_service_lock: + if _event_service is None: + _event_service = EventService() + return _event_service + + +# Stage A.3: 直接可导入的模块级单例别名（与 get_event_service() 共享同一实例） +# 兼容 services/__init__.py 中的 `from .event_service import event_service` 用法。 +event_service: EventService = get_event_service() + + +__all__ = ["EventService", "get_event_service", "event_service", "ListenerCallback"] diff --git a/src/web/backend/ws/__init__.py b/src/web/backend/ws/__init__.py new file mode 100644 index 0000000..e1657ac --- /dev/null +++ b/src/web/backend/ws/__init__.py @@ -0,0 +1,100 @@ +""" +WebSocket routers +聚合所有 ws/* 子模块的 router，统一挂载到 FastAPI app。 + +Stage A.2 / D.3: 端点骨架与跨端事件分发。 +""" + +from __future__ import annotations + +import sys +from pathlib import Path + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from fastapi import APIRouter + +# 子模块 routers（用 try/except 保护，避免单个模块失败影响整体导入） +_routers = [] +_loaded = {} + +try: + from src.web.backend.ws.chat import router as _chat_router + _routers.append(_chat_router) + _loaded['chat'] = True +except Exception as e: # noqa: BLE001 + _loaded['chat'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('ws', f'failed to import chat router: {e}') + except Exception: + pass + +try: + from src.web.backend.ws.debug import router as _debug_router + _routers.append(_debug_router) + _loaded['debug'] = True +except Exception as e: # noqa: BLE001 + _loaded['debug'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('ws', f'failed to import debug router: {e}') + except Exception: + pass + +try: + from src.web.backend.ws.events import router as _events_router + _routers.append(_events_router) + _loaded['events'] = True +except Exception as e: # noqa: BLE001 + _loaded['events'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('ws', f'failed to import events router: {e}') + except Exception: + pass + +# 兼容旧版 /ws/event 单数端点（前端 EmotionPanel/TimelineCanvas 仍引用）。 +# docs/api.md 标记为"已弃用，请用 /ws/events"，但需要保持可达以做向后兼容。 +try: + from src.web.backend.ws.event import router as _event_router + _routers.append(_event_router) + _loaded['event'] = True +except Exception as e: # noqa: BLE001 + _loaded['event'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('ws', f'failed to import event router: {e}') + except Exception: + pass + +try: + from src.web.backend.ws.proactive import router as _proactive_router + _routers.append(_proactive_router) + _loaded['proactive'] = True +except Exception as e: # noqa: BLE001 + _loaded['proactive'] = False + try: + from src.tools.debug_logger import get_debug_logger + get_debug_logger().log_warn('ws', f'failed to import proactive router: {e}') + except Exception: + pass + + +# 统一 router（保留空 router 以便 main.py 无脑 include_router） +router = APIRouter() +for r in _routers: + try: + router.include_router(r) + except Exception: + pass + + +def get_loaded_status() -> dict: + return dict(_loaded) + + +__all__ = ["router", "get_loaded_status"] diff --git a/src/web/backend/ws/chat.py b/src/web/backend/ws/chat.py new file mode 100644 index 0000000..2e25241 --- /dev/null +++ b/src/web/backend/ws/chat.py @@ -0,0 +1,347 @@ +""" +Chat WebSocket Router +- /ws/chat：流式聊天 +- 通过 ChatService.stream_chat 转发到客户端 +- Stage D.3 集成：connection 时把 chat_event 转发到 EventService +- Stage B.2 集成：把 echo 占位替换为 ChatAgent 真流式 +- Stage B.3 集成：聊天流式结束时推送 emotion_update 事件（前端缓存到 window.__lastEmotion） +""" + +from __future__ import annotations + +import asyncio +import json +import sys +import uuid +from pathlib import Path +from typing import Any, Optional + +from fastapi import APIRouter, WebSocket, WebSocketDisconnect + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from src.web.backend.ws.manager import manager + +try: + from src.web.backend.services.chat_service import ChatService +except Exception: # noqa: BLE001 + ChatService = None # type: ignore + +# Stage B.3: 情感雷达服务（懒加载、可选依赖） +try: + from src.web.backend.services.emotion_service import emotion_service +except Exception: # noqa: BLE001 + emotion_service = None # type: ignore + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_info(self, *args, **kwargs): pass + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + +debug_logger = get_debug_logger() + +router = APIRouter() +CHANNEL = "chat" + + +# ---------------------------------------------------------------------- +# Stage B.3: 工具：构造 + 发送 emotion_update +# ---------------------------------------------------------------------- +async def _publish_emotion_update( + conn_id: str, + user_id: str = "default", + last_message: str = "", +) -> None: + """ + 在聊天流式结束时调用： + 1) 调 emotion_service.get_latest_emotion(user_id) 拿到 8 维 dict + 2) 构造 emotion_update 事件 payload + 3) 通过 manager.send_personal 直接发到 chat 连接 + 4) 同时 es.emit_event('emotion_update', payload) 推给 /ws/events 订阅者 + 失败时静默降级（不阻断 chat 主循环）。 + """ + uid = (user_id or "default").strip() or "default" + payload: dict = {} + try: + if emotion_service is not None: + payload = emotion_service.build_emotion_update_event(uid) or {} + # 透传最近一次用户消息（哪怕情感分析没拿到 last_message） + if last_message and isinstance(payload, dict): + if not payload.get("last_message"): + payload["last_message"] = last_message[:2000] + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('ws.chat', f'build_emotion_update 失败: {exc}') + except Exception: + pass + payload = { + "type": "emotion_update", + "data": {}, + "cumulative": {}, + "historical_max": None, + "last_message": last_message or "", + "timestamp": "", + "user_id": uid, + } + + # 1) 直接发给当前 chat ws 客户端（前端 useChatStream 处理） + try: + await manager.send_personal(conn_id, payload) + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('ws.chat', f'send_personal emotion_update 失败: {exc}') + except Exception: + pass + + # 2) 同步 publish 给 /ws/events 订阅者（emotion_update 事件） + try: + from src.web.backend.services.event_service import get_event_service + es = get_event_service() + if es is not None: + # emit_event 期望 payload 是 dict；我们把外层 type 留出来 + publish_payload = {k: v for k, v in payload.items() if k != "type"} + publish_payload.setdefault("data", payload.get("data") or {}) + es.emit_event('emotion_update', publish_payload) + except Exception: + pass + + # 3) v3.1.0: 异步把情感数据持久化到最近一条 assistant 消息 + try: + from src.web.backend.services.chat_service import chat_service as _cs + if _cs is not None: + # payload.data 是 8 维 dict；连同 plutchik/cumulative 一并写库 + _cs.on_emotion({ + "data": payload.get("data") or {}, + "cumulative": payload.get("cumulative") or {}, + "historical_max": payload.get("historical_max"), + "dominant": payload.get("dominant"), + "last_message": payload.get("last_message") or "", + "timestamp": payload.get("timestamp") or "", + }) + except Exception: + pass + + +@router.websocket("/ws/chat") +async def chat_ws(websocket: WebSocket) -> None: + """ + /ws/chat 端点 + 客户端发送 {"type": "message", "content": "..."} + 服务端流式返回 Pydantic ChatChunk 的 model_dump()（含 type / chunk / done） + 聊天结束后追加一条 {"type": "emotion_update", ...} 帧（Stage B.3） + """ + try: + await websocket.accept() + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.chat', f'accept 失败: {e}', e) + except Exception: + pass + return + + conn_id = uuid.uuid4().hex + try: + await manager.connect(websocket, conn_id, CHANNEL) + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.chat', f'connect 失败: {e}', e) + except Exception: + pass + + # v3.1.0: 性能修复 - 复用模块级 ChatService 单例，避免每个 WS 连接都 + # 重新实例化 ChatAgent / ModelRouter / LangChainLLM / DatabaseManager + # 等重型组件（旧实现每连接 3 个 LLM 初始化 + 全套 agent 重建） + chat_service: Optional[ChatService] = None + try: + from src.web.backend.services.chat_service import chat_service as _singleton + chat_service = _singleton + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('ws.chat', f'复用 chat_service 单例失败: {exc}') + except Exception: + pass + + # v3.1.0: 解析 query param 中的 session_id / user_id 并绑定 + try: + q_session_id = websocket.query_params.get("session_id") + if q_session_id and chat_service is not None: + try: + chat_service.set_session_id(int(q_session_id)) + except (TypeError, ValueError): + pass + except Exception: + pass + + try: + await manager.send_personal(conn_id, { + "type": "system", + "message": "connected to /ws/chat", + }) + except Exception: + pass + + # v3.1.0: 若仍未绑 session_id（前端未传），首条消息来时由 chat_service 自动创建 + # ack 帧会把真实 session_id 告诉前端 + try: + await manager.send_personal(conn_id, { + "type": "ack", + "session_id": str(chat_service.get_session_id()) if chat_service and chat_service.get_session_id() else "", + "conn_id": conn_id, + }) + except Exception: + pass + + try: + while True: + try: + raw = await websocket.receive_text() + except WebSocketDisconnect: + break + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.chat', f'receive_text 失败: {e}', e) + except Exception: + pass + break + + try: + data = json.loads(raw) if raw else {} + except Exception: + data = {"type": "message", "content": raw} + + msg_type = (data.get("type") or "message").lower() + user_input = (data.get("content") or data.get("text") or "").strip() + user_id = (data.get("user_id") or "default").strip() or "default" + + if msg_type == "ping": + await manager.send_personal(conn_id, {"type": "pong"}) + continue + + if not user_input: + await manager.send_personal(conn_id, { + "type": "error", + "message": "empty input", + }) + continue + + try: + from src.web.backend.services.event_service import get_event_service + es = get_event_service() + es.emit_event('chat_event', { + "kind": "user_message", + "content": user_input, + }) + except Exception: + pass + + if chat_service is None: + await manager.send_personal(conn_id, { + "type": "error", + "message": "ChatService unavailable", + }) + continue + + # Stage B.2: 真流式调用 ChatService.chat_stream（字符串 chunk） + # 推送协议：每 chunk -> {"type": "chunk", "content": chunk} + # 结束 -> {"type": "done"} + # 异常 -> {"type": "error", "message": str(e)} + try: + # v3.1.0: 若首条消息自动创建了 session，主动把 session_id 通过 ack 帧告诉前端 + _had_session_at_start = bool( + chat_service and chat_service.get_session_id() + ) + async for chunk in chat_service.chat_stream( + user_input, context=data + ): + # 跳过空字符串 chunk（前端没有内容可显示） + if chunk is None or chunk == "": + continue + await manager.send_personal(conn_id, { + "type": "chunk", + "content": chunk, + }) + # 正常结束：done 帧 + await manager.send_personal(conn_id, {"type": "done"}) + + # v3.1.0: 若是首条消息并触发了 session 创建，补发 session_id + if chat_service is not None and not _had_session_at_start: + new_sid = chat_service.get_session_id() + if new_sid: + try: + await manager.send_personal(conn_id, { + "type": "session", + "session_id": str(new_sid), + }) + except Exception: + pass + + try: + from src.web.backend.services.event_service import get_event_service + es = get_event_service() + es.emit_event('chat_event', {"kind": "done", "input": user_input}) + except Exception: + pass + + # Stage B.3: 流式结束 → 推送 emotion_update + # 缓存失效 + 重新构造（确保拿到最新分析） + try: + if emotion_service is not None: + emotion_service.cache_clear(user_id) + except Exception: + pass + # v3.1.0: 先调 on_emotion 把情感数据写入 assistant 消息的 emotion_json + try: + if chat_service is not None: + # 占位 payload —— 真实值会由 _publish_emotion_update 推送； + # 这里只是尝试一次同步写入最新缓存（如果有） + pass + except Exception: + pass + await _publish_emotion_update( + conn_id=conn_id, + user_id=user_id, + last_message=user_input, + ) + except Exception as e: # noqa: BLE001 + # 兜底：流式过程中出现非预期异常，发送 error frame 后继续 + try: + debug_logger.log_error('ws.chat', f'chat_stream 失败: {e}', e) + except Exception: + pass + try: + await manager.send_personal(conn_id, { + "type": "error", + "message": str(e), + }) + except Exception: + pass + except WebSocketDisconnect: + pass + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.chat', f'主循环异常: {e}', e) + except Exception: + pass + finally: + # v3.1.0: 连接结束清理 - 重置单例的 _session_id， + # 防止下个连接继承上次的会话上下文（避免跨连接串话） + try: + if chat_service is not None and hasattr(chat_service, "set_session_id"): + chat_service.set_session_id(None) + except Exception: + pass + try: + manager.disconnect(conn_id) + except Exception: + pass + + +__all__ = ["router", "CHANNEL"] diff --git a/src/web/backend/ws/debug.py b/src/web/backend/ws/debug.py new file mode 100644 index 0000000..99c4067 --- /dev/null +++ b/src/web/backend/ws/debug.py @@ -0,0 +1,190 @@ +""" +Debug WebSocket Router +- /ws/debug：流式调试日志 +- Stage D.4: 连接时注册 DebugLogger 订阅者；断开时取消订阅 +- 同时也注册一个 EventService listener，把 'debug_log' 事件广播到本 channel +""" + +from __future__ import annotations + +import asyncio +import json +import sys +import uuid +from pathlib import Path +from typing import Any, Callable, Dict, Optional + +from fastapi import APIRouter, WebSocket, WebSocketDisconnect + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from src.web.backend.ws.manager import manager + +try: + from src.web.backend.services.event_service import get_event_service +except Exception: # noqa: BLE001 + def get_event_service(): + return None + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + _subscribers = [] + def subscribe(self, *a, **kw): return lambda: None + def unsubscribe(self, *a, **kw): return False + return _Stub() + +debug_logger = get_debug_logger() +router = APIRouter() +CHANNEL = "debug" + + +async def _safe_send_personal(conn_id: str, payload: Any) -> bool: + try: + return await manager.send_personal(conn_id, payload) + except Exception: + return False + + +@router.websocket("/ws/debug") +async def debug_ws(websocket: WebSocket) -> None: + """ + /ws/debug 端点 + - 接受连接后，订阅 DebugLogger；每条日志通过 WS 推送给客户端 + - 断开时取消订阅 + - 同时也注册一个 EventService listener，把 'debug_log' 事件广播到本 channel + """ + try: + await websocket.accept() + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.debug', f'accept 失败: {e}', e) + except Exception: + pass + return + + conn_id = uuid.uuid4().hex + try: + await manager.connect(websocket, conn_id, CHANNEL) + except Exception: + pass + + await _safe_send_personal(conn_id, {"type": "system", "message": "connected to /ws/debug"}) + + # 订阅 DebugLogger + queue: asyncio.Queue = asyncio.Queue(maxsize=500) + + def _subscriber(log_entry: Dict[str, Any]) -> None: + try: + try: + loop = asyncio.get_event_loop() + except RuntimeError: + loop = None + if loop is not None and loop.is_running(): + loop.call_soon_threadsafe(queue.put_nowait, log_entry) + else: + queue.put_nowait(log_entry) + except Exception: + pass + + unsubscribe = None + try: + if hasattr(debug_logger, 'subscribe'): + unsubscribe = debug_logger.subscribe(_subscriber) + except Exception: + unsubscribe = None + + # 注册 EventService listener + es_unsub = None + try: + es = get_event_service() + if es is not None: + async def _es_listener(_event_type: str, payload: Dict[str, Any]) -> None: + try: + await manager.broadcast(CHANNEL, { + "type": "event", + "event_type": _event_type, + "payload": payload, + }) + except Exception: + pass + + es_unsub = es.register_websocket_listener('debug_log', _es_listener) + except Exception: + es_unsub = None + + try: + while True: + # 使用 asyncio.wait 监听 receive_text 与 queue.get + receive_task = asyncio.create_task(websocket.receive_text()) + done_event = asyncio.create_task(queue.get()) + try: + done, _pending = await asyncio.wait( + {receive_task, done_event}, + return_when=asyncio.FIRST_COMPLETED, + ) + except Exception: + if not receive_task.done(): + receive_task.cancel() + if not done_event.done(): + done_event.cancel() + break + + if receive_task in done: + if not done_event.done(): + done_event.cancel() + try: + raw = receive_task.result() + except WebSocketDisconnect: + break + except Exception: + raw = None + if raw is None: + break + try: + data = json.loads(raw) if raw else {} + except Exception: + data = {"raw": raw} + if (data.get("type") or "").lower() == "ping": + await _safe_send_personal(conn_id, {"type": "pong"}) + continue + + if done_event in done: + if not receive_task.done(): + receive_task.cancel() + try: + log_entry = done_event.result() + except Exception: + log_entry = None + if log_entry is not None: + await _safe_send_personal(conn_id, { + "type": "log", + "entry": log_entry, + }) + except WebSocketDisconnect: + pass + except Exception: + pass + finally: + try: + if unsubscribe is not None: + unsubscribe() + except Exception: + pass + try: + if es_unsub is not None: + es_unsub() + except Exception: + pass + try: + manager.disconnect(conn_id) + except Exception: + pass + + +__all__ = ["router", "CHANNEL"] diff --git a/src/web/backend/ws/debug_broadcaster.py b/src/web/backend/ws/debug_broadcaster.py new file mode 100644 index 0000000..dd5ccf5 --- /dev/null +++ b/src/web/backend/ws/debug_broadcaster.py @@ -0,0 +1,140 @@ +""" +DebugBroadcaster +- 启动时把 DebugLogger 的日志广播到 /ws/debug 频道 +- 兼容两种集成方式： + 1) 通过 EventService（推荐）：注册 EventService 'debug_log' listener，由 events WS 转发 + 2) 直接通过 manager.broadcast('debug', ...) 向 /ws/debug 客户端推送 + +本模块只负责订阅 + 派发；具体 WebSocket 生命周期由 ws/debug.py 管理。 +""" + +from __future__ import annotations + +import asyncio +import sys +from pathlib import Path +from typing import Any, Callable, Dict, Optional + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + + +class DebugBroadcaster: + """ + 调试日志广播器 + - 启动时调用 attach() 注册订阅者 + - 关闭时调用 detach() 取消订阅 + - 同步 subscriber：直接 broadcast；async subscriber：调度到事件循环 + """ + + CHANNEL = "debug" + + def __init__(self) -> None: + self._unsub_debug: Optional[Callable[[], None]] = None + self._unsub_event_service: Optional[Callable[[], None]] = None + self._loop: Optional[asyncio.AbstractEventLoop] = None + + def attach(self, manager: Any = None, event_service: Any = None, + loop: Optional[asyncio.AbstractEventLoop] = None) -> bool: + """ + 注册订阅者。 + + Args: + manager: ConnectionManager 实例（来自 ws/manager.py）。Stage A.2 兼容： + 既支持新 manager（仅 channel -> broadcast）也支持老 manager 单例。 + event_service: EventService 实例（可选） + loop: 主事件循环 + + Returns: + True 表示至少成功注册了 DebugLogger 订阅者。 + """ + self._loop = loop + try: + from src.tools.debug_logger import get_debug_logger + except Exception: + return False + debug_logger = get_debug_logger() + try: + debug_logger.set_main_loop(loop) + except Exception: + pass + + if manager is not None and hasattr(debug_logger, 'subscribe'): + def _sync_subscriber(log_entry: Dict[str, Any]) -> None: + try: + if manager is None: + return + coro = manager.broadcast(self.CHANNEL, { + "type": "log", + "entry": log_entry, + }) + try: + running_loop = asyncio.get_running_loop() + except RuntimeError: + running_loop = None + if running_loop is not None and self._loop is running_loop: + running_loop.create_task(coro) + else: + target = self._loop + if target is not None and target.is_running(): + asyncio.run_coroutine_threadsafe(coro, target) + else: + try: + asyncio.run(coro) + except Exception: + pass + except Exception: + pass + + try: + unsub = debug_logger.subscribe(_sync_subscriber) + if unsub is not None and callable(unsub): + self._unsub_debug = unsub + except Exception: + self._unsub_debug = None + + if event_service is not None and hasattr(event_service, 'register_websocket_listener'): + try: + async def _es_listener(_event_type: str, payload: Dict[str, Any]) -> None: + if manager is not None: + await manager.broadcast(self.CHANNEL, payload) + + self._unsub_event_service = event_service.register_websocket_listener( + 'debug_log', _es_listener + ) + except Exception: + self._unsub_event_service = None + + return self._unsub_debug is not None or self._unsub_event_service is not None + + def detach(self) -> None: + """取消所有订阅。""" + if self._unsub_debug is not None: + try: + self._unsub_debug() + except Exception: + pass + self._unsub_debug = None + if self._unsub_event_service is not None: + try: + self._unsub_event_service() + except Exception: + pass + self._unsub_event_service = None + + +# 进程级单例 +_broadcaster: Optional[DebugBroadcaster] = None + + +def get_debug_broadcaster() -> DebugBroadcaster: + """获取全局 DebugBroadcaster 单例。""" + global _broadcaster + if _broadcaster is None: + _broadcaster = DebugBroadcaster() + return _broadcaster + + +__all__ = ["DebugBroadcaster", "get_debug_broadcaster"] diff --git a/src/web/backend/ws/event.py b/src/web/backend/ws/event.py new file mode 100644 index 0000000..8d23ef8 --- /dev/null +++ b/src/web/backend/ws/event.py @@ -0,0 +1,59 @@ +""" +``/ws/event`` 端点 — 事件通道 +``/ws/event`` endpoint — event channel. + +Stage A.2: 骨架实现。 +- 接受连接 → 注册到 ``manager`` 的 ``event`` channel。 +- 保持长连接；服务端**不主动发送**（事件由 EventManager 异步触发，Stage D 接入）。 +- 客户端发送的消息目前仅做 ack 回执。 +""" + +from __future__ import annotations + +import uuid + +from fastapi import APIRouter, WebSocket, WebSocketDisconnect + +from .manager import manager + + +router = APIRouter() + + +@router.websocket("/ws/event") +async def event_endpoint(websocket: WebSocket) -> None: + """ + 事件通道 WebSocket 端点。 + Event channel WebSocket endpoint. + + 协议（Stage A.2 骨架）： + - 客户端 connect → 服务端发送 ``{"type": "ack", "conn_id": ..., "channel": "event"}``。 + - 服务端**不主动推送**（Stage D 接入 EventManager 后由 ``manager.broadcast("event", ...)`` 驱动）。 + - 客户端发送任意消息 → 服务端回 ``{"type": "ack", "received": true}``。 + """ + await websocket.accept() + conn_id = uuid.uuid4().hex + await manager.connect(websocket, conn_id, "event") + + try: + await websocket.send_json( + {"type": "ack", "conn_id": conn_id, "channel": "event"} + ) + + while True: + # 仅做接收 / ack，不主动推送（Stage D 接入） + await websocket.receive_text() + await websocket.send_json({"type": "ack", "received": True}) + + except WebSocketDisconnect: + manager.disconnect(conn_id) + except Exception as exc: # noqa: BLE001 + print(f"[ws/event] 异常 conn_id={conn_id}: {exc}") + manager.disconnect(conn_id) + try: + await websocket.close() + except Exception: # noqa: BLE001 + pass + + +__all__ = ["router"] diff --git a/src/web/backend/ws/events.py b/src/web/backend/ws/events.py new file mode 100644 index 0000000..3e7f03d --- /dev/null +++ b/src/web/backend/ws/events.py @@ -0,0 +1,149 @@ +""" +Events WebSocket Router +- /ws/events：通用事件推送（scheduler_tick / proactive_message / chat_event / message / 通配） +- 通过 EventService.register_websocket_listener 订阅 +""" + +from __future__ import annotations + +import asyncio +import json +import sys +import uuid +from pathlib import Path +from typing import Any, Dict + +from fastapi import APIRouter, WebSocket, WebSocketDisconnect + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +from src.web.backend.ws.manager import manager + +try: + from src.web.backend.services.event_service import get_event_service +except Exception: # noqa: BLE001 + def get_event_service(): + return None + +router = APIRouter() +CHANNEL = "events" + + +@router.websocket("/ws/events") +async def events_ws(websocket: WebSocket) -> None: + """ + /ws/events 端点 + - 接受后注册 EventService listener（订阅常见 event types） + - 把事件原样推送给客户端 + - 断开时自动取消订阅 + """ + try: + await websocket.accept() + except Exception: + return + + conn_id = uuid.uuid4().hex + try: + await manager.connect(websocket, conn_id, CHANNEL) + except Exception: + pass + + try: + await manager.send_personal(conn_id, { + "type": "system", "message": "connected to /ws/events", + }) + except Exception: + pass + + unsubscribers = [] + + try: + es = get_event_service() + if es is not None: + channels = [ + 'scheduler_tick', + 'proactive_message', + 'chat_event', + 'message', + ] + + for ch in channels: + async def _make_listener(_ch: str): + async def _listener(event_type: str, payload: Dict[str, Any]) -> None: + try: + await manager.broadcast(CHANNEL, { + "type": "event", + "channel": _ch, + "event_type": event_type, + "payload": payload, + }) + except Exception: + pass + return _listener + + try: + listener = await _make_listener(ch) + unsub = es.register_websocket_listener(ch, listener) + if unsub is not None: + unsubscribers.append(unsub) + except Exception: + pass + + # 通配订阅（兜底） + async def _wildcard_listener(event_type: str, payload: Dict[str, Any]) -> None: + try: + await manager.broadcast(CHANNEL, { + "type": "event", + "channel": "*", + "event_type": event_type, + "payload": payload, + }) + except Exception: + pass + + try: + unsub = es.register_websocket_listener('*', _wildcard_listener) + if unsub is not None: + unsubscribers.append(unsub) + except Exception: + pass + except Exception: + pass + + try: + while True: + try: + raw = await websocket.receive_text() + except WebSocketDisconnect: + break + except Exception: + break + try: + data = json.loads(raw) if raw else {} + except Exception: + data = {"raw": raw} + if (data.get("type") or "").lower() == "ping": + try: + await manager.send_personal(conn_id, {"type": "pong"}) + except Exception: + pass + except WebSocketDisconnect: + pass + except Exception: + pass + finally: + for u in unsubscribers: + try: + u() + except Exception: + pass + try: + manager.disconnect(conn_id) + except Exception: + pass + + +__all__ = ["router", "CHANNEL"] diff --git a/src/web/backend/ws/manager.py b/src/web/backend/ws/manager.py new file mode 100644 index 0000000..21ec23f --- /dev/null +++ b/src/web/backend/ws/manager.py @@ -0,0 +1,240 @@ +""" +WebSocket Connection Manager +集中维护所有活跃的 WebSocket 连接（按 channel 分组）。 + +Stage A.2 兼容（conn_id-based API）： +- active_connections: Dict[conn_id -> WebSocket] +- channels: Dict[channel_name -> Set[conn_id]] +- 已被 event.py / proactive.py 等 Stage A.2 端点使用 + +Stage D.3 扩展： +- 新增 alias 方法 connect_by_channel / disconnect_by_channel / broadcast_by_channel + 供新端点 (chat/debug/events) 使用 +- 新增 connection_count / channels 查询方法 +- 单点失败不影响其他连接（try/except 保护） +""" + +from __future__ import annotations + +import asyncio +import json +import sys +import uuid +from pathlib import Path +from typing import Any, Dict, List, Optional, Set +from fastapi import WebSocket + +# 确保从项目根目录能正确 import src.* +_PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent.parent +if str(_PROJECT_ROOT) not in sys.path: + sys.path.insert(0, str(_PROJECT_ROOT)) + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_info(self, *args, **kwargs): pass + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + +debug_logger = get_debug_logger() + + +class ConnectionManager: + """ + WebSocket 连接管理器（单例友好的纯类）。 + + Attributes: + active_connections: 全局连接表，``conn_id -> WebSocket``。 + channels: 通道索引，``channel_name -> set(conn_id)``。 + _lock: 写操作的轻量锁（仅在 add/remove/broadcast 时短暂使用）。 + """ + + def __init__(self) -> None: + self.active_connections: Dict[str, WebSocket] = {} + self.channels: Dict[str, Set[str]] = {} + self._lock: asyncio.Lock = asyncio.Lock() + + # ------------------------------------------------------------------ + # 注册 / 注销（Stage A.2 API） + # ------------------------------------------------------------------ + async def connect(self, websocket: WebSocket, conn_id: str, channel: str) -> None: + """ + 注册一条连接（Stage A.2 老 API）。 + + 调用方需先 ``await websocket.accept()``，本方法不负责 accept。 + """ + self.register_channel(channel) + async with self._lock: + self.active_connections[conn_id] = websocket + self.channels[channel].add(conn_id) + try: + debug_logger.log_info('WSManager', f'客户端加入 [{channel}] conn_id={conn_id}') + except Exception: + pass + + def disconnect(self, conn_id: str) -> None: + """ + 注销一条连接（同步、幂等）。 + """ + self.active_connections.pop(conn_id, None) + for ch_name, members in list(self.channels.items()): + if conn_id in members: + members.discard(conn_id) + + # ------------------------------------------------------------------ + # Stage D.3: 接受 + 注册一行式 + # ------------------------------------------------------------------ + async def accept_and_connect(self, channel: str, websocket: WebSocket) -> str: + """ + 一步完成 accept + 注册到 channel，返回 conn_id。 + """ + try: + await websocket.accept() + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('WSManager', f'accept 失败: {e}', e) + except Exception: + pass + conn_id = uuid.uuid4().hex + await self.connect(websocket, conn_id, channel) + return conn_id + + async def close_and_disconnect(self, channel: str, websocket: WebSocket, + conn_id: Optional[str] = None) -> None: + """关闭 + 注销。""" + if conn_id is not None: + self.disconnect(conn_id) + return + # 兜底：反向查 conn_id + for cid, ws in list(self.active_connections.items()): + if ws is websocket: + self.disconnect(cid) + return + + # ------------------------------------------------------------------ + # 单播 + # ------------------------------------------------------------------ + async def send_personal(self, conn_id: str, message: Any) -> bool: + """ + 向指定 conn_id 发送一条消息（dict / str 均可）。 + """ + ws = self.active_connections.get(conn_id) + if ws is None: + return False + try: + if isinstance(message, (dict, list)): + await ws.send_text(json.dumps(message, ensure_ascii=False, default=str)) + elif isinstance(message, str): + await ws.send_text(message) + else: + await ws.send_text(str(message)) + return True + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('WSManager', + f'send_personal 失败 conn_id={conn_id}: {exc}') + except Exception: + pass + self.disconnect(conn_id) + return False + + # ------------------------------------------------------------------ + # 广播 + # ------------------------------------------------------------------ + async def broadcast(self, channel: str, message: Any) -> int: + """ + 向指定 channel 内的所有连接广播一条消息。 + 返回成功发送的数量。 + """ + if isinstance(message, (dict, list)): + text = json.dumps(message, ensure_ascii=False, default=str) + elif isinstance(message, str): + text = message + else: + text = str(message) + + members = list(self.channels.get(channel, set())) + sent = 0 + for conn_id in members: + ws = self.active_connections.get(conn_id) + if ws is None: + self.channels[channel].discard(conn_id) + continue + try: + await ws.send_text(text) + sent += 1 + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('WSManager', + f'broadcast 失败 channel={channel} conn_id={conn_id}: {exc}') + except Exception: + pass + self.disconnect(conn_id) + return sent + + async def broadcast_all(self, message: Any) -> int: + """向所有已注册连接广播一条消息。""" + if isinstance(message, (dict, list)): + text = json.dumps(message, ensure_ascii=False, default=str) + elif isinstance(message, str): + text = message + else: + text = str(message) + conn_ids = list(self.active_connections.keys()) + sent = 0 + for conn_id in conn_ids: + ws = self.active_connections.get(conn_id) + if ws is None: + continue + try: + await ws.send_text(text) + sent += 1 + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_warn('WSManager', + f'broadcast_all 失败 conn_id={conn_id}: {exc}') + except Exception: + pass + self.disconnect(conn_id) + return sent + + # ------------------------------------------------------------------ + # 通道管理 / 统计 + # ------------------------------------------------------------------ + def register_channel(self, channel: str) -> None: + """预注册一个通道名（幂等）。""" + if channel not in self.channels: + self.channels[channel] = set() + + def get_stats(self) -> dict: + per_channel: Dict[str, int] = { + ch: len(members) for ch, members in self.channels.items() + } + return { + "total_connections": len(self.active_connections), + "channels": per_channel, + } + + def connection_count(self, channel: Optional[str] = None) -> int: + """查询连接数。""" + if channel is None: + return len(self.active_connections) + return len(self.channels.get(str(channel), set())) + + def list_channels(self) -> List[str]: + return list(self.channels.keys()) + + +# 全局单例，方便其他模块直接 import +manager = ConnectionManager() + + +def get_connection_manager() -> ConnectionManager: + """获取全局 ConnectionManager 单例。""" + return manager + + +__all__ = ["ConnectionManager", "manager", "get_connection_manager"] diff --git a/src/web/backend/ws/proactive.py b/src/web/backend/ws/proactive.py new file mode 100644 index 0000000..04b962f --- /dev/null +++ b/src/web/backend/ws/proactive.py @@ -0,0 +1,235 @@ +""" +``/ws/proactive`` 端点 — 主动消息通道 +``/ws/proactive`` endpoint — proactive message channel. + +Stage A.2: 骨架实现（accept + ack）。 +Stage D.2: 完善 — 订阅 ``EventService`` 的 ``proactive_message`` 事件， + 包装 payload 为前端约定的格式后广播给客户端。 + - 频率节流：``PROACTIVE_MESSAGE_INTERVAL``（秒）可配置； + 低于阈值的重复消息不上推（前端可配置）。 + - 不修改 ``ProactiveEngine`` 现有逻辑，仅在端点层做节流。 +""" + +from __future__ import annotations + +import os +import time +import uuid +from typing import Any, Dict, Optional + +from fastapi import APIRouter, WebSocket, WebSocketDisconnect + +from .manager import manager + +try: + from src.web.backend.services.event_service import get_event_service +except Exception: # noqa: BLE001 + def get_event_service(): + return None + +try: + from src.tools.debug_logger import get_debug_logger +except Exception: # noqa: BLE001 + def get_debug_logger(): + class _Stub: + def log_info(self, *args, **kwargs): pass + def log_warn(self, *args, **kwargs): pass + def log_error(self, *args, **kwargs): pass + return _Stub() + +debug_logger = get_debug_logger() + + +router = APIRouter() +CHANNEL = "proactive" + + +# ---------------------------------------------------------------------- +# 频率节流配置 +# ---------------------------------------------------------------------- +def _get_proactive_interval() -> float: + """ + 读取 ``PROACTIVE_MESSAGE_INTERVAL`` 环境变量（秒）。 + 解析失败或未设置 → 返回 0（不节流，沿用 ProactiveEngine 自身节流）。 + """ + raw = os.getenv("PROACTIVE_MESSAGE_INTERVAL", "") + try: + if raw is None or str(raw).strip() == "": + return 0.0 + v = float(str(raw).strip()) + return max(0.0, v) + except (TypeError, ValueError): + return 0.0 + + +# 进程内节流状态（按 channel 共享即可） +_last_proactive_at: float = 0.0 +_PROACTIVE_INTERVAL_SEC: float = _get_proactive_interval() + + +def _should_throttle() -> bool: + """ + 是否被节流。返回 True 表示应跳过本次上推。 + """ + if _PROACTIVE_INTERVAL_SEC <= 0: + return False + now = time.time() + if (now - _last_proactive_at) < _PROACTIVE_INTERVAL_SEC: + return True + return False + + +def _mark_proactive_pushed() -> None: + global _last_proactive_at + _last_proactive_at = time.time() + + +# ---------------------------------------------------------------------- +# Payload 包装 +# ---------------------------------------------------------------------- +def _wrap_proactive_payload(payload: Optional[Dict[str, Any]]) -> Dict[str, Any]: + """ + 将 EventManager / ProactiveEngine 推送的 payload 包装为前端期望的格式。 + 字段： + - type: "proactive_message" + - content: 主动消息文本 + - timestamp: ISO 8601 时间戳 + - user: 接收用户 + - reason: 想法来源（habit_care / open_loop_recall / creative_share / weather_note） + - play_sound: 是否播放通知音（默认 True） + """ + if not isinstance(payload, dict): + payload = {} + content = str(payload.get("content") or payload.get("text") or "") + timestamp = str(payload.get("timestamp") or "") + user = str(payload.get("user") or "default") + reason = str(payload.get("reason") or payload.get("source") or "") + priority = str(payload.get("priority") or "normal") + return { + "type": "proactive_message", + "content": content, + "timestamp": timestamp, + "user": user, + "reason": reason, + "priority": priority, + "idea_id": str(payload.get("idea_id") or ""), + "source": str(payload.get("source") or "ProactiveEngine"), + "play_sound": bool(payload.get("play_sound", True)), + } + + +# ---------------------------------------------------------------------- +# 端点 +# ---------------------------------------------------------------------- +@router.websocket("/ws/proactive") +async def proactive_endpoint(websocket: WebSocket) -> None: + """ + 主动消息通道 WebSocket 端点。 + + 协议： + - 客户端 connect → 服务端发送 ``{"type": "ack", "conn_id": ..., "channel": "proactive"}``。 + - 客户端发送任意消息 → 服务端回 ``{"type": "ack", "received": true}``。 + - 主动推送：服务端通过 ``EventService`` 监听 ``proactive_message`` 事件， + 节流检查后包装为前端约定格式并广播。 + """ + await websocket.accept() + conn_id = uuid.uuid4().hex + try: + await manager.connect(websocket, conn_id, CHANNEL) + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error('ws.proactive', f'connect 失败: {e}', e) + except Exception: + pass + + # Stage D.2: 订阅 proactive_message 事件 + unsubscribers = [] + try: + es = get_event_service() + if es is not None: + async def _listener(event_type: str, payload: Dict[str, Any]) -> None: + if event_type != "proactive_message": + return + if _should_throttle(): + try: + debug_logger.log_info( + 'ws.proactive', + '主动消息被节流（间隔 < PROACTIVE_MESSAGE_INTERVAL）', + ) + except Exception: + pass + return + wrapped = _wrap_proactive_payload(payload) + try: + await manager.broadcast(CHANNEL, wrapped) + _mark_proactive_pushed() + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_error( + 'ws.proactive', f'广播失败: {e}', e, + ) + except Exception: + pass + + try: + unsub = es.register_websocket_listener( + "proactive_message", _listener, + ) + if unsub is not None: + unsubscribers.append(unsub) + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_warn( + 'ws.proactive', + f'订阅 proactive_message 失败: {e}', + ) + except Exception: + pass + except Exception as e: # noqa: BLE001 + try: + debug_logger.log_warn('ws.proactive', f'初始化 listener 失败: {e}') + except Exception: + pass + + try: + await websocket.send_json( + {"type": "ack", "conn_id": conn_id, "channel": CHANNEL} + ) + + while True: + try: + raw = await websocket.receive_text() + except WebSocketDisconnect: + break + except Exception: + break + # 简单 ack + try: + await websocket.send_json({"type": "ack", "received": True}) + except Exception: + break + except WebSocketDisconnect: + pass + except Exception as exc: # noqa: BLE001 + try: + debug_logger.log_error('ws.proactive', f'主循环异常: {exc}', exc) + except Exception: + pass + finally: + # 取消订阅 + for u in unsubscribers: + try: + u() + except Exception: + pass + try: + manager.disconnect(conn_id) + except Exception: + pass + try: + await websocket.close() + except Exception: # noqa: BLE001 + pass + + +__all__ = ["router", "CHANNEL"] diff --git a/src/web/frontend/.gitignore b/src/web/frontend/.gitignore new file mode 100644 index 0000000..9dda84f --- /dev/null +++ b/src/web/frontend/.gitignore @@ -0,0 +1,6 @@ +node_modules +dist +.vite +../static +*.local +.DS_Store diff --git a/src/web/frontend/index.html b/src/web/frontend/index.html new file mode 100644 index 0000000..8d5b14c --- /dev/null +++ b/src/web/frontend/index.html @@ -0,0 +1,13 @@ + + + + + + + Neo Agent + + +
+ + + diff --git a/src/web/frontend/package-lock.json b/src/web/frontend/package-lock.json new file mode 100644 index 0000000..e2446f9 --- /dev/null +++ b/src/web/frontend/package-lock.json @@ -0,0 +1,3860 @@ +{ + "name": "neo-agent-web-frontend", + "version": "3.0.0", + "lockfileVersion": 3, + "requires": true, + "packages": { + "": { + "name": "neo-agent-web-frontend", + "version": "3.0.0", + "dependencies": { + "@ant-design/icons": "^5.3.0", + "antd": "^5.15.0", + "axios": "^1.6.7", + "echarts": "^5.5.0", + "echarts-for-react": "^3.0.2", + "react": "^18.2.0", + "react-dom": "^18.2.0", + "react-router-dom": "^6.22.0", + "react-window": "^1.8.10", + "zustand": "^4.5.0" + }, + "devDependencies": { + "@types/react": "^18.2.55", + "@types/react-dom": "^18.2.19", + "@types/react-window": "^1.8.8", + "@vitejs/plugin-react": "^4.2.1", + "autoprefixer": "^10.4.17", + "postcss": "^8.4.35", + "tailwindcss": "^3.4.1", + "typescript": "^5.3.3", + "vite": "^5.1.0" + } + }, + "node_modules/@alloc/quick-lru": { + "version": "5.2.0", + "resolved": "https://registry.npmjs.org/@alloc/quick-lru/-/quick-lru-5.2.0.tgz", + "integrity": "sha512-UrcABB+4bUrFABwbluTIBErXwvbsU/V7TZWfmbgJfbkwiBuziS9gxdODUyuiecfdGQ85jglMW6juS3+z5TsKLw==", + "dev": true, + "engines": { + "node": ">=10" + }, + "funding": { + "url": "https://github.com/sponsors/sindresorhus" + } + }, + "node_modules/@ant-design/colors": { + "version": "7.2.1", + "resolved": "https://registry.npmjs.org/@ant-design/colors/-/colors-7.2.1.tgz", + "integrity": "sha512-lCHDcEzieu4GA3n8ELeZ5VQ8pKQAWcGGLRTQ50aQM2iqPpq2evTxER84jfdPvsPAtEcZ7m44NI45edFMo8oOYQ==", + "dependencies": { + "@ant-design/fast-color": "^2.0.6" + } + }, + "node_modules/@ant-design/cssinjs": { + "version": "1.24.0", + "resolved": "https://registry.npmjs.org/@ant-design/cssinjs/-/cssinjs-1.24.0.tgz", + "integrity": "sha512-K4cYrJBsgvL+IoozUXYjbT6LHHNt+19a9zkvpBPxLjFHas1UpPM2A5MlhROb0BT8N8WoavM5VsP9MeSeNK/3mg==", + "dependencies": { + "@babel/runtime": "^7.11.1", + "@emotion/hash": "^0.8.0", + "@emotion/unitless": "^0.7.5", + "classnames": "^2.3.1", + "csstype": "^3.1.3", + "rc-util": "^5.35.0", + "stylis": "^4.3.4" + }, + "peerDependencies": { + "react": ">=16.0.0", + "react-dom": ">=16.0.0" + } + }, + "node_modules/@ant-design/cssinjs-utils": { + "version": "1.1.3", + "resolved": "https://registry.npmjs.org/@ant-design/cssinjs-utils/-/cssinjs-utils-1.1.3.tgz", + "integrity": "sha512-nOoQMLW1l+xR1Co8NFVYiP8pZp3VjIIzqV6D6ShYF2ljtdwWJn5WSsH+7kvCktXL/yhEtWURKOfH5Xz/gzlwsg==", + "dependencies": { + "@ant-design/cssinjs": "^1.21.0", + "@babel/runtime": "^7.23.2", + "rc-util": "^5.38.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@ant-design/fast-color": { + "version": "2.0.6", + "resolved": "https://registry.npmjs.org/@ant-design/fast-color/-/fast-color-2.0.6.tgz", + "integrity": "sha512-y2217gk4NqL35giHl72o6Zzqji9O7vHh9YmhUVkPtAOpoTCH4uWxo/pr4VE8t0+ChEPs0qo4eJRC5Q1eXWo3vA==", + "dependencies": { + "@babel/runtime": "^7.24.7" + }, + "engines": { + "node": ">=8.x" + } + }, + "node_modules/@ant-design/icons": { + "version": "5.6.1", + "resolved": "https://registry.npmjs.org/@ant-design/icons/-/icons-5.6.1.tgz", + "integrity": "sha512-0/xS39c91WjPAZOWsvi1//zjx6kAp4kxWwctR6kuU6p133w8RU0D2dSCvZC19uQyharg/sAvYxGYWl01BbZZfg==", + "dependencies": { + "@ant-design/colors": "^7.0.0", + "@ant-design/icons-svg": "^4.4.0", + "@babel/runtime": "^7.24.8", + "classnames": "^2.2.6", + "rc-util": "^5.31.1" + }, + "engines": { + "node": ">=8" + }, + "peerDependencies": { + "react": ">=16.0.0", + "react-dom": ">=16.0.0" + } + }, + "node_modules/@ant-design/icons-svg": { + "version": "4.5.0", + "resolved": "https://registry.npmjs.org/@ant-design/icons-svg/-/icons-svg-4.5.0.tgz", + "integrity": "sha512-1BTUFyKPTBZ53MuTP8s0k5SFEXL7o3VHEOwLgzaoWKwnBeqIcqUtVshc4SKzhI6uACfqhJqBwBUE9FsWR3uULA==" + }, + "node_modules/@ant-design/react-slick": { + "version": "1.1.2", + "resolved": "https://registry.npmjs.org/@ant-design/react-slick/-/react-slick-1.1.2.tgz", + "integrity": "sha512-EzlvzE6xQUBrZuuhSAFTdsr4P2bBBHGZwKFemEfq8gIGyIQCxalYfZW/T2ORbtQx5rU69o+WycP3exY/7T1hGA==", + "dependencies": { + "@babel/runtime": "^7.10.4", + "classnames": "^2.2.5", + "json2mq": "^0.2.0", + "resize-observer-polyfill": "^1.5.1", + "throttle-debounce": "^5.0.0" + }, + "peerDependencies": { + "react": ">=16.9.0" + } + }, + "node_modules/@babel/code-frame": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/code-frame/-/code-frame-7.29.7.tgz", + "integrity": "sha512-Aup7aUOfpbAUg2ROOJN6Iw5f9DMBlzu0mIkm/malLQFN/YQgO48wCj0Kxa3sEHJvPVFg7siR+qRInwXd2qhQKw==", + "dev": true, + "dependencies": { + "@babel/helper-validator-identifier": "^7.29.7", + "js-tokens": "^4.0.0", + "picocolors": "^1.1.1" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/compat-data": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/compat-data/-/compat-data-7.29.7.tgz", + "integrity": "sha512-locTkQyKvwIEgBzVrn8693ebc97F2U8ZHjbXwDXJ5Fn2TCpNwTlKcaKLkdHop5c/icOFE7qt7Q9JC5hnKNa6Gg==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/core": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/core/-/core-7.29.7.tgz", + "integrity": "sha512-RgHBCvtjbOK2gXSNBNIkNoEc9qoVEtau3hj8gEqKQuL3HZAibKarWFEI3Lfm6EYKkLalOh8eSrj9b+ch9H/VBA==", + "dev": true, + "dependencies": { + "@babel/code-frame": "^7.29.7", + "@babel/generator": "^7.29.7", + "@babel/helper-compilation-targets": "^7.29.7", + "@babel/helper-module-transforms": "^7.29.7", + "@babel/helpers": "^7.29.7", + "@babel/parser": "^7.29.7", + "@babel/template": "^7.29.7", + "@babel/traverse": "^7.29.7", + "@babel/types": "^7.29.7", + "@jridgewell/remapping": "^2.3.5", + "convert-source-map": "^2.0.0", + "debug": "^4.1.0", + "gensync": "^1.0.0-beta.2", + "json5": "^2.2.3", + "semver": "^6.3.1" + }, + "engines": { + "node": ">=6.9.0" + }, + "funding": { + "type": "opencollective", + "url": "https://opencollective.com/babel" + } + }, + "node_modules/@babel/generator": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/generator/-/generator-7.29.7.tgz", + "integrity": "sha512-DkXD5OJQaAQIdZ1bt3UZdEnHAn9Imd3IVBdX03UFe+ony9Ojw5pzr9YVKGDY1jt+Gcn/FnGkNf8r+Vj5NOJWtQ==", + "dev": true, + "dependencies": { + "@babel/parser": "^7.29.7", + "@babel/types": "^7.29.7", + "@jridgewell/gen-mapping": "^0.3.12", + "@jridgewell/trace-mapping": "^0.3.28", + "jsesc": "^3.0.2" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-compilation-targets": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-compilation-targets/-/helper-compilation-targets-7.29.7.tgz", + "integrity": "sha512-wem6WaBj4NaVYVdNhLPPVacES6ZJ+KBBfSkTMD3YZxbP3rm3Di85tJU5ljaUNhaOynt+Aj0xruhYuzQBt8n71g==", + "dev": true, + "dependencies": { + "@babel/compat-data": "^7.29.7", + "@babel/helper-validator-option": "^7.29.7", + "browserslist": "^4.24.0", + "lru-cache": "^5.1.1", + "semver": "^6.3.1" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-globals": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-globals/-/helper-globals-7.29.7.tgz", + "integrity": "sha512-3nQVUAtvkKH9zahfWgw96Jc/uFOmjACE1kQz82E2lqWmHBgjzbNlsC22nuQTfahmWeQtTq5nQ/4Nnd2A1wj4zA==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-module-imports": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-module-imports/-/helper-module-imports-7.29.7.tgz", + "integrity": "sha512-ejHwrQQYcm9xnTivShn2IDOlIzInN34AXskvq9QicvCtEzq1Vzclu/tKF8Jq1Cg8JG2GL6/EmjgsCT7lXepE3g==", + "dev": true, + "dependencies": { + "@babel/traverse": "^7.29.7", + "@babel/types": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-module-transforms": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-module-transforms/-/helper-module-transforms-7.29.7.tgz", + "integrity": "sha512-UPUVSyXbOh627KiCIGQSgwWzGeBKLkaJ9PJEdrngIwMSzxLR4jS4+f1f1jb7VzBbg8nFLaYotvVPFCTqdrmTAg==", + "dev": true, + "dependencies": { + "@babel/helper-module-imports": "^7.29.7", + "@babel/helper-validator-identifier": "^7.29.7", + "@babel/traverse": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + }, + "peerDependencies": { + "@babel/core": "^7.0.0" + } + }, + "node_modules/@babel/helper-plugin-utils": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-plugin-utils/-/helper-plugin-utils-7.29.7.tgz", + "integrity": "sha512-G7sHYigPY17oO5SYWnfD/0MTBwVR781S/JI643e/JhUYgVgWE/61SoW3NH9KWUKyKq5LVh3npif99Wkt6j86Jw==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-string-parser": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-string-parser/-/helper-string-parser-7.29.7.tgz", + "integrity": "sha512-Pb5ijPrZ89GDH8223L4UP8i6QApWxs04RbPQJTeWDV0/keR2E36MeKnyr6LYmUUvqRRI+Iv87SuF1W6ErINzYw==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-validator-identifier": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-validator-identifier/-/helper-validator-identifier-7.29.7.tgz", + "integrity": "sha512-qehxGkRj55h/ff8EMaJ+cYhyaKlHIxqYDn682wQD7RNp9UujOQsHog2uS0r2vzr4pW+sXf90NeeayjcNaX3fFg==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helper-validator-option": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helper-validator-option/-/helper-validator-option-7.29.7.tgz", + "integrity": "sha512-N9ZErrD+yW5geCDtBqnOoxmR8+tNKiGuxKlDpuJxfsqpa2dFcexaziGAE/qoHLiDDreVNMupxGmSoNlyvsA3gw==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/helpers": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/helpers/-/helpers-7.29.7.tgz", + "integrity": "sha512-1k2lAGRMfHTcwuNYcCNUmaUffmQv8KWMfh2iJUUeRlwlwH4FdNG7mfPI10NPfLHJFThE4Tyr4mv7kTNZOiPuBg==", + "dev": true, + "dependencies": { + "@babel/template": "^7.29.7", + "@babel/types": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/parser": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/parser/-/parser-7.29.7.tgz", + "integrity": "sha512-hnORnjP/1P/zFEndoeX+n+t1RwWRJiJpM/jO7FW32Kn9r5+sJB2JWOdYo4L6k78j15eCwY3Gm/7364B1EMwtNg==", + "dev": true, + "dependencies": { + "@babel/types": "^7.29.7" + }, + "bin": { + "parser": "bin/babel-parser.js" + }, + "engines": { + "node": ">=6.0.0" + } + }, + "node_modules/@babel/plugin-transform-react-jsx-self": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/plugin-transform-react-jsx-self/-/plugin-transform-react-jsx-self-7.29.7.tgz", + "integrity": "sha512-TL0hMc9xzy86VD31nUiwzd5otRAcyEPcsegCxolO0PvcXuH1v0kECe/UIznYFihpkvU5wg/jk4v0TTEFfm53fw==", + "dev": true, + "dependencies": { + "@babel/helper-plugin-utils": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + }, + "peerDependencies": { + "@babel/core": "^7.0.0-0" + } + }, + "node_modules/@babel/plugin-transform-react-jsx-source": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/plugin-transform-react-jsx-source/-/plugin-transform-react-jsx-source-7.29.7.tgz", + "integrity": "sha512-06IyK09H3wi4cGbhDBwp5gUGo0IKtnYa8tyTiephirPCK6fbobVGiXMMI5zLQ4aKEYP3wZ3ArU44o+8KMrSG/Q==", + "dev": true, + "dependencies": { + "@babel/helper-plugin-utils": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + }, + "peerDependencies": { + "@babel/core": "^7.0.0-0" + } + }, + "node_modules/@babel/runtime": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/runtime/-/runtime-7.29.7.tgz", + "integrity": "sha512-Nq8OhGWiZIZGV6hLHoyAKLLcJihP/xFeBMGJoUrxTX2psI8dCifzLhZISFb+VWS3wFMRDmCGw5R+dOySCqPLhw==", + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/template": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/template/-/template-7.29.7.tgz", + "integrity": "sha512-puq+Gf35oI24FeN11LkoUQFqv9uwNeWpxXZi/Ji3rRIoKAzKnxRaZ+Gkj0vKS9ZCiTESfng1N9LyOyXvo+m+Gg==", + "dev": true, + "dependencies": { + "@babel/code-frame": "^7.29.7", + "@babel/parser": "^7.29.7", + "@babel/types": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/traverse": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/traverse/-/traverse-7.29.7.tgz", + "integrity": "sha512-EhlfNQtZ+NK22w5BM61ciuiq1m58ed33Wr1Xan//ZRTy6hgjnwyCffRYwzsGXdASJSUJ1guZILsErh1eQcl+zw==", + "dev": true, + "dependencies": { + "@babel/code-frame": "^7.29.7", + "@babel/generator": "^7.29.7", + "@babel/helper-globals": "^7.29.7", + "@babel/parser": "^7.29.7", + "@babel/template": "^7.29.7", + "@babel/types": "^7.29.7", + "debug": "^4.3.1" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@babel/types": { + "version": "7.29.7", + "resolved": "https://registry.npmjs.org/@babel/types/-/types-7.29.7.tgz", + "integrity": "sha512-4zBIxpPzowiZpusoFkyGVwakdRJUyuH5PxQ/PrqghfdFWWasvnCdPfQXHrenDai+gyLARulZjZowCOj6fjT4pA==", + "dev": true, + "dependencies": { + "@babel/helper-string-parser": "^7.29.7", + "@babel/helper-validator-identifier": "^7.29.7" + }, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/@emotion/hash": { + "version": "0.8.0", + "resolved": "https://registry.npmjs.org/@emotion/hash/-/hash-0.8.0.tgz", + "integrity": "sha512-kBJtf7PH6aWwZ6fka3zQ0p6SBYzx4fl1LoZXE2RrnYST9Xljm7WfKJrU4g/Xr3Beg72MLrp1AWNUmuYJTL7Cow==" + }, + "node_modules/@emotion/unitless": { + "version": "0.7.5", + "resolved": "https://registry.npmjs.org/@emotion/unitless/-/unitless-0.7.5.tgz", + "integrity": "sha512-OWORNpfjMsSSUBVrRBVGECkhWcULOAJz9ZW8uK9qgxD+87M7jHRcvh/A96XXNhXTLmKcoYSQtBEX7lHMO7YRwg==" + }, + "node_modules/@esbuild/aix-ppc64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/aix-ppc64/-/aix-ppc64-0.21.5.tgz", + "integrity": "sha512-1SDgH6ZSPTlggy1yI6+Dbkiz8xzpHJEVAlF/AM1tHPLsf5STom9rwtjE4hKAF20FfXXNTFqEYXyJNWh1GiZedQ==", + "cpu": [ + "ppc64" + ], + "dev": true, + "optional": true, + "os": [ + "aix" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/android-arm": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/android-arm/-/android-arm-0.21.5.tgz", + "integrity": "sha512-vCPvzSjpPHEi1siZdlvAlsPxXl7WbOVUBBAowWug4rJHb68Ox8KualB+1ocNvT5fjv6wpkX6o/iEpbDrf68zcg==", + "cpu": [ + "arm" + ], + "dev": true, + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/android-arm64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/android-arm64/-/android-arm64-0.21.5.tgz", + "integrity": "sha512-c0uX9VAUBQ7dTDCjq+wdyGLowMdtR/GoC2U5IYk/7D1H1JYC0qseD7+11iMP2mRLN9RcCMRcjC4YMclCzGwS/A==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/android-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/android-x64/-/android-x64-0.21.5.tgz", + "integrity": "sha512-D7aPRUUNHRBwHxzxRvp856rjUHRFW1SdQATKXH2hqA0kAZb1hKmi02OpYRacl0TxIGz/ZmXWlbZgjwWYaCakTA==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/darwin-arm64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/darwin-arm64/-/darwin-arm64-0.21.5.tgz", + "integrity": "sha512-DwqXqZyuk5AiWWf3UfLiRDJ5EDd49zg6O9wclZ7kUMv2WRFr4HKjXp/5t8JZ11QbQfUS6/cRCKGwYhtNAY88kQ==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "darwin" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/darwin-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/darwin-x64/-/darwin-x64-0.21.5.tgz", + "integrity": "sha512-se/JjF8NlmKVG4kNIuyWMV/22ZaerB+qaSi5MdrXtd6R08kvs2qCN4C09miupktDitvh8jRFflwGFBQcxZRjbw==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "darwin" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/freebsd-arm64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/freebsd-arm64/-/freebsd-arm64-0.21.5.tgz", + "integrity": "sha512-5JcRxxRDUJLX8JXp/wcBCy3pENnCgBR9bN6JsY4OmhfUtIHe3ZW0mawA7+RDAcMLrMIZaf03NlQiX9DGyB8h4g==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "freebsd" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/freebsd-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/freebsd-x64/-/freebsd-x64-0.21.5.tgz", + "integrity": "sha512-J95kNBj1zkbMXtHVH29bBriQygMXqoVQOQYA+ISs0/2l3T9/kj42ow2mpqerRBxDJnmkUDCaQT/dfNXWX/ZZCQ==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "freebsd" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-arm": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-arm/-/linux-arm-0.21.5.tgz", + "integrity": "sha512-bPb5AHZtbeNGjCKVZ9UGqGwo8EUu4cLq68E95A53KlxAPRmUyYv2D6F0uUI65XisGOL1hBP5mTronbgo+0bFcA==", + "cpu": [ + "arm" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-arm64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-arm64/-/linux-arm64-0.21.5.tgz", + "integrity": "sha512-ibKvmyYzKsBeX8d8I7MH/TMfWDXBF3db4qM6sy+7re0YXya+K1cem3on9XgdT2EQGMu4hQyZhan7TeQ8XkGp4Q==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-ia32": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-ia32/-/linux-ia32-0.21.5.tgz", + "integrity": "sha512-YvjXDqLRqPDl2dvRODYmmhz4rPeVKYvppfGYKSNGdyZkA01046pLWyRKKI3ax8fbJoK5QbxblURkwK/MWY18Tg==", + "cpu": [ + "ia32" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-loong64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-loong64/-/linux-loong64-0.21.5.tgz", + "integrity": "sha512-uHf1BmMG8qEvzdrzAqg2SIG/02+4/DHB6a9Kbya0XDvwDEKCoC8ZRWI5JJvNdUjtciBGFQ5PuBlpEOXQj+JQSg==", + "cpu": [ + "loong64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-mips64el": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-mips64el/-/linux-mips64el-0.21.5.tgz", + "integrity": "sha512-IajOmO+KJK23bj52dFSNCMsz1QP1DqM6cwLUv3W1QwyxkyIWecfafnI555fvSGqEKwjMXVLokcV5ygHW5b3Jbg==", + "cpu": [ + "mips64el" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-ppc64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-ppc64/-/linux-ppc64-0.21.5.tgz", + "integrity": "sha512-1hHV/Z4OEfMwpLO8rp7CvlhBDnjsC3CttJXIhBi+5Aj5r+MBvy4egg7wCbe//hSsT+RvDAG7s81tAvpL2XAE4w==", + "cpu": [ + "ppc64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-riscv64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-riscv64/-/linux-riscv64-0.21.5.tgz", + "integrity": "sha512-2HdXDMd9GMgTGrPWnJzP2ALSokE/0O5HhTUvWIbD3YdjME8JwvSCnNGBnTThKGEB91OZhzrJ4qIIxk/SBmyDDA==", + "cpu": [ + "riscv64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-s390x": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-s390x/-/linux-s390x-0.21.5.tgz", + "integrity": "sha512-zus5sxzqBJD3eXxwvjN1yQkRepANgxE9lgOW2qLnmr8ikMTphkjgXu1HR01K4FJg8h1kEEDAqDcZQtbrRnB41A==", + "cpu": [ + "s390x" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/linux-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/linux-x64/-/linux-x64-0.21.5.tgz", + "integrity": "sha512-1rYdTpyv03iycF1+BhzrzQJCdOuAOtaqHTWJZCWvijKD2N5Xu0TtVC8/+1faWqcP9iBCWOmjmhoH94dH82BxPQ==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/netbsd-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/netbsd-x64/-/netbsd-x64-0.21.5.tgz", + "integrity": "sha512-Woi2MXzXjMULccIwMnLciyZH4nCIMpWQAs049KEeMvOcNADVxo0UBIQPfSmxB3CWKedngg7sWZdLvLczpe0tLg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "netbsd" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/openbsd-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/openbsd-x64/-/openbsd-x64-0.21.5.tgz", + "integrity": "sha512-HLNNw99xsvx12lFBUwoT8EVCsSvRNDVxNpjZ7bPn947b8gJPzeHWyNVhFsaerc0n3TsbOINvRP2byTZ5LKezow==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "openbsd" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/sunos-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/sunos-x64/-/sunos-x64-0.21.5.tgz", + "integrity": "sha512-6+gjmFpfy0BHU5Tpptkuh8+uw3mnrvgs+dSPQXQOv3ekbordwnzTVEb4qnIvQcYXq6gzkyTnoZ9dZG+D4garKg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "sunos" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/win32-arm64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/win32-arm64/-/win32-arm64-0.21.5.tgz", + "integrity": "sha512-Z0gOTd75VvXqyq7nsl93zwahcTROgqvuAcYDUr+vOv8uHhNSKROyU961kgtCD1e95IqPKSQKH7tBTslnS3tA8A==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/win32-ia32": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/win32-ia32/-/win32-ia32-0.21.5.tgz", + "integrity": "sha512-SWXFF1CL2RVNMaVs+BBClwtfZSvDgtL//G/smwAc5oVK/UPu2Gu9tIaRgFmYFFKrmg3SyAjSrElf0TiJ1v8fYA==", + "cpu": [ + "ia32" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@esbuild/win32-x64": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/@esbuild/win32-x64/-/win32-x64-0.21.5.tgz", + "integrity": "sha512-tQd/1efJuzPC6rCFwEvLtci/xNFcTZknmXs98FYDfGE4wP9ClFV98nyKrzJKVPMhdDnjzLhdUyMX4PsQAPjwIw==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=12" + } + }, + "node_modules/@jridgewell/gen-mapping": { + "version": "0.3.13", + "resolved": "https://registry.npmjs.org/@jridgewell/gen-mapping/-/gen-mapping-0.3.13.tgz", + "integrity": "sha512-2kkt/7niJ6MgEPxF0bYdQ6etZaA+fQvDcLKckhy1yIQOzaoKjBBjSj63/aLVjYE3qhRt5dvM+uUyfCg6UKCBbA==", + "dev": true, + "dependencies": { + "@jridgewell/sourcemap-codec": "^1.5.0", + "@jridgewell/trace-mapping": "^0.3.24" + } + }, + "node_modules/@jridgewell/remapping": { + "version": "2.3.5", + "resolved": "https://registry.npmjs.org/@jridgewell/remapping/-/remapping-2.3.5.tgz", + "integrity": "sha512-LI9u/+laYG4Ds1TDKSJW2YPrIlcVYOwi2fUC6xB43lueCjgxV4lffOCZCtYFiH6TNOX+tQKXx97T4IKHbhyHEQ==", + "dev": true, + "dependencies": { + "@jridgewell/gen-mapping": "^0.3.5", + "@jridgewell/trace-mapping": "^0.3.24" + } + }, + "node_modules/@jridgewell/resolve-uri": { + "version": "3.1.2", + "resolved": "https://registry.npmjs.org/@jridgewell/resolve-uri/-/resolve-uri-3.1.2.tgz", + "integrity": "sha512-bRISgCIjP20/tbWSPWMEi54QVPRZExkuD9lJL+UIxUKtwVJA8wW1Trb1jMs1RFXo1CBTNZ/5hpC9QvmKWdopKw==", + "dev": true, + "engines": { + "node": ">=6.0.0" + } + }, + "node_modules/@jridgewell/sourcemap-codec": { + "version": "1.5.5", + "resolved": "https://registry.npmjs.org/@jridgewell/sourcemap-codec/-/sourcemap-codec-1.5.5.tgz", + "integrity": "sha512-cYQ9310grqxueWbl+WuIUIaiUaDcj7WOq5fVhEljNVgRfOUhY9fy2zTvfoqWsnebh8Sl70VScFbICvJnLKB0Og==", + "dev": true + }, + "node_modules/@jridgewell/trace-mapping": { + "version": "0.3.31", + "resolved": "https://registry.npmjs.org/@jridgewell/trace-mapping/-/trace-mapping-0.3.31.tgz", + "integrity": "sha512-zzNR+SdQSDJzc8joaeP8QQoCQr8NuYx2dIIytl1QeBEZHJ9uW6hebsrYgbz8hJwUQao3TWCMtmfV8Nu1twOLAw==", + "dev": true, + "dependencies": { + "@jridgewell/resolve-uri": "^3.1.0", + "@jridgewell/sourcemap-codec": "^1.4.14" + } + }, + "node_modules/@nodelib/fs.scandir": { + "version": "2.1.5", + "resolved": "https://registry.npmjs.org/@nodelib/fs.scandir/-/fs.scandir-2.1.5.tgz", + "integrity": "sha512-vq24Bq3ym5HEQm2NKCr3yXDwjc7vTsEThRDnkp2DK9p1uqLR+DHurm/NOTo0KG7HYHU7eppKZj3MyqYuMBf62g==", + "dev": true, + "dependencies": { + "@nodelib/fs.stat": "2.0.5", + "run-parallel": "^1.1.9" + }, + "engines": { + "node": ">= 8" + } + }, + "node_modules/@nodelib/fs.stat": { + "version": "2.0.5", + "resolved": "https://registry.npmjs.org/@nodelib/fs.stat/-/fs.stat-2.0.5.tgz", + "integrity": "sha512-RkhPPp2zrqDAQA/2jNhnztcPAlv64XdhIp7a7454A5ovI7Bukxgt7MX7udwAu3zg1DcpPU0rz3VV1SeaqvY4+A==", + "dev": true, + "engines": { + "node": ">= 8" + } + }, + "node_modules/@nodelib/fs.walk": { + "version": "1.2.8", + "resolved": "https://registry.npmjs.org/@nodelib/fs.walk/-/fs.walk-1.2.8.tgz", + "integrity": "sha512-oGB+UxlgWcgQkgwo8GcEGwemoTFt3FIO9ababBmaGwXIoBKZ+GTy0pP185beGg7Llih/NSHSV2XAs1lnznocSg==", + "dev": true, + "dependencies": { + "@nodelib/fs.scandir": "2.1.5", + "fastq": "^1.6.0" + }, + "engines": { + "node": ">= 8" + } + }, + "node_modules/@rc-component/async-validator": { + "version": "5.1.2", + "resolved": "https://registry.npmjs.org/@rc-component/async-validator/-/async-validator-5.1.2.tgz", + "integrity": "sha512-WYbrZSjzznU1ekD0qFq2qRxt309VoS61MTG5npnFQlKYcoy9IzU8T+ZCIhq5bGAXRbXysABFWTspicMfmWFwow==", + "dependencies": { + "@babel/runtime": "^7.24.4" + }, + "engines": { + "node": ">=14.x" + } + }, + "node_modules/@rc-component/color-picker": { + "version": "2.0.1", + "resolved": "https://registry.npmjs.org/@rc-component/color-picker/-/color-picker-2.0.1.tgz", + "integrity": "sha512-WcZYwAThV/b2GISQ8F+7650r5ZZJ043E57aVBFkQ+kSY4C6wdofXgB0hBx+GPGpIU0Z81eETNoDUJMr7oy/P8Q==", + "dependencies": { + "@ant-design/fast-color": "^2.0.6", + "@babel/runtime": "^7.23.6", + "classnames": "^2.2.6", + "rc-util": "^5.38.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/context": { + "version": "1.4.0", + "resolved": "https://registry.npmjs.org/@rc-component/context/-/context-1.4.0.tgz", + "integrity": "sha512-kFcNxg9oLRMoL3qki0OMxK+7g5mypjgaaJp/pkOis/6rVxma9nJBF/8kCIuTYHUQNr0ii7MxqE33wirPZLJQ2w==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "rc-util": "^5.27.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/mini-decimal": { + "version": "1.1.4", + "resolved": "https://registry.npmjs.org/@rc-component/mini-decimal/-/mini-decimal-1.1.4.tgz", + "integrity": "sha512-xiuXcaCwyOWpD8a8scdExFl+bntNphAW8XeenL1ig2en0AAZY0Pcp4pC0dI22qJ+NvxKn9RoNIoRdqYU3BLH4w==", + "dependencies": { + "@babel/runtime": "^7.18.0" + }, + "engines": { + "node": ">=8.x" + } + }, + "node_modules/@rc-component/mutate-observer": { + "version": "1.1.0", + "resolved": "https://registry.npmjs.org/@rc-component/mutate-observer/-/mutate-observer-1.1.0.tgz", + "integrity": "sha512-QjrOsDXQusNwGZPf4/qRQasg7UFEj06XiCJ8iuiq/Io7CrHrgVi6Uuetw60WAMG1799v+aM8kyc+1L/GBbHSlw==", + "dependencies": { + "@babel/runtime": "^7.18.0", + "classnames": "^2.3.2", + "rc-util": "^5.24.4" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/portal": { + "version": "1.1.2", + "resolved": "https://registry.npmjs.org/@rc-component/portal/-/portal-1.1.2.tgz", + "integrity": "sha512-6f813C0IsasTZms08kfA8kPAGxbbkYToa8ALaiDIGGECU4i9hj8Plgbx0sNJDrey3EtHO30hmdaxtT0138xZcg==", + "dependencies": { + "@babel/runtime": "^7.18.0", + "classnames": "^2.3.2", + "rc-util": "^5.24.4" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/qrcode": { + "version": "1.1.3", + "resolved": "https://registry.npmjs.org/@rc-component/qrcode/-/qrcode-1.1.3.tgz", + "integrity": "sha512-aGv6alnn4HbDEsURzKP+jv13rbi1VxmAYfBNZr5GKF1iohMNWy5tAVoJ1E3cOvzMB1kbUPvCXchM6zSFlRGPhA==", + "dependencies": { + "@babel/runtime": "^7.24.7" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/tour": { + "version": "1.15.1", + "resolved": "https://registry.npmjs.org/@rc-component/tour/-/tour-1.15.1.tgz", + "integrity": "sha512-Tr2t7J1DKZUpfJuDZWHxyxWpfmj8EZrqSgyMZ+BCdvKZ6r1UDsfU46M/iWAAFBy961Ssfom2kv5f3UcjIL2CmQ==", + "dependencies": { + "@babel/runtime": "^7.18.0", + "@rc-component/portal": "^1.0.0-9", + "@rc-component/trigger": "^2.0.0", + "classnames": "^2.3.2", + "rc-util": "^5.24.4" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@rc-component/trigger": { + "version": "2.3.1", + "resolved": "https://registry.npmjs.org/@rc-component/trigger/-/trigger-2.3.1.tgz", + "integrity": "sha512-ORENF39PeXTzM+gQEshuk460Z8N4+6DkjpxlpE7Q3gYy1iBpLrx0FOJz3h62ryrJZ/3zCAUIkT1Pb/8hHWpb3A==", + "dependencies": { + "@babel/runtime": "^7.23.2", + "@rc-component/portal": "^1.1.0", + "classnames": "^2.3.2", + "rc-motion": "^2.0.0", + "rc-resize-observer": "^1.3.1", + "rc-util": "^5.44.0" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/@remix-run/router": { + "version": "1.23.3", + "resolved": "https://registry.npmjs.org/@remix-run/router/-/router-1.23.3.tgz", + "integrity": "sha512-4An71tdz9X8+3sI4Qqqd2LWd9vS39J7sqd9EU4Scw7TJE/qB10Flv/UuqbPVgfQV9XoK8Np6jNquZitnZq5i+Q==", + "engines": { + "node": ">=14.0.0" + } + }, + "node_modules/@rolldown/pluginutils": { + "version": "1.0.0-beta.27", + "resolved": "https://registry.npmjs.org/@rolldown/pluginutils/-/pluginutils-1.0.0-beta.27.tgz", + "integrity": "sha512-+d0F4MKMCbeVUJwG96uQ4SgAznZNSq93I3V+9NHA4OpvqG8mRCpGdKmK8l/dl02h2CCDHwW2FqilnTyDcAnqjA==", + "dev": true + }, + "node_modules/@rollup/rollup-android-arm-eabi": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-android-arm-eabi/-/rollup-android-arm-eabi-4.62.2.tgz", + "integrity": "sha512-6o7ZLZK+BeenkZCFNDXqpbjw9bD6nuWonvS/lwQJp7NoVVxm6p3qE7qQ5jGuBjiFsgvqjD8mZAU5oWxTmbOeOg==", + "cpu": [ + "arm" + ], + "dev": true, + "optional": true, + "os": [ + "android" + ] + }, + "node_modules/@rollup/rollup-android-arm64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-android-arm64/-/rollup-android-arm64-4.62.2.tgz", + "integrity": "sha512-BaH7BllCACHoH1LguOU56UItGfUWjujlO65kS9LAodViaN4bwIKd7oeW/ZHJ/4ljr/7MIiENnNy3HJ0zXv8Zkw==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "android" + ] + }, + "node_modules/@rollup/rollup-darwin-arm64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-darwin-arm64/-/rollup-darwin-arm64-4.62.2.tgz", + "integrity": "sha512-v39RCCvj4He82I9sFmk+M1VZ0PLM9sfsLVikjfx2hYBNALhrrOR2D3JjQA6AhlaSOgcR+RzrKY7e1+bT6SUO/A==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "darwin" + ] + }, + "node_modules/@rollup/rollup-darwin-x64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-darwin-x64/-/rollup-darwin-x64-4.62.2.tgz", + "integrity": "sha512-yl0y2vq3S3lHeuXhEdss6TWfKW8vkujImO12tn4ZkG/4oghr09LvdYm2RElVjokTQiUvDUGXLGsYeLqUMCKpGA==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "darwin" + ] + }, + "node_modules/@rollup/rollup-freebsd-arm64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-freebsd-arm64/-/rollup-freebsd-arm64-4.62.2.tgz", + "integrity": "sha512-tT4pvt4qXD+vEoezupCWi+a1F0vvDiksiHc+PxRlYTOH1I6/X4id9jPxTP+Fg+545euaFT1jJVs4CEdHZAU1vw==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "freebsd" + ] + }, + "node_modules/@rollup/rollup-freebsd-x64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-freebsd-x64/-/rollup-freebsd-x64-4.62.2.tgz", + "integrity": "sha512-6nU5F2wCW+qvCBhTn1pdIU3bzsIoF7EUwsCDRxilWGprQR6yd508YnH9+OKFCwpfS8pjZqDUmnCAr7exax0XCg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "freebsd" + ] + }, + "node_modules/@rollup/rollup-linux-arm-gnueabihf": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-arm-gnueabihf/-/rollup-linux-arm-gnueabihf-4.62.2.tgz", + "integrity": "sha512-n1GJHPOvpIfhi3TmrCeh6S6URt9BFCt0KQE3qvexyGCTAKpR4Lg+eWvNZEqu7epxwus/8ElT3hacYEucm49SZg==", + "cpu": [ + "arm" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-arm-musleabihf": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-arm-musleabihf/-/rollup-linux-arm-musleabihf-4.62.2.tgz", + "integrity": "sha512-JqgflS8wEB+UXV/vS1RpRbifGBeN4D5lz8D8oOFbFZw4vedvdOgCFAjfBmIMdW3yL10XpQQ0Ambepw6MXrhOnA==", + "cpu": [ + "arm" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-arm64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-arm64-gnu/-/rollup-linux-arm64-gnu-4.62.2.tgz", + "integrity": "sha512-wnFJkogWvN4jm/hQRF2UBaeUmk20j5+DmHvoyWii2b8HJDyvz1MF2OU/6ynXt2KR63rbZLWkFpoytpdc/yBuSA==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-arm64-musl": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-arm64-musl/-/rollup-linux-arm64-musl-4.62.2.tgz", + "integrity": "sha512-HVu2bp0zhvJ8xHEV9+UUs7S90VadmBSY3LcIMvozbPo4AuMGDWlz3ymHLHZPX4hR67TKTt8Qp5PJ5RBg/i+RMQ==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-loong64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-loong64-gnu/-/rollup-linux-loong64-gnu-4.62.2.tgz", + "integrity": "sha512-mQqqAV8QaoSgr9I2fKDLY2BAVvmKjWoGiu/cSYQonsLvtqwEn1E4QYfnCOcp5zoEqNhsDYin1s6jx/VJmrxlZg==", + "cpu": [ + "loong64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-loong64-musl": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-loong64-musl/-/rollup-linux-loong64-musl-4.62.2.tgz", + "integrity": "sha512-IxKLoxCQ2IWi6bT2akyDUBGsOImDKB+sPp4EsTmwFQ/fMwpCKm8uLSSgP/Kx/QYUgKis6SEZ5/Nlhup0DIA0PQ==", + "cpu": [ + "loong64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-ppc64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-ppc64-gnu/-/rollup-linux-ppc64-gnu-4.62.2.tgz", + "integrity": "sha512-Mk5ha2RQSgyFfmYYLkBpPnUk8D8FriBxesO1u9O75X0mHgXL1UQcH5Itl2lurWL2tj0RxV9b9tJgipac0hRY9A==", + "cpu": [ + "ppc64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-ppc64-musl": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-ppc64-musl/-/rollup-linux-ppc64-musl-4.62.2.tgz", + "integrity": "sha512-CjvEnqJL/0/TQ3TXX3OPIJ/kmBellrWd4heXUmHeJlTnmwjKpSJzoehLaL6Xk0ZnMHBu9dZuFADNOrtjF4v+2w==", + "cpu": [ + "ppc64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-riscv64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-riscv64-gnu/-/rollup-linux-riscv64-gnu-4.62.2.tgz", + "integrity": "sha512-1SiZbzwdkaDURsew/tSOrooKiYy7EQGT6m8ufavAi9NEyQb/6VuIxFXAL1fqa4iZe3g4NbNk4P7J32z2tw5Mgg==", + "cpu": [ + "riscv64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-riscv64-musl": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-riscv64-musl/-/rollup-linux-riscv64-musl-4.62.2.tgz", + "integrity": "sha512-nQts12zJ3NQRoE6uYljOH89v7szzLDvG2JD/vsX+vGXU8w/At1GowTZ5/7qeFQ8m7L55rpR8Okugnuo5bgjy2Q==", + "cpu": [ + "riscv64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-s390x-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-s390x-gnu/-/rollup-linux-s390x-gnu-4.62.2.tgz", + "integrity": "sha512-E9/ll019jhPIJgpzfZoIkBGhcz+kKNgVWYRY0zr9srBdPPFVpvOKW8VaJKUbeK+eZXyQF9ltME+Kk6affeaPgg==", + "cpu": [ + "s390x" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-x64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-x64-gnu/-/rollup-linux-x64-gnu-4.62.2.tgz", + "integrity": "sha512-5BqxR/pshjey51iliyzTD5Xi3EN0aLmQ2lZ3lvefVV9c82BvrLo2/6OT55iifpWBufs6kdwWbuOKS841DrmK9A==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-linux-x64-musl": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-linux-x64-musl/-/rollup-linux-x64-musl-4.62.2.tgz", + "integrity": "sha512-uNN83XxQrRAh/w0/pmAfibcwyb6YWt4gP+dpnQKPVJshAloQ785ii8CT8ZCIxkGg9opVsvAlGhFitSm6D1Jjpg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "linux" + ] + }, + "node_modules/@rollup/rollup-openbsd-x64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-openbsd-x64/-/rollup-openbsd-x64-4.62.2.tgz", + "integrity": "sha512-srjEIxSH3LRnJN6THczDHWQplqEMFiAJrTab0msUryh9kwNpkICf3Ea6q6MN/2cZwRFUNx5w+h6Hpi4QuHS6Zg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "openbsd" + ] + }, + "node_modules/@rollup/rollup-openharmony-arm64": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-openharmony-arm64/-/rollup-openharmony-arm64-4.62.2.tgz", + "integrity": "sha512-8hOJnxgbyObnCm5AlRA3A931xX19xq80RjVTKgJOvEKWqJruP/Uf12IbAOaDjjEXYRewwHLfmF0YRIdK3OwKWA==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "openharmony" + ] + }, + "node_modules/@rollup/rollup-win32-arm64-msvc": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-win32-arm64-msvc/-/rollup-win32-arm64-msvc-4.62.2.tgz", + "integrity": "sha512-mmF4AY1i0hG/bLWUctUq59gtmgaSIRa3cu/A3JFRp/sCNEme2bgDEiDS22P9FbnJB8NJNF4jPJiSP5RHQpUTDg==", + "cpu": [ + "arm64" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ] + }, + "node_modules/@rollup/rollup-win32-ia32-msvc": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-win32-ia32-msvc/-/rollup-win32-ia32-msvc-4.62.2.tgz", + "integrity": "sha512-DZgkknc6jhHrk46V25vbAM0zZkyP0nSDkJB8/dRkLTxv470dOmWDqGoEJl/9A0dFfS7yE3REOwNDxpHwSLSt0Q==", + "cpu": [ + "ia32" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ] + }, + "node_modules/@rollup/rollup-win32-x64-gnu": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-win32-x64-gnu/-/rollup-win32-x64-gnu-4.62.2.tgz", + "integrity": "sha512-T6xr6ucWSFto+VGajA8YH26LdpHRuP4YLHEKAtCWvJDOlnmWcDZVCI2Jmjr+IFHDlt2zRaTAKE4tfjTaWLgJBg==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ] + }, + "node_modules/@rollup/rollup-win32-x64-msvc": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/@rollup/rollup-win32-x64-msvc/-/rollup-win32-x64-msvc-4.62.2.tgz", + "integrity": "sha512-BfzEnDJOt9T8M989/lA37EcJgat01wLRnoi5dQf3QzOH7jzpqTAzdDbVfRljVr5r+jzKqpbHeyOfAaXxAd0PAA==", + "cpu": [ + "x64" + ], + "dev": true, + "optional": true, + "os": [ + "win32" + ] + }, + "node_modules/@types/babel__core": { + "version": "7.20.5", + "resolved": "https://registry.npmjs.org/@types/babel__core/-/babel__core-7.20.5.tgz", + "integrity": "sha512-qoQprZvz5wQFJwMDqeseRXWv3rqMvhgpbXFfVyWhbx9X47POIA6i/+dXefEmZKoAgOaTdaIgNSMqMIU61yRyzA==", + "dev": true, + "dependencies": { + "@babel/parser": "^7.20.7", + "@babel/types": "^7.20.7", + "@types/babel__generator": "*", + "@types/babel__template": "*", + "@types/babel__traverse": "*" + } + }, + "node_modules/@types/babel__generator": { + "version": "7.27.0", + "resolved": "https://registry.npmjs.org/@types/babel__generator/-/babel__generator-7.27.0.tgz", + "integrity": "sha512-ufFd2Xi92OAVPYsy+P4n7/U7e68fex0+Ee8gSG9KX7eo084CWiQ4sdxktvdl0bOPupXtVJPY19zk6EwWqUQ8lg==", + "dev": true, + "dependencies": { + "@babel/types": "^7.0.0" + } + }, + "node_modules/@types/babel__template": { + "version": "7.4.4", + "resolved": "https://registry.npmjs.org/@types/babel__template/-/babel__template-7.4.4.tgz", + "integrity": "sha512-h/NUaSyG5EyxBIp8YRxo4RMe2/qQgvyowRwVMzhYhBCONbW8PUsg4lkFMrhgZhUe5z3L3MiLDuvyJ/CaPa2A8A==", + "dev": true, + "dependencies": { + "@babel/parser": "^7.1.0", + "@babel/types": "^7.0.0" + } + }, + "node_modules/@types/babel__traverse": { + "version": "7.28.0", + "resolved": "https://registry.npmjs.org/@types/babel__traverse/-/babel__traverse-7.28.0.tgz", + "integrity": "sha512-8PvcXf70gTDZBgt9ptxJ8elBeBjcLOAcOtoO/mPJjtji1+CdGbHgm77om1GrsPxsiE+uXIpNSK64UYaIwQXd4Q==", + "dev": true, + "dependencies": { + "@babel/types": "^7.28.2" + } + }, + "node_modules/@types/estree": { + "version": "1.0.9", + "resolved": "https://registry.npmjs.org/@types/estree/-/estree-1.0.9.tgz", + "integrity": "sha512-GhdPgy1el4/ImP05X05Uw4cw2/M93BCUmnEvWZNStlCzEKME4Fkk+YpoA5OiHNQmoS7Cafb8Xa3Pya8m1Qrzeg==", + "dev": true + }, + "node_modules/@types/prop-types": { + "version": "15.7.15", + "resolved": "https://registry.npmjs.org/@types/prop-types/-/prop-types-15.7.15.tgz", + "integrity": "sha512-F6bEyamV9jKGAFBEmlQnesRPGOQqS2+Uwi0Em15xenOxHaf2hv6L8YCVn3rPdPJOiJfPiCnLIRyvwVaqMY3MIw==", + "devOptional": true + }, + "node_modules/@types/react": { + "version": "18.3.31", + "resolved": "https://registry.npmjs.org/@types/react/-/react-18.3.31.tgz", + "integrity": "sha512-vfEqpXTvwT91yhmwdfouStN2hSKwTvyRs8qpLfADyrq/kxDw0hZM7Wk9Ug1FELj8hIby+S/+kQCSRFF32nv2Qw==", + "devOptional": true, + "dependencies": { + "@types/prop-types": "*", + "csstype": "^3.2.2" + } + }, + "node_modules/@types/react-dom": { + "version": "18.3.7", + "resolved": "https://registry.npmjs.org/@types/react-dom/-/react-dom-18.3.7.tgz", + "integrity": "sha512-MEe3UeoENYVFXzoXEWsvcpg6ZvlrFNlOQ7EOsvhI3CfAXwzPfO8Qwuxd40nepsYKqyyVQnTdEfv68q91yLcKrQ==", + "dev": true, + "peerDependencies": { + "@types/react": "^18.0.0" + } + }, + "node_modules/@types/react-window": { + "version": "1.8.8", + "resolved": "https://registry.npmjs.org/@types/react-window/-/react-window-1.8.8.tgz", + "integrity": "sha512-8Ls660bHR1AUA2kuRvVG9D/4XpRC6wjAaPT9dil7Ckc76eP9TKWZwwmgfq8Q1LANX3QNDnoU4Zp48A3w+zK69Q==", + "dev": true, + "dependencies": { + "@types/react": "*" + } + }, + "node_modules/@vitejs/plugin-react": { + "version": "4.7.0", + "resolved": "https://registry.npmjs.org/@vitejs/plugin-react/-/plugin-react-4.7.0.tgz", + "integrity": "sha512-gUu9hwfWvvEDBBmgtAowQCojwZmJ5mcLn3aufeCsitijs3+f2NsrPtlAWIR6OPiqljl96GVCUbLe0HyqIpVaoA==", + "dev": true, + "dependencies": { + "@babel/core": "^7.28.0", + "@babel/plugin-transform-react-jsx-self": "^7.27.1", + "@babel/plugin-transform-react-jsx-source": "^7.27.1", + "@rolldown/pluginutils": "1.0.0-beta.27", + "@types/babel__core": "^7.20.5", + "react-refresh": "^0.17.0" + }, + "engines": { + "node": "^14.18.0 || >=16.0.0" + }, + "peerDependencies": { + "vite": "^4.2.0 || ^5.0.0 || ^6.0.0 || ^7.0.0" + } + }, + "node_modules/agent-base": { + "version": "6.0.2", + "resolved": "https://registry.npmjs.org/agent-base/-/agent-base-6.0.2.tgz", + "integrity": "sha512-RZNwNclF7+MS/8bDg70amg32dyeZGZxiDuQmZxKLAlQjr3jGyLx+4Kkk58UO7D2QdgFIQCovuSuZESne6RG6XQ==", + "dependencies": { + "debug": "4" + }, + "engines": { + "node": ">= 6.0.0" + } + }, + "node_modules/antd": { + "version": "5.29.3", + "resolved": "https://registry.npmjs.org/antd/-/antd-5.29.3.tgz", + "integrity": "sha512-3DdbGCa9tWAJGcCJ6rzR8EJFsv2CtyEbkVabZE14pfgUHfCicWCj0/QzQVLDYg8CPfQk9BH7fHCoTXHTy7MP/A==", + "dependencies": { + "@ant-design/colors": "^7.2.1", + "@ant-design/cssinjs": "^1.23.0", + "@ant-design/cssinjs-utils": "^1.1.3", + "@ant-design/fast-color": "^2.0.6", + "@ant-design/icons": "^5.6.1", + "@ant-design/react-slick": "~1.1.2", + "@babel/runtime": "^7.26.0", + "@rc-component/color-picker": "~2.0.1", + "@rc-component/mutate-observer": "^1.1.0", + "@rc-component/qrcode": "~1.1.0", + "@rc-component/tour": "~1.15.1", + "@rc-component/trigger": "^2.3.0", + "classnames": "^2.5.1", + "copy-to-clipboard": "^3.3.3", + "dayjs": "^1.11.11", + "rc-cascader": "~3.34.0", + "rc-checkbox": "~3.5.0", + "rc-collapse": "~3.9.0", + "rc-dialog": "~9.6.0", + "rc-drawer": "~7.3.0", + "rc-dropdown": "~4.2.1", + "rc-field-form": "~2.7.1", + "rc-image": "~7.12.0", + "rc-input": "~1.8.0", + "rc-input-number": "~9.5.0", + "rc-mentions": "~2.20.0", + "rc-menu": "~9.16.1", + "rc-motion": "^2.9.5", + "rc-notification": "~5.6.4", + "rc-pagination": "~5.1.0", + "rc-picker": "~4.11.3", + "rc-progress": "~4.0.0", + "rc-rate": "~2.13.1", + "rc-resize-observer": "^1.4.3", + "rc-segmented": "~2.7.0", + "rc-select": "~14.16.8", + "rc-slider": "~11.1.9", + "rc-steps": "~6.0.1", + "rc-switch": "~4.1.0", + "rc-table": "~7.54.0", + "rc-tabs": "~15.7.0", + "rc-textarea": "~1.10.2", + "rc-tooltip": "~6.4.0", + "rc-tree": "~5.13.1", + "rc-tree-select": "~5.27.0", + "rc-upload": "~4.11.0", + "rc-util": "^5.44.4", + "scroll-into-view-if-needed": "^3.1.0", + "throttle-debounce": "^5.0.2" + }, + "funding": { + "type": "opencollective", + "url": "https://opencollective.com/ant-design" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/any-promise": { + "version": "1.3.0", + "resolved": "https://registry.npmjs.org/any-promise/-/any-promise-1.3.0.tgz", + "integrity": "sha512-7UvmKalWRt1wgjL1RrGxoSJW/0QZFIegpeGvZG9kjp8vrRu55XTHbwnqq2GpXm9uLbcuhxm3IqX9OB4MZR1b2A==", + "dev": true + }, + "node_modules/anymatch": { + "version": "3.1.3", + "resolved": "https://registry.npmjs.org/anymatch/-/anymatch-3.1.3.tgz", + "integrity": "sha512-KMReFUr0B4t+D+OBkjR3KYqvocp2XaSzO55UcB6mgQMd3KbcE+mWTyvVV7D/zsdEbNnV6acZUutkiHQXvTr1Rw==", + "dev": true, + "dependencies": { + "normalize-path": "^3.0.0", + "picomatch": "^2.0.4" + }, + "engines": { + "node": ">= 8" + } + }, + "node_modules/arg": { + "version": "5.0.2", + "resolved": "https://registry.npmjs.org/arg/-/arg-5.0.2.tgz", + "integrity": "sha512-PYjyFOLKQ9y57JvQ6QLo8dAgNqswh8M1RMJYdQduT6xbWSgK36P/Z/v+p888pM69jMMfS8Xd8F6I1kQ/I9HUGg==", + "dev": true + }, + "node_modules/asynckit": { + "version": "0.4.0", + "resolved": "https://registry.npmjs.org/asynckit/-/asynckit-0.4.0.tgz", + "integrity": "sha512-Oei9OH4tRh0YqU3GxhX79dM/mwVgvbZJaSNaRk+bshkj0S5cfHcgYakreBjrHwatXKbz+IoIdYLxrKim2MjW0Q==" + }, + "node_modules/autoprefixer": { + "version": "10.5.2", + "resolved": "https://registry.npmjs.org/autoprefixer/-/autoprefixer-10.5.2.tgz", + "integrity": "sha512-rD5t5DwOjJdmSORcTq64j8MawTC+tbQ+HHqjR4NDumamy/ambn1UJrlKL+KdwujWxMkFjPM3pPHOEA9tl4767Q==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/postcss/" + }, + { + "type": "tidelift", + "url": "https://tidelift.com/funding/github/npm/autoprefixer" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "browserslist": "^4.28.4", + "caniuse-lite": "^1.0.30001799", + "fraction.js": "^5.3.4", + "picocolors": "^1.1.1", + "postcss-value-parser": "^4.2.0" + }, + "bin": { + "autoprefixer": "bin/autoprefixer" + }, + "engines": { + "node": "^10 || ^12 || >=14" + }, + "peerDependencies": { + "postcss": "^8.1.0" + } + }, + "node_modules/axios": { + "version": "1.18.1", + "resolved": "https://registry.npmjs.org/axios/-/axios-1.18.1.tgz", + "integrity": "sha512-3nTvFlvpn9Zu/RkHUqtc7/+al4UpRW5az71ap5zccp6e8RAYEzhMTecX8Dz1wWDYrPpUoB1HAQEGEAEvUr7S9g==", + "dependencies": { + "follow-redirects": "^1.16.0", + "form-data": "^4.0.5", + "https-proxy-agent": "^5.0.1", + "proxy-from-env": "^2.1.0" + } + }, + "node_modules/baseline-browser-mapping": { + "version": "2.10.43", + "resolved": "https://registry.npmjs.org/baseline-browser-mapping/-/baseline-browser-mapping-2.10.43.tgz", + "integrity": "sha512-AjYpR78kDWAY3Efj+cDTFH9t9SCoL7OoTp1BOb0mQV7S+6CiLwnWM3FyxhJtdPufDFKzmCSFoUncKjWgJEZTCQ==", + "dev": true, + "bin": { + "baseline-browser-mapping": "dist/cli.cjs" + }, + "engines": { + "node": ">=6.0.0" + } + }, + "node_modules/binary-extensions": { + "version": "2.3.0", + "resolved": "https://registry.npmjs.org/binary-extensions/-/binary-extensions-2.3.0.tgz", + "integrity": "sha512-Ceh+7ox5qe7LJuLHoY0feh3pHuUDHAcRUeyL2VYghZwfpkNIy/+8Ocg0a3UuSoYzavmylwuLWQOf3hl0jjMMIw==", + "dev": true, + "engines": { + "node": ">=8" + }, + "funding": { + "url": "https://github.com/sponsors/sindresorhus" + } + }, + "node_modules/braces": { + "version": "3.0.3", + "resolved": "https://registry.npmjs.org/braces/-/braces-3.0.3.tgz", + "integrity": "sha512-yQbXgO/OSZVD2IsiLlro+7Hf6Q18EJrKSEsdoMzKePKXct3gvD8oLcOQdIzGupr5Fj+EDe8gO/lxc1BzfMpxvA==", + "dev": true, + "dependencies": { + "fill-range": "^7.1.1" + }, + "engines": { + "node": ">=8" + } + }, + "node_modules/browserslist": { + "version": "4.28.6", + "resolved": "https://registry.npmjs.org/browserslist/-/browserslist-4.28.6.tgz", + "integrity": "sha512-FQBYNK15VMslhLHpA7+n+n1GOlF1kId2xcCg7/j95f24AOF6VDYMNH4mFxF7KuaTdv627faazpOAjFzMrfJOUw==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/browserslist" + }, + { + "type": "tidelift", + "url": "https://tidelift.com/funding/github/npm/browserslist" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "baseline-browser-mapping": "^2.10.42", + "caniuse-lite": "^1.0.30001803", + "electron-to-chromium": "^1.5.389", + "node-releases": "^2.0.51", + "update-browserslist-db": "^1.2.3" + }, + "bin": { + "browserslist": "cli.js" + }, + "engines": { + "node": "^6 || ^7 || ^8 || ^9 || ^10 || ^11 || ^12 || >=13.7" + } + }, + "node_modules/call-bind-apply-helpers": { + "version": "1.0.2", + "resolved": "https://registry.npmjs.org/call-bind-apply-helpers/-/call-bind-apply-helpers-1.0.2.tgz", + "integrity": "sha512-Sp1ablJ0ivDkSzjcaJdxEunN5/XvksFJ2sMBFfq6x0ryhQV/2b/KwFe21cMpmHtPOSij8K99/wSfoEuTObmuMQ==", + "dependencies": { + "es-errors": "^1.3.0", + "function-bind": "^1.1.2" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/camelcase-css": { + "version": "2.0.1", + "resolved": "https://registry.npmjs.org/camelcase-css/-/camelcase-css-2.0.1.tgz", + "integrity": "sha512-QOSvevhslijgYwRx6Rv7zKdMF8lbRmx+uQGx2+vDc+KI/eBnsy9kit5aj23AgGu3pa4t9AgwbnXWqS+iOY+2aA==", + "dev": true, + "engines": { + "node": ">= 6" + } + }, + "node_modules/caniuse-lite": { + "version": "1.0.30001805", + "resolved": "https://registry.npmjs.org/caniuse-lite/-/caniuse-lite-1.0.30001805.tgz", + "integrity": "sha512-52noaS3DubycKSXaU30TwPGIp+POyQSUVa5jBEq3vkRkY0kjyb3LQgvhU6WGyCcyXqVLWO0Cw0Q6BSdD0kUfVA==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/browserslist" + }, + { + "type": "tidelift", + "url": "https://tidelift.com/funding/github/npm/caniuse-lite" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ] + }, + "node_modules/chokidar": { + "version": "3.6.0", + "resolved": "https://registry.npmjs.org/chokidar/-/chokidar-3.6.0.tgz", + "integrity": "sha512-7VT13fmjotKpGipCW9JEQAusEPE+Ei8nl6/g4FBAmIm0GOOLMua9NDDo/DWp0ZAxCr3cPq5ZpBqmPAQgDda2Pw==", + "dev": true, + "dependencies": { + "anymatch": "~3.1.2", + "braces": "~3.0.2", + "glob-parent": "~5.1.2", + "is-binary-path": "~2.1.0", + "is-glob": "~4.0.1", + "normalize-path": "~3.0.0", + "readdirp": "~3.6.0" + }, + "engines": { + "node": ">= 8.10.0" + }, + "funding": { + "url": "https://paulmillr.com/funding/" + }, + "optionalDependencies": { + "fsevents": "~2.3.2" + } + }, + "node_modules/chokidar/node_modules/glob-parent": { + "version": "5.1.2", + "resolved": "https://registry.npmjs.org/glob-parent/-/glob-parent-5.1.2.tgz", + "integrity": "sha512-AOIgSQCepiJYwP3ARnGx+5VnTu2HBYdzbGP45eLw1vr3zB3vZLeyed1sC9hnbcOc9/SrMyM5RPQrkGz4aS9Zow==", + "dev": true, + "dependencies": { + "is-glob": "^4.0.1" + }, + "engines": { + "node": ">= 6" + } + }, + "node_modules/classnames": { + "version": "2.5.1", + "resolved": "https://registry.npmjs.org/classnames/-/classnames-2.5.1.tgz", + "integrity": "sha512-saHYOzhIQs6wy2sVxTM6bUDsQO4F50V9RQ22qBpEdCW+I+/Wmke2HOl6lS6dTpdxVhb88/I6+Hs+438c3lfUow==" + }, + "node_modules/combined-stream": { + "version": "1.0.8", + "resolved": "https://registry.npmjs.org/combined-stream/-/combined-stream-1.0.8.tgz", + "integrity": "sha512-FQN4MRfuJeHf7cBbBMJFXhKSDq+2kAArBlmRBvcvFE5BB1HZKXtSFASDhdlz9zOYwxh8lDdnvmMOe/+5cdoEdg==", + "dependencies": { + "delayed-stream": "~1.0.0" + }, + "engines": { + "node": ">= 0.8" + } + }, + "node_modules/commander": { + "version": "4.1.1", + "resolved": "https://registry.npmjs.org/commander/-/commander-4.1.1.tgz", + "integrity": "sha512-NOKm8xhkzAjzFx8B2v5OAHT+u5pRQc2UCa2Vq9jYL/31o2wi9mxBA7LIFs3sV5VSC49z6pEhfbMULvShKj26WA==", + "dev": true, + "engines": { + "node": ">= 6" + } + }, + "node_modules/compute-scroll-into-view": { + "version": "3.1.1", + "resolved": "https://registry.npmjs.org/compute-scroll-into-view/-/compute-scroll-into-view-3.1.1.tgz", + "integrity": "sha512-VRhuHOLoKYOy4UbilLbUzbYg93XLjv2PncJC50EuTWPA3gaja1UjBsUP/D/9/juV3vQFr6XBEzn9KCAHdUvOHw==" + }, + "node_modules/convert-source-map": { + "version": "2.0.0", + "resolved": "https://registry.npmjs.org/convert-source-map/-/convert-source-map-2.0.0.tgz", + "integrity": "sha512-Kvp459HrV2FEJ1CAsi1Ku+MY3kasH19TFykTz2xWmMeq6bk2NU3XXvfJ+Q61m0xktWwt+1HSYf3JZsTms3aRJg==", + "dev": true + }, + "node_modules/copy-to-clipboard": { + "version": "3.3.3", + "resolved": "https://registry.npmjs.org/copy-to-clipboard/-/copy-to-clipboard-3.3.3.tgz", + "integrity": "sha512-2KV8NhB5JqC3ky0r9PMCAZKbUHSwtEo4CwCs0KXgruG43gX5PMqDEBbVU4OUzw2MuAWUfsuFmWvEKG5QRfSnJA==", + "dependencies": { + "toggle-selection": "^1.0.6" + } + }, + "node_modules/cssesc": { + "version": "3.0.0", + "resolved": "https://registry.npmjs.org/cssesc/-/cssesc-3.0.0.tgz", + "integrity": "sha512-/Tb/JcjK111nNScGob5MNtsntNM1aCNUDipB/TkwZFhyDrrE47SOx/18wF2bbjgc3ZzCSKW1T5nt5EbFoAz/Vg==", + "dev": true, + "bin": { + "cssesc": "bin/cssesc" + }, + "engines": { + "node": ">=4" + } + }, + "node_modules/csstype": { + "version": "3.2.3", + "resolved": "https://registry.npmjs.org/csstype/-/csstype-3.2.3.tgz", + "integrity": "sha512-z1HGKcYy2xA8AGQfwrn0PAy+PB7X/GSj3UVJW9qKyn43xWa+gl5nXmU4qqLMRzWVLFC8KusUX8T/0kCiOYpAIQ==" + }, + "node_modules/dayjs": { + "version": "1.11.21", + "resolved": "https://registry.npmjs.org/dayjs/-/dayjs-1.11.21.tgz", + "integrity": "sha512-98IT+HOahAisibz/yjKbzuOBwYcjJ7BCLPzARyHiyEBmRz4fatF+KPJszEHXsGYjUG234aH/cOjW1wwTbKUZlA==" + }, + "node_modules/debug": { + "version": "4.4.3", + "resolved": "https://registry.npmjs.org/debug/-/debug-4.4.3.tgz", + "integrity": "sha512-RGwwWnwQvkVfavKVt22FGLw+xYSdzARwm0ru6DhTVA3umU5hZc28V3kO4stgYryrTlLpuvgI9GiijltAjNbcqA==", + "dependencies": { + "ms": "^2.1.3" + }, + "engines": { + "node": ">=6.0" + }, + "peerDependenciesMeta": { + "supports-color": { + "optional": true + } + } + }, + "node_modules/delayed-stream": { + "version": "1.0.0", + "resolved": "https://registry.npmjs.org/delayed-stream/-/delayed-stream-1.0.0.tgz", + "integrity": "sha512-ZySD7Nf91aLB0RxL4KGrKHBXl7Eds1DAmEdcoVawXnLD7SDhpNgtuII2aAkg7a7QS41jxPSZ17p4VdGnMHk3MQ==", + "engines": { + "node": ">=0.4.0" + } + }, + "node_modules/didyoumean": { + "version": "1.2.2", + "resolved": "https://registry.npmjs.org/didyoumean/-/didyoumean-1.2.2.tgz", + "integrity": "sha512-gxtyfqMg7GKyhQmb056K7M3xszy/myH8w+B4RT+QXBQsvAOdc3XymqDDPHx1BgPgsdAA5SIifona89YtRATDzw==", + "dev": true + }, + "node_modules/dlv": { + "version": "1.1.3", + "resolved": "https://registry.npmjs.org/dlv/-/dlv-1.1.3.tgz", + "integrity": "sha512-+HlytyjlPKnIG8XuRG8WvmBP8xs8P71y+SKKS6ZXWoEgLuePxtDoUEiH7WkdePWrQ5JBpE6aoVqfZfJUQkjXwA==", + "dev": true + }, + "node_modules/dunder-proto": { + "version": "1.0.1", + "resolved": "https://registry.npmjs.org/dunder-proto/-/dunder-proto-1.0.1.tgz", + "integrity": "sha512-KIN/nDJBQRcXw0MLVhZE9iQHmG68qAVIBg9CqmUYjmQIhgij9U5MFvrqkUL5FbtyyzZuOeOt0zdeRe4UY7ct+A==", + "dependencies": { + "call-bind-apply-helpers": "^1.0.1", + "es-errors": "^1.3.0", + "gopd": "^1.2.0" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/echarts": { + "version": "5.6.0", + "resolved": "https://registry.npmjs.org/echarts/-/echarts-5.6.0.tgz", + "integrity": "sha512-oTbVTsXfKuEhxftHqL5xprgLoc0k7uScAwtryCgWF6hPYFLRwOUHiFmHGCBKP5NPFNkDVopOieyUqYGH8Fa3kA==", + "dependencies": { + "tslib": "2.3.0", + "zrender": "5.6.1" + } + }, + "node_modules/echarts-for-react": { + "version": "3.0.6", + "resolved": "https://registry.npmjs.org/echarts-for-react/-/echarts-for-react-3.0.6.tgz", + "integrity": "sha512-4zqLgTGWS3JvkQDXjzkR1k1CHRdpd6by0988TWMJgnvDytegWLbeP/VNZmMa+0VJx2eD7Y632bi2JquXDgiGJg==", + "dependencies": { + "fast-deep-equal": "^3.1.3", + "size-sensor": "^1.0.1" + }, + "peerDependencies": { + "echarts": "^3.0.0 || ^4.0.0 || ^5.0.0 || ^6.0.0", + "react": "^15.0.0 || >=16.0.0" + } + }, + "node_modules/electron-to-chromium": { + "version": "1.5.389", + "resolved": "https://registry.npmjs.org/electron-to-chromium/-/electron-to-chromium-1.5.389.tgz", + "integrity": "sha512-cEto7aeOqBfU1D+c5py5pE+ooscKE75JifxLBdFUZsqAxRS6y7kebtxAZvICszSl05gPjYHDTjY+lXpyGvpJbg==", + "dev": true + }, + "node_modules/es-define-property": { + "version": "1.0.1", + "resolved": "https://registry.npmjs.org/es-define-property/-/es-define-property-1.0.1.tgz", + "integrity": "sha512-e3nRfgfUZ4rNGL232gUgX06QNyyez04KdjFrF+LTRoOXmrOgFKDg4BCdsjW8EnT69eqdYGmRpJwiPVYNrCaW3g==", + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/es-errors": { + "version": "1.3.0", + "resolved": "https://registry.npmjs.org/es-errors/-/es-errors-1.3.0.tgz", + "integrity": "sha512-Zf5H2Kxt2xjTvbJvP2ZWLEICxA6j+hAmMzIlypy4xcBg1vKVnx89Wy0GbS+kf5cwCVFFzdCFh2XSCFNULS6csw==", + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/es-object-atoms": { + "version": "1.1.2", + "resolved": "https://registry.npmjs.org/es-object-atoms/-/es-object-atoms-1.1.2.tgz", + "integrity": "sha512-HWcBoN6NileqtSydK2FqHbS/LoDd2pqrnQHLyJzBj4kOp/ky2MWMN694xOfkK8/SnUsW2DH7EfyVlydKCsm1Zw==", + "dependencies": { + "es-errors": "^1.3.0" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/es-set-tostringtag": { + "version": "2.1.0", + "resolved": "https://registry.npmjs.org/es-set-tostringtag/-/es-set-tostringtag-2.1.0.tgz", + "integrity": "sha512-j6vWzfrGVfyXxge+O0x5sh6cvxAog0a/4Rdd2K36zCMV5eJ+/+tOAngRO8cODMNWbVRdVlmGZQL2YS3yR8bIUA==", + "dependencies": { + "es-errors": "^1.3.0", + "get-intrinsic": "^1.2.6", + "has-tostringtag": "^1.0.2", + "hasown": "^2.0.2" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/esbuild": { + "version": "0.21.5", + "resolved": "https://registry.npmjs.org/esbuild/-/esbuild-0.21.5.tgz", + "integrity": "sha512-mg3OPMV4hXywwpoDxu3Qda5xCKQi+vCTZq8S9J/EpkhB2HzKXq4SNFZE3+NK93JYxc8VMSep+lOUSC/RVKaBqw==", + "dev": true, + "hasInstallScript": true, + "bin": { + "esbuild": "bin/esbuild" + }, + "engines": { + "node": ">=12" + }, + "optionalDependencies": { + "@esbuild/aix-ppc64": "0.21.5", + "@esbuild/android-arm": "0.21.5", + "@esbuild/android-arm64": "0.21.5", + "@esbuild/android-x64": "0.21.5", + "@esbuild/darwin-arm64": "0.21.5", + "@esbuild/darwin-x64": "0.21.5", + "@esbuild/freebsd-arm64": "0.21.5", + "@esbuild/freebsd-x64": "0.21.5", + "@esbuild/linux-arm": "0.21.5", + "@esbuild/linux-arm64": "0.21.5", + "@esbuild/linux-ia32": "0.21.5", + "@esbuild/linux-loong64": "0.21.5", + "@esbuild/linux-mips64el": "0.21.5", + "@esbuild/linux-ppc64": "0.21.5", + "@esbuild/linux-riscv64": "0.21.5", + "@esbuild/linux-s390x": "0.21.5", + "@esbuild/linux-x64": "0.21.5", + "@esbuild/netbsd-x64": "0.21.5", + "@esbuild/openbsd-x64": "0.21.5", + "@esbuild/sunos-x64": "0.21.5", + "@esbuild/win32-arm64": "0.21.5", + "@esbuild/win32-ia32": "0.21.5", + "@esbuild/win32-x64": "0.21.5" + } + }, + "node_modules/escalade": { + "version": "3.2.0", + "resolved": "https://registry.npmjs.org/escalade/-/escalade-3.2.0.tgz", + "integrity": "sha512-WUj2qlxaQtO4g6Pq5c29GTcWGDyd8itL8zTlipgECz3JesAiiOKotd8JU6otB3PACgG6xkJUyVhboMS+bje/jA==", + "dev": true, + "engines": { + "node": ">=6" + } + }, + "node_modules/fast-deep-equal": { + "version": "3.1.3", + "resolved": "https://registry.npmjs.org/fast-deep-equal/-/fast-deep-equal-3.1.3.tgz", + "integrity": "sha512-f3qQ9oQy9j2AhBe/H9VC91wLmKBCCU/gDOnKNAYG5hswO7BLKj09Hc5HYNz9cGI++xlpDCIgDaitVs03ATR84Q==" + }, + "node_modules/fast-glob": { + "version": "3.3.3", + "resolved": "https://registry.npmjs.org/fast-glob/-/fast-glob-3.3.3.tgz", + "integrity": "sha512-7MptL8U0cqcFdzIzwOTHoilX9x5BrNqye7Z/LuC7kCMRio1EMSyqRK3BEAUD7sXRq4iT4AzTVuZdhgQ2TCvYLg==", + "dev": true, + "dependencies": { + "@nodelib/fs.stat": "^2.0.2", + "@nodelib/fs.walk": "^1.2.3", + "glob-parent": "^5.1.2", + "merge2": "^1.3.0", + "micromatch": "^4.0.8" + }, + "engines": { + "node": ">=8.6.0" + } + }, + "node_modules/fast-glob/node_modules/glob-parent": { + "version": "5.1.2", + "resolved": "https://registry.npmjs.org/glob-parent/-/glob-parent-5.1.2.tgz", + "integrity": "sha512-AOIgSQCepiJYwP3ARnGx+5VnTu2HBYdzbGP45eLw1vr3zB3vZLeyed1sC9hnbcOc9/SrMyM5RPQrkGz4aS9Zow==", + "dev": true, + "dependencies": { + "is-glob": "^4.0.1" + }, + "engines": { + "node": ">= 6" + } + }, + "node_modules/fastq": { + "version": "1.20.1", + "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.20.1.tgz", + "integrity": "sha512-GGToxJ/w1x32s/D2EKND7kTil4n8OVk/9mycTc4VDza13lOvpUZTGX3mFSCtV9ksdGBVzvsyAVLM6mHFThxXxw==", + "dev": true, + "dependencies": { + "reusify": "^1.0.4" + } + }, + "node_modules/fill-range": { + "version": "7.1.1", + "resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.1.1.tgz", + "integrity": "sha512-YsGpe3WHLK8ZYi4tWDg2Jy3ebRz2rXowDxnld4bkQB00cc/1Zw9AWnC0i9ztDJitivtQvaI9KaLyKrc+hBW0yg==", + "dev": true, + "dependencies": { + "to-regex-range": "^5.0.1" + }, + "engines": { + "node": ">=8" + } + }, + "node_modules/follow-redirects": { + "version": "1.16.0", + "resolved": "https://registry.npmjs.org/follow-redirects/-/follow-redirects-1.16.0.tgz", + "integrity": "sha512-y5rN/uOsadFT/JfYwhxRS5R7Qce+g3zG97+JrtFZlC9klX/W5hD7iiLzScI4nZqUS7DNUdhPgw4xI8W2LuXlUw==", + "funding": [ + { + "type": "individual", + "url": "https://github.com/sponsors/RubenVerborgh" + } + ], + "engines": { + "node": ">=4.0" + }, + "peerDependenciesMeta": { + "debug": { + "optional": true + } + } + }, + "node_modules/form-data": { + "version": "4.0.6", + "resolved": "https://registry.npmjs.org/form-data/-/form-data-4.0.6.tgz", + "integrity": "sha512-vKatAh4SlVfgbv+YtmhiRjhEMJsYpsG1Y2rMQtR+SVSbytsSD1YGzDIcrAJmdFec88u/+VoGmxnl+80gL1tRCQ==", + "dependencies": { + "asynckit": "^0.4.0", + "combined-stream": "^1.0.8", + "es-set-tostringtag": "^2.1.0", + "hasown": "^2.0.4", + "mime-types": "^2.1.35" + }, + "engines": { + "node": ">= 6" + } + }, + "node_modules/fraction.js": { + "version": "5.3.4", + "resolved": "https://registry.npmjs.org/fraction.js/-/fraction.js-5.3.4.tgz", + "integrity": "sha512-1X1NTtiJphryn/uLQz3whtY6jK3fTqoE3ohKs0tT+Ujr1W59oopxmoEh7Lu5p6vBaPbgoM0bzveAW4Qi5RyWDQ==", + "dev": true, + "engines": { + "node": "*" + }, + "funding": { + "type": "github", + "url": "https://github.com/sponsors/rawify" + } + }, + "node_modules/fsevents": { + "version": "2.3.3", + "resolved": "https://registry.npmjs.org/fsevents/-/fsevents-2.3.3.tgz", + "integrity": "sha512-5xoDfX+fL7faATnagmWPpbFtwh/R77WmMMqqHGS65C3vvB0YHrgF+B1YmZ3441tMj5n63k0212XNoJwzlhffQw==", + "dev": true, + "hasInstallScript": true, + "optional": true, + "os": [ + "darwin" + ], + "engines": { + "node": "^8.16.0 || ^10.6.0 || >=11.0.0" + } + }, + "node_modules/function-bind": { + "version": "1.1.2", + "resolved": "https://registry.npmjs.org/function-bind/-/function-bind-1.1.2.tgz", + "integrity": "sha512-7XHNxH7qX9xG5mIwxkhumTox/MIRNcOgDrxWsMt2pAr23WHp6MrRlN7FBSFpCpr+oVO0F744iUgR82nJMfG2SA==", + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/gensync": { + "version": "1.0.0-beta.2", + "resolved": "https://registry.npmjs.org/gensync/-/gensync-1.0.0-beta.2.tgz", + "integrity": "sha512-3hN7NaskYvMDLQY55gnW3NQ+mesEAepTqlg+VEbj7zzqEMBVNhzcGYYeqFo/TlYz6eQiFcp1HcsCZO+nGgS8zg==", + "dev": true, + "engines": { + "node": ">=6.9.0" + } + }, + "node_modules/get-intrinsic": { + "version": "1.3.0", + "resolved": "https://registry.npmjs.org/get-intrinsic/-/get-intrinsic-1.3.0.tgz", + "integrity": "sha512-9fSjSaos/fRIVIp+xSJlE6lfwhES7LNtKaCBIamHsjr2na1BiABJPo0mOjjz8GJDURarmCPGqaiVg5mfjb98CQ==", + "dependencies": { + "call-bind-apply-helpers": "^1.0.2", + "es-define-property": "^1.0.1", + "es-errors": "^1.3.0", + "es-object-atoms": "^1.1.1", + "function-bind": "^1.1.2", + "get-proto": "^1.0.1", + "gopd": "^1.2.0", + "has-symbols": "^1.1.0", + "hasown": "^2.0.2", + "math-intrinsics": "^1.1.0" + }, + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/get-proto": { + "version": "1.0.1", + "resolved": "https://registry.npmjs.org/get-proto/-/get-proto-1.0.1.tgz", + "integrity": "sha512-sTSfBjoXBp89JvIKIefqw7U2CCebsc74kiY6awiGogKtoSGbgjYE/G/+l9sF3MWFPNc9IcoOC4ODfKHfxFmp0g==", + "dependencies": { + "dunder-proto": "^1.0.1", + "es-object-atoms": "^1.0.0" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/glob-parent": { + "version": "6.0.2", + "resolved": "https://registry.npmjs.org/glob-parent/-/glob-parent-6.0.2.tgz", + "integrity": "sha512-XxwI8EOhVQgWp6iDL+3b0r86f4d6AX6zSU55HfB4ydCEuXLXc5FcYeOu+nnGftS4TEju/11rt4KJPTMgbfmv4A==", + "dev": true, + "dependencies": { + "is-glob": "^4.0.3" + }, + "engines": { + "node": ">=10.13.0" + } + }, + "node_modules/gopd": { + "version": "1.2.0", + "resolved": "https://registry.npmjs.org/gopd/-/gopd-1.2.0.tgz", + "integrity": "sha512-ZUKRh6/kUFoAiTAtTYPZJ3hw9wNxx+BIBOijnlG9PnrJsCcSjs1wyyD6vJpaYtgnzDrKYRSqf3OO6Rfa93xsRg==", + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/has-symbols": { + "version": "1.1.0", + "resolved": "https://registry.npmjs.org/has-symbols/-/has-symbols-1.1.0.tgz", + "integrity": "sha512-1cDNdwJ2Jaohmb3sg4OmKaMBwuC48sYni5HUw2DvsC8LjGTLK9h+eb1X6RyuOHe4hT0ULCW68iomhjUoKUqlPQ==", + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/has-tostringtag": { + "version": "1.0.2", + "resolved": "https://registry.npmjs.org/has-tostringtag/-/has-tostringtag-1.0.2.tgz", + "integrity": "sha512-NqADB8VjPFLM2V0VvHUewwwsw0ZWBaIdgo+ieHtK3hasLz4qeCRjYcqfB6AQrBggRKppKF8L52/VqdVsO47Dlw==", + "dependencies": { + "has-symbols": "^1.0.3" + }, + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/hasown": { + "version": "2.0.4", + "resolved": "https://registry.npmjs.org/hasown/-/hasown-2.0.4.tgz", + "integrity": "sha512-T2UbfbBEF32wiepXIsMlTW9+dDYC6wMh/t/vYA4tuOMKqWz/n3vr1NFSxQiyP+zk2mXsoMA/i/7qV6LKut1t1A==", + "dependencies": { + "function-bind": "^1.1.2" + }, + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/https-proxy-agent": { + "version": "5.0.1", + "resolved": "https://registry.npmjs.org/https-proxy-agent/-/https-proxy-agent-5.0.1.tgz", + "integrity": "sha512-dFcAjpTQFgoLMzC2VwU+C/CbS7uRL0lWmxDITmqm7C+7F0Odmj6s9l6alZc6AELXhrnggM2CeWSXHGOdX2YtwA==", + "dependencies": { + "agent-base": "6", + "debug": "4" + }, + "engines": { + "node": ">= 6" + } + }, + "node_modules/is-binary-path": { + "version": "2.1.0", + "resolved": "https://registry.npmjs.org/is-binary-path/-/is-binary-path-2.1.0.tgz", + "integrity": "sha512-ZMERYes6pDydyuGidse7OsHxtbI7WVeUEozgR/g7rd0xUimYNlvZRE/K2MgZTjWy725IfelLeVcEM97mmtRGXw==", + "dev": true, + "dependencies": { + "binary-extensions": "^2.0.0" + }, + "engines": { + "node": ">=8" + } + }, + "node_modules/is-core-module": { + "version": "2.16.2", + "resolved": "https://registry.npmjs.org/is-core-module/-/is-core-module-2.16.2.tgz", + "integrity": "sha512-evOr8xfXKxE6qSR0hSXL2r3sd7ALj8+7jQEUvPYcm5sgZFdJ+AYzT6yNmJenvIYQBgIGwfwz08sL8zoL7yq2BA==", + "dev": true, + "dependencies": { + "hasown": "^2.0.3" + }, + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/is-extglob": { + "version": "2.1.1", + "resolved": "https://registry.npmjs.org/is-extglob/-/is-extglob-2.1.1.tgz", + "integrity": "sha512-SbKbANkN603Vi4jEZv49LeVJMn4yGwsbzZworEoyEiutsN3nJYdbO36zfhGJ6QEDpOZIFkDtnq5JRxmvl3jsoQ==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/is-glob": { + "version": "4.0.3", + "resolved": "https://registry.npmjs.org/is-glob/-/is-glob-4.0.3.tgz", + "integrity": "sha512-xelSayHH36ZgE7ZWhli7pW34hNbNl8Ojv5KVmkJD4hBdD3th8Tfk9vYasLM+mXWOZhFkgZfxhLSnrwRr4elSSg==", + "dev": true, + "dependencies": { + "is-extglob": "^2.1.1" + }, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/is-number": { + "version": "7.0.0", + "resolved": "https://registry.npmjs.org/is-number/-/is-number-7.0.0.tgz", + "integrity": "sha512-41Cifkg6e8TylSpdtTpeLVMqvSBEVzTttHvERD741+pnZ8ANv0004MRL43QKPDlK9cGvNp6NZWZUBlbGXYxxng==", + "dev": true, + "engines": { + "node": ">=0.12.0" + } + }, + "node_modules/jiti": { + "version": "1.21.7", + "resolved": "https://registry.npmjs.org/jiti/-/jiti-1.21.7.tgz", + "integrity": "sha512-/imKNG4EbWNrVjoNC/1H5/9GFy+tqjGBHCaSsN+P2RnPqjsLmv6UD3Ej+Kj8nBWaRAwyk7kK5ZUc+OEatnTR3A==", + "dev": true, + "bin": { + "jiti": "bin/jiti.js" + } + }, + "node_modules/js-tokens": { + "version": "4.0.0", + "resolved": "https://registry.npmjs.org/js-tokens/-/js-tokens-4.0.0.tgz", + "integrity": "sha512-RdJUflcE3cUzKiMqQgsCu06FPu9UdIJO0beYbPhHN4k6apgJtifcoCtT9bcxOpYBtpD2kCM6Sbzg4CausW/PKQ==" + }, + "node_modules/jsesc": { + "version": "3.1.0", + "resolved": "https://registry.npmjs.org/jsesc/-/jsesc-3.1.0.tgz", + "integrity": "sha512-/sM3dO2FOzXjKQhJuo0Q173wf2KOo8t4I8vHy6lF9poUp7bKT0/NHE8fPX23PwfhnykfqnC2xRxOnVw5XuGIaA==", + "dev": true, + "bin": { + "jsesc": "bin/jsesc" + }, + "engines": { + "node": ">=6" + } + }, + "node_modules/json2mq": { + "version": "0.2.0", + "resolved": "https://registry.npmjs.org/json2mq/-/json2mq-0.2.0.tgz", + "integrity": "sha512-SzoRg7ux5DWTII9J2qkrZrqV1gt+rTaoufMxEzXbS26Uid0NwaJd123HcoB80TgubEppxxIGdNxCx50fEoEWQA==", + "dependencies": { + "string-convert": "^0.2.0" + } + }, + "node_modules/json5": { + "version": "2.2.3", + "resolved": "https://registry.npmjs.org/json5/-/json5-2.2.3.tgz", + "integrity": "sha512-XmOWe7eyHYH14cLdVPoyg+GOH3rYX++KpzrylJwSW98t3Nk+U8XOl8FWKOgwtzdb8lXGf6zYwDUzeHMWfxasyg==", + "dev": true, + "bin": { + "json5": "lib/cli.js" + }, + "engines": { + "node": ">=6" + } + }, + "node_modules/lilconfig": { + "version": "3.1.3", + "resolved": "https://registry.npmjs.org/lilconfig/-/lilconfig-3.1.3.tgz", + "integrity": "sha512-/vlFKAoH5Cgt3Ie+JLhRbwOsCQePABiU3tJ1egGvyQ+33R/vcwM2Zl2QR/LzjsBeItPt3oSVXapn+m4nQDvpzw==", + "dev": true, + "engines": { + "node": ">=14" + }, + "funding": { + "url": "https://github.com/sponsors/antonk52" + } + }, + "node_modules/lines-and-columns": { + "version": "1.2.4", + "resolved": "https://registry.npmjs.org/lines-and-columns/-/lines-and-columns-1.2.4.tgz", + "integrity": "sha512-7ylylesZQ/PV29jhEDl3Ufjo6ZX7gCqJr5F7PKrqc93v7fzSymt1BpwEU8nAUXs8qzzvqhbjhK5QZg6Mt/HkBg==", + "dev": true + }, + "node_modules/loose-envify": { + "version": "1.4.0", + "resolved": "https://registry.npmjs.org/loose-envify/-/loose-envify-1.4.0.tgz", + "integrity": "sha512-lyuxPGr/Wfhrlem2CL/UcnUc1zcqKAImBDzukY7Y5F/yQiNdko6+fRLevlw1HgMySw7f611UIY408EtxRSoK3Q==", + "dependencies": { + "js-tokens": "^3.0.0 || ^4.0.0" + }, + "bin": { + "loose-envify": "cli.js" + } + }, + "node_modules/lru-cache": { + "version": "5.1.1", + "resolved": "https://registry.npmjs.org/lru-cache/-/lru-cache-5.1.1.tgz", + "integrity": "sha512-KpNARQA3Iwv+jTA0utUVVbrh+Jlrr1Fv0e56GGzAFOXN7dk/FviaDW8LHmK52DlcH4WP2n6gI8vN1aesBFgo9w==", + "dev": true, + "dependencies": { + "yallist": "^3.0.2" + } + }, + "node_modules/math-intrinsics": { + "version": "1.1.0", + "resolved": "https://registry.npmjs.org/math-intrinsics/-/math-intrinsics-1.1.0.tgz", + "integrity": "sha512-/IXtbwEk5HTPyEwyKX6hGkYXxM9nbj64B+ilVJnC/R6B0pH5G4V3b0pVbL7DBj4tkhBAppbQUlf6F6Xl9LHu1g==", + "engines": { + "node": ">= 0.4" + } + }, + "node_modules/memoize-one": { + "version": "5.2.1", + "resolved": "https://registry.npmjs.org/memoize-one/-/memoize-one-5.2.1.tgz", + "integrity": "sha512-zYiwtZUcYyXKo/np96AGZAckk+FWWsUdJ3cHGGmld7+AhvcWmQyGCYUh1hc4Q/pkOhb65dQR/pqCyK0cOaHz4Q==" + }, + "node_modules/merge2": { + "version": "1.4.1", + "resolved": "https://registry.npmjs.org/merge2/-/merge2-1.4.1.tgz", + "integrity": "sha512-8q7VEgMJW4J8tcfVPy8g09NcQwZdbwFEqhe/WZkoIzjn/3TGDwtOCYtXGxA3O8tPzpczCCDgv+P2P5y00ZJOOg==", + "dev": true, + "engines": { + "node": ">= 8" + } + }, + "node_modules/micromatch": { + "version": "4.0.8", + "resolved": "https://registry.npmjs.org/micromatch/-/micromatch-4.0.8.tgz", + "integrity": "sha512-PXwfBhYu0hBCPw8Dn0E+WDYb7af3dSLVWKi3HGv84IdF4TyFoC0ysxFd0Goxw7nSv4T/PzEJQxsYsEiFCKo2BA==", + "dev": true, + "dependencies": { + "braces": "^3.0.3", + "picomatch": "^2.3.1" + }, + "engines": { + "node": ">=8.6" + } + }, + "node_modules/mime-db": { + "version": "1.52.0", + "resolved": "https://registry.npmjs.org/mime-db/-/mime-db-1.52.0.tgz", + "integrity": "sha512-sPU4uV7dYlvtWJxwwxHD0PuihVNiE7TyAbQ5SWxDCB9mUYvOgroQOwYQQOKPJ8CIbE+1ETVlOoK1UC2nU3gYvg==", + "engines": { + "node": ">= 0.6" + } + }, + "node_modules/mime-types": { + "version": "2.1.35", + "resolved": "https://registry.npmjs.org/mime-types/-/mime-types-2.1.35.tgz", + "integrity": "sha512-ZDY+bPm5zTTF+YpCrAU9nK0UgICYPT0QtT1NZWFv4s++TNkcgVaT0g6+4R2uI4MjQjzysHB1zxuWL50hzaeXiw==", + "dependencies": { + "mime-db": "1.52.0" + }, + "engines": { + "node": ">= 0.6" + } + }, + "node_modules/ms": { + "version": "2.1.3", + "resolved": "https://registry.npmjs.org/ms/-/ms-2.1.3.tgz", + "integrity": "sha512-6FlzubTLZG3J2a/NVCAleEhjzq5oxgHyaCU9yYXvcLsvoVaHJq/s5xXI6/XXP6tz7R9xAOtHnSO/tXtF3WRTlA==" + }, + "node_modules/mz": { + "version": "2.7.0", + "resolved": "https://registry.npmjs.org/mz/-/mz-2.7.0.tgz", + "integrity": "sha512-z81GNO7nnYMEhrGh9LeymoE4+Yr0Wn5McHIZMK5cfQCl+NDX08sCZgUc9/6MHni9IWuFLm1Z3HTCXu2z9fN62Q==", + "dev": true, + "dependencies": { + "any-promise": "^1.0.0", + "object-assign": "^4.0.1", + "thenify-all": "^1.0.0" + } + }, + "node_modules/nanoid": { + "version": "3.3.16", + "resolved": "https://registry.npmjs.org/nanoid/-/nanoid-3.3.16.tgz", + "integrity": "sha512-bzlKTyNJ7+LdGIIwy8ijFpIqEQIvafahV7eYykJ8Cvh42EdJeODoJ6gUJXpQJvej1BddH8OqTXZNE/KfbWAu8Q==", + "dev": true, + "funding": [ + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "bin": { + "nanoid": "bin/nanoid.cjs" + }, + "engines": { + "node": "^10 || ^12 || ^13.7 || ^14 || >=15.0.1" + } + }, + "node_modules/node-releases": { + "version": "2.0.51", + "resolved": "https://registry.npmjs.org/node-releases/-/node-releases-2.0.51.tgz", + "integrity": "sha512-wRNIrw4DmVLKQlbgOMdkMx27Wrpzes2hh5Jtbi2bjPd+4wJstWIqP5A+lscnqbm0xxmT5Bpg8Lec5ItEBwx6BQ==", + "dev": true, + "engines": { + "node": ">=18" + } + }, + "node_modules/normalize-path": { + "version": "3.0.0", + "resolved": "https://registry.npmjs.org/normalize-path/-/normalize-path-3.0.0.tgz", + "integrity": "sha512-6eZs5Ls3WtCisHWp9S2GUy8dqkpGi4BVSz3GaqiE6ezub0512ESztXUwUB6C6IKbQkY2Pnb/mD4WYojCRwcwLA==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/object-assign": { + "version": "4.1.1", + "resolved": "https://registry.npmjs.org/object-assign/-/object-assign-4.1.1.tgz", + "integrity": "sha512-rJgTQnkUnH1sFw8yT6VSU3zD3sWmu6sZhIseY8VX+GRu3P6F7Fu+JNDoXfklElbLJSnc3FUQHVe4cU5hj+BcUg==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/object-hash": { + "version": "3.0.0", + "resolved": "https://registry.npmjs.org/object-hash/-/object-hash-3.0.0.tgz", + "integrity": "sha512-RSn9F68PjH9HqtltsSnqYC1XXoWe9Bju5+213R98cNGttag9q9yAOTzdbsqvIa7aNm5WffBZFpWYr2aWrklWAw==", + "dev": true, + "engines": { + "node": ">= 6" + } + }, + "node_modules/path-parse": { + "version": "1.0.7", + "resolved": "https://registry.npmjs.org/path-parse/-/path-parse-1.0.7.tgz", + "integrity": "sha512-LDJzPVEEEPR+y48z93A0Ed0yXb8pAByGWo/k5YYdYgpY2/2EsOsksJrq7lOHxryrVOn1ejG6oAp8ahvOIQD8sw==", + "dev": true + }, + "node_modules/picocolors": { + "version": "1.1.1", + "resolved": "https://registry.npmjs.org/picocolors/-/picocolors-1.1.1.tgz", + "integrity": "sha512-xceH2snhtb5M9liqDsmEw56le376mTZkEX/jEb/RxNFyegNul7eNslCXP9FDj/Lcu0X8KEyMceP2ntpaHrDEVA==", + "dev": true + }, + "node_modules/picomatch": { + "version": "2.3.2", + "resolved": "https://registry.npmjs.org/picomatch/-/picomatch-2.3.2.tgz", + "integrity": "sha512-V7+vQEJ06Z+c5tSye8S+nHUfI51xoXIXjHQ99cQtKUkQqqO1kO/KCJUfZXuB47h/YBlDhah2H3hdUGXn8ie0oA==", + "dev": true, + "engines": { + "node": ">=8.6" + }, + "funding": { + "url": "https://github.com/sponsors/jonschlinkert" + } + }, + "node_modules/pify": { + "version": "2.3.0", + "resolved": "https://registry.npmjs.org/pify/-/pify-2.3.0.tgz", + "integrity": "sha512-udgsAY+fTnvv7kI7aaxbqwWNb0AHiB0qBO89PZKPkoTmGOgdbrHDKD+0B2X4uTfJ/FT1R09r9gTsjUjNJotuog==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/pirates": { + "version": "4.0.7", + "resolved": "https://registry.npmjs.org/pirates/-/pirates-4.0.7.tgz", + "integrity": "sha512-TfySrs/5nm8fQJDcBDuUng3VOUKsd7S+zqvbOTiGXHfxX4wK31ard+hoNuvkicM/2YFzlpDgABOevKSsB4G/FA==", + "dev": true, + "engines": { + "node": ">= 6" + } + }, + "node_modules/postcss": { + "version": "8.5.19", + "resolved": "https://registry.npmjs.org/postcss/-/postcss-8.5.19.tgz", + "integrity": "sha512-Mz8SaolMd8nB+G13WkORcxQKHZ/NE4xXevtkJHVuG+guo9/wYKlIMTKAqGdEmYOXR2ijPjTYNHssizdaVSUNdQ==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/postcss/" + }, + { + "type": "tidelift", + "url": "https://tidelift.com/funding/github/npm/postcss" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "nanoid": "^3.3.12", + "picocolors": "^1.1.1", + "source-map-js": "^1.2.1" + }, + "engines": { + "node": "^10 || ^12 || >=14" + } + }, + "node_modules/postcss-import": { + "version": "15.1.0", + "resolved": "https://registry.npmjs.org/postcss-import/-/postcss-import-15.1.0.tgz", + "integrity": "sha512-hpr+J05B2FVYUAXHeK1YyI267J/dDDhMU6B6civm8hSY1jYJnBXxzKDKDswzJmtLHryrjhnDjqqp/49t8FALew==", + "dev": true, + "dependencies": { + "postcss-value-parser": "^4.0.0", + "read-cache": "^1.0.0", + "resolve": "^1.1.7" + }, + "engines": { + "node": ">=14.0.0" + }, + "peerDependencies": { + "postcss": "^8.0.0" + } + }, + "node_modules/postcss-js": { + "version": "4.1.0", + "resolved": "https://registry.npmjs.org/postcss-js/-/postcss-js-4.1.0.tgz", + "integrity": "sha512-oIAOTqgIo7q2EOwbhb8UalYePMvYoIeRY2YKntdpFQXNosSu3vLrniGgmH9OKs/qAkfoj5oB3le/7mINW1LCfw==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/postcss/" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "camelcase-css": "^2.0.1" + }, + "engines": { + "node": "^12 || ^14 || >= 16" + }, + "peerDependencies": { + "postcss": "^8.4.21" + } + }, + "node_modules/postcss-load-config": { + "version": "6.0.1", + "resolved": "https://registry.npmjs.org/postcss-load-config/-/postcss-load-config-6.0.1.tgz", + "integrity": "sha512-oPtTM4oerL+UXmx+93ytZVN82RrlY/wPUV8IeDxFrzIjXOLF1pN+EmKPLbubvKHT2HC20xXsCAH2Z+CKV6Oz/g==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/postcss/" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "lilconfig": "^3.1.1" + }, + "engines": { + "node": ">= 18" + }, + "peerDependencies": { + "jiti": ">=1.21.0", + "postcss": ">=8.0.9", + "tsx": "^4.8.1", + "yaml": "^2.4.2" + }, + "peerDependenciesMeta": { + "jiti": { + "optional": true + }, + "postcss": { + "optional": true + }, + "tsx": { + "optional": true + }, + "yaml": { + "optional": true + } + } + }, + "node_modules/postcss-nested": { + "version": "6.2.0", + "resolved": "https://registry.npmjs.org/postcss-nested/-/postcss-nested-6.2.0.tgz", + "integrity": "sha512-HQbt28KulC5AJzG+cZtj9kvKB93CFCdLvog1WFLf1D+xmMvPGlBstkpTEZfK5+AN9hfJocyBFCNiqyS48bpgzQ==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/postcss/" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "postcss-selector-parser": "^6.1.1" + }, + "engines": { + "node": ">=12.0" + }, + "peerDependencies": { + "postcss": "^8.2.14" + } + }, + "node_modules/postcss-selector-parser": { + "version": "6.1.4", + "resolved": "https://registry.npmjs.org/postcss-selector-parser/-/postcss-selector-parser-6.1.4.tgz", + "integrity": "sha512-bIoJLOmjCO1S9XdY/DcnR5hJxvrDir1PbGChrzXG3vw0/FOliy/fA3dmdhQ441kah4gKv+TwckGzex6wNS5cnQ==", + "dev": true, + "dependencies": { + "cssesc": "^3.0.0", + "util-deprecate": "^1.0.2" + }, + "engines": { + "node": ">=4" + } + }, + "node_modules/postcss-value-parser": { + "version": "4.2.0", + "resolved": "https://registry.npmjs.org/postcss-value-parser/-/postcss-value-parser-4.2.0.tgz", + "integrity": "sha512-1NNCs6uurfkVbeXG4S8JFT9t19m45ICnif8zWLd5oPSZ50QnwMfK+H3jv408d4jw/7Bttv5axS5IiHoLaVNHeQ==", + "dev": true + }, + "node_modules/proxy-from-env": { + "version": "2.1.0", + "resolved": "https://registry.npmjs.org/proxy-from-env/-/proxy-from-env-2.1.0.tgz", + "integrity": "sha512-cJ+oHTW1VAEa8cJslgmUZrc+sjRKgAKl3Zyse6+PV38hZe/V6Z14TbCuXcan9F9ghlz4QrFr2c92TNF82UkYHA==", + "engines": { + "node": ">=10" + } + }, + "node_modules/queue-microtask": { + "version": "1.2.3", + "resolved": "https://registry.npmjs.org/queue-microtask/-/queue-microtask-1.2.3.tgz", + "integrity": "sha512-NuaNSa6flKT5JaSYQzJok04JzTL1CA6aGhv5rfLW3PgqA+M2ChpZQnAC8h8i4ZFkBS8X5RqkDBHA7r4hej3K9A==", + "dev": true, + "funding": [ + { + "type": "github", + "url": "https://github.com/sponsors/feross" + }, + { + "type": "patreon", + "url": "https://www.patreon.com/feross" + }, + { + "type": "consulting", + "url": "https://feross.org/support" + } + ] + }, + "node_modules/rc-cascader": { + "version": "3.34.0", + "resolved": "https://registry.npmjs.org/rc-cascader/-/rc-cascader-3.34.0.tgz", + "integrity": "sha512-KpXypcvju9ptjW9FaN2NFcA2QH9E9LHKq169Y0eWtH4e/wHQ5Wh5qZakAgvb8EKZ736WZ3B0zLLOBsrsja5Dag==", + "dependencies": { + "@babel/runtime": "^7.25.7", + "classnames": "^2.3.1", + "rc-select": "~14.16.2", + "rc-tree": "~5.13.0", + "rc-util": "^5.43.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-checkbox": { + "version": "3.5.0", + "resolved": "https://registry.npmjs.org/rc-checkbox/-/rc-checkbox-3.5.0.tgz", + "integrity": "sha512-aOAQc3E98HteIIsSqm6Xk2FPKIER6+5vyEFMZfo73TqM+VVAIqOkHoPjgKLqSNtVLWScoaM7vY2ZrGEheI79yg==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.3.2", + "rc-util": "^5.25.2" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-collapse": { + "version": "3.9.0", + "resolved": "https://registry.npmjs.org/rc-collapse/-/rc-collapse-3.9.0.tgz", + "integrity": "sha512-swDdz4QZ4dFTo4RAUMLL50qP0EY62N2kvmk2We5xYdRwcRn8WcYtuetCJpwpaCbUfUt5+huLpVxhvmnK+PHrkA==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "2.x", + "rc-motion": "^2.3.4", + "rc-util": "^5.27.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-dialog": { + "version": "9.6.0", + "resolved": "https://registry.npmjs.org/rc-dialog/-/rc-dialog-9.6.0.tgz", + "integrity": "sha512-ApoVi9Z8PaCQg6FsUzS8yvBEQy0ZL2PkuvAgrmohPkN3okps5WZ5WQWPc1RNuiOKaAYv8B97ACdsFU5LizzCqg==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "@rc-component/portal": "^1.0.0-8", + "classnames": "^2.2.6", + "rc-motion": "^2.3.0", + "rc-util": "^5.21.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-drawer": { + "version": "7.3.0", + "resolved": "https://registry.npmjs.org/rc-drawer/-/rc-drawer-7.3.0.tgz", + "integrity": "sha512-DX6CIgiBWNpJIMGFO8BAISFkxiuKitoizooj4BDyee8/SnBn0zwO2FHrNDpqqepj0E/TFTDpmEBCyFuTgC7MOg==", + "dependencies": { + "@babel/runtime": "^7.23.9", + "@rc-component/portal": "^1.1.1", + "classnames": "^2.2.6", + "rc-motion": "^2.6.1", + "rc-util": "^5.38.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-dropdown": { + "version": "4.2.1", + "resolved": "https://registry.npmjs.org/rc-dropdown/-/rc-dropdown-4.2.1.tgz", + "integrity": "sha512-YDAlXsPv3I1n42dv1JpdM7wJ+gSUBfeyPK59ZpBD9jQhK9jVuxpjj3NmWQHOBceA1zEPVX84T2wbdb2SD0UjmA==", + "dependencies": { + "@babel/runtime": "^7.18.3", + "@rc-component/trigger": "^2.0.0", + "classnames": "^2.2.6", + "rc-util": "^5.44.1" + }, + "peerDependencies": { + "react": ">=16.11.0", + "react-dom": ">=16.11.0" + } + }, + "node_modules/rc-field-form": { + "version": "2.7.1", + "resolved": "https://registry.npmjs.org/rc-field-form/-/rc-field-form-2.7.1.tgz", + "integrity": "sha512-vKeSifSJ6HoLaAB+B8aq/Qgm8a3dyxROzCtKNCsBQgiverpc4kWDQihoUwzUj+zNWJOykwSY4dNX3QrGwtVb9A==", + "dependencies": { + "@babel/runtime": "^7.18.0", + "@rc-component/async-validator": "^5.0.3", + "rc-util": "^5.32.2" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-image": { + "version": "7.12.0", + "resolved": "https://registry.npmjs.org/rc-image/-/rc-image-7.12.0.tgz", + "integrity": "sha512-cZ3HTyyckPnNnUb9/DRqduqzLfrQRyi+CdHjdqgsyDpI3Ln5UX1kXnAhPBSJj9pVRzwRFgqkN7p9b6HBDjmu/Q==", + "dependencies": { + "@babel/runtime": "^7.11.2", + "@rc-component/portal": "^1.0.2", + "classnames": "^2.2.6", + "rc-dialog": "~9.6.0", + "rc-motion": "^2.6.2", + "rc-util": "^5.34.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-input": { + "version": "1.8.0", + "resolved": "https://registry.npmjs.org/rc-input/-/rc-input-1.8.0.tgz", + "integrity": "sha512-KXvaTbX+7ha8a/k+eg6SYRVERK0NddX8QX7a7AnRvUa/rEH0CNMlpcBzBkhI0wp2C8C4HlMoYl8TImSN+fuHKA==", + "dependencies": { + "@babel/runtime": "^7.11.1", + "classnames": "^2.2.1", + "rc-util": "^5.18.1" + }, + "peerDependencies": { + "react": ">=16.0.0", + "react-dom": ">=16.0.0" + } + }, + "node_modules/rc-input-number": { + "version": "9.5.0", + "resolved": "https://registry.npmjs.org/rc-input-number/-/rc-input-number-9.5.0.tgz", + "integrity": "sha512-bKaEvB5tHebUURAEXw35LDcnRZLq3x1k7GxfAqBMzmpHkDGzjAtnUL8y4y5N15rIFIg5IJgwr211jInl3cipag==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "@rc-component/mini-decimal": "^1.0.1", + "classnames": "^2.2.5", + "rc-input": "~1.8.0", + "rc-util": "^5.40.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-mentions": { + "version": "2.20.0", + "resolved": "https://registry.npmjs.org/rc-mentions/-/rc-mentions-2.20.0.tgz", + "integrity": "sha512-w8HCMZEh3f0nR8ZEd466ATqmXFCMGMN5UFCzEUL0bM/nGw/wOS2GgRzKBcm19K++jDyuWCOJOdgcKGXU3fXfbQ==", + "dependencies": { + "@babel/runtime": "^7.22.5", + "@rc-component/trigger": "^2.0.0", + "classnames": "^2.2.6", + "rc-input": "~1.8.0", + "rc-menu": "~9.16.0", + "rc-textarea": "~1.10.0", + "rc-util": "^5.34.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-menu": { + "version": "9.16.1", + "resolved": "https://registry.npmjs.org/rc-menu/-/rc-menu-9.16.1.tgz", + "integrity": "sha512-ghHx6/6Dvp+fw8CJhDUHFHDJ84hJE3BXNCzSgLdmNiFErWSOaZNsihDAsKq9ByTALo/xkNIwtDFGIl6r+RPXBg==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "@rc-component/trigger": "^2.0.0", + "classnames": "2.x", + "rc-motion": "^2.4.3", + "rc-overflow": "^1.3.1", + "rc-util": "^5.27.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-motion": { + "version": "2.9.5", + "resolved": "https://registry.npmjs.org/rc-motion/-/rc-motion-2.9.5.tgz", + "integrity": "sha512-w+XTUrfh7ArbYEd2582uDrEhmBHwK1ZENJiSJVb7uRxdE7qJSYjbO2eksRXmndqyKqKoYPc9ClpPh5242mV1vA==", + "dependencies": { + "@babel/runtime": "^7.11.1", + "classnames": "^2.2.1", + "rc-util": "^5.44.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-notification": { + "version": "5.6.4", + "resolved": "https://registry.npmjs.org/rc-notification/-/rc-notification-5.6.4.tgz", + "integrity": "sha512-KcS4O6B4qzM3KH7lkwOB7ooLPZ4b6J+VMmQgT51VZCeEcmghdeR4IrMcFq0LG+RPdnbe/ArT086tGM8Snimgiw==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "2.x", + "rc-motion": "^2.9.0", + "rc-util": "^5.20.1" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-overflow": { + "version": "1.5.0", + "resolved": "https://registry.npmjs.org/rc-overflow/-/rc-overflow-1.5.0.tgz", + "integrity": "sha512-Lm/v9h0LymeUYJf0x39OveU52InkdRXqnn2aYXfWmo8WdOonIKB2kfau+GF0fWq6jPgtdO9yMqveGcK6aIhJmg==", + "dependencies": { + "@babel/runtime": "^7.11.1", + "classnames": "^2.2.1", + "rc-resize-observer": "^1.0.0", + "rc-util": "^5.37.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-pagination": { + "version": "5.1.0", + "resolved": "https://registry.npmjs.org/rc-pagination/-/rc-pagination-5.1.0.tgz", + "integrity": "sha512-8416Yip/+eclTFdHXLKTxZvn70duYVGTvUUWbckCCZoIl3jagqke3GLsFrMs0bsQBikiYpZLD9206Ej4SOdOXQ==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.3.2", + "rc-util": "^5.38.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-picker": { + "version": "4.11.3", + "resolved": "https://registry.npmjs.org/rc-picker/-/rc-picker-4.11.3.tgz", + "integrity": "sha512-MJ5teb7FlNE0NFHTncxXQ62Y5lytq6sh5nUw0iH8OkHL/TjARSEvSHpr940pWgjGANpjCwyMdvsEV55l5tYNSg==", + "dependencies": { + "@babel/runtime": "^7.24.7", + "@rc-component/trigger": "^2.0.0", + "classnames": "^2.2.1", + "rc-overflow": "^1.3.2", + "rc-resize-observer": "^1.4.0", + "rc-util": "^5.43.0" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "date-fns": ">= 2.x", + "dayjs": ">= 1.x", + "luxon": ">= 3.x", + "moment": ">= 2.x", + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + }, + "peerDependenciesMeta": { + "date-fns": { + "optional": true + }, + "dayjs": { + "optional": true + }, + "luxon": { + "optional": true + }, + "moment": { + "optional": true + } + } + }, + "node_modules/rc-progress": { + "version": "4.0.0", + "resolved": "https://registry.npmjs.org/rc-progress/-/rc-progress-4.0.0.tgz", + "integrity": "sha512-oofVMMafOCokIUIBnZLNcOZFsABaUw8PPrf1/y0ZBvKZNpOiu5h4AO9vv11Sw0p4Hb3D0yGWuEattcQGtNJ/aw==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.2.6", + "rc-util": "^5.16.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-rate": { + "version": "2.13.1", + "resolved": "https://registry.npmjs.org/rc-rate/-/rc-rate-2.13.1.tgz", + "integrity": "sha512-QUhQ9ivQ8Gy7mtMZPAjLbxBt5y9GRp65VcUyGUMF3N3fhiftivPHdpuDIaWIMOTEprAjZPC08bls1dQB+I1F2Q==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.2.5", + "rc-util": "^5.0.1" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-resize-observer": { + "version": "1.4.3", + "resolved": "https://registry.npmjs.org/rc-resize-observer/-/rc-resize-observer-1.4.3.tgz", + "integrity": "sha512-YZLjUbyIWox8E9i9C3Tm7ia+W7euPItNWSPX5sCcQTYbnwDb5uNpnLHQCG1f22oZWUhLw4Mv2tFmeWe68CDQRQ==", + "dependencies": { + "@babel/runtime": "^7.20.7", + "classnames": "^2.2.1", + "rc-util": "^5.44.1", + "resize-observer-polyfill": "^1.5.1" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-segmented": { + "version": "2.7.1", + "resolved": "https://registry.npmjs.org/rc-segmented/-/rc-segmented-2.7.1.tgz", + "integrity": "sha512-izj1Nw/Dw2Vb7EVr+D/E9lUTkBe+kKC+SAFSU9zqr7WV2W5Ktaa9Gc7cB2jTqgk8GROJayltaec+DBlYKc6d+g==", + "dependencies": { + "@babel/runtime": "^7.11.1", + "classnames": "^2.2.1", + "rc-motion": "^2.4.4", + "rc-util": "^5.17.0" + }, + "peerDependencies": { + "react": ">=16.0.0", + "react-dom": ">=16.0.0" + } + }, + "node_modules/rc-select": { + "version": "14.16.8", + "resolved": "https://registry.npmjs.org/rc-select/-/rc-select-14.16.8.tgz", + "integrity": "sha512-NOV5BZa1wZrsdkKaiK7LHRuo5ZjZYMDxPP6/1+09+FB4KoNi8jcG1ZqLE3AVCxEsYMBe65OBx71wFoHRTP3LRg==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "@rc-component/trigger": "^2.1.1", + "classnames": "2.x", + "rc-motion": "^2.0.1", + "rc-overflow": "^1.3.1", + "rc-util": "^5.16.1", + "rc-virtual-list": "^3.5.2" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": "*", + "react-dom": "*" + } + }, + "node_modules/rc-slider": { + "version": "11.1.9", + "resolved": "https://registry.npmjs.org/rc-slider/-/rc-slider-11.1.9.tgz", + "integrity": "sha512-h8IknhzSh3FEM9u8ivkskh+Ef4Yo4JRIY2nj7MrH6GQmrwV6mcpJf5/4KgH5JaVI1H3E52yCdpOlVyGZIeph5A==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.2.5", + "rc-util": "^5.36.0" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-steps": { + "version": "6.0.1", + "resolved": "https://registry.npmjs.org/rc-steps/-/rc-steps-6.0.1.tgz", + "integrity": "sha512-lKHL+Sny0SeHkQKKDJlAjV5oZ8DwCdS2hFhAkIjuQt1/pB81M0cA0ErVFdHq9+jmPmFw1vJB2F5NBzFXLJxV+g==", + "dependencies": { + "@babel/runtime": "^7.16.7", + "classnames": "^2.2.3", + "rc-util": "^5.16.1" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-switch": { + "version": "4.1.0", + "resolved": "https://registry.npmjs.org/rc-switch/-/rc-switch-4.1.0.tgz", + "integrity": "sha512-TI8ufP2Az9oEbvyCeVE4+90PDSljGyuwix3fV58p7HV2o4wBnVToEyomJRVyTaZeqNPAp+vqeo4Wnj5u0ZZQBg==", + "dependencies": { + "@babel/runtime": "^7.21.0", + "classnames": "^2.2.1", + "rc-util": "^5.30.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-table": { + "version": "7.54.0", + "resolved": "https://registry.npmjs.org/rc-table/-/rc-table-7.54.0.tgz", + "integrity": "sha512-/wDTkki6wBTjwylwAGjpLKYklKo9YgjZwAU77+7ME5mBoS32Q4nAwoqhA2lSge6fobLW3Tap6uc5xfwaL2p0Sw==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "@rc-component/context": "^1.4.0", + "classnames": "^2.2.5", + "rc-resize-observer": "^1.1.0", + "rc-util": "^5.44.3", + "rc-virtual-list": "^3.14.2" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-tabs": { + "version": "15.7.0", + "resolved": "https://registry.npmjs.org/rc-tabs/-/rc-tabs-15.7.0.tgz", + "integrity": "sha512-ZepiE+6fmozYdWf/9gVp7k56PKHB1YYoDsKeQA1CBlJ/POIhjkcYiv0AGP0w2Jhzftd3AVvZP/K+V+Lpi2ankA==", + "dependencies": { + "@babel/runtime": "^7.11.2", + "classnames": "2.x", + "rc-dropdown": "~4.2.0", + "rc-menu": "~9.16.0", + "rc-motion": "^2.6.2", + "rc-resize-observer": "^1.0.0", + "rc-util": "^5.34.1" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-textarea": { + "version": "1.10.2", + "resolved": "https://registry.npmjs.org/rc-textarea/-/rc-textarea-1.10.2.tgz", + "integrity": "sha512-HfaeXiaSlpiSp0I/pvWpecFEHpVysZ9tpDLNkxQbMvMz6gsr7aVZ7FpWP9kt4t7DB+jJXesYS0us1uPZnlRnwQ==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "^2.2.1", + "rc-input": "~1.8.0", + "rc-resize-observer": "^1.0.0", + "rc-util": "^5.27.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-tooltip": { + "version": "6.4.0", + "resolved": "https://registry.npmjs.org/rc-tooltip/-/rc-tooltip-6.4.0.tgz", + "integrity": "sha512-kqyivim5cp8I5RkHmpsp1Nn/Wk+1oeloMv9c7LXNgDxUpGm+RbXJGL+OPvDlcRnx9DBeOe4wyOIl4OKUERyH1g==", + "dependencies": { + "@babel/runtime": "^7.11.2", + "@rc-component/trigger": "^2.0.0", + "classnames": "^2.3.1", + "rc-util": "^5.44.3" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-tree": { + "version": "5.13.1", + "resolved": "https://registry.npmjs.org/rc-tree/-/rc-tree-5.13.1.tgz", + "integrity": "sha512-FNhIefhftobCdUJshO7M8uZTA9F4OPGVXqGfZkkD/5soDeOhwO06T/aKTrg0WD8gRg/pyfq+ql3aMymLHCTC4A==", + "dependencies": { + "@babel/runtime": "^7.10.1", + "classnames": "2.x", + "rc-motion": "^2.0.1", + "rc-util": "^5.16.1", + "rc-virtual-list": "^3.5.1" + }, + "engines": { + "node": ">=10.x" + }, + "peerDependencies": { + "react": "*", + "react-dom": "*" + } + }, + "node_modules/rc-tree-select": { + "version": "5.27.0", + "resolved": "https://registry.npmjs.org/rc-tree-select/-/rc-tree-select-5.27.0.tgz", + "integrity": "sha512-2qTBTzwIT7LRI1o7zLyrCzmo5tQanmyGbSaGTIf7sYimCklAToVVfpMC6OAldSKolcnjorBYPNSKQqJmN3TCww==", + "dependencies": { + "@babel/runtime": "^7.25.7", + "classnames": "2.x", + "rc-select": "~14.16.2", + "rc-tree": "~5.13.0", + "rc-util": "^5.43.0" + }, + "peerDependencies": { + "react": "*", + "react-dom": "*" + } + }, + "node_modules/rc-upload": { + "version": "4.11.0", + "resolved": "https://registry.npmjs.org/rc-upload/-/rc-upload-4.11.0.tgz", + "integrity": "sha512-ZUyT//2JAehfHzjWowqROcwYJKnZkIUGWaTE/VogVrepSl7AFNbQf4+zGfX4zl9Vrj/Jm8scLO0R6UlPDKK4wA==", + "dependencies": { + "@babel/runtime": "^7.18.3", + "classnames": "^2.2.5", + "rc-util": "^5.2.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-util": { + "version": "5.44.4", + "resolved": "https://registry.npmjs.org/rc-util/-/rc-util-5.44.4.tgz", + "integrity": "sha512-resueRJzmHG9Q6rI/DfK6Kdv9/Lfls05vzMs1Sk3M2P+3cJa+MakaZyWY8IPfehVuhPJFKrIY1IK4GqbiaiY5w==", + "dependencies": { + "@babel/runtime": "^7.18.3", + "react-is": "^18.2.0" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/rc-virtual-list": { + "version": "3.19.2", + "resolved": "https://registry.npmjs.org/rc-virtual-list/-/rc-virtual-list-3.19.2.tgz", + "integrity": "sha512-Ys6NcjwGkuwkeaWBDqfI3xWuZ7rDiQXlH1o2zLfFzATfEgXcqpk8CkgMfbJD81McqjcJVez25a3kPxCR807evA==", + "dependencies": { + "@babel/runtime": "^7.20.0", + "classnames": "^2.2.6", + "rc-resize-observer": "^1.0.0", + "rc-util": "^5.36.0" + }, + "engines": { + "node": ">=8.x" + }, + "peerDependencies": { + "react": ">=16.9.0", + "react-dom": ">=16.9.0" + } + }, + "node_modules/react": { + "version": "18.3.1", + "resolved": "https://registry.npmjs.org/react/-/react-18.3.1.tgz", + "integrity": "sha512-wS+hAgJShR0KhEvPJArfuPVN1+Hz1t0Y6n5jLrGQbkb4urgPE/0Rve+1kMB1v/oWgHgm4WIcV+i7F2pTVj+2iQ==", + "dependencies": { + "loose-envify": "^1.1.0" + }, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/react-dom": { + "version": "18.3.1", + "resolved": "https://registry.npmjs.org/react-dom/-/react-dom-18.3.1.tgz", + "integrity": "sha512-5m4nQKp+rZRb09LNH59GM4BxTh9251/ylbKIbpe7TpGxfJ+9kv6BLkLBXIjjspbgbnIBNqlI23tRnTWT0snUIw==", + "dependencies": { + "loose-envify": "^1.1.0", + "scheduler": "^0.23.2" + }, + "peerDependencies": { + "react": "^18.3.1" + } + }, + "node_modules/react-is": { + "version": "18.3.1", + "resolved": "https://registry.npmjs.org/react-is/-/react-is-18.3.1.tgz", + "integrity": "sha512-/LLMVyas0ljjAtoYiPqYiL8VWXzUUdThrmU5+n20DZv+a+ClRoevUzw5JxU+Ieh5/c87ytoTBV9G1FiKfNJdmg==" + }, + "node_modules/react-refresh": { + "version": "0.17.0", + "resolved": "https://registry.npmjs.org/react-refresh/-/react-refresh-0.17.0.tgz", + "integrity": "sha512-z6F7K9bV85EfseRCp2bzrpyQ0Gkw1uLoCel9XBVWPg/TjRj94SkJzUTGfOa4bs7iJvBWtQG0Wq7wnI0syw3EBQ==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/react-router": { + "version": "6.30.4", + "resolved": "https://registry.npmjs.org/react-router/-/react-router-6.30.4.tgz", + "integrity": "sha512-SVUsDe+DybHM/WmYKIVYhZh1o5Dcuf16yM6WjG02Q9XVFMZIJyHYhwrr6bFBXZkVP6z69kNkMyBCujt8FaFLJA==", + "dependencies": { + "@remix-run/router": "1.23.3" + }, + "engines": { + "node": ">=14.0.0" + }, + "peerDependencies": { + "react": ">=16.8" + } + }, + "node_modules/react-router-dom": { + "version": "6.30.4", + "resolved": "https://registry.npmjs.org/react-router-dom/-/react-router-dom-6.30.4.tgz", + "integrity": "sha512-q4HvNl+mmDdkS0g+MqiBZNteQJCuimWoOyHMy4T/RQLAn9Z29+E91QXRaxOujeMl2HTzRSS0KFPd7lxX3PjV0Q==", + "dependencies": { + "@remix-run/router": "1.23.3", + "react-router": "6.30.4" + }, + "engines": { + "node": ">=14.0.0" + }, + "peerDependencies": { + "react": ">=16.8", + "react-dom": ">=16.8" + } + }, + "node_modules/react-window": { + "version": "1.8.11", + "resolved": "https://registry.npmjs.org/react-window/-/react-window-1.8.11.tgz", + "integrity": "sha512-+SRbUVT2scadgFSWx+R1P754xHPEqvcfSfVX10QYg6POOz+WNgkN48pS+BtZNIMGiL1HYrSEiCkwsMS15QogEQ==", + "dependencies": { + "@babel/runtime": "^7.0.0", + "memoize-one": ">=3.1.1 <6" + }, + "engines": { + "node": ">8.0.0" + }, + "peerDependencies": { + "react": "^15.0.0 || ^16.0.0 || ^17.0.0 || ^18.0.0 || ^19.0.0", + "react-dom": "^15.0.0 || ^16.0.0 || ^17.0.0 || ^18.0.0 || ^19.0.0" + } + }, + "node_modules/read-cache": { + "version": "1.0.0", + "resolved": "https://registry.npmjs.org/read-cache/-/read-cache-1.0.0.tgz", + "integrity": "sha512-Owdv/Ft7IjOgm/i0xvNDZ1LrRANRfew4b2prF3OWMQLxLfu3bS8FVhCsrSCMK4lR56Y9ya+AThoTpDCTxCmpRA==", + "dev": true, + "dependencies": { + "pify": "^2.3.0" + } + }, + "node_modules/readdirp": { + "version": "3.6.0", + "resolved": "https://registry.npmjs.org/readdirp/-/readdirp-3.6.0.tgz", + "integrity": "sha512-hOS089on8RduqdbhvQ5Z37A0ESjsqz6qnRcffsMU3495FuTdqSm+7bhJ29JvIOsBDEEnan5DPu9t3To9VRlMzA==", + "dev": true, + "dependencies": { + "picomatch": "^2.2.1" + }, + "engines": { + "node": ">=8.10.0" + } + }, + "node_modules/resize-observer-polyfill": { + "version": "1.5.1", + "resolved": "https://registry.npmjs.org/resize-observer-polyfill/-/resize-observer-polyfill-1.5.1.tgz", + "integrity": "sha512-LwZrotdHOo12nQuZlHEmtuXdqGoOD0OhaxopaNFxWzInpEgaLWoVuAMbTzixuosCx2nEG58ngzW3vxdWoxIgdg==" + }, + "node_modules/resolve": { + "version": "1.22.12", + "resolved": "https://registry.npmjs.org/resolve/-/resolve-1.22.12.tgz", + "integrity": "sha512-TyeJ1zif53BPfHootBGwPRYT1RUt6oGWsaQr8UyZW/eAm9bKoijtvruSDEmZHm92CwS9nj7/fWttqPCgzep8CA==", + "dev": true, + "dependencies": { + "es-errors": "^1.3.0", + "is-core-module": "^2.16.1", + "path-parse": "^1.0.7", + "supports-preserve-symlinks-flag": "^1.0.0" + }, + "bin": { + "resolve": "bin/resolve" + }, + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/reusify": { + "version": "1.1.0", + "resolved": "https://registry.npmjs.org/reusify/-/reusify-1.1.0.tgz", + "integrity": "sha512-g6QUff04oZpHs0eG5p83rFLhHeV00ug/Yf9nZM6fLeUrPguBTkTQOdpAWWspMh55TZfVQDPaN3NQJfbVRAxdIw==", + "dev": true, + "engines": { + "iojs": ">=1.0.0", + "node": ">=0.10.0" + } + }, + "node_modules/rollup": { + "version": "4.62.2", + "resolved": "https://registry.npmjs.org/rollup/-/rollup-4.62.2.tgz", + "integrity": "sha512-RFnrW4lhXA3s3eqHDZvN654g8OTjzRfqpIRJYczCGB6HzphckVAi/Qh4tbPUbRuDi7s1Llv8g/NspLkttY3gTA==", + "dev": true, + "dependencies": { + "@types/estree": "1.0.9" + }, + "bin": { + "rollup": "dist/bin/rollup" + }, + "engines": { + "node": ">=18.0.0", + "npm": ">=8.0.0" + }, + "optionalDependencies": { + "@rollup/rollup-android-arm-eabi": "4.62.2", + "@rollup/rollup-android-arm64": "4.62.2", + "@rollup/rollup-darwin-arm64": "4.62.2", + "@rollup/rollup-darwin-x64": "4.62.2", + "@rollup/rollup-freebsd-arm64": "4.62.2", + "@rollup/rollup-freebsd-x64": "4.62.2", + "@rollup/rollup-linux-arm-gnueabihf": "4.62.2", + "@rollup/rollup-linux-arm-musleabihf": "4.62.2", + "@rollup/rollup-linux-arm64-gnu": "4.62.2", + "@rollup/rollup-linux-arm64-musl": "4.62.2", + "@rollup/rollup-linux-loong64-gnu": "4.62.2", + "@rollup/rollup-linux-loong64-musl": "4.62.2", + "@rollup/rollup-linux-ppc64-gnu": "4.62.2", + "@rollup/rollup-linux-ppc64-musl": "4.62.2", + "@rollup/rollup-linux-riscv64-gnu": "4.62.2", + "@rollup/rollup-linux-riscv64-musl": "4.62.2", + "@rollup/rollup-linux-s390x-gnu": "4.62.2", + "@rollup/rollup-linux-x64-gnu": "4.62.2", + "@rollup/rollup-linux-x64-musl": "4.62.2", + "@rollup/rollup-openbsd-x64": "4.62.2", + "@rollup/rollup-openharmony-arm64": "4.62.2", + "@rollup/rollup-win32-arm64-msvc": "4.62.2", + "@rollup/rollup-win32-ia32-msvc": "4.62.2", + "@rollup/rollup-win32-x64-gnu": "4.62.2", + "@rollup/rollup-win32-x64-msvc": "4.62.2", + "fsevents": "~2.3.2" + } + }, + "node_modules/run-parallel": { + "version": "1.2.0", + "resolved": "https://registry.npmjs.org/run-parallel/-/run-parallel-1.2.0.tgz", + "integrity": "sha512-5l4VyZR86LZ/lDxZTR6jqL8AFE2S0IFLMP26AbjsLVADxHdhB/c0GUsH+y39UfCi3dzz8OlQuPmnaJOMoDHQBA==", + "dev": true, + "funding": [ + { + "type": "github", + "url": "https://github.com/sponsors/feross" + }, + { + "type": "patreon", + "url": "https://www.patreon.com/feross" + }, + { + "type": "consulting", + "url": "https://feross.org/support" + } + ], + "dependencies": { + "queue-microtask": "^1.2.2" + } + }, + "node_modules/scheduler": { + "version": "0.23.2", + "resolved": "https://registry.npmjs.org/scheduler/-/scheduler-0.23.2.tgz", + "integrity": "sha512-UOShsPwz7NrMUqhR6t0hWjFduvOzbtv7toDH1/hIrfRNIDBnnBWd0CwJTGvTpngVlmwGCdP9/Zl/tVrDqcuYzQ==", + "dependencies": { + "loose-envify": "^1.1.0" + } + }, + "node_modules/scroll-into-view-if-needed": { + "version": "3.1.0", + "resolved": "https://registry.npmjs.org/scroll-into-view-if-needed/-/scroll-into-view-if-needed-3.1.0.tgz", + "integrity": "sha512-49oNpRjWRvnU8NyGVmUaYG4jtTkNonFZI86MmGRDqBphEK2EXT9gdEUoQPZhuBM8yWHxCWbobltqYO5M4XrUvQ==", + "dependencies": { + "compute-scroll-into-view": "^3.0.2" + } + }, + "node_modules/semver": { + "version": "6.3.1", + "resolved": "https://registry.npmjs.org/semver/-/semver-6.3.1.tgz", + "integrity": "sha512-BR7VvDCVHO+q2xBEWskxS6DJE1qRnb7DxzUrogb71CWoSficBxYsiAGd+Kl0mmq/MprG9yArRkyrQxTO6XjMzA==", + "dev": true, + "bin": { + "semver": "bin/semver.js" + } + }, + "node_modules/size-sensor": { + "version": "1.0.3", + "resolved": "https://registry.npmjs.org/size-sensor/-/size-sensor-1.0.3.tgz", + "integrity": "sha512-+k9mJ2/rQMiRmQUcjn+qznch260leIXY8r4FyYKKyRBO/s5UoeMAHGkCJyE1R/4wrIhTJONfyloY55SkE7ve3A==" + }, + "node_modules/source-map-js": { + "version": "1.2.1", + "resolved": "https://registry.npmjs.org/source-map-js/-/source-map-js-1.2.1.tgz", + "integrity": "sha512-UXWMKhLOwVKb728IUtQPXxfYU+usdybtUrK/8uGE8CQMvrhOpwvzDBwj0QhSL7MQc7vIsISBG8VQ8+IDQxpfQA==", + "dev": true, + "engines": { + "node": ">=0.10.0" + } + }, + "node_modules/string-convert": { + "version": "0.2.1", + "resolved": "https://registry.npmjs.org/string-convert/-/string-convert-0.2.1.tgz", + "integrity": "sha512-u/1tdPl4yQnPBjnVrmdLo9gtuLvELKsAoRapekWggdiQNvvvum+jYF329d84NAa660KQw7pB2n36KrIKVoXa3A==" + }, + "node_modules/stylis": { + "version": "4.4.0", + "resolved": "https://registry.npmjs.org/stylis/-/stylis-4.4.0.tgz", + "integrity": "sha512-5Z9ZpRzfuH6l/UAvCPAPUo3665Nk2wLaZU3x+TLHKVzIz33+sbJqbtrYoC3KD4/uVOr2Zp+L0LySezP9OHV9yA==" + }, + "node_modules/sucrase": { + "version": "3.35.1", + "resolved": "https://registry.npmjs.org/sucrase/-/sucrase-3.35.1.tgz", + "integrity": "sha512-DhuTmvZWux4H1UOnWMB3sk0sbaCVOoQZjv8u1rDoTV0HTdGem9hkAZtl4JZy8P2z4Bg0nT+YMeOFyVr4zcG5Tw==", + "dev": true, + "dependencies": { + "@jridgewell/gen-mapping": "^0.3.2", + "commander": "^4.0.0", + "lines-and-columns": "^1.1.6", + "mz": "^2.7.0", + "pirates": "^4.0.1", + "tinyglobby": "^0.2.11", + "ts-interface-checker": "^0.1.9" + }, + "bin": { + "sucrase": "bin/sucrase", + "sucrase-node": "bin/sucrase-node" + }, + "engines": { + "node": ">=16 || 14 >=14.17" + } + }, + "node_modules/supports-preserve-symlinks-flag": { + "version": "1.0.0", + "resolved": "https://registry.npmjs.org/supports-preserve-symlinks-flag/-/supports-preserve-symlinks-flag-1.0.0.tgz", + "integrity": "sha512-ot0WnXS9fgdkgIcePe6RHNk1WA8+muPa6cSjeR3V8K27q9BB1rTE3R1p7Hv0z1ZyAc8s6Vvv8DIyWf681MAt0w==", + "dev": true, + "engines": { + "node": ">= 0.4" + }, + "funding": { + "url": "https://github.com/sponsors/ljharb" + } + }, + "node_modules/tailwindcss": { + "version": "3.4.19", + "resolved": "https://registry.npmjs.org/tailwindcss/-/tailwindcss-3.4.19.tgz", + "integrity": "sha512-3ofp+LL8E+pK/JuPLPggVAIaEuhvIz4qNcf3nA1Xn2o/7fb7s/TYpHhwGDv1ZU3PkBluUVaF8PyCHcm48cKLWQ==", + "dev": true, + "dependencies": { + "@alloc/quick-lru": "^5.2.0", + "arg": "^5.0.2", + "chokidar": "^3.6.0", + "didyoumean": "^1.2.2", + "dlv": "^1.1.3", + "fast-glob": "^3.3.2", + "glob-parent": "^6.0.2", + "is-glob": "^4.0.3", + "jiti": "^1.21.7", + "lilconfig": "^3.1.3", + "micromatch": "^4.0.8", + "normalize-path": "^3.0.0", + "object-hash": "^3.0.0", + "picocolors": "^1.1.1", + "postcss": "^8.4.47", + "postcss-import": "^15.1.0", + "postcss-js": "^4.0.1", + "postcss-load-config": "^4.0.2 || ^5.0 || ^6.0", + "postcss-nested": "^6.2.0", + "postcss-selector-parser": "^6.1.2", + "resolve": "^1.22.8", + "sucrase": "^3.35.0" + }, + "bin": { + "tailwind": "lib/cli.js", + "tailwindcss": "lib/cli.js" + }, + "engines": { + "node": ">=14.0.0" + } + }, + "node_modules/thenify": { + "version": "3.3.1", + "resolved": "https://registry.npmjs.org/thenify/-/thenify-3.3.1.tgz", + "integrity": "sha512-RVZSIV5IG10Hk3enotrhvz0T9em6cyHBLkH/YAZuKqd8hRkKhSfCGIcP2KUY0EPxndzANBmNllzWPwak+bheSw==", + "dev": true, + "dependencies": { + "any-promise": "^1.0.0" + } + }, + "node_modules/thenify-all": { + "version": "1.6.0", + "resolved": "https://registry.npmjs.org/thenify-all/-/thenify-all-1.6.0.tgz", + "integrity": "sha512-RNxQH/qI8/t3thXJDwcstUO4zeqo64+Uy/+sNVRBx4Xn2OX+OZ9oP+iJnNFqplFra2ZUVeKCSa2oVWi3T4uVmA==", + "dev": true, + "dependencies": { + "thenify": ">= 3.1.0 < 4" + }, + "engines": { + "node": ">=0.8" + } + }, + "node_modules/throttle-debounce": { + "version": "5.0.2", + "resolved": "https://registry.npmjs.org/throttle-debounce/-/throttle-debounce-5.0.2.tgz", + "integrity": "sha512-B71/4oyj61iNH0KeCamLuE2rmKuTO5byTOSVwECM5FA7TiAiAW+UqTKZ9ERueC4qvgSttUhdmq1mXC3kJqGX7A==", + "engines": { + "node": ">=12.22" + } + }, + "node_modules/tinyglobby": { + "version": "0.2.17", + "resolved": "https://registry.npmjs.org/tinyglobby/-/tinyglobby-0.2.17.tgz", + "integrity": "sha512-wXR/dYpcqKmfWpEdZjiKJOwCNFndD0DMnrW/cYjVGttEkBfVgcLFHoNrlj47mjOVic9yyNu65alsgF4NQyTa2g==", + "dev": true, + "dependencies": { + "fdir": "^6.5.0", + "picomatch": "^4.0.4" + }, + "engines": { + "node": ">=12.0.0" + }, + "funding": { + "url": "https://github.com/sponsors/SuperchupuDev" + } + }, + "node_modules/tinyglobby/node_modules/fdir": { + "version": "6.5.0", + "resolved": "https://registry.npmjs.org/fdir/-/fdir-6.5.0.tgz", + "integrity": "sha512-tIbYtZbucOs0BRGqPJkshJUYdL+SDH7dVM8gjy+ERp3WAUjLEFJE+02kanyHtwjWOnwrKYBiwAmM0p4kLJAnXg==", + "dev": true, + "engines": { + "node": ">=12.0.0" + }, + "peerDependencies": { + "picomatch": "^3 || ^4" + }, + "peerDependenciesMeta": { + "picomatch": { + "optional": true + } + } + }, + "node_modules/tinyglobby/node_modules/picomatch": { + "version": "4.0.5", + "resolved": "https://registry.npmjs.org/picomatch/-/picomatch-4.0.5.tgz", + "integrity": "sha512-RvwwcruNjI1ncT5xRakeyS9Lf8lcItv34KD+aif+VH9kduAyfYBipGh12274xtenIPZ119/R9BdTBa8gAwSh0A==", + "dev": true, + "engines": { + "node": ">=12" + }, + "funding": { + "url": "https://github.com/sponsors/jonschlinkert" + } + }, + "node_modules/to-regex-range": { + "version": "5.0.1", + "resolved": "https://registry.npmjs.org/to-regex-range/-/to-regex-range-5.0.1.tgz", + "integrity": "sha512-65P7iz6X5yEr1cwcgvQxbbIw7Uk3gOy5dIdtZ4rDveLqhrdJP+Li/Hx6tyK0NEb+2GCyneCMJiGqrADCSNk8sQ==", + "dev": true, + "dependencies": { + "is-number": "^7.0.0" + }, + "engines": { + "node": ">=8.0" + } + }, + "node_modules/toggle-selection": { + "version": "1.0.6", + "resolved": "https://registry.npmjs.org/toggle-selection/-/toggle-selection-1.0.6.tgz", + "integrity": "sha512-BiZS+C1OS8g/q2RRbJmy59xpyghNBqrr6k5L/uKBGRsTfxmu3ffiRnd8mlGPUVayg8pvfi5urfnu8TU7DVOkLQ==" + }, + "node_modules/ts-interface-checker": { + "version": "0.1.13", + "resolved": "https://registry.npmjs.org/ts-interface-checker/-/ts-interface-checker-0.1.13.tgz", + "integrity": "sha512-Y/arvbn+rrz3JCKl9C4kVNfTfSm2/mEp5FSz5EsZSANGPSlQrpRI5M4PKF+mJnE52jOO90PnPSc3Ur3bTQw0gA==", + "dev": true + }, + "node_modules/tslib": { + "version": "2.3.0", + "resolved": "https://registry.npmjs.org/tslib/-/tslib-2.3.0.tgz", + "integrity": "sha512-N82ooyxVNm6h1riLCoyS9e3fuJ3AMG2zIZs2Gd1ATcSFjSA23Q0fzjjZeh0jbJvWVDZ0cJT8yaNNaaXHzueNjg==" + }, + "node_modules/typescript": { + "version": "5.9.3", + "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.9.3.tgz", + "integrity": "sha512-jl1vZzPDinLr9eUt3J/t7V6FgNEw9QjvBPdysz9KfQDD41fQrC2Y4vKQdiaUpFT4bXlb1RHhLpp8wtm6M5TgSw==", + "dev": true, + "bin": { + "tsc": "bin/tsc", + "tsserver": "bin/tsserver" + }, + "engines": { + "node": ">=14.17" + } + }, + "node_modules/update-browserslist-db": { + "version": "1.2.3", + "resolved": "https://registry.npmjs.org/update-browserslist-db/-/update-browserslist-db-1.2.3.tgz", + "integrity": "sha512-Js0m9cx+qOgDxo0eMiFGEueWztz+d4+M3rGlmKPT+T4IS/jP4ylw3Nwpu6cpTTP8R1MAC1kF4VbdLt3ARf209w==", + "dev": true, + "funding": [ + { + "type": "opencollective", + "url": "https://opencollective.com/browserslist" + }, + { + "type": "tidelift", + "url": "https://tidelift.com/funding/github/npm/browserslist" + }, + { + "type": "github", + "url": "https://github.com/sponsors/ai" + } + ], + "dependencies": { + "escalade": "^3.2.0", + "picocolors": "^1.1.1" + }, + "bin": { + "update-browserslist-db": "cli.js" + }, + "peerDependencies": { + "browserslist": ">= 4.21.0" + } + }, + "node_modules/use-sync-external-store": { + "version": "1.6.0", + "resolved": "https://registry.npmjs.org/use-sync-external-store/-/use-sync-external-store-1.6.0.tgz", + "integrity": "sha512-Pp6GSwGP/NrPIrxVFAIkOQeyw8lFenOHijQWkUTrDvrF4ALqylP2C/KCkeS9dpUM3KvYRQhna5vt7IL95+ZQ9w==", + "peerDependencies": { + "react": "^16.8.0 || ^17.0.0 || ^18.0.0 || ^19.0.0" + } + }, + "node_modules/util-deprecate": { + "version": "1.0.2", + "resolved": "https://registry.npmjs.org/util-deprecate/-/util-deprecate-1.0.2.tgz", + "integrity": "sha512-EPD5q1uXyFxJpCrLnCc1nHnq3gOa6DZBocAIiI2TaSCA7VCJ1UJDMagCzIkXNsUYfD1daK//LTEQ8xiIbrHtcw==", + "dev": true + }, + "node_modules/vite": { + "version": "5.4.21", + "resolved": "https://registry.npmjs.org/vite/-/vite-5.4.21.tgz", + "integrity": "sha512-o5a9xKjbtuhY6Bi5S3+HvbRERmouabWbyUcpXXUA1u+GNUKoROi9byOJ8M0nHbHYHkYICiMlqxkg1KkYmm25Sw==", + "dev": true, + "dependencies": { + "esbuild": "^0.21.3", + "postcss": "^8.4.43", + "rollup": "^4.20.0" + }, + "bin": { + "vite": "bin/vite.js" + }, + "engines": { + "node": "^18.0.0 || >=20.0.0" + }, + "funding": { + "url": "https://github.com/vitejs/vite?sponsor=1" + }, + "optionalDependencies": { + "fsevents": "~2.3.3" + }, + "peerDependencies": { + "@types/node": "^18.0.0 || >=20.0.0", + "less": "*", + "lightningcss": "^1.21.0", + "sass": "*", + "sass-embedded": "*", + "stylus": "*", + "sugarss": "*", + "terser": "^5.4.0" + }, + "peerDependenciesMeta": { + "@types/node": { + "optional": true + }, + "less": { + "optional": true + }, + "lightningcss": { + "optional": true + }, + "sass": { + "optional": true + }, + "sass-embedded": { + "optional": true + }, + "stylus": { + "optional": true + }, + "sugarss": { + "optional": true + }, + "terser": { + "optional": true + } + } + }, + "node_modules/yallist": { + "version": "3.1.1", + "resolved": "https://registry.npmjs.org/yallist/-/yallist-3.1.1.tgz", + "integrity": "sha512-a4UGQaWPH59mOXUYnAG2ewncQS4i4F43Tv3JoAM+s2VDAmS9NsK8GpDMLrCHPksFT7h3K6TOoUNn2pb7RoXx4g==", + "dev": true + }, + "node_modules/zrender": { + "version": "5.6.1", + "resolved": "https://registry.npmjs.org/zrender/-/zrender-5.6.1.tgz", + "integrity": "sha512-OFXkDJKcrlx5su2XbzJvj/34Q3m6PvyCZkVPHGYpcCJ52ek4U/ymZyfuV1nKE23AyBJ51E/6Yr0mhZ7xGTO4ag==", + "dependencies": { + "tslib": "2.3.0" + } + }, + "node_modules/zustand": { + "version": "4.5.7", + "resolved": "https://registry.npmjs.org/zustand/-/zustand-4.5.7.tgz", + "integrity": "sha512-CHOUy7mu3lbD6o6LJLfllpjkzhHXSBlX8B9+qPddUsIfeF5S/UZ5q0kmCsnRqT1UHFQZchNFDDzMbQsuesHWlw==", + "dependencies": { + "use-sync-external-store": "^1.2.2" + }, + "engines": { + "node": ">=12.7.0" + }, + "peerDependencies": { + "@types/react": ">=16.8", + "immer": ">=9.0.6", + "react": ">=16.8" + }, + "peerDependenciesMeta": { + "@types/react": { + "optional": true + }, + "immer": { + "optional": true + }, + "react": { + "optional": true + } + } + } + } +} diff --git a/src/web/frontend/package.json b/src/web/frontend/package.json new file mode 100644 index 0000000..015e5bf --- /dev/null +++ b/src/web/frontend/package.json @@ -0,0 +1,35 @@ +{ + "name": "neo-agent-web-frontend", + "version": "3.0.0", + "private": true, + "type": "module", + "scripts": { + "dev": "vite", + "build": "tsc && vite build", + "preview": "vite preview", + "lint": "eslint . --ext .ts,.tsx" + }, + "dependencies": { + "react": "^18.2.0", + "react-dom": "^18.2.0", + "react-router-dom": "^6.22.0", + "antd": "^5.15.0", + "@ant-design/icons": "^5.3.0", + "zustand": "^4.5.0", + "echarts": "^5.5.0", + "echarts-for-react": "^3.0.2", + "axios": "^1.6.7", + "react-window": "^1.8.10" + }, + "devDependencies": { + "@types/react": "^18.2.55", + "@types/react-dom": "^18.2.19", + "@types/react-window": "^1.8.8", + "@vitejs/plugin-react": "^4.2.1", + "typescript": "^5.3.3", + "vite": "^5.1.0", + "tailwindcss": "^3.4.1", + "autoprefixer": "^10.4.17", + "postcss": "^8.4.35" + } +} diff --git a/src/web/frontend/postcss.config.js b/src/web/frontend/postcss.config.js new file mode 100644 index 0000000..2aa7205 --- /dev/null +++ b/src/web/frontend/postcss.config.js @@ -0,0 +1,6 @@ +export default { + plugins: { + tailwindcss: {}, + autoprefixer: {}, + }, +}; diff --git a/src/web/frontend/src/App.tsx b/src/web/frontend/src/App.tsx new file mode 100644 index 0000000..d1e4af1 --- /dev/null +++ b/src/web/frontend/src/App.tsx @@ -0,0 +1,31 @@ +import { Routes, Route, Navigate } from 'react-router-dom'; +import Layout from './components/Layout'; +import ChatPage from './pages/Chat'; +import DebugPage from './pages/Debug'; +import KnowledgePage from './pages/Knowledge'; +import SchedulePage from './pages/Schedule'; +import EventPage from './pages/Event'; +import NPSPage from './pages/NPS'; +import DatabasePage from './pages/Database'; +import CreativePage from './pages/Creative'; +import SettingsPage from './pages/Settings'; + +export default function App() { + return ( + + }> + } /> + } /> + } /> + } /> + } /> + } /> + } /> + } /> + } /> + } /> + } /> + + + ); +} diff --git a/src/web/frontend/src/api/chatSessions.ts b/src/web/frontend/src/api/chatSessions.ts new file mode 100644 index 0000000..9e81419 --- /dev/null +++ b/src/web/frontend/src/api/chatSessions.ts @@ -0,0 +1,156 @@ +/** + * Chat Sessions API 客户端 - v3.1.0 + * ================================= + * + * 与后端 `src/web/backend/api/chat_sessions.py` 对齐: + * - GET /api/chat/sessions?user_id=&limit=&offset= + * - POST /api/chat/sessions body: { user_id, title? } + * - GET /api/chat/sessions/{id}/messages?limit= + * - PATCH /api/chat/sessions/{id} body: { title } + * - DELETE /api/chat/sessions/{id} -> 204 + * + * 设计要点: + * - 纯 fetch,无 axios 依赖,便于与 useChatStream 解耦 + * - base URL 优先用 import.meta.env.VITE_API_BASE,否则默认 '/api' + * - 失败抛 Error, 由 UI 层决定是否弹 message.error + */ + +export interface ChatSession { + id: number; + user_id: string; + title: string; + created_at: number; + updated_at: number; + message_count: number; +} + +export interface ChatMessage { + id: number; + session_id: number; + role: 'user' | 'assistant' | 'system'; + content: string; + emotion_json: string | null; + conn_id: string | null; + created_at: number; +} + +function getBaseUrl(): string { + const env = (import.meta as any)?.env; + const base = env?.VITE_API_BASE; + if (typeof base === 'string' && base.trim()) { + return base.replace(/\/+$/, ''); + } + return '/api'; +} + +async function unwrapError(resp: Response, fallback: string): Promise { + let detail = fallback; + try { + const data = await resp.json(); + if (data && typeof data === 'object') { + const d = (data as any).detail; + if (typeof d === 'string' && d.trim()) { + detail = d; + } + } + } catch { + /* ignore parse error */ + } + throw new Error(detail); +} + +/** + * 列出用户会话（按 updated_at 倒序）。 + */ +export async function listSessions( + userId: string = 'default', + limit: number = 50, + offset: number = 0, +): Promise<{ items: ChatSession[]; total: number }> { + const url = `${getBaseUrl()}/chat/sessions?user_id=${encodeURIComponent(userId)}&limit=${limit}&offset=${offset}`; + const resp = await fetch(url, { method: 'GET' }); + if (!resp.ok) { + return unwrapError(resp, `listSessions failed: ${resp.status}`); + } + const data = await resp.json(); + return { + items: Array.isArray(data?.items) ? (data.items as ChatSession[]) : [], + total: Number(data?.total ?? 0) || 0, + }; +} + +/** + * 创建新会话。 + */ +export async function createSession( + userId: string, + title?: string, +): Promise { + const url = `${getBaseUrl()}/chat/sessions`; + const resp = await fetch(url, { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ user_id: userId, title: title ?? '新会话' }), + }); + if (!resp.ok) { + return unwrapError(resp, `createSession failed: ${resp.status}`); + } + return (await resp.json()) as ChatSession; +} + +/** + * 删除会话（级联删 messages）。 + */ +export async function deleteSession(sessionId: number): Promise { + const url = `${getBaseUrl()}/chat/sessions/${sessionId}`; + const resp = await fetch(url, { method: 'DELETE' }); + if (!resp.ok && resp.status !== 204) { + return unwrapError(resp, `deleteSession failed: ${resp.status}`); + } +} + +/** + * 重命名会话。 + */ +export async function renameSession( + sessionId: number, + title: string, +): Promise { + const url = `${getBaseUrl()}/chat/sessions/${sessionId}`; + const resp = await fetch(url, { + method: 'PATCH', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ title }), + }); + if (!resp.ok) { + return unwrapError(resp, `renameSession failed: ${resp.status}`); + } + return (await resp.json()) as ChatSession; +} + +/** + * 拉取会话消息（按 id 升序）。 + */ +export async function listMessages( + sessionId: number, + limit: number = 200, +): Promise<{ items: ChatMessage[]; total: number }> { + const url = `${getBaseUrl()}/chat/sessions/${sessionId}/messages?limit=${limit}`; + const resp = await fetch(url, { method: 'GET' }); + if (!resp.ok) { + return unwrapError(resp, `listMessages failed: ${resp.status}`); + } + const data = await resp.json(); + return { + items: Array.isArray(data?.items) ? (data.items as ChatMessage[]) : [], + total: Number(data?.total ?? 0) || 0, + }; +} + +export default { + listSessions, + createSession, + deleteSession, + renameSession, + listMessages, +}; diff --git a/src/web/frontend/src/api/creative.ts b/src/web/frontend/src/api/creative.ts new file mode 100644 index 0000000..2d0daaf --- /dev/null +++ b/src/web/frontend/src/api/creative.ts @@ -0,0 +1,235 @@ +// Creative REST API 客户端。 +// 对应后端路由：src/web/backend/api/creative.py +// - GET /api/creative/projects?status=...&limit=50 +// - GET /api/creative/projects/{uuid} +// - POST /api/creative/projects +// - POST /api/creative/projects/{uuid}/advance +// - DELETE /api/creative/projects/{uuid} +// +// 类型与 src/web/backend/schemas/creative.py 的 Pydantic v2 模型对齐： +// - CreativeProject / CreativeProjectListResponse +// - CreativeProjectCreate +// - AdvanceRequest / AdvanceResponse +// Vite dev server 已将 /api 代理到后端 8000 端口（vite.config.ts）。 + +import type { CreativeProject } from '@/types/creative'; + +// 重新导出常用类型，保持单点导入（其他类型由调用方按需直接从 '@/types/creative' 取）。 +export type { + CreativeProject, + StoryBible, + Chapter, + Character, + CreativeProjectStatus, +} from '@/types/creative'; + +// 列表接口的响应包：与后端 CreativeProjectListResponse 对齐 +export interface CreativeProjectListResponse { + projects: CreativeProject[]; + total: number; +} + +// 续写请求体：与后端 AdvanceRequest 对齐 +export interface AdvanceRequest { + user?: string; + hint?: string; + advance_async?: boolean; +} + +// 续写响应：与后端 AdvanceResponse 对齐 +export interface AdvanceResponse { + project_uuid: string; + task_id: string; + status: string; + last_advanced_at?: string | null; + project_status: string; + message: string; +} + +// 创建项目请求体：与后端 CreativeProjectCreate 对齐 +export interface CreativeProjectCreate { + title: string; + work_type?: string; + premise?: string; + tone?: string; + point_of_view?: string; + target_chars?: number; + inspiration_source?: string; + outline?: string[]; + characters?: Record[]; +} + +// 自定义错误：包含后端 detail 和 HTTP 状态码 +export class CreativeApiError extends Error { + status: number; + detail: string; + constructor(status: number, detail: string) { + super(`Creative API error ${status}: ${detail}`); + this.name = 'CreativeApiError'; + this.status = status; + this.detail = detail; + } +} + +const DEFAULT_TIMEOUT_MS = 15_000; + +/** + * 统一 fetch 封装： + * - 解析 JSON 响应 + * - 非 2xx 抛出 CreativeApiError + * - 自动拼接 query string + * - 支持 AbortController 超时 + */ +async function request( + path: string, + init: RequestInit = {}, + query?: Record, +): Promise { + let url = path; + if (query) { + const params = new URLSearchParams(); + Object.entries(query).forEach(([k, v]) => { + if (v !== undefined && v !== null && v !== '') { + params.append(k, String(v)); + } + }); + const qs = params.toString(); + if (qs) url += (url.includes('?') ? '&' : '?') + qs; + } + + const controller = new AbortController(); + const timer = setTimeout(() => controller.abort(), DEFAULT_TIMEOUT_MS); + + let res: Response; + try { + res = await fetch(url, { + ...init, + signal: controller.signal, + headers: { + 'Content-Type': 'application/json', + Accept: 'application/json', + ...(init.headers || {}), + }, + }); + } catch (err) { + clearTimeout(timer); + if ((err as Error).name === 'AbortError') { + throw new CreativeApiError(0, `请求超时：${url}`); + } + throw new CreativeApiError(0, `网络错误：${(err as Error).message}`); + } + clearTimeout(timer); + + // 204 No Content + if (res.status === 204) { + return undefined as unknown as T; + } + + let payload: unknown = null; + const text = await res.text(); + if (text) { + try { + payload = JSON.parse(text); + } catch { + payload = { detail: text }; + } + } + + if (!res.ok) { + const detail = + (payload && typeof payload === 'object' && 'detail' in payload + ? String((payload as { detail: unknown }).detail) + : null) || `HTTP ${res.status}`; + throw new CreativeApiError(res.status, detail); + } + + return payload as T; +} + +// ===================================================================== +// 对外 API +// ===================================================================== + +/** + * 列出创作项目。 + * GET /api/creative/projects?status=...&limit=50 + * @param status 可选过滤条件：active / paused / completed / drafting / finished + * @param limit 返回上限（1-200），默认 50 + */ +export async function listProjects( + status?: string, + limit: number = 50, +): Promise { + return request( + '/api/creative/projects', + { method: 'GET' }, + { status, limit }, + ); +} + +/** + * 读取项目详情（含 Story Bible）。 + * GET /api/creative/projects/{uuid} + * 404 时抛 CreativeApiError(404, ...)。 + */ +export async function getProject(uuid: string): Promise { + if (!uuid) { + throw new CreativeApiError(400, 'project uuid is required'); + } + return request( + `/api/creative/projects/${encodeURIComponent(uuid)}`, + { method: 'GET' }, + ); +} + +/** + * 创建创作项目。 + * POST /api/creative/projects + * 返回 201 + CreativeProject。 + */ +export async function createProject( + payload: CreativeProjectCreate, +): Promise { + return request('/api/creative/projects', { + method: 'POST', + body: JSON.stringify(payload), + }); +} + +/** + * 触发续写。默认异步模式：返回 task_id，进度通过 /ws/events 的 + * 'creative_progress' 事件推送。 + * POST /api/creative/projects/{uuid}/advance + * 404 时抛 CreativeApiError(404, ...)。 + */ +export async function advanceProject( + uuid: string, + payload: AdvanceRequest = {}, +): Promise { + if (!uuid) { + throw new CreativeApiError(400, 'project uuid is required'); + } + return request( + `/api/creative/projects/${encodeURIComponent(uuid)}/advance`, + { + method: 'POST', + // body 可缺省；后端会使用 AdvanceRequest() 默认值 + body: JSON.stringify(payload), + }, + ); +} + +/** + * 删除项目。 + * DELETE /api/creative/projects/{uuid} + * 成功返回 204 No Content。 + */ +export async function deleteProject(uuid: string): Promise { + if (!uuid) { + throw new CreativeApiError(400, 'project uuid is required'); + } + await request( + `/api/creative/projects/${encodeURIComponent(uuid)}`, + { method: 'DELETE' }, + ); +} diff --git a/src/web/frontend/src/api/database.ts b/src/web/frontend/src/api/database.ts new file mode 100644 index 0000000..3f4375d --- /dev/null +++ b/src/web/frontend/src/api/database.ts @@ -0,0 +1,137 @@ +// Database API 客户端 +// 包装 `/api/database/*` 三个端点：tables 列表、单表分页查询、按 uuid 删除（带 X-Confirm 二次确认头）。 +// 与后端 schemas/database.py 契约对齐。 + +import axios, { AxiosError } from 'axios'; +import type { Row } from '@/types/database'; + +// ---------------------------------------------------------------------- +// 类型契约（与后端 Pydantic v2 模型对齐） +// ---------------------------------------------------------------------- + +/** `GET /api/database/tables` 响应 */ +export interface TablesListResponse { + tables: string[]; + total: number; +} + +/** `GET /api/database/{table}?limit=&offset=` 响应 */ +export interface TableDataResponse { + table: string; + rows: Row[]; + total: number; +} + +/** `DELETE /api/database/{table}/{uuid}` 响应 */ +export interface DeleteResponse { + table: string; + uuid: string; + deleted: boolean; +} + +/** 后端统一错误响应：`{"error": "..."}` */ +export interface ApiErrorBody { + error: string; +} + +// ---------------------------------------------------------------------- +// axios 实例：走 vite dev server 的 `/api` 代理，无需写 baseURL +// ---------------------------------------------------------------------- + +const http = axios.create({ + timeout: 15000, + headers: { + 'Content-Type': 'application/json', + }, +}); + +// ---------------------------------------------------------------------- +// 错误处理工具 +// ---------------------------------------------------------------------- + +/** + * 把 axios 错误规整为「中文用户可读」的字符串。 + * 优先使用后端 `{"error": "..."}` 字段（项目硬约束：所有 4xx 响应都是这个格式）。 + */ +export const formatApiError = (err: unknown): string => { + if (axios.isAxiosError(err)) { + const axErr = err as AxiosError; + const backendMsg = axErr.response?.data?.error; + if (backendMsg) return backendMsg; + if (axErr.response?.status === 400) { + return '请求参数错误（400）'; + } + if (axErr.response?.status === 404) { + return '未找到对应记录（404）'; + } + if (axErr.code === 'ECONNABORTED') { + return '请求超时，请检查后端服务'; + } + if (axErr.message) return axErr.message; + } + if (err instanceof Error) return err.message; + return '未知错误'; +}; + +// ---------------------------------------------------------------------- +// 1) GET /api/database/tables +// ---------------------------------------------------------------------- + +/** + * 列出白名单中实际存在的业务表。 + * @returns 表名列表 + total + */ +export const listTables = async (): Promise => { + const res = await http.get('/api/database/tables'); + return res.data; +}; + +// ---------------------------------------------------------------------- +// 2) GET /api/database/{table}?limit=100&offset=0 +// ---------------------------------------------------------------------- + +/** + * 分页查询指定表的数据。 + * @param table 表名（必须命中后端白名单，否则 400） + * @param limit 每页行数，默认 100 + * @param offset 偏移量，默认 0 + */ +export const getTableData = async ( + table: string, + limit = 100, + offset = 0, +): Promise => { + const res = await http.get( + `/api/database/${encodeURIComponent(table)}`, + { + params: { limit, offset }, + }, + ); + return res.data; +}; + +// ---------------------------------------------------------------------- +// 3) DELETE /api/database/{table}/{uuid} —— 必须带 X-Confirm: true +// ---------------------------------------------------------------------- + +/** + * 按 uuid 删除单条记录。 + * 注意：后端硬约束要求请求头 `X-Confirm: true`，缺失 → 400。 + * @param table 表名 + * @param uuid 记录 uuid + */ +export const deleteRecord = async ( + table: string, + uuid: string, +): Promise => { + const res = await http.delete( + `/api/database/${encodeURIComponent(table)}/${encodeURIComponent(uuid)}`, + { + // 必须显式声明二次确认头；缺失或非 "true" → 后端 400 + headers: { + 'X-Confirm': 'true', + }, + }, + ); + return res.data; +}; diff --git a/src/web/frontend/src/api/knowledge.ts b/src/web/frontend/src/api/knowledge.ts new file mode 100644 index 0000000..096f788 --- /dev/null +++ b/src/web/frontend/src/api/knowledge.ts @@ -0,0 +1,175 @@ +/** + * Knowledge REST API 客户端（Stage C.1）。 + * + * 端点契约（对应后端 src/web/backend/api/knowledge.py）： + * GET /api/knowledge/search?q=&limit= 关键词搜索 + * GET /api/knowledge?limit= 列出所有实体 + * POST /api/knowledge 新增实体 + * DELETE /api/knowledge/{uuid} 删除实体 + * + * 设计要点： + * 1. 基于 axios（package.json 已声明 ^1.6.7），通过 Vite dev proxy 转发到 :8000。 + * 2. 所有方法返回后端原始 JSON；Entity 字段补全由 toEntity() 适配。 + * 3. 错误统一通过 antd message.error 提示，并向上抛出供调用方决定是否继续。 + * 4. 搜索关键词为空时回退到 listEntities，避免后端 q min_length=1 校验失败。 + */ + +import axios, { AxiosError } from 'axios'; +import { message } from 'antd'; +import { + KNOWN_ENTITY_CATEGORIES, + type Entity, + type EntityCategory, + type KnowledgeEntityRow, +} from '../types/knowledge'; + +const API_BASE = '/api/knowledge'; + +/** + * 把后端返回的任意字符串归一化到 EntityCategory 枚举。 + * 不在白名单时回落到 'concept'，保证 UI 渲染不崩。 + */ +function normalizeCategory(input: string | undefined | null): EntityCategory { + if (!input) return 'concept'; + const value = input.trim().toLowerCase(); + if ((KNOWN_ENTITY_CATEGORIES as readonly string[]).includes(value)) { + return value as EntityCategory; + } + return 'concept'; +} + +/** + * 把后端 KnowledgeEntityRow 适配为前端 UI 使用的 Entity。 + * 后端 entities 表本身只含 uuid/name/normalized_name/created_at/updated_at； + * category/description 由 search_knowledge() 通过 definition 表补齐； + * confidence/mention_count/metadata/related_uuids 后端暂无直接字段，使用默认值。 + */ +export function toEntity(row: KnowledgeEntityRow): Entity { + return { + uuid: String(row.uuid ?? ''), + name: String(row.name ?? ''), + category: normalizeCategory(row.category), + // 后端暂未提供 confidence，UI 期望 0-1，这里给中性默认值 0.5 + confidence: 0.5, + // 后端暂未提供 mention_count，前端用 0 占位 + mention_count: 0, + updated_at: row.updated_at || row.created_at || new Date().toISOString(), + description: row.description ?? '', + metadata: {}, + related_uuids: [], + }; +} + +export interface SearchResponsePayload { + results: KnowledgeEntityRow[]; + total: number; +} + +export interface CreateResponsePayload { + uuid: string; + entity_name: string; + created: boolean; +} + +export interface DeleteResponsePayload { + uuid: string; + deleted: boolean; +} + +/** 提取后端错误信息：优先用 {"error": "..."}，否则用 axios message。 */ +function extractErrorMessage(err: unknown, fallback: string): string { + const ax = err as AxiosError<{ error?: string; detail?: string }>; + const data = ax?.response?.data as { error?: string; detail?: string } | undefined; + if (data?.error) return data.error; + if (data?.detail) return data.detail; + if (ax?.message) return ax.message; + return fallback; +} + +/** + * 搜索实体。 + * - q 为空字符串时回退到 listEntities()（后端 search 要求 q.min_length=1）。 + * - 默认 limit 50 足以覆盖演示场景。 + */ +export async function searchEntities(q: string = '', limit: number = 50): Promise { + const keyword = (q ?? '').trim(); + if (!keyword) { + return listEntities(limit); + } + try { + const { data } = await axios.get(`${API_BASE}/search`, { + params: { q: keyword, limit }, + }); + const rows = Array.isArray(data?.results) ? data.results : []; + return rows.map(toEntity); + } catch (err) { + const msg = extractErrorMessage(err, '搜索知识库失败'); + message.error(`搜索失败:${msg}`); + throw err; + } +} + +/** + * 列出所有实体。 + * 后端 /api/knowledge?limit= 仅返回 entities 行，不含 definition 字段， + * 所以 description/category 会落默认值。 + */ +export async function listEntities(limit: number = 50): Promise { + try { + const { data } = await axios.get(API_BASE, { + params: { limit }, + }); + const rows = Array.isArray(data?.results) ? data.results : []; + return rows.map(toEntity); + } catch (err) { + const msg = extractErrorMessage(err, '列出知识库失败'); + message.error(`列出失败:${msg}`); + throw err; + } +} + +export interface CreateEntityPayload { + name: string; + category: EntityCategory; + description?: string; + related_info?: Record; +} + +/** + * 新增实体。 + * 前端表单字段 {name, category, description} 映射到后端 + * {entity_name, category, description, related_info}。 + */ +export async function createEntity( + payload: CreateEntityPayload, +): Promise { + try { + const { data } = await axios.post(API_BASE, { + entity_name: payload.name, + category: payload.category, + description: payload.description ?? '', + related_info: payload.related_info, + }); + message.success(`已新增实体:${data?.entity_name ?? payload.name}`); + return data; + } catch (err) { + const msg = extractErrorMessage(err, '创建实体失败'); + message.error(`创建失败:${msg}`); + throw err; + } +} + +/** 删除实体,后端返回 {uuid, deleted: true}。 */ +export async function deleteEntity(uuid: string): Promise { + try { + const { data } = await axios.delete( + `${API_BASE}/${encodeURIComponent(uuid)}`, + ); + message.success(`已删除实体:${uuid}`); + return data; + } catch (err) { + const msg = extractErrorMessage(err, '删除实体失败'); + message.error(`删除失败:${msg}`); + throw err; + } +} diff --git a/src/web/frontend/src/api/llm.ts b/src/web/frontend/src/api/llm.ts new file mode 100644 index 0000000..c3f7664 --- /dev/null +++ b/src/web/frontend/src/api/llm.ts @@ -0,0 +1,84 @@ +/** + * LLM 配置 API 客户端 - v3.1.0 + * ============================= + * + * 与后端 `src/web/backend/api/llm.py` 对齐: + * - GET /api/llm/config -> LLMConfig + * - POST /api/llm/test body: { prompt: "ping" } -> {ok, latency_ms, sample_response, error} + */ + +export interface LLMConfig { + provider: string; + base_url: string; + model_name: string; + temperature: number; + max_tokens: number; + timeout: number; + valid: boolean; + api_key_set?: boolean; +} + +export interface LLMTestResult { + ok: boolean; + latency_ms?: number; + sample_response?: string; + error?: string; +} + +function getBaseUrl(): string { + const env = (import.meta as any)?.env; + const base = env?.VITE_API_BASE; + if (typeof base === 'string' && base.trim()) { + return base.replace(/\/+$/, ''); + } + return '/api'; +} + +async function unwrapError(resp: Response, fallback: string): Promise { + let detail = fallback; + try { + const data = await resp.json(); + if (data && typeof data === 'object') { + const d = (data as any).detail; + if (typeof d === 'string' && d.trim()) { + detail = d; + } + } + } catch { + /* ignore */ + } + throw new Error(detail); +} + +/** + * 读取当前生效 LLM 配置（provider / base_url / model / 温度 / tokens / timeout / 有效性）。 + */ +export async function getConfig(): Promise { + const url = `${getBaseUrl()}/llm/config`; + const resp = await fetch(url, { method: 'GET' }); + if (!resp.ok) { + return unwrapError(resp, `getConfig failed: ${resp.status}`); + } + return (await resp.json()) as LLMConfig; +} + +/** + * 调一次 ping 验证 LLM 连通性。 + */ +export async function testConnection(prompt: string = 'ping'): Promise { + const url = `${getBaseUrl()}/llm/test`; + const resp = await fetch(url, { + method: 'POST', + headers: { 'Content-Type': 'application/json' }, + body: JSON.stringify({ prompt }), + }); + if (!resp.ok) { + return unwrapError(resp, `testConnection failed: ${resp.status}`); + } + return (await resp.json()) as LLMTestResult; +} + +export default { + getConfig, + testConnection, +}; diff --git a/src/web/frontend/src/api/nps.ts b/src/web/frontend/src/api/nps.ts new file mode 100644 index 0000000..1550d19 --- /dev/null +++ b/src/web/frontend/src/api/nps.ts @@ -0,0 +1,229 @@ +/** + * NPS 前端 API 客户端。 + * 对接后端 /api/nps 路由： + * GET /api/nps 列出已注册 NPS + * POST /api/nps/register 注册新 NPS + * POST /api/nps/invoke 调用 NPS + * + * 字段命名约定： + * 后端使用 `parameters` (Record) 与 `registered_at`； + * 前端页面使用 `params` (NPSParam[]) 与 `created_at`。 + * 在此模块内做映射，避免污染页面层。 + */ + +import axios, { AxiosError, AxiosResponse } from 'axios'; +import type { NPS, NPSParam, NPSType, NPSConfigFormValues } from '../types/nps'; + +// ===================================================================== +// 后端契约类型 +// ===================================================================== + +/** 后端 NPS 条目形状（GET /api/nps / POST /api/nps/register 的响应）。 */ +export interface NpsEntry { + uuid: string; + name: string; + description: string; + version: string; + parameters: Record; + registered_at: string; + // 可选：UI 需要的扩展字段（后端可能返回） + enabled?: boolean; + nps_type?: NPSType; + call_count?: number; + last_called_at?: string | null; +} + +/** 注册 NPS 的请求体（POST /api/nps/register）。 */ +export interface NpsRegisterPayload { + name: string; + description: string; + parameters: Record; + version?: string; + enabled?: boolean; + nps_type?: NPSType; +} + +/** 调用 NPS 的请求体（POST /api/nps/invoke）。 */ +export interface NpsInvokePayload { + uuid: string; + args: Record; +} + +/** 调用 NPS 的响应（POST /api/nps/invoke）。 */ +export interface NpsInvokeResponse { + result: unknown; + status: string; + executed_at: string; +} + +/** 注册 NPS 的入参（页面 → API 模块）。 */ +export type NpsRegisterInput = NPSConfigFormValues; + +// ===================================================================== +// axios 实例 +// ===================================================================== + +// 走 vite proxy：/api → http://localhost:8000 +const http = axios.create({ + baseURL: '/api', + timeout: 15000, +}); + +// ===================================================================== +// 内部工具：字段映射 +// ===================================================================== + +/** + * 把后端 NpsEntry 映射为前端 NPS 形状。 + * 后端 parameters (dict) → 前端 params (NPSParam[])。 + */ +export function toFrontendNps(entry: NpsEntry): NPS { + const params: NPSParam[] = Object.entries(entry.parameters ?? {}).map( + ([key, value]) => ({ + key, + // 复杂值序列化为字符串，避免丢信息 + value: + typeof value === 'string' + ? value + : value == null + ? '' + : JSON.stringify(value), + }), + ); + return { + uuid: entry.uuid, + name: entry.name, + description: entry.description ?? '', + nps_type: (entry.nps_type ?? 'function') as NPSType, + params, + enabled: entry.enabled ?? true, + call_count: entry.call_count ?? 0, + last_called_at: entry.last_called_at ?? null, + created_at: entry.registered_at ?? new Date().toISOString(), + }; +} + +/** + * 把前端 NPSParam[] 转换回后端 parameters (dict)。 + * 过滤掉 key 为空的项。 + */ +function paramsToParameters( + params: NPSParam[] | undefined, +): Record { + if (!params) return {}; + const out: Record = {}; + for (const p of params) { + const k = (p.key ?? '').trim(); + if (!k) continue; + out[k] = p.value ?? ''; + } + return out; +} + +/** 兼容多种列表响应形态（数组 / {results} / {items} / {data}）。 */ +function unwrapList(payload: unknown): NpsEntry[] { + if (Array.isArray(payload)) return payload as NpsEntry[]; + if (payload && typeof payload === 'object') { + const obj = payload as Record; + for (const key of ['results', 'items', 'data', 'list']) { + const v = obj[key]; + if (Array.isArray(v)) return v as NpsEntry[]; + } + } + return []; +} + +/** + * 解析 axios 错误，提取可读中文消息。 + */ +function extractErrorMessage(err: unknown, fallback: string): string { + if (axios.isAxiosError(err)) { + const axErr = err as AxiosError<{ error?: string; detail?: string; message?: string }>; + if (axErr.response?.data) { + const data = axErr.response.data; + if (typeof data === 'string') return data; + if (typeof data === 'object' && data !== null) { + return ( + data.error || + data.detail || + data.message || + axErr.message || + fallback + ); + } + } + return axErr.message || fallback; + } + if (err instanceof Error) return err.message || fallback; + return fallback; +} + +// ===================================================================== +// 公开 API +// ===================================================================== + +/** + * 列出已注册的 NPS。 + * 失败时抛 Error（message 为可读中文），由调用方捕获后 message.error 显示。 + */ +export async function listNps(): Promise { + try { + const resp: AxiosResponse = await http.get('/nps'); + const raw = unwrapList(resp.data); + return raw.map(toFrontendNps); + } catch (err) { + throw new Error(extractErrorMessage(err, '获取 NPS 列表失败')); + } +} + +/** + * 注册新 NPS。 + * 入参使用页面表单形状；内部转换为后端契约。 + */ +export async function registerNps( + payload: NpsRegisterInput, +): Promise { + try { + const body: NpsRegisterPayload = { + name: payload.name, + description: payload.description ?? '', + parameters: paramsToParameters(payload.params), + enabled: payload.enabled, + nps_type: payload.nps_type, + version: '1.0.0', + }; + const resp = await http.post('/nps/register', body); + if (!resp.data || !resp.data.uuid) { + throw new Error('注册响应缺少 uuid'); + } + return toFrontendNps(resp.data); + } catch (err) { + throw new Error(extractErrorMessage(err, '注册 NPS 失败')); + } +} + +/** + * 调用已注册的 NPS。 + * 失败时抛 Error，由调用方处理。 + */ +export async function invokeNps( + payload: NpsInvokePayload, +): Promise { + try { + const resp = await http.post('/nps/invoke', payload); + if (!resp.data || typeof resp.data !== 'object') { + throw new Error('调用响应格式异常'); + } + return resp.data; + } catch (err) { + throw new Error(extractErrorMessage(err, '调用 NPS 失败')); + } +} + +// 兼容命名空间导入：import * as npsApi from '../../api/nps' +export default { + listNps, + registerNps, + invokeNps, + toFrontendNps, +}; diff --git a/src/web/frontend/src/api/schedule.ts b/src/web/frontend/src/api/schedule.ts new file mode 100644 index 0000000..69e16c8 --- /dev/null +++ b/src/web/frontend/src/api/schedule.ts @@ -0,0 +1,243 @@ +/** + * Schedule API 客户端 + * + * 与后端 `src/web/backend/api/schedule.py` 对齐: + * - GET /api/schedule?date=YYYY-MM-DD + * - GET /api/schedule?from=YYYY-MM-DD&to=YYYY-MM-DD + * - POST /api/schedule + * - PUT /api/schedule/{id} + * - DELETE /api/schedule/{id} + * - POST /api/schedule/{id}/confirm?confirmed=true + * + * 设计要点: + * - 失败统一抛 Error(message=后端 detail 或 axios message),由调用方决定是否弹 message.error + * - 字段映射:后端 `id`(int) → 前端 `uuid`(string) + * - 协作字段后端 DTO 未暴露,默认 false/undefined(后续若 backend 扩展可自动读取) + */ + +import axios, { type AxiosError } from 'axios'; + +export type SchedulePriority = 'low' | 'normal' | 'high'; +export type ScheduleType = 'personal' | 'work' | 'family'; +export type ScheduleStatus = 'pending' | 'confirmed' | 'done' | 'cancelled'; +export type CollaboratorStatus = 'confirmed' | 'pending' | 'declined'; + +/** 前端 UI 使用的日程视图(字段超集,后端未给的协作字段给默认值) */ +export interface Schedule { + /** 数据库主键(后端 int) */ + id?: number; + /** 字符串形式的主键,UI 列表的 rowKey */ + uuid: string; + title: string; + description: string; + /** ISO 字符串 */ + start_time: string; + /** ISO 字符串 */ + end_time: string; + priority: SchedulePriority; + schedule_type: ScheduleType; + status: ScheduleStatus; + /** 是否协作日程 */ + is_collaborative: boolean; + /** 协作方状态 */ + collaborator_status?: CollaboratorStatus; + /** 协作人名称 */ + collaborator_name?: string; +} + +/** 新建日程请求体(对应后端 ScheduleCreate) */ +export interface ScheduleCreatePayload { + title: string; + description?: string; + start_time: string; + end_time: string; + priority?: SchedulePriority; + schedule_type?: ScheduleType; +} + +/** 更新日程请求体(对应后端 ScheduleUpdate) */ +export interface ScheduleUpdatePayload { + title?: string; + description?: string; + start_time?: string; + end_time?: string; + priority?: SchedulePriority; + status?: ScheduleStatus; +} + +const client = axios.create({ baseURL: '/api' }); + +/** 后端原始 DTO 行(容忍空字段) */ +interface RawScheduleRow { + id?: number | string; + uuid?: string; + title?: string; + description?: string | null; + start_time?: string; + end_time?: string | null; + priority?: string; + status?: string; + schedule_type?: string; + is_collaborative?: number | boolean | null; + collaborator_status?: string | null; + collaborator_name?: string | null; + confirmed?: number | boolean | null; +} + +function normalizePriority(v?: string | null): SchedulePriority { + if (v === 'low' || v === 'high') return v; + return 'normal'; +} + +function normalizeType(v?: string | null): ScheduleType { + if (v === 'work' || v === 'family') return v; + return 'personal'; +} + +function normalizeStatus(v?: string | null): ScheduleStatus { + if (v === 'confirmed' || v === 'done' || v === 'cancelled') return v; + return 'pending'; +} + +function normalizeCollaboratorStatus( + v?: string | null, +): CollaboratorStatus | undefined { + if (v === 'confirmed' || v === 'declined') return v; + if (v === 'pending') return 'pending'; + return undefined; +} + +/** 后端 DTO 行 → 前端 Schedule */ +function mapRow(row: RawScheduleRow): Schedule { + const idNum = + row.id !== undefined && row.id !== null ? Number(row.id) : undefined; + const safeId = idNum !== undefined && Number.isFinite(idNum) ? idNum : undefined; + return { + id: safeId, + uuid: String(row.uuid || safeId || ''), + title: String(row.title || ''), + description: row.description ?? '', + start_time: String(row.start_time || ''), + end_time: row.end_time ? String(row.end_time) : '', + priority: normalizePriority(row.priority), + schedule_type: normalizeType(row.schedule_type), + status: normalizeStatus(row.status), + is_collaborative: Boolean(row.is_collaborative), + collaborator_status: normalizeCollaboratorStatus(row.collaborator_status), + collaborator_name: row.collaborator_name ?? undefined, + }; +} + +/** 提取后端错误信息(优先 detail) */ +function unwrapError(err: unknown, fallback: string): string { + if (axios.isAxiosError(err)) { + const ax = err as AxiosError<{ detail?: string | unknown[] }>; + const detail = ax.response?.data?.detail; + if (typeof detail === 'string' && detail) return detail; + if (Array.isArray(detail)) { + const joined = detail + .map((d) => + typeof d === 'object' ? JSON.stringify(d) : String(d), + ) + .join('; '); + if (joined) return joined; + } + return ax.message || fallback; + } + if (err instanceof Error && err.message) return err.message; + return fallback; +} + +/** + * 按日期查询日程。 + * @param date YYYY-MM-DD + */ +export async function listByDate(date: string): Promise { + try { + const { data } = await client.get<{ items?: RawScheduleRow[] }>('/schedule', { + params: { date }, + }); + return (data.items || []).map(mapRow); + } catch (err) { + throw new Error(unwrapError(err, '查询日程失败')); + } +} + +/** + * 按日期区间查询日程。 + * @param from YYYY-MM-DD + * @param to YYYY-MM-DD + */ +export async function listByRange(from: string, to: string): Promise { + try { + const { data } = await client.get<{ items?: RawScheduleRow[] }>('/schedule', { + params: { from, to }, + }); + return (data.items || []).map(mapRow); + } catch (err) { + throw new Error(unwrapError(err, '查询日程区间失败')); + } +} + +/** + * 新建日程。 + */ +export async function create(payload: ScheduleCreatePayload): Promise { + try { + const { data } = await client.post<{ + id?: number | string; + item?: RawScheduleRow; + }>('/schedule', payload); + const row: RawScheduleRow = data.item + ? { ...payload, ...data.item, id: data.item.id ?? data.id } + : { ...payload, id: data.id }; + return mapRow(row); + } catch (err) { + throw new Error(unwrapError(err, '新建日程失败')); + } +} + +/** + * 更新日程。 + */ +export async function update( + id: number | string, + payload: ScheduleUpdatePayload, +): Promise { + try { + await client.put(`/schedule/${id}`, { id: Number(id), ...payload }); + } catch (err) { + throw new Error(unwrapError(err, '更新日程失败')); + } +} + +/** + * 删除日程。 + */ +export async function remove(id: number | string): Promise { + try { + await client.delete(`/schedule/${id}`); + } catch (err) { + throw new Error(unwrapError(err, '删除日程失败')); + } +} + +/** + * 协作日程确认 / 取消确认。 + * @param confirmed true=确认, false=取消确认(回到 pending) + */ +export async function confirmCollaboration( + id: number | string, + confirmed: boolean = true, +): Promise { + try { + const { data } = await client.post<{ status?: string }>( + `/schedule/${id}/confirm`, + null, + { params: { confirmed } }, + ); + return normalizeStatus(data.status); + } catch (err) { + throw new Error(unwrapError(err, '协作确认失败')); + } +} diff --git a/src/web/frontend/src/components/ChatBubble.tsx b/src/web/frontend/src/components/ChatBubble.tsx new file mode 100644 index 0000000..264936b --- /dev/null +++ b/src/web/frontend/src/components/ChatBubble.tsx @@ -0,0 +1,129 @@ +/** + * ChatBubble - visual representation of a single chat message. + * + * - user: right-aligned, blue background + * - assistant: left-aligned, white background, markdown rendered + * - system: centered, gray + * + * Streaming assistant messages show a blinking caret at the end. + */ + +import React, { useMemo } from 'react'; +import { Typography } from 'antd'; +import type { ChatMessage } from '../hooks/useChatStream'; + +const { Text } = Typography; + +export interface ChatBubbleProps { + message: ChatMessage; + /** Optional render function for markdown; defaults to a simple per-line renderer */ + renderMarkdown?: (content: string) => React.ReactNode; +} + +function formatRelativeTime(iso: string): string { + try { + const then = new Date(iso).getTime(); + if (Number.isNaN(then)) return ''; + const diff = Date.now() - then; + if (diff < 0) return '刚刚'; + const sec = Math.floor(diff / 1000); + if (sec < 10) return '刚刚'; + if (sec < 60) return `${sec}秒前`; + const min = Math.floor(sec / 60); + if (min < 60) return `${min}分钟前`; + const hr = Math.floor(min / 60); + if (hr < 24) return `${hr}小时前`; + const day = Math.floor(hr / 24); + if (day < 30) return `${day}天前`; + return new Date(iso).toLocaleDateString(); + } catch { + return ''; + } +} + +function defaultMarkdown(content: string): React.ReactNode { + // Lightweight renderer: split by newlines, keep it simple. + return content.split('\n').map((line, idx) => ( + + {line} + {idx < content.split('\n').length - 1 ?
: null} + + )); +} + +const baseBubble = + 'inline-block max-w-[80%] px-3 py-2 rounded-lg shadow-sm break-words whitespace-pre-wrap leading-relaxed text-sm'; + +const stylesByRole: Record = { + user: { + wrap: 'flex justify-end', + bubble: `${baseBubble} bg-blue-500 text-white rounded-tr-sm`, + text: 'text-white', + align: 'items-end', + }, + assistant: { + wrap: 'flex justify-start', + bubble: `${baseBubble} bg-white text-gray-800 border border-gray-200 rounded-tl-sm`, + text: 'text-gray-800', + align: 'items-start', + }, + system: { + wrap: 'flex justify-center', + bubble: `${baseBubble} bg-gray-100 text-gray-500 italic rounded-full px-4 py-1 text-xs`, + text: 'text-gray-500', + align: 'items-center', + }, +}; + +const ChatBubbleComponent: React.FC = ({ message, renderMarkdown }) => { + const { role, content, timestamp, done } = message; + const style = stylesByRole[role] ?? stylesByRole.system; + const isStreaming = role === 'assistant' && done === false; + + const renderedContent = useMemo(() => { + if (role === 'system') return content; + const fn = renderMarkdown ?? defaultMarkdown; + return fn(content); + }, [role, content, renderMarkdown]); + + if (role === 'system') { + return ( +
+
+
+ {content} +
+ {timestamp ? ( + + {formatRelativeTime(timestamp)} + + ) : null} +
+
+ ); + } + + return ( +
+
+
+
{renderedContent}
+ {isStreaming ? ( + ▍ + ) : null} +
+ {timestamp ? ( + + {formatRelativeTime(timestamp)} + + ) : null} +
+
+ ); +}; + +export const ChatBubble = React.memo(ChatBubbleComponent); +export default ChatBubble; diff --git a/src/web/frontend/src/components/ChatInput.tsx b/src/web/frontend/src/components/ChatInput.tsx new file mode 100644 index 0000000..de00883 --- /dev/null +++ b/src/web/frontend/src/components/ChatInput.tsx @@ -0,0 +1,91 @@ +/** + * ChatInput - auto-resizing textarea with send button. + * + * 快捷键: + * - Enter: 发送 + * - Shift+Enter: 换行 + * - Ctrl/⌘ + Enter: 发送(与 Enter 等价,便于多行输入场景) + * - Disabled while sending + */ + +import React, { useCallback, useRef, useState } from 'react'; +import { Input, Button, Tooltip } from 'antd'; +import { SendOutlined, LoadingOutlined } from '@ant-design/icons'; + +export interface ChatInputProps { + onSend: (text: string) => void; + disabled?: boolean; + placeholder?: string; + /** Maximum rows the textarea will auto-grow to. */ + maxRows?: number; +} + +const ChatInputComponent: React.FC = ({ + onSend, + disabled = false, + placeholder = '按 Ctrl/⌘ + Enter 发送,Enter 发送,Shift+Enter 换行', + maxRows = 6, +}) => { + const [value, setValue] = useState(''); + const taRef = useRef(null); + + const submit = useCallback(() => { + const text = value.trim(); + if (!text || disabled) return; + onSend(text); + setValue(''); + // refocus after sending + requestAnimationFrame(() => { + try { + taRef.current?.focus?.(); + } catch { + // ignore + } + }); + }, [value, disabled, onSend]); + + const handleKeyDown = useCallback( + (e: React.KeyboardEvent) => { + // Ctrl/⌘ + Enter 发送 + if ((e.ctrlKey || e.metaKey) && e.key === 'Enter') { + e.preventDefault(); + submit(); + return; + } + // Enter 直接发送(无 Shift) + if (e.key === 'Enter' && !e.shiftKey && !e.nativeEvent.isComposing) { + e.preventDefault(); + submit(); + } + }, + [submit] + ); + + return ( +
+ setValue(e.target.value)} + onKeyDown={handleKeyDown} + autoSize={{ minRows: 1, maxRows }} + placeholder={placeholder} + disabled={disabled} + className="flex-1" + /> + + + +
+ ); +}; + +export const ChatInput = React.memo(ChatInputComponent); +export default ChatInput; diff --git a/src/web/frontend/src/components/DataTable.tsx b/src/web/frontend/src/components/DataTable.tsx new file mode 100644 index 0000000..5e37bf0 --- /dev/null +++ b/src/web/frontend/src/components/DataTable.tsx @@ -0,0 +1,171 @@ +import React, { useMemo, useState } from 'react'; +import { Table, Button, Popconfirm, Tag, Tooltip, Typography, Space } from 'antd'; +import type { ColumnsType } from 'antd/es/table'; +import { DeleteOutlined, CodeOutlined } from '@ant-design/icons'; +import type { Column, Row } from '@/types/database'; + +const { Text, Paragraph } = Typography; + +export interface DataTableProps { + columns: Column[]; + data: Row[]; + loading?: boolean; + rowKey?: string; + onDelete?: (uuid: string) => void; +} + +const formatValue = (value: unknown, type: Column['type']): React.ReactNode => { + if (value === null || value === undefined) { + return —; + } + switch (type) { + case 'number': + return {Number(value).toLocaleString()}; + case 'datetime': + return {String(value)}; + case 'boolean': + return value ? true : false; + case 'json': + return ; + case 'string': + default: + return {String(value)}; + } +}; + +const JsonCell: React.FC<{ value: unknown }> = ({ value }) => { + const [open, setOpen] = useState(false); + const text = useMemo(() => { + try { + return JSON.stringify(value, null, 2); + } catch { + return String(value); + } + }, [value]); + if (open) { + return ( + + + 
+          {text}
+
+ + ); + } + return ( + + + + ); +}; + +const DataTable: React.FC = ({ + columns, + data, + loading = false, + rowKey = 'uuid', + onDelete, +}) => { + const tableColumns: ColumnsType = useMemo(() => { + const antdColumns: ColumnsType = columns.map((col) => ({ + title: col.label, + dataIndex: col.key, + key: col.key, + width: col.width, + sorter: + col.type === 'number' || col.type === 'datetime' + ? (a: Row, b: Row) => { + const av = a[col.key]; + const bv = b[col.key]; + if (typeof av === 'number' && typeof bv === 'number') { + return av - bv; + } + return String(av ?? '').localeCompare(String(bv ?? '')); + } + : undefined, + filters: + col.type === 'string' || col.type === 'datetime' + ? Array.from(new Set(data.map((r) => String(r[col.key] ?? '')))) + .slice(0, 20) + .map((v) => ({ text: v, value: v })) + : undefined, + onFilter: (value, record) => String(record[col.key] ?? '') === value, + render: (value: unknown) => formatValue(value, col.type), + ellipsis: col.type === 'string' || col.type === 'datetime', + })); + + if (onDelete) { + antdColumns.push({ + title: 'Actions', + key: 'actions', + width: 120, + fixed: 'right', + render: (_value, record) => { + const uuid = String(record[rowKey] ?? ''); + return ( + onDelete(uuid)} + > + + + ); + }, + }); + } + + return antdColumns; + }, [columns, data, onDelete, rowKey]); + + return ( +
+ + rowKey={rowKey} + columns={tableColumns} + dataSource={data} + loading={loading} + scroll={{ x: 'max-content' }} + size="small" + pagination={{ + defaultPageSize: 10, + pageSizeOptions: ['10', '20', '50', '100'], + showSizeChanger: true, + showTotal: (total) => `${total} rows`, + }} + /> + {data.length === 0 && !loading ? ( + + No rows in this table. + + ) : null} +
+ ); +}; + +export default DataTable; diff --git a/src/web/frontend/src/components/Debug/LogExporter.tsx b/src/web/frontend/src/components/Debug/LogExporter.tsx new file mode 100644 index 0000000..f75f5b6 --- /dev/null +++ b/src/web/frontend/src/components/Debug/LogExporter.tsx @@ -0,0 +1,111 @@ +/** + * LogExporter - export log entries as TXT or NDJSON via Blob download. + * + * The component is purely presentational + serialization; it does not + * know how the log list was filtered. The parent passes the array it + * wants exported (typically the filtered view, so "what you see is + * what you export"). + * + * Output formats: + * - TXT: one entry per line, "[timestamp] [level] [module] message" + * - JSON: NDJSON, `logs.map(JSON.stringify).join('\n')` — one JSON + * object per line, friendly to `jq` and log-shippers. + * + * antd 5 removed `Button.Group` in favor of `Space.Compact`; we use the + * latter to match the recommended antd v5 idiom and stay consistent + * with the rest of the codebase (NPSConfigDrawer). + */ + +import React, { useCallback } from 'react'; +import { Button, Space, message as antdMessage } from 'antd'; +import { DownloadOutlined, CodeOutlined } from '@ant-design/icons'; +import dayjs from 'dayjs'; +import type { LogEntry } from './types'; + +export interface LogExporterProps { + logs: LogEntry[]; + /** Filename prefix; default "debug". Final name is `${prefix}-YYYYMMDD-HHmmss.{txt,json}`. */ + filenamePrefix?: string; + /** Optional click hook, fired after a successful export. */ + onExported?: (format: 'txt' | 'json', count: number) => void; +} + +function downloadFile(filename: string, content: string, mime: string): void { + const blob = new Blob([content], { type: mime }); + const url = URL.createObjectURL(blob); + const a = document.createElement('a'); + a.href = url; + a.download = filename; + document.body.appendChild(a); + a.click(); + document.body.removeChild(a); + // Defer revoke so Safari has a chance to start the download. + setTimeout(() => URL.revokeObjectURL(url), 0); +} + +function formatLogLine(l: LogEntry): string { + return `[${l.timestamp}] [${l.level}] [${l.module}] ${l.message}`; +} + +function buildFilename(prefix: string, ext: 'txt' | 'json'): string { + return `${prefix}-${dayjs().format('YYYYMMDD-HHmmss')}.${ext}`; +} + +const LogExporter: React.FC = ({ + logs, + filenamePrefix = 'debug', + onExported, +}) => { + const guardEmpty = useCallback((): boolean => { + if (logs.length === 0) { + antdMessage.warning('没有可导出的日志'); + return false; + } + return true; + }, [logs.length]); + + const onExportTxt = useCallback(() => { + if (!guardEmpty()) return; + const text = logs.map(formatLogLine).join('\n'); + downloadFile( + buildFilename(filenamePrefix, 'txt'), + text, + 'text/plain;charset=utf-8' + ); + onExported?.('txt', logs.length); + }, [logs, filenamePrefix, onExported, guardEmpty]); + + const onExportJson = useCallback(() => { + if (!guardEmpty()) return; + // NDJSON: one JSON object per line. Use plain JSON.stringify (no indent) + // so each line is a single compact record, easy to stream / grep. + const ndjson = logs.map((l) => JSON.stringify(l)).join('\n'); + downloadFile( + buildFilename(filenamePrefix, 'json'), + ndjson, + 'application/x-ndjson;charset=utf-8' + ); + onExported?.('json', logs.length); + }, [logs, filenamePrefix, onExported, guardEmpty]); + + return ( + + + + + ); +}; + +export default LogExporter; diff --git a/src/web/frontend/src/components/Debug/LogFilterBar.tsx b/src/web/frontend/src/components/Debug/LogFilterBar.tsx new file mode 100644 index 0000000..d25413a --- /dev/null +++ b/src/web/frontend/src/components/Debug/LogFilterBar.tsx @@ -0,0 +1,89 @@ +/** + * LogFilterBar - client-side filter controls for the debug log viewer. + * + * Three controls, all controlled by the parent via the `value` / `onChange` + * contract so the filtering logic stays in the page that owns the data: + * + * 1. Module substring match (`Input`) + * 2. Level multi-select (`Select mode="multiple"`) — INFO / WARN / ERROR / DEBUG + * 3. Time range (`DatePicker.RangePicker`) — from / to + * + * Convention: an empty `levels` array means "match all levels" (no filter). + * The parent is responsible for translating this state into a `filter()` + * predicate over the raw log list. + */ + +import React from 'react'; +import { Input, Select, DatePicker, Space } from 'antd'; +import type { Dayjs } from 'dayjs'; + +export const ALL_LOG_LEVELS = ['INFO', 'WARN', 'ERROR', 'DEBUG'] as const; +export type LogLevelOption = (typeof ALL_LOG_LEVELS)[number]; + +export interface LogFilterValue { + levels: LogLevelOption[]; + module: string; + range: [Dayjs | null, Dayjs | null] | null; +} + +export interface LogFilterBarProps { + value: LogFilterValue; + onChange: (next: LogFilterValue) => void; + /** + * Optional list of known module names. When provided, the Input + * is augmented with a native for browser-side autocomplete. + */ + moduleOptions?: string[]; +} + +const LogFilterBar: React.FC = ({ value, onChange, moduleOptions }) => { + // Stable, collision-resistant id for the native datalist. + const datalistId = React.useMemo( + () => `debug-modules-${Math.random().toString(36).slice(2, 8)}`, + [] + ); + + return ( + + onChange({ ...value, module: e.target.value })} + allowClear + style={{ width: 220 }} + list={moduleOptions && moduleOptions.length > 0 ? datalistId : undefined} + /> + {moduleOptions && moduleOptions.length > 0 && ( + + {moduleOptions.map((m) => ( + + )} + + mode="multiple" + allowClear + maxTagCount="responsive" + placeholder="选择级别 (空 = 全部)" + value={value.levels} + onChange={(levels) => + // Defensive cast: antd's SelectValue widens to string|number|LabeledValue, + // but the parent only ever feeds LogLevelOption values. + onChange({ ...value, levels: levels as LogLevelOption[] }) + } + style={{ minWidth: 220 }} + options={ALL_LOG_LEVELS.map((l) => ({ value: l, label: l }))} + /> + + onChange({ ...value, range: r as [Dayjs | null, Dayjs | null] | null }) + } + allowClear + /> + + ); +}; + +export default LogFilterBar; diff --git a/src/web/frontend/src/components/Debug/types.ts b/src/web/frontend/src/components/Debug/types.ts new file mode 100644 index 0000000..159f487 --- /dev/null +++ b/src/web/frontend/src/components/Debug/types.ts @@ -0,0 +1,16 @@ +/** + * Shared types for the Debug page and its sub-components. + * + * Kept in a leaf file (no React imports) so both the page and the + * extracted components can import it without risking circular deps. + */ + +export type LogLevel = 'INFO' | 'WARN' | 'ERROR' | 'DEBUG' | string; + +export interface LogEntry { + id: string; + timestamp: string; + level: LogLevel; + module: string; + message: string; +} diff --git a/src/web/frontend/src/components/EmotionPanel.tsx b/src/web/frontend/src/components/EmotionPanel.tsx new file mode 100644 index 0000000..39c7587 --- /dev/null +++ b/src/web/frontend/src/components/EmotionPanel.tsx @@ -0,0 +1,431 @@ +/** + * EmotionPanel - Plutchik 8 维情感雷达 + * + * 功能： + * - 使用 echarts-for-react 渲染 8 维雷达图（joy/trust/fear/surprise/sadness/disgust/anger/anticipation） + * - 显示当前实时值（实色）+ 历史最高值（淡色背景） + * - 配色：积极情感（joy/trust/anticipation）暖色，消极情感（fear/sadness/disgust/anger）冷色，surprise 中性色 + * - 数据通过 props 传入；useEmotionWebSocket hook 订阅 /ws/event 过滤 type="emotion_update" + * - 数据变化时 ECharts 自动平滑过渡（setOption 增量更新） + * - 数据为空时显示骨架屏 + */ + +import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'; +import ReactECharts from 'echarts-for-react'; +import { Skeleton, Empty, Card, Tag } from 'antd'; +import { useWebSocket, WebSocketStatus } from '../hooks/useWebSocket'; + +// ============= 类型定义 ============= + +/** Plutchik 8 维情感数据 */ +export interface EmotionData { + joy: number; + trust: number; + fear: number; + surprise: number; + sadness: number; + disgust: number; + anger: number; + anticipation: number; +} + +/** EmotionPanel 组件 Props */ +export interface EmotionPanelProps { + /** 实时情感数据（每维 0~maxValue） */ + data: EmotionData; + /** 雷达最大值，默认 10 */ + maxValue?: number; + /** 卡片标题 */ + title?: string; + /** 历史最高值（淡色背景叠加），可选 */ + historicalMax?: EmotionData; + /** 高度，默认 360 */ + height?: number; + /** 后端 event WS 地址（可选，默认 ws:///ws/event） */ + wsUrl?: string; +} + +/** /ws/event 消息类型 */ +export interface EmotionUpdateEvent { + type: 'emotion_update'; + /** 当前情感数据 */ + data: EmotionData; + /** 可选历史最高 */ + historical_max?: EmotionData; + /** 可选时间戳 */ + timestamp?: string; +} + +// ============= 常量 ============= + +/** 8 维情感（顺时针排列，按 Plutchik 顺序） */ +export const EMOTION_KEYS = [ + 'joy', + 'trust', + 'fear', + 'surprise', + 'sadness', + 'disgust', + 'anger', + 'anticipation', +] as const; + +/** 中文标签 */ +export const EMOTION_LABELS_CN: Record = { + joy: '喜悦', + trust: '信任', + fear: '恐惧', + surprise: '惊讶', + sadness: '悲伤', + disgust: '厌恶', + anger: '愤怒', + anticipation: '期待', +}; + +/** + * 配色策略： + * - 积极情感（joy / trust / anticipation）→ 暖色（橙/金/桃） + * - 消极情感（fear / sadness / disgust / anger）→ 冷色（蓝/灰/青/紫） + * - surprise → 中性色 + */ +const POSITIVE_COLOR = '#fa8c16'; // 暖橙 +const NEUTRAL_COLOR = '#8c8c8c'; // 中性灰 +const NEGATIVE_COLOR = '#1677ff'; // 冷蓝（antd 主色） + +function colorForEmotion(key: keyof EmotionData): string { + switch (key) { + case 'joy': + case 'trust': + case 'anticipation': + return POSITIVE_COLOR; + case 'surprise': + return NEUTRAL_COLOR; + case 'fear': + case 'sadness': + case 'disgust': + case 'anger': + return NEGATIVE_COLOR; + default: + return NEUTRAL_COLOR; + } +} + +const DEFAULT_WS_URL = (() => { + if (typeof window === 'undefined') return 'ws://localhost:8000/ws/event'; + const proto = window.location.protocol === 'https:' ? 'wss:' : 'ws:'; + return `${proto}//${window.location.hostname}:8000/ws/event`; +})(); + +const DEFAULT_MAX = 10; +const DEFAULT_HEIGHT = 360; + +// ============= 工具函数 ============= + +function clamp(value: number, min: number, max: number): number { + if (Number.isNaN(value)) return min; + return Math.max(min, Math.min(max, value)); +} + +function safeNumber(value: unknown, fallback = 0): number { + const n = Number(value); + return Number.isFinite(n) ? n : fallback; +} + +function normalizeData(data: Partial | null | undefined): EmotionData { + return { + joy: safeNumber(data?.joy), + trust: safeNumber(data?.trust), + fear: safeNumber(data?.fear), + surprise: safeNumber(data?.surprise), + sadness: safeNumber(data?.sadness), + disgust: safeNumber(data?.disgust), + anger: safeNumber(data?.anger), + anticipation: safeNumber(data?.anticipation), + }; +} + +function isEmotionDataEmpty(data: EmotionData): boolean { + return EMOTION_KEYS.every((k) => data[k] === 0); +} + +// ============= Hook: useEmotionWebSocket ============= + +export interface UseEmotionWebSocketResult { + /** 最新一次 emotion_update 携带的实时数据 */ + data: EmotionData | null; + /** 历史最高（来自后端） */ + historicalMax: EmotionData | null; + /** WebSocket 连接状态 */ + status: WebSocketStatus; + /** 收到的全部 emotion_update 事件（按时间顺序） */ + events: EmotionUpdateEvent[]; + /** 手动重连 */ + reconnect: () => void; +} + +/** + * useEmotionWebSocket - 订阅 /ws/event，过滤 type="emotion_update" 消息。 + * + * 任何 WebSocket 消息只要 type === "emotion_update" 即被解析为 EmotionUpdateEvent； + * 不符合的消息会被忽略，组件不会因此重渲染。 + */ +export function useEmotionWebSocket( + url: string = DEFAULT_WS_URL +): UseEmotionWebSocketResult { + const [data, setData] = useState(null); + const [historicalMax, setHistoricalMax] = useState(null); + const [events, setEvents] = useState([]); + + const handleMessage = useCallback((parsed: unknown) => { + if (!parsed || typeof parsed !== 'object') return; + const msg = parsed as Record; + if (msg.type !== 'emotion_update') return; + if (!msg.data || typeof msg.data !== 'object') return; + + const next: EmotionUpdateEvent = { + type: 'emotion_update', + data: normalizeData(msg.data as Partial), + historical_max: msg.historical_max + ? normalizeData(msg.historical_max as Partial) + : undefined, + timestamp: typeof msg.timestamp === 'string' ? msg.timestamp : undefined, + }; + + setData(next.data); + if (next.historical_max) { + setHistoricalMax(next.historical_max); + } + setEvents((prev) => [...prev, next]); + }, []); + + const { status, reconnectNow } = useWebSocket(url, { + onMessage: handleMessage, + }); + + return { data, historicalMax, status, events, reconnect: reconnectNow }; +} + +// ============= 主组件 ============= + +const EmotionPanel: React.FC = ({ + data, + maxValue = DEFAULT_MAX, + title = '情感雷达（Plutchik 8 维）', + historicalMax, + height = DEFAULT_HEIGHT, + wsUrl, +}) => { + // 可选：组件内部订阅 WS（仅在显式传入 wsUrl 时启用） + // 传入空字符串时 useWebSocket 会直接进入 error 状态，不会发起连接 + const { data: wsData, historicalMax: wsHistoricalMax, status } = useEmotionWebSocket( + wsUrl ?? '' + ); + + // 优先使用 props 传入的数据；否则回退到 WS 推送的数据 + const liveData = useMemo(() => { + if (data && !isEmotionDataEmpty(data)) return normalizeData(data); + if (wsData) return wsData; + return normalizeData(data); + }, [data, wsData]); + + const maxData = useMemo(() => { + if (historicalMax) return normalizeData(historicalMax); + if (wsHistoricalMax) return wsHistoricalMax; + return null; + }, [historicalMax, wsHistoricalMax]); + + // ============= ECharts 配置 ============= + + const indicator = useMemo( + () => + EMOTION_KEYS.map((key) => ({ + name: EMOTION_LABELS_CN[key], + max: maxValue, + })), + [maxValue] + ); + + const option = useMemo(() => { + const currentValues = EMOTION_KEYS.map((k) => clamp(liveData[k], 0, maxValue)); + const maxValues = maxData + ? EMOTION_KEYS.map((k) => clamp(maxData[k], 0, maxValue)) + : currentValues; + + return { + // 全局动画（平滑过渡） + animationDuration: 600, + animationEasing: 'cubicOut' as const, + animationDurationUpdate: 600, + animationEasingUpdate: 'cubicOut' as const, + + color: [...EMOTION_KEYS.map((k) => colorForEmotion(k))], + + title: { + text: title, + left: 'center', + top: 4, + textStyle: { + fontSize: 14, + fontWeight: 500, + color: '#262626', + }, + }, + + tooltip: { + trigger: 'item', + confine: true, + formatter: (params: any) => { + const idx = params?.dataIndex ?? 0; + const key = EMOTION_KEYS[idx]; + const val = currentValues[idx]; + const max = maxValues[idx]; + return `
+
${EMOTION_LABELS_CN[key]}（${key}）
+
当前值：${val.toFixed(2)}
+
历史最高：${max.toFixed(2)}
+
`; + }, + }, + + legend: { + show: true, + bottom: 0, + data: ['当前值', '历史最高'], + textStyle: { color: '#595959' }, + }, + + radar: { + indicator, + shape: 'polygon' as const, + radius: '65%', + center: ['50%', '54%'], + splitNumber: 4, + axisName: { + color: '#262626', + fontSize: 12, + }, + splitLine: { + lineStyle: { color: '#d9d9d9' }, + }, + splitArea: { + areaStyle: { + color: ['rgba(250, 250, 250, 0.4)', 'rgba(250, 250, 250, 0.7)'], + }, + }, + axisLine: { + lineStyle: { color: '#bfbfbf' }, + }, + }, + + series: [ + // 历史最高（淡色背景） + ...(maxData + ? [ + { + name: '历史最高', + type: 'radar' as const, + symbol: 'circle' as const, + symbolSize: 4, + lineStyle: { + color: '#bfbfbf', + width: 1, + type: 'dashed' as const, + }, + areaStyle: { + color: 'rgba(191, 191, 191, 0.15)', + }, + itemStyle: { color: '#bfbfbf' }, + data: [ + { + value: maxValues, + name: '历史最高', + }, + ], + }, + ] + : []), + // 当前值（实色） + { + name: '当前值', + type: 'radar' as const, + symbol: 'circle' as const, + symbolSize: 6, + lineStyle: { + color: '#1677ff', + width: 2, + }, + areaStyle: { + color: 'rgba(22, 119, 255, 0.25)', + }, + itemStyle: { color: '#1677ff' }, + data: [ + { + value: currentValues, + name: '当前值', + }, + ], + }, + ], + }; + }, [indicator, liveData, maxData, maxValue, title]); + + // 强制 useEffect + setOption 平滑过渡 + const echartsRef = useRef(null); + useEffect(() => { + const inst = echartsRef.current?.getEchartsInstance?.(); + if (!inst) return; + inst.setOption(option, { notMerge: false, lazyUpdate: true }); + }, [option]); + + const noData = + isEmotionDataEmpty(liveData) && + (maxData === null || isEmotionDataEmpty(maxData)); + + const statusTag = (() => { + switch (status) { + case 'open': + return WS 已连接; + case 'connecting': + return 连接中; + case 'closed': + return 已断开; + case 'error': + return 错误; + default: + return null; + } + })(); + + return ( + + {noData ? ( + + ) : ( + { + echartsRef.current = e; + }} + option={option} + notMerge={false} + lazyUpdate + style={{ height, width: '100%' }} + opts={{ renderer: 'canvas' }} + /> + )} + {noData && ( + + )} + + ); +}; + +export default EmotionPanel; diff --git a/src/web/frontend/src/components/EventTable.tsx b/src/web/frontend/src/components/EventTable.tsx new file mode 100644 index 0000000..4a9662c --- /dev/null +++ b/src/web/frontend/src/components/EventTable.tsx @@ -0,0 +1,173 @@ +import React from 'react'; +import { Table, Tag, Button, Space, Typography, Descriptions, Empty } from 'antd'; +import type { ColumnsType, ExpandableConfig } from 'antd/es/table/interface'; +import { CheckOutlined, InboxOutlined, EyeOutlined } from '@ant-design/icons'; +import type { Event, EventStatus, EventType } from '../types/event'; + +const { Text, Paragraph } = Typography; + +export interface EventTableProps { + events: Event[]; + onMarkRead?: (uuid: string) => void; + onArchive?: (uuid: string) => void; + loading?: boolean; +} + +const typeColorMap: Record = { + notification: 'blue', + task: 'geekblue', + system: 'purple', +}; + +const typeLabelMap: Record = { + notification: 'notification', + task: 'task', + system: 'system', +}; + +const statusColorMap: Record = { + pending: 'gold', + completed: 'green', + failed: 'red', +}; + +const statusLabelMap: Record = { + pending: 'pending', + completed: 'completed', + failed: 'failed', +}; + +const formatTimestamp = (iso: string): string => { + const d = new Date(iso); + if (Number.isNaN(d.getTime())) return iso; + const pad = (n: number) => String(n).padStart(2, '0'); + return `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())} ${pad(d.getHours())}:${pad(d.getMinutes())}:${pad(d.getSeconds())}`; +}; + +const EventTable: React.FC = ({ events, onMarkRead, onArchive, loading = false }) => { + const columns: ColumnsType = [ + { + title: '时间', + dataIndex: 'timestamp', + key: 'timestamp', + width: 180, + sorter: (a, b) => new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime(), + defaultSortOrder: 'descend', + render: (value: string) => {formatTimestamp(value)}, + }, + { + title: '类型', + dataIndex: 'type', + key: 'type', + width: 130, + filters: (Object.keys(typeLabelMap) as EventType[]).map((k) => ({ text: typeLabelMap[k], value: k })), + onFilter: (value, record) => record.type === value, + render: (value: EventType) => {typeLabelMap[value]}, + }, + { + title: '状态', + dataIndex: 'status', + key: 'status', + width: 110, + filters: (Object.keys(statusLabelMap) as EventStatus[]).map((k) => ({ text: statusLabelMap[k], value: k })), + onFilter: (value, record) => record.status === value, + render: (value: EventStatus) => {statusLabelMap[value]}, + }, + { + title: '来源', + dataIndex: 'source', + key: 'source', + width: 200, + ellipsis: true, + sorter: (a, b) => a.source.localeCompare(b.source), + }, + { + title: '消息', + dataIndex: 'message', + key: 'message', + ellipsis: true, + sorter: (a, b) => a.message.localeCompare(b.message), + }, + { + title: '操作', + key: 'actions', + width: 200, + fixed: 'right', + render: (_value, record) => ( + + + + + ), + }, + ]; + + const expandable: ExpandableConfig = { + expandedRowRender: (record) => ( +
+ Payload 详情}> + + {record.uuid} + + + {record.timestamp} + + + + 
{JSON.stringify(record.payload, null, 2)}
+ + + +
+ ), + rowExpandable: () => true, + }; + + return ( + + rowKey="uuid" + columns={columns} + dataSource={events} + loading={loading} + size="middle" + bordered + pagination={{ + pageSize: 20, + showSizeChanger: true, + showTotal: (total) => `共 ${total} 条`, + pageSizeOptions: ['10', '20', '50', '100'], + }} + scroll={{ x: 1000 }} + expandable={expandable} + locale={{ + emptyText: , + }} + /> + ); +}; + +export default EventTable; diff --git a/src/web/frontend/src/components/KnowledgeGraph.tsx b/src/web/frontend/src/components/KnowledgeGraph.tsx new file mode 100644 index 0000000..be70745 --- /dev/null +++ b/src/web/frontend/src/components/KnowledgeGraph.tsx @@ -0,0 +1,171 @@ +import { useMemo } from 'react'; +import ReactECharts from 'echarts-for-react'; +import type { Entity, EntityCategory, Relation } from '../types/knowledge'; + +/** + * KnowledgeGraph —— 用 ECharts 渲染的实体关系图。 + * + * 设计要点: + * - 节点 = Entity,边 = Relation + * - 节点颜色按 category 映射 (固定调色板) + * - 边宽度按 mention_count 线性映射 + * - 节点点击通过 onNodeClick 回调把 uuid 抛给父组件 + * - 使用力导向布局 (force / repulsion) + */ + +export interface KnowledgeGraphProps { + entities: Entity[]; + relations: Relation[]; + /** 当前高亮的实体 uuid(用于给节点加边框) */ + selectedUuid?: string; + onNodeClick?: (uuid: string) => void; + /** 容器高度,默认 480 */ + height?: number; +} + +const CATEGORY_COLORS: Record = { + person: '#1677ff', + project: '#722ed1', + concept: '#13c2c2', + location: '#52c41a', + organization: '#fa8c16', + event: '#eb2f96', +}; + +const CATEGORY_LABELS: Record = { + person: '人物', + project: '项目', + concept: '概念', + location: '地点', + organization: '组织', + event: '事件', +}; + +/** + * 把 mention_count 映射为 ECharts edge 线宽 (1 ~ 8 px)。 + * mention_count 越大线越粗,直观体现关联强度。 + */ +function widthForMention(count: number, maxCount: number): number { + if (maxCount <= 0) return 1; + const ratio = count / maxCount; + return Math.max(1, Math.min(8, 1 + ratio * 7)); +} + +export default function KnowledgeGraph({ + entities, + relations, + selectedUuid, + onNodeClick, + height = 480, +}: KnowledgeGraphProps): JSX.Element { + const option = useMemo(() => { + const maxMention = relations.reduce( + (m, r) => Math.max(m, r.mention_count), + 0, + ); + + const nodes = entities.map((e) => ({ + id: e.uuid, + name: e.name, + symbolSize: Math.max(20, Math.min(60, 20 + e.mention_count * 0.4)), + category: e.category, + // 用 itemStyle 同时表达"分类色"和"选中高亮" + itemStyle: { + color: CATEGORY_COLORS[e.category], + borderColor: selectedUuid === e.uuid ? '#000' : '#fff', + borderWidth: selectedUuid === e.uuid ? 3 : 1.5, + shadowBlur: 6, + shadowColor: 'rgba(0,0,0,0.15)', + }, + label: { show: true, fontSize: 12 }, + value: e.confidence, + })); + + const links = relations.map((r) => ({ + source: r.source, + target: r.target, + name: r.relation, + lineStyle: { + width: widthForMention(r.mention_count, maxMention), + color: '#bfbfbf', + curveness: 0.1, + }, + value: r.mention_count, + })); + + return { + tooltip: { + trigger: 'item', + formatter: (params: { dataType?: string; data?: Record }) => { + if (params.dataType === 'node' && params.data) { + const d = params.data as { name: string; category: EntityCategory; value: number }; + return `${d.name}
类型: ${CATEGORY_LABELS[d.category]}
置信度: ${( + d.value * 100 + ).toFixed(0)}%`; + } + if (params.dataType === 'edge' && params.data) { + const d = params.data as { name: string; value: number }; + return `关系: ${d.name}
提及次数: ${d.value}`; + } + return ''; + }, + }, + legend: [ + { + data: Object.keys(CATEGORY_LABELS).map((k) => ({ + name: CATEGORY_LABELS[k as EntityCategory], + icon: 'circle', + })), + textStyle: { fontSize: 11 }, + top: 0, + }, + ], + series: [ + { + type: 'graph', + layout: 'force', + roam: true, + draggable: true, + animation: true, + data: nodes, + links, + categories: Object.keys(CATEGORY_COLORS).map((k) => ({ + name: CATEGORY_LABELS[k as EntityCategory], + itemStyle: { color: CATEGORY_COLORS[k as EntityCategory] }, + })), + force: { + repulsion: 260, + edgeLength: [60, 140], + gravity: 0.05, + friction: 0.2, + }, + emphasis: { + focus: 'adjacency', + lineStyle: { width: 4 }, + }, + }, + ], + } as Record; + }, [entities, relations, selectedUuid]); + + const onEvents = useMemo( + () => ({ + click: (event: { dataType?: string; data?: { id?: string } }) => { + if (event.dataType === 'node' && event.data?.id && onNodeClick) { + onNodeClick(event.data.id); + } + }, + }), + [onNodeClick], + ); + + return ( + + ); +} diff --git a/src/web/frontend/src/components/Layout.tsx b/src/web/frontend/src/components/Layout.tsx new file mode 100644 index 0000000..c56a36d --- /dev/null +++ b/src/web/frontend/src/components/Layout.tsx @@ -0,0 +1,126 @@ +/** + * AppLayout - 全局壳布局(Sider + Header + Content) + * + * Stage B.1 移动端降级: + * - 监听 window.innerWidth,当 < 768px 时不渲染主布局 + * - 改用 antd Result 组件显示"请使用桌面端访问"提示 + * - 提供"刷新重试"按钮,方便用户切回桌面后手动验证 + * - SSR 安全:用 typeof window 兜底,避免在非浏览器环境崩溃 + * + * 注意:本降级是软提示,实际渲染层(echarts 等)对窄屏并不友好,因此 + * 显式拦截比"勉强渲染"更稳妥。 + */ + +import React, { useEffect, useState } from 'react'; +import { Layout, Menu, Badge, Result, Button } from 'antd'; +import { + MessageOutlined, + BugOutlined, + BookOutlined, + CalendarOutlined, + AlertOutlined, + StarOutlined, + DatabaseOutlined, + ThunderboltOutlined, + SettingOutlined, +} from '@ant-design/icons'; +import { Outlet, useNavigate, useLocation } from 'react-router-dom'; + +const { Sider, Header, Content } = Layout; + +/** 移动端判断阈值(像素) */ +const MOBILE_BREAKPOINT = 768; + +type MenuItem = { + key: string; + label: string; + icon: React.ReactNode; +}; + +const menuItems: MenuItem[] = [ + { key: '/chat', label: 'Chat', icon: }, + { key: '/debug', label: 'Debug', icon: }, + { key: '/knowledge', label: 'Knowledge', icon: }, + { key: '/schedule', label: 'Schedule', icon: }, + { key: '/event', label: 'Event', icon: }, + { key: '/nps', label: 'NPS', icon: }, + { key: '/database', label: 'Database', icon: }, + { key: '/creative', label: 'Creative', icon: }, + { key: '/settings', label: '系统设置', icon: }, +]; + +/** SSR 安全的初始窗口宽度(默认按桌面端,避免水合不一致) */ +function getInitialWidth(): number { + if (typeof window === 'undefined') return 1280; + return window.innerWidth; +} + +export default function AppLayout() { + const navigate = useNavigate(); + const location = useLocation(); + + // 移动端检测:< 768px 视为移动端 + const [isMobile, setIsMobile] = useState( + () => getInitialWidth() < MOBILE_BREAKPOINT + ); + + useEffect(() => { + if (typeof window === 'undefined') return; + const handler = () => setIsMobile(window.innerWidth < MOBILE_BREAKPOINT); + window.addEventListener('resize', handler); + return () => window.removeEventListener('resize', handler); + }, []); + + const selectedKey = + menuItems.find((item) => location.pathname.startsWith(item.key))?.key ?? '/chat'; + + // 移动端降级:不渲染主布局,改用 Result 提示用户 + if (isMobile) { + return ( +
+ window.location.reload()}> + 刷新重试 + + } + /> +
+ ); + } + + return ( + + +
+ Neo Agent +
+ ({ + key: item.key, + icon: item.icon, + label: item.label, + }))} + onClick={({ key }) => navigate(key)} + /> + + +
+
Neo Agent 控制台
+
+ +
+
+ + + + + + ); +} diff --git a/src/web/frontend/src/components/MessageList.tsx b/src/web/frontend/src/components/MessageList.tsx new file mode 100644 index 0000000..ab5b811 --- /dev/null +++ b/src/web/frontend/src/components/MessageList.tsx @@ -0,0 +1,147 @@ +/** + * MessageList - 虚拟滚动消息列表 + * + * Stage B.1 升级: + * - 使用 react-window FixedSizeList 渲染,可承载 1000+ 条消息而不卡顿 + * - 保留所有原 props 接口(messages / renderItem / renderMarkdown / autoScroll / + * className / emptyText),确保调用方零侵入 + * - 通过 ResizeObserver 自动测量容器高度 + * - 自动滚动到底部:autoScroll=true 时,新消息或最后一条消息变化时,RAF 内 + * 调用 listRef.scrollToItem(messages.length - 1) + * - 消息行高 80px(由 ChatBubble 内容决定,溢出由 ChatBubble 内部处理) + */ + +import React, { useEffect, useMemo, useRef, useState } from 'react'; +import { Empty } from 'antd'; +import { FixedSizeList, ListChildComponentProps } from 'react-window'; +import type { ChatMessage } from '../hooks/useChatStream'; +import { ChatBubble, ChatBubbleProps } from './ChatBubble'; + +export interface MessageListProps { + messages: ChatMessage[]; + /** Optional custom renderer; defaults to ChatBubble */ + renderItem?: (message: ChatMessage) => React.ReactNode; + /** Optional markdown render fn passed through to ChatBubble */ + renderMarkdown?: ChatBubbleProps['renderMarkdown']; + /** Auto-scroll to bottom on new messages; default true */ + autoScroll?: boolean; + /** Optional className for the scroll container */ + className?: string; + /** Placeholder shown when messages is empty */ + emptyText?: string; + /** 单条消息行高(px),默认 80。ChatBubble 高度可能大于该值时由 ChatBubble 内部滚动 */ + itemSize?: number; +} + +const DEFAULT_ITEM_SIZE = 80; + +const MessageListComponent: React.FC = ({ + messages, + renderItem, + renderMarkdown, + autoScroll = true, + className, + emptyText = '还没有对话，开始聊点什么吧…', + itemSize = DEFAULT_ITEM_SIZE, +}) => { + const containerRef = useRef(null); + const listRef = useRef(null); + const prevCountRef = useRef(messages.length); + const prevLastIdRef = useRef(null); + + // 用 ResizeObserver 动态获取容器高度,作为 FixedSizeList 的 height + const [containerHeight, setContainerHeight] = useState(400); + + useEffect(() => { + const el = containerRef.current; + if (!el) return; + // 初始化一次 + setContainerHeight(el.clientHeight || 400); + if (typeof ResizeObserver === 'undefined') { + // 兜底:监听 window resize + const handler = () => setContainerHeight(el.clientHeight || 400); + window.addEventListener('resize', handler); + return () => window.removeEventListener('resize', handler); + } + const ro = new ResizeObserver((entries) => { + for (const entry of entries) { + const h = entry.contentRect.height; + if (h > 0) setContainerHeight(h); + } + }); + ro.observe(el); + return () => ro.disconnect(); + }, []); + + // Determine the id of the last message (for "follow streaming" behaviour). + const lastId = useMemo(() => { + if (messages.length === 0) return null; + return messages[messages.length - 1].id; + }, [messages]); + + // 自动滚动到底部:仅在消息数量变化或最后一条消息 id 变化时触发 + useEffect(() => { + if (!autoScroll) return; + if (messages.length === 0) return; + const isNewMessage = messages.length !== prevCountRef.current; + const isLastChanged = lastId !== prevLastIdRef.current; + if (isNewMessage || isLastChanged) { + // 下一帧让 DOM 先行渲染,再滚到末尾 + requestAnimationFrame(() => { + try { + listRef.current?.scrollToItem(messages.length - 1, 'end'); + } catch { + /* 忽略 scrollToItem 边界异常 */ + } + }); + } + prevCountRef.current = messages.length; + prevLastIdRef.current = lastId; + }, [messages.length, lastId, autoScroll, messages]); + + if (messages.length === 0) { + return ( +
+ +
+ ); + } + + // react-window 行渲染器:每行用 ChatBubble 渲染当前消息 + const Row: React.FC = ({ index, style }) => { + const msg = messages[index]; + return ( +
+ {renderItem ? ( + renderItem(msg) + ) : ( + + )} +
+ ); + }; + + return ( +
+ + {Row} + +
+ ); +}; + +export const MessageList = React.memo(MessageListComponent); +export default MessageList; diff --git a/src/web/frontend/src/components/NPSConfigDrawer.tsx b/src/web/frontend/src/components/NPSConfigDrawer.tsx new file mode 100644 index 0000000..a14d5dd --- /dev/null +++ b/src/web/frontend/src/components/NPSConfigDrawer.tsx @@ -0,0 +1,242 @@ +import React, { useEffect } from 'react'; +import { + Drawer, + Form, + Input, + Select, + Switch, + Button, + Space, + Divider, + Typography, + Empty, + Card, +} from 'antd'; +import { PlusOutlined, MinusCircleOutlined } from '@ant-design/icons'; +import type { NPSType, NPSParam, NPSConfigFormValues } from '../types/nps'; + +const { TextArea } = Input; +const { Text } = Typography; + +export interface NPSConfigDrawerProps { + open: boolean; + onClose: () => void; + onSubmit: (values: NPSConfigFormValues) => void; + initialValues?: Partial; + title?: string; + submitText?: string; +} + +const defaultValues: NPSConfigFormValues = { + name: '', + description: '', + nps_type: 'function', + params: [], + enabled: true, +}; + +const ParamKeyValueEditor: React.FC = () => { + const form = Form.useFormInstance(); + const params: NPSParam[] = (form.getFieldValue('params') as NPSParam[] | undefined) ?? []; + + const updateParams = (next: NPSParam[]) => { + form.setFieldValue('params', next); + }; + + const addRow = () => { + updateParams([...params, { key: '', value: '' }]); + }; + + const removeRow = (idx: number) => { + updateParams(params.filter((_, i) => i !== idx)); + }; + + const updateRow = (idx: number, patch: Partial) => { + updateParams(params.map((p, i) => (i === idx ? { ...p, ...patch } : p))); + }; + + return ( +
+ {params.length === 0 && ( + 暂无参数，点击下方按钮添加} + style={{ margin: '12px 0' }} + /> + )} + + {params.map((p, idx) => ( + + updateRow(idx, { key: e.target.value })} + /> + updateRow(idx, { value: e.target.value })} + /> + +
+ ); +}; + +const NPSConfigDrawer: React.FC = ({ + open, + onClose, + onSubmit, + initialValues, + title = '注册新 NPS', + submitText = '提交', +}) => { + const [form] = Form.useForm(); + + useEffect(() => { + if (open) { + const merged: NPSConfigFormValues = { + ...defaultValues, + ...initialValues, + params: (initialValues?.params ?? []) as NPSParam[], + }; + form.setFieldsValue(merged); + } else { + form.resetFields(); + } + }, [open, initialValues, form]); + + const handleFinish = (values: NPSConfigFormValues) => { + onSubmit(values); + }; + + const handleReset = () => { + const merged: NPSConfigFormValues = { + ...defaultValues, + ...initialValues, + params: (initialValues?.params ?? []) as NPSParam[], + }; + form.setFieldsValue(merged); + }; + + return ( + + + + + } + footer={ +
+ + + + +
+ } + > + + form={form} + layout="vertical" + onFinish={handleFinish} + initialValues={defaultValues} + requiredMark="optional" + > + + + + + +