1 line
20 KiB
JSON
1 line
20 KiB
JSON
{"repo": "anbeime/antinet", "files": {"backend/main.py": {"c61af6c5507b61a08b5f588514b27f86ef60d85c": {"t": 1785768246.1310368, "n": 17536, "h": "2e278ff3d6c2812d", "c": "#!/usr/bin/env python3\r\n# backend/main.py - 主API服务 (重构版)\r\n\"\"\"\r\n知易智能知识管家 - 后端API服务\r\n参考 SiYuan 架构重构，配置和中间件集中管理\r\n\"\"\"\r\n\r\nimport os\r\nimport sys\r\nfrom pathlib import Path\r\nfrom contextlib import asynccontextmanager\r\nfrom dotenv import load_dotenv\r\n\r\n# ============================================================\r\n# 1. 路径设置（必须在导入之前）\r\n# ============================================================\r\nif getattr(sys, 'frozen', False):\r\n # PyInstaller 打包后，exe 已在 backend/ 目录下\r\n backend_dir = os.path.dirname(sys.executable)\r\n project_root = os.path.dirname(backend_dir)\r\nelse:\r\n backend_dir = os.path.dirname(os.path.abspath(__file__))\r\n project_root = os.path.dirname(backend_dir)\r\nload_dotenv(os.path.join(project_root, '.env'))\r\n\r\nif backend_dir not in sys.path:\r\n sys.path.insert(0, backend_dir)\r\nif project_root not in sys.path:\r\n sys.path.insert(0, project_root)\r\n\r\n# ============================================================\r\n# 2. NPU 库路径配置（使用新的配置模块）\r\n# ============================================================\r\ntry:\r\n from conf.npu import NPUConfig\r\n\r\n npu_config = NPUConfig()\r\n lib_path = npu_config.get_libs_path()\r\n\r\n print(f\"[SETUP] QAI 库路径: {lib_path}\")\r\n\r\n # 设置环境变量\r\n paths_to_add = [lib_path]\r\n current_path = os.environ.get('PATH', '')\r\n for p in paths_to_add:\r\n if p not in current_path:\r\n current_path = p + ';' + current_path\r\n os.environ['PATH'] = current_path\r\n os.environ['QAI_LIBS_PATH'] = lib_path\r\n os.environ['QNN_LOG_LEVEL'] = npu_config.QNN_LOG_LEVEL\r\n\r\n # 添加 DLL 目录\r\n for p in paths_to_add:\r\n if os.path.exists(p):\r\n os.add_dll_directory(p)\r\n print(f\"[SETUP] 添加 DLL 目录: {p}\")\r\n\r\n print(f\"[SETUP] NPU 配置: backend={npu_config.QNN_BACKEND}, device={npu_config.QNN_DEVICE}\")\r\nexcept Exception as e:\r\n print(f\"[WARN] NPU 初始化失败（不影响 Genie HTTP 调用）: {e}\")\r\n\r\n# ============================================================\r\n# 3. FastAPI 应用创建\r\n# ============================================================\r\nfrom fastapi import FastAPI\r\nfrom fastapi.responses import JSONResponse\r\n\r\n# 使用新的配置模块\r\nfrom conf.app import AppConfig\r\n\r\napp_config = AppConfig()\r\n\r\n@asynccontextmanager\r\nasync def lifespan(app: FastAPI):\r\n \"\"\"FastAPI 生命周期管理（替代已弃用的 on_event）\"\"\"\r\n # 启动时\r\n try:\r\n from services.reminder_service import start_reminder_service\r\n start_reminder_service()\r\n print(\"[OK] 提醒服务已启动\")\r\n except Exception as e:\r\n print(f\"[WARN] 提醒服务启动失败: {e}\")\r\n yield\r\n # 关闭时\r\n try:\r\n from services.reminder_service import stop_reminder_service\r\n stop_reminder_service()\r\n print(\"[OK] 提醒服务已停止\")\r\n except Exception as e:\r\n print(f\"[WARN] 提醒服务停止失败: {e}\")\r\n\r\napp = FastAPI(\r\n title=app_config.APP_NAME,\r\n version=app_config.APP_VERSION,\r\n description=app_config.APP_DESCRIPTION,\r\n lifespan=lifespan,\r\n max_form_memory_size=app_config.MAX_UPLOAD_SIZE,\r\n)\r\n\r\n# ============================================================\r\n# 4. 中间件设置（使用新的中间件模块）\r\n# ============================================================\r\nfrom middleware import create_middleware_stack\r\n\r\ncreate_middleware_stack(app)\r\nprint(\"[OK] 中间件栈已配置\")\r\n\r\n# ============================================================\r\n# 5. 数据库初始化\r\n# ============================================================\r\nfrom conf.database import DatabaseConfig\r\n\r\ndb_config = DatabaseConfig()\r\nprint(f\"[Database] 数据库路径: {db_config.DB_PATH}\")\r\n\r\nfrom database import DatabaseManager\r\n\r\ndb_manager = DatabaseManager(db_config.DB_PATH)\r\nprint(\"[OK] 数据库初始化完成\")\r\n\r\n# ============================================================\r\n# 6. AI 服务初始化（使用新的 AI 服务模块）\r\n# ============================================================\r\nfrom services.ai.factory import AIServiceFactory, create_ai_service\r\n\r\nAIServiceFactory.create_default_services()\r\n\r\n# 仅本地模型 (NPU/Genie)，不再注册云端 Sensenova\r\n# 如需使用 Sensenova 云端模型，可取消下方注释\r\n# _sensenova_cfg = {\r\n# 'api_key': 'sk-aMuGLXz1jMSznP9zSUxOfS4uTG7wlsFI',\r\n# 'base_url': 'https://token.sensenova.cn/v1',\r\n# 'model': 'sensenova-6.7-flash-lite',\r\n# 'timeout': 60,\r\n# 'max_tokens': 2048,\r\n# 'temperature': 1.0,\r\n# }\r\n# _sensenova = create_ai_service('openai', _sensenova_cfg)\r\n# if _sensenova:\r\n# AIServiceFactory.register('sensenova', _sensenova, set_default=False)\r\n# print(\"[OK] Sensenova 服务已注册（chat/skill/workflow 专用）\")\r\n\r\n# 注册 NVIDIA NIM 服务（API Key 通过环境变量 NVIDIA_API_KEY 设置）\r\n_nim_cfg = {\r\n 'api_key': os.getenv('NVIDIA_API_KEY', ''),\r\n 'base_url': 'https://integrate.api.nvidia.com/v1',\r\n 'model': 'minimaxai/minimax-m2.7',\r\n 'timeout': 60,\r\n 'max_tokens': 8192,\r\n 'temperature': 1.0,\r\n}\r\n_nim = create_ai_service('nim', _nim_cfg)\r\nif _nim and _nim_cfg['api_key']:\r\n AIServiceFactory.register('nim', _nim, set_default=False)\r\n print(\"[OK] NVIDIA NIM 服务已注册\")\r\nelif _nim:\r\n print(\"[SKIP] NVIDIA NIM 服务跳过：未设置 NVIDIA_API_KEY 环境变量\")\r\n\r\nprint(\"[OK] AI 服务工厂已初始化（仅本地模型）\")\n\r\n# ============================================================\r\n# 6.5 AI 模型预热（使用 QAI/QAIRT SDK 的本地模型，无须 Ollama）\r\n# ============================================================\r\n\r\n# ============================================================\r\n# 7. 提醒服务已通过 lifespan 事件管理（见上方 app = FastAPI(lifespan=...) 定义）\r\n# ============================================================\r\n\r\n# ============================================================\r\n# 8. 路由注册（简化版）\r\n# ============================================================\r\ndef register_router(module_name: str):\r\n \"\"\"直接注册路由模块\"\"\"\r\n try:\r\n # 动态导入\r\n parts = module_name.split('.')\r\n if len(parts) == 2:\r\n # routes.xxx 格式\r\n router_module = __import__(module_name, fromlist=['router'])\r\n router = getattr(router_module, 'router', None)\r\n if router:\r\n # 直接使用路由自己的 prefix\r\n actual_prefix = getattr(router, 'prefix', '') or '(无前缀)'\r\n app.include_router(router)\r\n print(f\"[OK] 路由已注册: {module_name} -> {actual_prefix}\")\r\n return True\r\n else:\r\n print(f\"[WARN] {module_name} 无 router 属性\")\r\n return False\r\n except Exception as e:\r\n import traceback\r\n print(f\"[WARN] 无法导入 {module_name}: {e}\")\r\n traceback.print_exc()\r\n return False\r\n\r\n# 核心路由 - 使用各自的 prefix\r\nprint(\"[INFO] 开始注册路由...\")\r\nregister_router(\"routes.knowledge_routes\")\r\nregister_router(\"routes.chat_routes\")\r\nregister_router(\"routes.data_routes\")\r\nregister_router(\"routes.agent_routes\")\r\nregister_router(\"routes.skill_routes\")\r\nregister_router(\"routes.npu_routes\")\r\nregister_router(\"routes.pdf_routes\")\r\nregister_router(\"routes.pdf_opendataloader_routes\") # OpenDataLoader 长 PDF 专用路由\r\nregister_router(\"routes.ocr_routes\") # OCR 路由已存在但未注册\r\nregister_router(\"routes.excel_routes\")\r\nregister_router(\"routes.ppt_routes\")\r\nregister_router(\"routes.multi_model_routes\")\r\nregister_router(\"routes.genie_playground_routes\")\r\nregister_router(\"routes.gtd_routes\")\r\nregister_router(\"routes.backlink_routes\")\r\nregister_router(\"routes.integration_routes\")\r\nregister_router(\"routes.moc_routes\")\r\nregister_router(\"routes.vision_routes\")\r\nregister_router(\"routes.invoice_routes\")\r\n\r\nprint(\"[INFO] 注册增强版聊天路由...\")\r\nregister_router(\"routes.enhanced_chat_routes\")\r\nregister_router(\"routes.hermes_chat_routes\") # Hermes + 8 Agent 协同\r\nregister_router(\"routes.evolving_chat_routes\") # 自进化聊天\r\nregister_router(\"routes.chat_context_routes\")\r\nregister_router(\"routes.md2pdf_routes\")\r\nregister_router(\"routes.card_pdf_routes\")\r\nregister_router(\"routes.libreoffice_routes\")\r\n# register_router(\"routes.vector_search\") # 无router属性，是工具模块\r\nregister_router(\"routes.wiki\")\r\nregister_router(\"routes.markdown_converter_routes\")\r\nregister_router(\"routes.meeting_routes\")\r\nregister_router(\"routes.speech_routes\")\r\nregister_router(\"routes.research_routes\")\r\nregister_router(\"routes.ppt_structure_routes\")\r\nregister_router(\"routes.image_routes\") # 图片上传路由\r\nregister_router(\"routes.analysis_routes\")\r\nregister_router(\"routes.report_routes\")\r\nregister_router(\"routes.collaboration_routes\") # 实时协作 (WebSocket + REST)\r\nregister_router(\"routes.mindmap_routes\") # 思维导图\r\nregister_router(\"routes.remotion_routes\") # Remotion 动态演示\r\nregister_router(\"routes.pdf_edit_routes\") # PDF 文本编辑\r\nregister_router(\"routes.design_system_routes\") # 统一设计系统\r\nregister_router(\"routes.ppt_preview_routes\") # PPT 预览（增强版）\r\nregister_router(\"routes.ppt_native_routes\") # PPT 原生形状生成（SVG→DrawingML）\r\nregister_router(\"routes.collab_docs_routes\") # 协作文档 CRUD + 版本 + 权限\r\nregister_router(\"routes.file_browser_routes\") # 文件浏览器 + 卡片索引联动层\r\nregister_router(\"routes.auth_routes\") # JWT 认证\r\n\r\n# ============================================================\r\n# 9. 初始化各模块的数据库连接\r\n# ============================================================\r\nprint(\"[INFO] 开始初始化各模块数据库连接...\")\r\n\r\n# 自动设置所有路由模块的db_manager\r\nimport routes\r\nfor attr_name in dir(routes):\r\n mod = getattr(routes, attr_name, None)\r\n if hasattr(mod, 'db_manager') and hasattr(mod, 'set_db_manager'):\r\n try:\r\n mod.set_db_manager(db_manager)\r\n print(f\"[OK] {attr_name} 数据库已连接\")\r\n except Exception as e:\r\n print(f\"[WARN] {attr_name} 设置失败: {e}\")\r\n elif hasattr(mod, 'db_manager'):\r\n try:\r\n mod.db_manager = db_manager\r\n print(f\"[OK] {attr_name} db_manager已设置\")\r\n except Exception as e:\r\n pass\r\n\r\n# 单独设置其他需要db_manager的模块\r\ntry:\r\n from routes import auto_card\r\n if hasattr(auto_card, 'set_db_manager'):\r\n auto_card.set_db_manager(db_manager)\r\n print(\"[OK] auto_card 数据库已连接\")\r\n auto_card.db_manager = db_manager\r\nexcept Exception as e:\r\n print(f\"[WARN] auto_card: {e}\")\r\n\r\ntry:\r\n from routes import data_routes\r\n if hasattr(data_routes, 'set_db_manager'):\r\n data_routes.set_db_manager(db_manager)\r\n print(\"[OK] data_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] data_routes: {e}\")\r\n\r\ntry:\r\n from routes import enhanced_chat_routes\r\n enhanced_chat_routes.set_db_manager(db_manager)\r\n print(\"[OK] enhanced_chat 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] enhanced_chat: {e}\")\r\n\r\ntry:\r\n from routes import chat_routes\r\n chat_routes.db_manager = db_manager\r\n print(\"[OK] chat_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] chat_routes: {e}\")\r\n\r\ntry:\r\n from routes import ppt_routes\r\n ppt_routes.set_db_manager(db_manager)\r\n print(\"[OK] ppt_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] ppt_routes: {e}\")\r\n\r\ntry:\r\n from routes import invoice_routes\r\n invoice_routes.set_db_manager(db_manager)\r\n print(\"[OK] invoice_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] invoice_routes: {e}\")\r\n\r\ntry:\r\n from routes import vector_search\r\n vector_search.set_db_manager(db_manager)\r\n vector_search.init_on_startup()\r\n print(\"[OK] vector_search 数据库已连接，embedding 已初始化\")\r\n\r\n from routes import rag_pipeline\r\n rag_pipeline.set_db_manager(db_manager)\r\n print(\"[OK] rag_pipeline 数据库已连接\")\r\n\r\n from routes import auth_routes\r\n auth_routes.set_db_manager(db_manager)\r\n print(\"[OK] auth_routes 数据库已连接\")\r\n\r\n from middleware import audit\r\n audit.set_db_manager(db_manager)\r\n print(\"[OK] audit 中间件数据库已连接\")\nexcept Exception as e:\r\n print(f\"[WARN] vector_search: {e}\")\r\n\r\n# 初始化协作文档路由数据库\r\ntry:\r\n from routes import collab_docs_routes\r\n collab_docs_routes.set_db_manager(db_manager)\r\n print(\"[OK] collab_docs_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] collab_docs_routes: {e}\")\r\n\r\ntry:\r\n from routes import collaboration_routes\r\n collaboration_routes.set_db_manager(db_manager)\r\n print(\"[OK] collaboration_routes 数据库已连接\")\r\nexcept Exception as e:\r\n print(f\"[WARN] collaboration_routes: {e}\")\r\n\r\nprint(\"[INFO] 数据库连接初始化完成\")\r\n\r\n# ============================================================\r\n# 10. 健康检查端点\r\n# ============================================================\r\n@app.get(\"/api/health\")\r\nasync def health_check():\r\n \"\"\"健康检查端点\"\"\"\r\n return {\r\n \"status\": \"ok\",\r\n \"app\": app_config.APP_NAME,\r\n \"version\": app_config.APP_VERSION,\r\n \"database_initialized\": True,\r\n }\r\n\r\n@app.get(\"/api/debug/routes\")\r\nasync def debug_routes():\r\n \"\"\"调试端点 - 列出所有已注册的路由\"\"\"\r\n all_routes = []\r\n for route in app.routes:\r\n if hasattr(route, 'path'):\r\n all_routes.append(route.path)\r\n return {\r\n \"total_routes\": len(all_routes),\r\n \"routes\": sorted(all_routes)\r\n }\r\n\r\n# ============================================================\r\n# 11. 启动服务\r\n# ============================================================\r\nif __name__ == \"__main__\":\r\n import socket\r\n import subprocess\r\n import uvicorn\r\n import sys\r\n import traceback\r\n\r\n def _excepthook(exc_type, exc_value, exc_tb):\r\n msg = \"\".join(traceback.format_exception(exc_type, exc_value, exc_tb))\r\n sys.stderr.write(f\"[UNCAUGHT] {msg}\\n\")\r\n try:\r\n with open(\"uncaught.log\", \"a\", encoding=\"utf-8\") as f:\r\n f.write(f\"\\n[{__import__('datetime').datetime.now()}] {msg}\\n\")\r\n except Exception:\r\n pass\r\n sys.excepthook = _excepthook\r\n\r\n try:\r\n from services.global_error_handler import install_global_handlers\r\n install_global_handlers()\r\n except Exception:\r\n pass\r\n \r\n # 检查端口是否被占用，如果是则释放\r\n def check_and_free_port(port: int):\r\n try:\r\n with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:\r\n s.settimeout(1)\r\n result = s.connect_ex(('127.0.0.1', port))\r\n if result == 0:\r\n print(f\"[PORT] 端口 {port} 已被占用，尝试释放...\")\r\n output = subprocess.run(\r\n ['powershell', '-Command',\r\n f\"Get-NetTCPConnection -LocalPort {port} -ErrorAction SilentlyContinue | Select-Object -ExpandProperty OwningProcess | ForEach-Object {{ Stop-Process -Id $_ -Force -ErrorAction SilentlyContinue }}\"],\r\n capture_output=True, text=True, timeout=10\r\n )\r\n import time\r\n time.sleep(2)\r\n # 再次检查\r\n with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s2:\r\n s2.settimeout(1)\r\n result2 = s2.connect_ex(('127.0.0.1', port))\r\n if result2 == 0:\r\n print(f\"[WARN] 无法释放端口 {port}，请手动关闭占用进程\")\r\n return False\r\n else:\r\n print(f\"[OK] 端口 {port} 已释放\")\r\n return True\r\n else:\r\n print(f\"[OK] 端口 {port} 可用\")\r\n return True\r\n except Exception as e:\r\n print(f\"[WARN] 检查端口 {port} 时出错: {e}\")\r\n return True # 继续尝试启动\r\n \r\n check_and_free_port(app_config.PORT)\r\n \r\n # 打印所有已注册路由（便于诊断 404）\r\n print(f\"\\n{'─'*50}\")\r\n print(f\"已注册路由 ({len([r for r in app.routes if hasattr(r, 'path')])} 条):\")\r\n for route in sorted(app.routes, key=lambda r: getattr(r, 'path', '')):\r\n if hasattr(route, 'path') and hasattr(route, 'methods'):\r\n print(f\" {route.path:50s} {route.methods}\")\r\n print(f\"{'─'*50}\")\r\n \r\n print(f\"\\n{'='*50}\")\r\n print(f\"启动 {app_config.APP_NAME} v{app_config.APP_VERSION}\")\r\n print(f\"服务地址: http://{app_config.HOST}:{app_config.PORT}\")\r\n print(f\"{'='*50}\\n\")\r\n \r\n _auto_restart = os.environ.get(\"ZHIYI_AUTO_RESTART\", \"1\") == \"1\"\r\n _attempt = 0\r\n while True:\r\n try:\r\n uvicorn.run(\r\n app, # 直接传递 app 对象（PyInstaller frozen 模式需要，避免 \"main:app\" 字符串导入失败）\r\n host=app_config.HOST,\r\n port=app_config.PORT,\r\n reload=False,\r\n log_level=\"info\"\r\n )\r\n break\r\n except KeyboardInterrupt:\r\n raise\r\n except SystemExit as e:\r\n if not _auto_restart or e.code == 0:\r\n break\r\n _attempt += 1\r\n print(f\"[WATCHDOG] uvicorn SystemExit({e.code}); restart #{_attempt} in 2s\")\r\n check_and_free_port(app_config.PORT)\r\n import time as _t; _t.sleep(2)\r\n continue\r\n except BaseException as e:\r\n _attempt += 1\r\n print(f\"[WATCHDOG] uvicorn crashed: {e!r}; restart #{_attempt} in 2s\")\r\n traceback.print_exc()\r\n check_and_free_port(app_config.PORT)\r\n import time as _t; _t.sleep(2)\r\n continue\r\n\n# 初始化技能热插拔系统\ntry:\n from services.skill_hotplug import init_hotplug_manager\n hotplug_mgr = init_hotplug_manager()\n hotplug_mgr.start_watching()\n print(\"[OK] 技能热插拔系统已初始化\")\nexcept Exception as e:\n print(f\"[WARN] 技能热插拔系统初始化失败: {e}\")\r\nregister_router(\"routes.skill_hotplug_routes\") # 技能热插拔管理\r\nregister_router(\"routes.chain_word_routes\") # 链词机制\r\n\n# 初始化链词提取器\ntry:\n from services.chain_word_extractor import get_chain_word_extractor\n extractor = get_chain_word_extractor()\n print(\"[OK] 链词提取器已初始化\")\nexcept Exception as e:\n print(f\"[WARN] 链词提取器初始化失败: {e}\")\r\nregister_router(\"routes.ppf_routes\") # PPF 自动化处理流程\r\n\n# 初始化 PPF 处理器\ntry:\n from services.ppf_processor import init_ppf_processor\n init_ppf_processor(db_manager=db_manager)\n print(\"[OK] PPF 处理器已初始化\")\nexcept Exception as e:\n print(f\"[WARN] PPF 处理器初始化失败: {e}\")\r\n"}}}} |