diff --git a/auto_code_fixer.py b/auto_code_fixer.py index 9e1d33e..2fe30af 100644 --- a/auto_code_fixer.py +++ b/auto_code_fixer.py @@ -9,7 +9,6 @@ import re import subprocess from pathlib import Path - class CodeIssue: """代码问题记录""" @@ -33,7 +32,6 @@ class CodeIssue: def __repr__(self) -> None: return f"{self.file_path}:{self.line_no} [{self.severity}] {self.issue_type}: {self.message}" - class CodeFixer: """代码自动修复器""" @@ -431,7 +429,6 @@ class CodeFixer: return "\n".join(report) - def git_commit_and_push(project_path: str) -> tuple[bool, str]: """Git 提交和推送""" try: @@ -470,7 +467,6 @@ def git_commit_and_push(project_path: str) -> tuple[bool, str]: except Exception as e: return False, f"Git 操作异常: {e}" - def main() -> None: project_path = "/root/.openclaw/workspace/projects/insightflow" @@ -514,6 +510,5 @@ def main() -> None: return report - if __name__ == "__main__": main() diff --git a/auto_fix_code.py b/auto_fix_code.py index 43097b3..fa3695f 100644 --- a/auto_fix_code.py +++ b/auto_fix_code.py @@ -6,7 +6,6 @@ Auto-fix script for InsightFlow code issues import re from pathlib import Path - def fix_file(filepath): """Fix common issues in a Python file""" with open(filepath, 'r', encoding='utf-8') as f: @@ -75,7 +74,6 @@ def fix_file(filepath): return True, changes return False, [] - def main(): backend_dir = Path('/root/.openclaw/workspace/projects/insightflow/backend') py_files = list(backend_dir.glob('*.py')) @@ -97,6 +95,5 @@ def main(): for c in all_changes[:20]: print(f" {c}") - if __name__ == '__main__': main() diff --git a/backend/__pycache__/ai_manager.cpython-312.pyc b/backend/__pycache__/ai_manager.cpython-312.pyc index 0e33da3..ada8cd5 100644 Binary files a/backend/__pycache__/ai_manager.cpython-312.pyc and b/backend/__pycache__/ai_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/api_key_manager.cpython-312.pyc b/backend/__pycache__/api_key_manager.cpython-312.pyc index 8da9471..6f314ee 100644 Binary files a/backend/__pycache__/api_key_manager.cpython-312.pyc and b/backend/__pycache__/api_key_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/collaboration_manager.cpython-312.pyc b/backend/__pycache__/collaboration_manager.cpython-312.pyc index 23551f1..508acca 100644 Binary files a/backend/__pycache__/collaboration_manager.cpython-312.pyc and b/backend/__pycache__/collaboration_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/db_manager.cpython-312.pyc b/backend/__pycache__/db_manager.cpython-312.pyc index 3b6828e..c089df8 100644 Binary files a/backend/__pycache__/db_manager.cpython-312.pyc and b/backend/__pycache__/db_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/developer_ecosystem_manager.cpython-312.pyc b/backend/__pycache__/developer_ecosystem_manager.cpython-312.pyc index 2a56c60..ec9c28b 100644 Binary files a/backend/__pycache__/developer_ecosystem_manager.cpython-312.pyc and b/backend/__pycache__/developer_ecosystem_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/document_processor.cpython-312.pyc b/backend/__pycache__/document_processor.cpython-312.pyc index c95ae78..7fd7b05 100644 Binary files a/backend/__pycache__/document_processor.cpython-312.pyc and b/backend/__pycache__/document_processor.cpython-312.pyc differ diff --git a/backend/__pycache__/enterprise_manager.cpython-312.pyc b/backend/__pycache__/enterprise_manager.cpython-312.pyc index b08c44d..f77349f 100644 Binary files a/backend/__pycache__/enterprise_manager.cpython-312.pyc and b/backend/__pycache__/enterprise_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/entity_aligner.cpython-312.pyc b/backend/__pycache__/entity_aligner.cpython-312.pyc index 7496bc3..472c084 100644 Binary files a/backend/__pycache__/entity_aligner.cpython-312.pyc and b/backend/__pycache__/entity_aligner.cpython-312.pyc differ diff --git a/backend/__pycache__/export_manager.cpython-312.pyc b/backend/__pycache__/export_manager.cpython-312.pyc index 2ab372a..a4d5e46 100644 Binary files a/backend/__pycache__/export_manager.cpython-312.pyc and b/backend/__pycache__/export_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/growth_manager.cpython-312.pyc b/backend/__pycache__/growth_manager.cpython-312.pyc index 9a408ba..a19d399 100644 Binary files a/backend/__pycache__/growth_manager.cpython-312.pyc and b/backend/__pycache__/growth_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/image_processor.cpython-312.pyc b/backend/__pycache__/image_processor.cpython-312.pyc index 325396c..5afba9b 100644 Binary files a/backend/__pycache__/image_processor.cpython-312.pyc and b/backend/__pycache__/image_processor.cpython-312.pyc differ diff --git a/backend/__pycache__/init_db.cpython-312.pyc b/backend/__pycache__/init_db.cpython-312.pyc index 54ce54f..99c0cb7 100644 Binary files a/backend/__pycache__/init_db.cpython-312.pyc and b/backend/__pycache__/init_db.cpython-312.pyc differ diff --git a/backend/__pycache__/knowledge_reasoner.cpython-312.pyc b/backend/__pycache__/knowledge_reasoner.cpython-312.pyc index 49659d3..6797e97 100644 Binary files a/backend/__pycache__/knowledge_reasoner.cpython-312.pyc and b/backend/__pycache__/knowledge_reasoner.cpython-312.pyc differ diff --git a/backend/__pycache__/llm_client.cpython-312.pyc b/backend/__pycache__/llm_client.cpython-312.pyc index 96ebdb1..8a37a69 100644 Binary files a/backend/__pycache__/llm_client.cpython-312.pyc and b/backend/__pycache__/llm_client.cpython-312.pyc differ diff --git a/backend/__pycache__/localization_manager.cpython-312.pyc b/backend/__pycache__/localization_manager.cpython-312.pyc index f832bd2..56dc93f 100644 Binary files a/backend/__pycache__/localization_manager.cpython-312.pyc and b/backend/__pycache__/localization_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/main.cpython-312.pyc b/backend/__pycache__/main.cpython-312.pyc index db04ade..c31be7d 100644 Binary files a/backend/__pycache__/main.cpython-312.pyc and b/backend/__pycache__/main.cpython-312.pyc differ diff --git a/backend/__pycache__/multimodal_entity_linker.cpython-312.pyc b/backend/__pycache__/multimodal_entity_linker.cpython-312.pyc index b0ebce0..688a940 100644 Binary files a/backend/__pycache__/multimodal_entity_linker.cpython-312.pyc and b/backend/__pycache__/multimodal_entity_linker.cpython-312.pyc differ diff --git a/backend/__pycache__/multimodal_processor.cpython-312.pyc b/backend/__pycache__/multimodal_processor.cpython-312.pyc index 556c2e9..6a706bd 100644 Binary files a/backend/__pycache__/multimodal_processor.cpython-312.pyc and b/backend/__pycache__/multimodal_processor.cpython-312.pyc differ diff --git a/backend/__pycache__/neo4j_manager.cpython-312.pyc b/backend/__pycache__/neo4j_manager.cpython-312.pyc index baed9bb..fdf459c 100644 Binary files a/backend/__pycache__/neo4j_manager.cpython-312.pyc and b/backend/__pycache__/neo4j_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/ops_manager.cpython-312.pyc b/backend/__pycache__/ops_manager.cpython-312.pyc index 0e08d9c..2660b89 100644 Binary files a/backend/__pycache__/ops_manager.cpython-312.pyc and b/backend/__pycache__/ops_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/oss_uploader.cpython-312.pyc b/backend/__pycache__/oss_uploader.cpython-312.pyc index e35b950..f990747 100644 Binary files a/backend/__pycache__/oss_uploader.cpython-312.pyc and b/backend/__pycache__/oss_uploader.cpython-312.pyc differ diff --git a/backend/__pycache__/performance_manager.cpython-312.pyc b/backend/__pycache__/performance_manager.cpython-312.pyc index 0545cb7..2d5de05 100644 Binary files a/backend/__pycache__/performance_manager.cpython-312.pyc and b/backend/__pycache__/performance_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/plugin_manager.cpython-312.pyc b/backend/__pycache__/plugin_manager.cpython-312.pyc index 2e42c6e..fe43dd6 100644 Binary files a/backend/__pycache__/plugin_manager.cpython-312.pyc and b/backend/__pycache__/plugin_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/rate_limiter.cpython-312.pyc b/backend/__pycache__/rate_limiter.cpython-312.pyc index 17535b5..33cab14 100644 Binary files a/backend/__pycache__/rate_limiter.cpython-312.pyc and b/backend/__pycache__/rate_limiter.cpython-312.pyc differ diff --git a/backend/__pycache__/search_manager.cpython-312.pyc b/backend/__pycache__/search_manager.cpython-312.pyc index a8f6ece..9f98e2f 100644 Binary files a/backend/__pycache__/search_manager.cpython-312.pyc and b/backend/__pycache__/search_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/security_manager.cpython-312.pyc b/backend/__pycache__/security_manager.cpython-312.pyc index db594a8..398e696 100644 Binary files a/backend/__pycache__/security_manager.cpython-312.pyc and b/backend/__pycache__/security_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/subscription_manager.cpython-312.pyc b/backend/__pycache__/subscription_manager.cpython-312.pyc index 48a8354..b402cb4 100644 Binary files a/backend/__pycache__/subscription_manager.cpython-312.pyc and b/backend/__pycache__/subscription_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/tenant_manager.cpython-312.pyc b/backend/__pycache__/tenant_manager.cpython-312.pyc index 3363b77..c77cf26 100644 Binary files a/backend/__pycache__/tenant_manager.cpython-312.pyc and b/backend/__pycache__/tenant_manager.cpython-312.pyc differ diff --git a/backend/__pycache__/test_multimodal.cpython-312.pyc b/backend/__pycache__/test_multimodal.cpython-312.pyc index a75526f..f4a5507 100644 Binary files a/backend/__pycache__/test_multimodal.cpython-312.pyc and b/backend/__pycache__/test_multimodal.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase7_task6_8.cpython-312.pyc b/backend/__pycache__/test_phase7_task6_8.cpython-312.pyc index f1a3764..ec99ff9 100644 Binary files a/backend/__pycache__/test_phase7_task6_8.cpython-312.pyc and b/backend/__pycache__/test_phase7_task6_8.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task1.cpython-312.pyc b/backend/__pycache__/test_phase8_task1.cpython-312.pyc index 40d7b75..5b56557 100644 Binary files a/backend/__pycache__/test_phase8_task1.cpython-312.pyc and b/backend/__pycache__/test_phase8_task1.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task2.cpython-312.pyc b/backend/__pycache__/test_phase8_task2.cpython-312.pyc index 0e823bc..f71cdb6 100644 Binary files a/backend/__pycache__/test_phase8_task2.cpython-312.pyc and b/backend/__pycache__/test_phase8_task2.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task4.cpython-312.pyc b/backend/__pycache__/test_phase8_task4.cpython-312.pyc index f4a2651..dcd9965 100644 Binary files a/backend/__pycache__/test_phase8_task4.cpython-312.pyc and b/backend/__pycache__/test_phase8_task4.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task5.cpython-312.pyc b/backend/__pycache__/test_phase8_task5.cpython-312.pyc index 7ef10ed..7582de9 100644 Binary files a/backend/__pycache__/test_phase8_task5.cpython-312.pyc and b/backend/__pycache__/test_phase8_task5.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task6.cpython-312.pyc b/backend/__pycache__/test_phase8_task6.cpython-312.pyc index 63c112b..cce35a6 100644 Binary files a/backend/__pycache__/test_phase8_task6.cpython-312.pyc and b/backend/__pycache__/test_phase8_task6.cpython-312.pyc differ diff --git a/backend/__pycache__/test_phase8_task8.cpython-312.pyc b/backend/__pycache__/test_phase8_task8.cpython-312.pyc index ba0f0cc..4e3ddf0 100644 Binary files a/backend/__pycache__/test_phase8_task8.cpython-312.pyc and b/backend/__pycache__/test_phase8_task8.cpython-312.pyc differ diff --git a/backend/__pycache__/tingwu_client.cpython-312.pyc b/backend/__pycache__/tingwu_client.cpython-312.pyc index dc92539..92ce6f2 100644 Binary files a/backend/__pycache__/tingwu_client.cpython-312.pyc and b/backend/__pycache__/tingwu_client.cpython-312.pyc differ diff --git a/backend/__pycache__/workflow_manager.cpython-312.pyc b/backend/__pycache__/workflow_manager.cpython-312.pyc index 7c08712..e85663d 100644 Binary files a/backend/__pycache__/workflow_manager.cpython-312.pyc and b/backend/__pycache__/workflow_manager.cpython-312.pyc differ diff --git a/backend/ai_manager.py b/backend/ai_manager.py index 7f34801..e207866 100644 --- a/backend/ai_manager.py +++ b/backend/ai_manager.py @@ -27,7 +27,6 @@ import httpx # Database path DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db") - class ModelType(StrEnum): """模型类型""" @@ -36,7 +35,6 @@ class ModelType(StrEnum): SUMMARIZATION = "summarization" # 摘要 PREDICTION = "prediction" # 预测 - class ModelStatus(StrEnum): """模型状态""" @@ -46,7 +44,6 @@ class ModelStatus(StrEnum): FAILED = "failed" ARCHIVED = "archived" - class MultimodalProvider(StrEnum): """多模态模型提供商""" @@ -55,7 +52,6 @@ class MultimodalProvider(StrEnum): GEMINI = "gemini-pro-vision" KIMI_VL = "kimi-vl" - class PredictionType(StrEnum): """预测类型""" @@ -64,7 +60,6 @@ class PredictionType(StrEnum): ENTITY_GROWTH = "entity_growth" # 实体增长预测 RELATION_EVOLUTION = "relation_evolution" # 关系演变预测 - @dataclass class CustomModel: """自定义模型""" @@ -84,7 +79,6 @@ class CustomModel: trained_at: str | None created_by: str - @dataclass class TrainingSample: """训练样本""" @@ -96,7 +90,6 @@ class TrainingSample: metadata: dict created_at: str - @dataclass class MultimodalAnalysis: """多模态分析结果""" @@ -113,7 +106,6 @@ class MultimodalAnalysis: cost: float created_at: str - @dataclass class KnowledgeGraphRAG: """基于知识图谱的 RAG 配置""" @@ -130,7 +122,6 @@ class KnowledgeGraphRAG: created_at: str updated_at: str - @dataclass class RAGQuery: """RAG 查询记录""" @@ -146,7 +137,6 @@ class RAGQuery: latency_ms: int created_at: str - @dataclass class PredictionModel: """预测模型""" @@ -166,7 +156,6 @@ class PredictionModel: created_at: str updated_at: str - @dataclass class PredictionResult: """预测结果""" @@ -182,7 +171,6 @@ class PredictionResult: is_correct: bool | None created_at: str - @dataclass class SmartSummary: """智能摘要""" @@ -200,7 +188,6 @@ class SmartSummary: tokens_used: int created_at: str - class AIManager: """AI 能力管理主类""" @@ -1526,11 +1513,9 @@ class AIManager: created_at=row["created_at"], ) - # Singleton instance _ai_manager = None - def get_ai_manager() -> AIManager: global _ai_manager if _ai_manager is None: diff --git a/backend/api_key_manager.py b/backend/api_key_manager.py index f4f073c..042b3e9 100644 --- a/backend/api_key_manager.py +++ b/backend/api_key_manager.py @@ -15,13 +15,11 @@ from enum import Enum DB_PATH = os.getenv("DB_PATH", "/app/data/insightflow.db") - class ApiKeyStatus(Enum): ACTIVE = "active" REVOKED = "revoked" EXPIRED = "expired" - @dataclass class ApiKey: id: str @@ -39,7 +37,6 @@ class ApiKey: revoked_reason: str | None total_calls: int = 0 - class ApiKeyManager: """API Key 管理器""" @@ -531,11 +528,9 @@ class ApiKeyManager: total_calls=row["total_calls"], ) - # 全局实例 _api_key_manager: ApiKeyManager | None = None - def get_api_key_manager() -> ApiKeyManager: """获取 API Key 管理器实例""" global _api_key_manager diff --git a/backend/collaboration_manager.py b/backend/collaboration_manager.py index 4e14d02..5d83edd 100644 --- a/backend/collaboration_manager.py +++ b/backend/collaboration_manager.py @@ -11,7 +11,6 @@ from datetime import datetime, timedelta from enum import Enum from typing import Any - class SharePermission(Enum): """分享权限级别""" @@ -20,7 +19,6 @@ class SharePermission(Enum): EDIT = "edit" # 可编辑 ADMIN = "admin" # 管理员 - class CommentTargetType(Enum): """评论目标类型""" @@ -29,7 +27,6 @@ class CommentTargetType(Enum): TRANSCRIPT = "transcript" # 转录文本评论 PROJECT = "project" # 项目级评论 - class ChangeType(Enum): """变更类型""" @@ -39,7 +36,6 @@ class ChangeType(Enum): MERGE = "merge" # 合并 SPLIT = "split" # 拆分 - @dataclass class ProjectShare: """项目分享链接""" @@ -58,7 +54,6 @@ class ProjectShare: allow_download: bool # 允许下载 allow_export: bool # 允许导出 - @dataclass class Comment: """评论/批注""" @@ -79,7 +74,6 @@ class Comment: mentions: list[str] # 提及的用户 attachments: list[dict] # 附件 - @dataclass class ChangeRecord: """变更记录""" @@ -101,7 +95,6 @@ class ChangeRecord: reverted_at: str | None # 回滚时间 reverted_by: str | None # 回滚者 - @dataclass class TeamMember: """团队成员""" @@ -117,7 +110,6 @@ class TeamMember: last_active_at: str | None # 最后活跃时间 permissions: list[str] # 具体权限列表 - @dataclass class TeamSpace: """团队空间""" @@ -132,7 +124,6 @@ class TeamSpace: project_count: int settings: dict[str, Any] # 团队设置 - class CollaborationManager: """协作管理主类""" @@ -986,11 +977,9 @@ class CollaborationManager: ) self.db.conn.commit() - # 全局协作管理器实例 _collaboration_manager = None - def get_collaboration_manager(db_manager=None) -> None: """获取协作管理器单例""" global _collaboration_manager diff --git a/backend/db_manager.py b/backend/db_manager.py index d7c16f5..b34e67c 100644 --- a/backend/db_manager.py +++ b/backend/db_manager.py @@ -17,7 +17,6 @@ DB_PATH = os.getenv("DB_PATH", "/app/data/insightflow.db") # Constants UUID_LENGTH = 8 # UUID 截断长度 - @dataclass class Project: id: str @@ -26,7 +25,6 @@ class Project: created_at: str = "" updated_at: str = "" - @dataclass class Entity: id: str @@ -47,7 +45,6 @@ class Entity: if self.attributes is None: self.attributes = {} - @dataclass class AttributeTemplate: """属性模板定义""" @@ -68,7 +65,6 @@ class AttributeTemplate: if self.options is None: self.options = [] - @dataclass class EntityAttribute: """实体属性值""" @@ -89,7 +85,6 @@ class EntityAttribute: if self.options is None: self.options = [] - @dataclass class AttributeHistory: """属性变更历史""" @@ -103,7 +98,6 @@ class AttributeHistory: changed_at: str = "" change_reason: str = "" - @dataclass class EntityMention: id: str @@ -114,7 +108,6 @@ class EntityMention: text_snippet: str confidence: float = 1.0 - class DatabaseManager: def __init__(self, db_path: str = DB_PATH) -> None: self.db_path = db_path @@ -1463,11 +1456,9 @@ class DatabaseManager: conn.close() return stats - # Singleton instance _db_manager = None - def get_db_manager() -> DatabaseManager: global _db_manager if _db_manager is None: diff --git a/backend/developer_ecosystem_manager.py b/backend/developer_ecosystem_manager.py index 2b170e6..4549a8e 100644 --- a/backend/developer_ecosystem_manager.py +++ b/backend/developer_ecosystem_manager.py @@ -21,7 +21,6 @@ from enum import StrEnum # Database path DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db") - class SDKLanguage(StrEnum): """SDK 语言类型""" @@ -32,7 +31,6 @@ class SDKLanguage(StrEnum): JAVA = "java" RUST = "rust" - class SDKStatus(StrEnum): """SDK 状态""" @@ -42,7 +40,6 @@ class SDKStatus(StrEnum): DEPRECATED = "deprecated" # 已弃用 ARCHIVED = "archived" # 已归档 - class TemplateCategory(StrEnum): """模板分类""" @@ -53,7 +50,6 @@ class TemplateCategory(StrEnum): TECH = "tech" # 科技 GENERAL = "general" # 通用 - class TemplateStatus(StrEnum): """模板状态""" @@ -63,7 +59,6 @@ class TemplateStatus(StrEnum): PUBLISHED = "published" # 已发布 UNLISTED = "unlisted" # 未列出 - class PluginStatus(StrEnum): """插件状态""" @@ -74,7 +69,6 @@ class PluginStatus(StrEnum): PUBLISHED = "published" # 已发布 SUSPENDED = "suspended" # 已暂停 - class PluginCategory(StrEnum): """插件分类""" @@ -85,7 +79,6 @@ class PluginCategory(StrEnum): SECURITY = "security" # 安全 CUSTOM = "custom" # 自定义 - class DeveloperStatus(StrEnum): """开发者认证状态""" @@ -95,7 +88,6 @@ class DeveloperStatus(StrEnum): CERTIFIED = "certified" # 已认证（高级） SUSPENDED = "suspended" # 已暂停 - @dataclass class SDKRelease: """SDK 发布""" @@ -121,7 +113,6 @@ class SDKRelease: published_at: str | None created_by: str - @dataclass class SDKVersion: """SDK 版本历史""" @@ -138,7 +129,6 @@ class SDKVersion: download_count: int created_at: str - @dataclass class TemplateMarketItem: """模板市场项目""" @@ -170,7 +160,6 @@ class TemplateMarketItem: updated_at: str published_at: str | None - @dataclass class TemplateReview: """模板评价""" @@ -186,7 +175,6 @@ class TemplateReview: created_at: str updated_at: str - @dataclass class PluginMarketItem: """插件市场项目""" @@ -225,7 +213,6 @@ class PluginMarketItem: reviewed_at: str | None review_notes: str | None - @dataclass class PluginReview: """插件评价""" @@ -241,7 +228,6 @@ class PluginReview: created_at: str updated_at: str - @dataclass class DeveloperProfile: """开发者档案""" @@ -265,7 +251,6 @@ class DeveloperProfile: updated_at: str verified_at: str | None - @dataclass class DeveloperRevenue: """开发者收益""" @@ -283,7 +268,6 @@ class DeveloperRevenue: transaction_id: str created_at: str - @dataclass class CodeExample: """代码示例""" @@ -306,7 +290,6 @@ class CodeExample: created_at: str updated_at: str - @dataclass class APIDocumentation: """API 文档生成记录""" @@ -320,7 +303,6 @@ class APIDocumentation: generated_at: str generated_by: str - @dataclass class DeveloperPortalConfig: """开发者门户配置""" @@ -344,7 +326,6 @@ class DeveloperPortalConfig: created_at: str updated_at: str - class DeveloperEcosystemManager: """开发者生态系统管理主类""" @@ -2075,11 +2056,9 @@ class DeveloperEcosystemManager: updated_at=row["updated_at"], ) - # Singleton instance _developer_ecosystem_manager = None - def get_developer_ecosystem_manager() -> DeveloperEcosystemManager: """获取开发者生态系统管理器单例""" global _developer_ecosystem_manager diff --git a/backend/document_processor.py b/backend/document_processor.py index 88c29a5..8725b51 100644 --- a/backend/document_processor.py +++ b/backend/document_processor.py @@ -7,7 +7,6 @@ Document Processor - Phase 3 import io import os - class DocumentProcessor: """文档处理器 - 提取 PDF/DOCX 文本""" @@ -157,10 +156,8 @@ class DocumentProcessor: ext = os.path.splitext(filename.lower())[1] return ext in self.supported_formats - # 简单的文本提取器（不需要外部依赖） - class SimpleTextExtractor: """简单的文本提取器，用于测试""" @@ -176,7 +173,6 @@ class SimpleTextExtractor: return content.decode("latin-1", errors="ignore") - if __name__ == "__main__": # 测试 processor = DocumentProcessor() diff --git a/backend/enterprise_manager.py b/backend/enterprise_manager.py index cafdd1e..2bde7ac 100644 --- a/backend/enterprise_manager.py +++ b/backend/enterprise_manager.py @@ -21,7 +21,6 @@ from typing import Any logger = logging.getLogger(__name__) - class SSOProvider(StrEnum): """SSO 提供商类型""" @@ -33,7 +32,6 @@ class SSOProvider(StrEnum): GOOGLE = "google" # Google Workspace CUSTOM_SAML = "custom_saml" # 自定义 SAML - class SSOStatus(StrEnum): """SSO 配置状态""" @@ -42,7 +40,6 @@ class SSOStatus(StrEnum): ACTIVE = "active" # 已启用 ERROR = "error" # 配置错误 - class SCIMSyncStatus(StrEnum): """SCIM 同步状态""" @@ -51,7 +48,6 @@ class SCIMSyncStatus(StrEnum): SUCCESS = "success" # 同步成功 FAILED = "failed" # 同步失败 - class AuditLogExportFormat(StrEnum): """审计日志导出格式""" @@ -60,7 +56,6 @@ class AuditLogExportFormat(StrEnum): PDF = "pdf" XLSX = "xlsx" - class DataRetentionAction(StrEnum): """数据保留策略动作""" @@ -68,7 +63,6 @@ class DataRetentionAction(StrEnum): DELETE = "delete" # 删除 ANONYMIZE = "anonymize" # 匿名化 - class ComplianceStandard(StrEnum): """合规标准""" @@ -78,7 +72,6 @@ class ComplianceStandard(StrEnum): HIPAA = "hipaa" PCI_DSS = "pci_dss" - @dataclass class SSOConfig: """SSO 配置数据类""" @@ -111,7 +104,6 @@ class SSOConfig: last_tested_at: datetime | None last_error: str | None - @dataclass class SCIMConfig: """SCIM 配置数据类""" @@ -136,7 +128,6 @@ class SCIMConfig: created_at: datetime updated_at: datetime - @dataclass class SCIMUser: """SCIM 用户数据类""" @@ -156,7 +147,6 @@ class SCIMUser: created_at: datetime updated_at: datetime - @dataclass class AuditLogExport: """审计日志导出记录""" @@ -181,7 +171,6 @@ class AuditLogExport: completed_at: datetime | None error_message: str | None - @dataclass class DataRetentionPolicy: """数据保留策略""" @@ -209,7 +198,6 @@ class DataRetentionPolicy: created_at: datetime updated_at: datetime - @dataclass class DataRetentionJob: """数据保留任务""" @@ -227,7 +215,6 @@ class DataRetentionJob: details: dict[str, Any] created_at: datetime - @dataclass class SAMLAuthRequest: """SAML 认证请求""" @@ -242,7 +229,6 @@ class SAMLAuthRequest: used: bool used_at: datetime | None - @dataclass class SAMLAuthResponse: """SAML 认证响应""" @@ -259,7 +245,6 @@ class SAMLAuthResponse: processed_at: datetime | None created_at: datetime - class EnterpriseManager: """企业级功能管理器""" @@ -2246,11 +2231,9 @@ class EnterpriseManager: ), ) - # 全局实例 _enterprise_manager = None - def get_enterprise_manager(db_path: str = "insightflow.db") -> EnterpriseManager: """获取 EnterpriseManager 单例""" global _enterprise_manager diff --git a/backend/entity_aligner.py b/backend/entity_aligner.py index 4247092..c645a35 100644 --- a/backend/entity_aligner.py +++ b/backend/entity_aligner.py @@ -15,7 +15,6 @@ import numpy as np KIMI_API_KEY = os.getenv("KIMI_API_KEY", "") KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding") - @dataclass class EntityEmbedding: entity_id: str @@ -23,7 +22,6 @@ class EntityEmbedding: definition: str embedding: list[float] - class EntityAligner: """实体对齐器 - 使用 embedding 进行相似度匹配""" @@ -327,10 +325,8 @@ class EntityAligner: return [] - # 简单的字符串相似度计算（不使用 embedding） - def simple_similarity(str1: str, str2: str) -> float: """ 计算两个字符串的简单相似度 @@ -361,7 +357,6 @@ def simple_similarity(str1: str, str2: str) -> float: return SequenceMatcher(None, s1, s2).ratio() - if __name__ == "__main__": # 测试 aligner = EntityAligner() diff --git a/backend/export_manager.py b/backend/export_manager.py index f57683e..5109a77 100644 --- a/backend/export_manager.py +++ b/backend/export_manager.py @@ -36,7 +36,6 @@ try: except ImportError: REPORTLAB_AVAILABLE = False - @dataclass class ExportEntity: id: str @@ -47,7 +46,6 @@ class ExportEntity: mention_count: int attributes: dict[str, Any] - @dataclass class ExportRelation: id: str @@ -57,7 +55,6 @@ class ExportRelation: confidence: float evidence: str - @dataclass class ExportTranscript: id: str @@ -67,7 +64,6 @@ class ExportTranscript: segments: list[dict] entity_mentions: list[dict] - class ExportManager: """导出管理器 - 处理各种导出需求""" @@ -633,11 +629,9 @@ class ExportManager: return json.dumps(data, ensure_ascii=False, indent=2) - # 全局导出管理器实例 _export_manager = None - def get_export_manager(db_manager=None) -> None: """获取导出管理器实例""" global _export_manager diff --git a/backend/growth_manager.py b/backend/growth_manager.py index c667b9b..7ef3800 100644 --- a/backend/growth_manager.py +++ b/backend/growth_manager.py @@ -28,7 +28,6 @@ import httpx # Database path DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db") - class EventType(StrEnum): """事件类型""" @@ -44,7 +43,6 @@ class EventType(StrEnum): INVITE_ACCEPTED = "invite_accepted" # 接受邀请 REFERRAL_REWARD = "referral_reward" # 推荐奖励 - class ExperimentStatus(StrEnum): """实验状态""" @@ -54,7 +52,6 @@ class ExperimentStatus(StrEnum): COMPLETED = "completed" # 已完成 ARCHIVED = "archived" # 已归档 - class TrafficAllocationType(StrEnum): """流量分配类型""" @@ -62,7 +59,6 @@ class TrafficAllocationType(StrEnum): STRATIFIED = "stratified" # 分层分配 TARGETED = "targeted" # 定向分配 - class EmailTemplateType(StrEnum): """邮件模板类型""" @@ -74,7 +70,6 @@ class EmailTemplateType(StrEnum): REFERRAL = "referral" # 推荐邀请 NEWSLETTER = "newsletter" # 新闻通讯 - class EmailStatus(StrEnum): """邮件状态""" @@ -88,7 +83,6 @@ class EmailStatus(StrEnum): BOUNCED = "bounced" # 退信 FAILED = "failed" # 失败 - class WorkflowTriggerType(StrEnum): """工作流触发类型""" @@ -100,7 +94,6 @@ class WorkflowTriggerType(StrEnum): MILESTONE = "milestone" # 里程碑 CUSTOM_EVENT = "custom_event" # 自定义事件 - class ReferralStatus(StrEnum): """推荐状态""" @@ -109,7 +102,6 @@ class ReferralStatus(StrEnum): REWARDED = "rewarded" # 已奖励 EXPIRED = "expired" # 已过期 - @dataclass class AnalyticsEvent: """分析事件""" @@ -128,7 +120,6 @@ class AnalyticsEvent: utm_medium: str | None utm_campaign: str | None - @dataclass class UserProfile: """用户画像""" @@ -148,7 +139,6 @@ class UserProfile: created_at: datetime updated_at: datetime - @dataclass class Funnel: """转化漏斗""" @@ -161,7 +151,6 @@ class Funnel: created_at: datetime updated_at: datetime - @dataclass class FunnelAnalysis: """漏斗分析结果""" @@ -174,7 +163,6 @@ class FunnelAnalysis: overall_conversion: float # 总体转化率 drop_off_points: list[dict] # 流失点 - @dataclass class Experiment: """A/B 测试实验""" @@ -199,7 +187,6 @@ class Experiment: updated_at: datetime created_by: str - @dataclass class ExperimentResult: """实验结果""" @@ -217,7 +204,6 @@ class ExperimentResult: uplift: float # 提升幅度 created_at: datetime - @dataclass class EmailTemplate: """邮件模板""" @@ -238,7 +224,6 @@ class EmailTemplate: created_at: datetime updated_at: datetime - @dataclass class EmailCampaign: """邮件营销活动""" @@ -260,7 +245,6 @@ class EmailCampaign: completed_at: datetime | None created_at: datetime - @dataclass class EmailLog: """邮件发送记录""" @@ -282,7 +266,6 @@ class EmailLog: error_message: str | None created_at: datetime - @dataclass class AutomationWorkflow: """自动化工作流""" @@ -299,7 +282,6 @@ class AutomationWorkflow: created_at: datetime updated_at: datetime - @dataclass class ReferralProgram: """推荐计划""" @@ -319,7 +301,6 @@ class ReferralProgram: created_at: datetime updated_at: datetime - @dataclass class Referral: """推荐记录""" @@ -340,7 +321,6 @@ class Referral: expires_at: datetime created_at: datetime - @dataclass class TeamIncentive: """团队升级激励""" @@ -358,7 +338,6 @@ class TeamIncentive: is_active: bool created_at: datetime - class GrowthManager: """运营与增长管理主类""" @@ -2211,11 +2190,9 @@ class GrowthManager: created_at=row["created_at"], ) - # Singleton instance _growth_manager = None - def get_growth_manager() -> GrowthManager: global _growth_manager if _growth_manager is None: diff --git a/backend/image_processor.py b/backend/image_processor.py index 31c4f93..e5b12b4 100644 --- a/backend/image_processor.py +++ b/backend/image_processor.py @@ -36,7 +36,6 @@ try: except ImportError: PYTESSERACT_AVAILABLE = False - @dataclass class ImageEntity: """图片中检测到的实体""" @@ -46,7 +45,6 @@ class ImageEntity: confidence: float bbox: tuple[int, int, int, int] | None = None # (x, y, width, height) - @dataclass class ImageRelation: """图片中检测到的关系""" @@ -56,7 +54,6 @@ class ImageRelation: relation_type: str confidence: float - @dataclass class ImageProcessingResult: """图片处理结果""" @@ -72,7 +69,6 @@ class ImageProcessingResult: success: bool error_message: str = "" - @dataclass class BatchProcessingResult: """批量图片处理结果""" @@ -82,7 +78,6 @@ class BatchProcessingResult: success_count: int failed_count: int - class ImageProcessor: """图片处理器 - 处理各种类型图片""" @@ -561,11 +556,9 @@ class ImageProcessor: print(f"Thumbnail generation error: {e}") return image_data - # Singleton instance _image_processor = None - def get_image_processor(temp_dir: str | None = None) -> ImageProcessor: """获取图片处理器单例""" global _image_processor diff --git a/backend/knowledge_reasoner.py b/backend/knowledge_reasoner.py index 2cdff42..c6bc2dd 100644 --- a/backend/knowledge_reasoner.py +++ b/backend/knowledge_reasoner.py @@ -15,7 +15,6 @@ import httpx KIMI_API_KEY = os.getenv("KIMI_API_KEY", "") KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding") - class ReasoningType(Enum): """推理类型""" @@ -25,7 +24,6 @@ class ReasoningType(Enum): COMPARATIVE = "comparative" # 对比推理 SUMMARY = "summary" # 总结推理 - @dataclass class ReasoningResult: """推理结果""" @@ -37,7 +35,6 @@ class ReasoningResult: related_entities: list[str] # 相关实体 gaps: list[str] # 知识缺口 - @dataclass class InferencePath: """推理路径""" @@ -47,7 +44,6 @@ class InferencePath: path: list[dict] # 路径上的节点和关系 strength: float # 路径强度 - class KnowledgeReasoner: """知识推理引擎""" @@ -525,11 +521,9 @@ class KnowledgeReasoner: "confidence": 0.5, } - # Singleton instance _reasoner = None - def get_knowledge_reasoner() -> KnowledgeReasoner: global _reasoner if _reasoner is None: diff --git a/backend/llm_client.py b/backend/llm_client.py index 6d6a6d5..068257d 100644 --- a/backend/llm_client.py +++ b/backend/llm_client.py @@ -15,13 +15,11 @@ import httpx KIMI_API_KEY = os.getenv("KIMI_API_KEY", "") KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding") - @dataclass class ChatMessage: role: str content: str - @dataclass class EntityExtractionResult: name: str @@ -29,7 +27,6 @@ class EntityExtractionResult: definition: str confidence: float - @dataclass class RelationExtractionResult: source: str @@ -37,7 +34,6 @@ class RelationExtractionResult: type: str confidence: float - class LLMClient: """Kimi API 客户端""" @@ -267,11 +263,9 @@ class LLMClient: messages = [ChatMessage(role="user", content=prompt)] return await self.chat(messages, temperature=0.3) - # Singleton instance _llm_client = None - def get_llm_client() -> LLMClient: global _llm_client if _llm_client is None: diff --git a/backend/localization_manager.py b/backend/localization_manager.py index f7c4dbe..8a29b73 100644 --- a/backend/localization_manager.py +++ b/backend/localization_manager.py @@ -35,7 +35,6 @@ except ImportError: logger = logging.getLogger(__name__) - class LanguageCode(StrEnum): """支持的语言代码""" @@ -52,7 +51,6 @@ class LanguageCode(StrEnum): AR = "ar" HI = "hi" - class RegionCode(StrEnum): """区域代码""" @@ -64,7 +62,6 @@ class RegionCode(StrEnum): LATIN_AMERICA = "latam" MIDDLE_EAST = "me" - class DataCenterRegion(StrEnum): """数据中心区域""" @@ -78,7 +75,6 @@ class DataCenterRegion(StrEnum): CN_NORTH = "cn-north" CN_EAST = "cn-east" - class PaymentProvider(StrEnum): """支付提供商""" @@ -95,7 +91,6 @@ class PaymentProvider(StrEnum): SEPA = "sepa" UNIONPAY = "unionpay" - class CalendarType(StrEnum): """日历类型""" @@ -107,7 +102,6 @@ class CalendarType(StrEnum): PERSIAN = "persian" BUDDHIST = "buddhist" - @dataclass class Translation: id: str @@ -122,7 +116,6 @@ class Translation: reviewed_by: str | None reviewed_at: datetime | None - @dataclass class LanguageConfig: code: str @@ -140,7 +133,6 @@ class LanguageConfig: first_day_of_week: int calendar_type: str - @dataclass class DataCenter: id: str @@ -155,7 +147,6 @@ class DataCenter: created_at: datetime updated_at: datetime - @dataclass class TenantDataCenterMapping: id: str @@ -167,7 +158,6 @@ class TenantDataCenterMapping: created_at: datetime updated_at: datetime - @dataclass class LocalizedPaymentMethod: id: str @@ -185,7 +175,6 @@ class LocalizedPaymentMethod: created_at: datetime updated_at: datetime - @dataclass class CountryConfig: code: str @@ -207,7 +196,6 @@ class CountryConfig: vat_rate: float | None is_active: bool - @dataclass class TimezoneConfig: id: str @@ -218,7 +206,6 @@ class TimezoneConfig: region: str is_active: bool - @dataclass class CurrencyConfig: code: str @@ -230,7 +217,6 @@ class CurrencyConfig: thousands_separator: str is_active: bool - @dataclass class LocalizationSettings: id: str @@ -250,7 +236,6 @@ class LocalizationSettings: created_at: datetime updated_at: datetime - class LocalizationManager: DEFAULT_LANGUAGES = { LanguageCode.EN: { @@ -1755,10 +1740,8 @@ class LocalizationManager: ), ) - _localization_manager = None - def get_localization_manager(db_path: str = "insightflow.db") -> LocalizationManager: global _localization_manager if _localization_manager is None: diff --git a/backend/main.py b/backend/main.py index 1d22ae2..fbd1562 100644 --- a/backend/main.py +++ b/backend/main.py @@ -34,8 +34,6 @@ from fastapi import ( from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import JSONResponse, PlainTextResponse, StreamingResponse from fastapi.staticfiles import StaticFiles -from ops_manager import OpsManager -from plugin_manager import PluginManager from pydantic import BaseModel, Field # Configure logger @@ -112,7 +110,6 @@ except ImportError: REASONER_AVAILABLE = False try: - from export_manager import get_export_manager EXPORT_AVAILABLE = True except ImportError: @@ -177,7 +174,6 @@ except ImportError as e: # Phase 7 Task 7: Plugin Manager try: - from plugin_manager import ( BotHandler, Plugin, PluginStatus, @@ -303,7 +299,6 @@ except ImportError as e: # Phase 8 Task 8: Operations & Monitoring Manager try: - from ops_manager import ( AlertChannelType, AlertRuleType, AlertSeverity, @@ -423,7 +418,6 @@ ADMIN_PATHS = { # Master Key（用于管理所有 API Keys） MASTER_KEY = os.getenv("INSIGHTFLOW_MASTER_KEY", "") - async def verify_api_key(request: Request, x_api_key: str | None = Header(None, alias="X-API-Key")): """ 验证 API Key 的依赖函数 @@ -479,7 +473,6 @@ async def verify_api_key(request: Request, x_api_key: str | None = Header(None, return {"type": "api_key", "key_id": api_key.id, "permissions": api_key.permissions} - async def rate_limit_middleware(request: Request, call_next): """ 限流中间件 @@ -566,7 +559,6 @@ async def rate_limit_middleware(request: Request, call_next): return response - # 添加限流中间件 app.middleware("http")(rate_limit_middleware) @@ -574,14 +566,12 @@ app.middleware("http")(rate_limit_middleware) # API Key 相关模型 - class ApiKeyCreate(BaseModel): name: str = Field(..., description="API Key 名称/描述") permissions: list[str] = Field(default=["read"], description="权限列表: read, write, delete") rate_limit: int = Field(default=60, description="每分钟请求限制") expires_days: int | None = Field(default=None, description="过期天数（可选）") - class ApiKeyResponse(BaseModel): id: str key_preview: str @@ -594,23 +584,19 @@ class ApiKeyResponse(BaseModel): last_used_at: str | None total_calls: int - class ApiKeyCreateResponse(BaseModel): api_key: str = Field(..., description="API Key（仅显示一次，请妥善保存）") info: ApiKeyResponse - class ApiKeyListResponse(BaseModel): keys: list[ApiKeyResponse] total: int - class ApiKeyUpdate(BaseModel): name: str | None = None permissions: list[str] | None = None rate_limit: int | None = None - class ApiCallStats(BaseModel): total_calls: int success_calls: int @@ -619,13 +605,11 @@ class ApiCallStats(BaseModel): max_response_time_ms: int min_response_time_ms: int - class ApiStatsResponse(BaseModel): summary: ApiCallStats endpoints: list[dict] daily: list[dict] - class ApiCallLog(BaseModel): id: int endpoint: str @@ -637,22 +621,18 @@ class ApiCallLog(BaseModel): error_message: str created_at: str - class ApiLogsResponse(BaseModel): logs: list[ApiCallLog] total: int - class RateLimitStatus(BaseModel): limit: int remaining: int reset_time: int window: str - # 原有模型（保留） - class EntityModel(BaseModel): id: str name: str @@ -660,14 +640,12 @@ class EntityModel(BaseModel): definition: str | None = "" aliases: list[str] = [] - class TranscriptSegment(BaseModel): start: float end: float text: str speaker: str | None = "Speaker A" - class AnalysisResult(BaseModel): transcript_id: str project_id: str @@ -676,52 +654,42 @@ class AnalysisResult(BaseModel): full_text: str created_at: str - class ProjectCreate(BaseModel): name: str description: str = "" - class EntityUpdate(BaseModel): name: str | None = None type: str | None = None definition: str | None = None aliases: list[str] | None = None - class RelationCreate(BaseModel): source_entity_id: str target_entity_id: str relation_type: str evidence: str | None = "" - class TranscriptUpdate(BaseModel): full_text: str - class AgentQuery(BaseModel): query: str stream: bool = False - class AgentCommand(BaseModel): command: str - class EntityMergeRequest(BaseModel): source_entity_id: str target_entity_id: str - class GlossaryTermCreate(BaseModel): term: str pronunciation: str | None = "" - # ==================== Phase 7: Workflow Pydantic Models ==================== - class WorkflowCreate(BaseModel): name: str = Field(..., description="工作流名称") description: str = Field(default="", description="工作流描述") @@ -735,7 +703,6 @@ class WorkflowCreate(BaseModel): config: dict = Field(default_factory=dict, description="工作流配置") webhook_ids: list[str] = Field(default_factory=list, description="关联的Webhook ID列表") - class WorkflowUpdate(BaseModel): name: str | None = None description: str | None = None @@ -746,7 +713,6 @@ class WorkflowUpdate(BaseModel): config: dict | None = None webhook_ids: list[str] | None = None - class WorkflowResponse(BaseModel): id: str name: str @@ -767,12 +733,10 @@ class WorkflowResponse(BaseModel): success_count: int fail_count: int - class WorkflowListResponse(BaseModel): workflows: list[WorkflowResponse] total: int - class WorkflowTaskCreate(BaseModel): name: str = Field(..., description="任务名称") task_type: str = Field( @@ -786,7 +750,6 @@ class WorkflowTaskCreate(BaseModel): retry_count: int = Field(default=3, description="重试次数") retry_delay: int = Field(default=5, description="重试延迟(秒)") - class WorkflowTaskUpdate(BaseModel): name: str | None = None task_type: str | None = None @@ -797,7 +760,6 @@ class WorkflowTaskUpdate(BaseModel): retry_count: int | None = None retry_delay: int | None = None - class WorkflowTaskResponse(BaseModel): id: str workflow_id: str @@ -812,7 +774,6 @@ class WorkflowTaskResponse(BaseModel): created_at: str updated_at: str - class WebhookCreate(BaseModel): name: str = Field(..., description="Webhook名称") webhook_type: str = Field(..., description="Webhook类型: feishu, dingtalk, slack, custom") @@ -821,7 +782,6 @@ class WebhookCreate(BaseModel): headers: dict = Field(default_factory=dict, description="自定义请求头") template: str = Field(default="", description="消息模板") - class WebhookUpdate(BaseModel): name: str | None = None webhook_type: str | None = None @@ -831,7 +791,6 @@ class WebhookUpdate(BaseModel): template: str | None = None is_active: bool | None = None - class WebhookResponse(BaseModel): id: str name: str @@ -846,12 +805,10 @@ class WebhookResponse(BaseModel): success_count: int fail_count: int - class WebhookListResponse(BaseModel): webhooks: list[WebhookResponse] total: int - class WorkflowLogResponse(BaseModel): id: str workflow_id: str @@ -865,16 +822,13 @@ class WorkflowLogResponse(BaseModel): error_message: str created_at: str - class WorkflowLogListResponse(BaseModel): logs: list[WorkflowLogResponse] total: int - class WorkflowTriggerRequest(BaseModel): input_data: dict = Field(default_factory=dict, description="工作流输入数据") - class WorkflowTriggerResponse(BaseModel): success: bool workflow_id: str @@ -882,7 +836,6 @@ class WorkflowTriggerResponse(BaseModel): results: dict duration_ms: int - class WorkflowStatsResponse(BaseModel): total: int success: int @@ -891,7 +844,6 @@ class WorkflowStatsResponse(BaseModel): avg_duration_ms: float daily: list[dict] - # API Keys KIMI_API_KEY = os.getenv("KIMI_API_KEY", "") KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding") @@ -899,34 +851,28 @@ KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding") # Phase 3: Entity Aligner singleton _aligner: "EntityAligner | None" = None - def get_aligner() -> "EntityAligner | None": global _aligner if _aligner is None and ALIGNER_AVAILABLE: _aligner = EntityAligner() return _aligner - # Phase 3: Document Processor singleton _doc_processor: "DocumentProcessor | None" = None - def get_doc_processor() -> "DocumentProcessor | None": global _doc_processor if _doc_processor is None and DOC_PROCESSOR_AVAILABLE: _doc_processor = DocumentProcessor() return _doc_processor - # Phase 7 Task 4: Collaboration Manager singleton _collaboration_manager: "CollaborationManager | None" = None - # Forward declaration for type hints class CollaborationManager: pass - def get_collab_manager() -> "CollaborationManager | None": global _collaboration_manager if _collaboration_manager is None and COLLABORATION_AVAILABLE: @@ -934,10 +880,8 @@ def get_collab_manager() -> "CollaborationManager | None": _collaboration_manager = get_collaboration_manager(db) return _collaboration_manager - # Phase 2: Entity Edit API - @app.put("/api/v1/entities/{entity_id}", tags=["Entities"]) async def update_entity(entity_id: str, update: EntityUpdate, _=Depends(verify_api_key)): """更新实体信息（名称、类型、定义、别名）""" @@ -961,7 +905,6 @@ async def update_entity(entity_id: str, update: EntityUpdate, _=Depends(verify_a "aliases": updated.aliases, } - @app.delete("/api/v1/entities/{entity_id}", tags=["Entities"]) async def delete_entity(entity_id: str, _=Depends(verify_api_key)): """删除实体""" @@ -976,7 +919,6 @@ async def delete_entity(entity_id: str, _=Depends(verify_api_key)): db.delete_entity(entity_id) return {"success": True, "message": f"Entity {entity_id} deleted"} - @app.post("/api/v1/entities/{entity_id}/merge", tags=["Entities"]) async def merge_entities_endpoint( entity_id: str, @@ -1008,10 +950,8 @@ async def merge_entities_endpoint( }, } - # Phase 2: Relation Edit API - @app.post("/api/v1/projects/{project_id}/relations", tags=["Relations"]) async def create_relation_endpoint( project_id: str, @@ -1047,7 +987,6 @@ async def create_relation_endpoint( "success": True, } - @app.delete("/api/v1/relations/{relation_id}", tags=["Relations"]) async def delete_relation(relation_id: str, _=Depends(verify_api_key)): """删除关系""" @@ -1058,7 +997,6 @@ async def delete_relation(relation_id: str, _=Depends(verify_api_key)): db.delete_relation(relation_id) return {"success": True, "message": f"Relation {relation_id} deleted"} - @app.put("/api/v1/relations/{relation_id}", tags=["Relations"]) async def update_relation(relation_id: str, relation: RelationCreate, _=Depends(verify_api_key)): """更新关系""" @@ -1079,10 +1017,8 @@ async def update_relation(relation_id: str, relation: RelationCreate, _=Depends( "success": True, } - # Phase 2: Transcript Edit API - @app.get("/api/v1/transcripts/{transcript_id}", tags=["Transcripts"]) async def get_transcript(transcript_id: str, _=Depends(verify_api_key)): """获取转录详情""" @@ -1097,7 +1033,6 @@ async def get_transcript(transcript_id: str, _=Depends(verify_api_key)): return transcript - @app.put("/api/v1/transcripts/{transcript_id}", tags=["Transcripts"]) async def update_transcript( transcript_id: str, @@ -1122,10 +1057,8 @@ async def update_transcript( "success": True, } - # Phase 2: Manual Entity Creation - class ManualEntityCreate(BaseModel): name: str type: str = "OTHER" @@ -1134,7 +1067,6 @@ class ManualEntityCreate(BaseModel): start_pos: int | None = None end_pos: int | None = None - @app.post("/api/v1/projects/{project_id}/entities", tags=["Entities"]) async def create_manual_entity( project_id: str, @@ -1189,7 +1121,6 @@ async def create_manual_entity( "success": True, } - def transcribe_audio(audio_data: bytes, filename: str) -> dict: """转录音频：OSS上传 + 听悟转录""" @@ -1220,7 +1151,6 @@ def transcribe_audio(audio_data: bytes, filename: str) -> dict: logger.warning(f"Tingwu failed: {e}") return mock_transcribe() - def mock_transcribe() -> dict: """Mock 转录结果""" return { @@ -1235,7 +1165,6 @@ def mock_transcribe() -> dict: ], } - def extract_entities_with_llm(text: str) -> tuple[list[dict], list[dict]]: """使用 Kimi API 提取实体和关系 @@ -1292,7 +1221,6 @@ def extract_entities_with_llm(text: str) -> tuple[list[dict], list[dict]]: return [], [] - def align_entity(project_id: str, name: str, db, definition: str = "") -> Optional["Entity"]: """实体对齐 - Phase 3: 使用 embedding 对齐""" # 1. 首先尝试精确匹配 @@ -1314,10 +1242,8 @@ def align_entity(project_id: str, name: str, db, definition: str = "") -> Option return None - # API Endpoints - @app.post("/api/v1/projects", response_model=dict, tags=["Projects"]) async def create_project(project: ProjectCreate, _=Depends(verify_api_key)): """创建新项目""" @@ -1329,7 +1255,6 @@ async def create_project(project: ProjectCreate, _=Depends(verify_api_key)): p = db.create_project(project_id, project.name, project.description) return {"id": p.id, "name": p.name, "description": p.description} - @app.get("/api/v1/projects", tags=["Projects"]) async def list_projects(_=Depends(verify_api_key)): """列出所有项目""" @@ -1340,7 +1265,6 @@ async def list_projects(_=Depends(verify_api_key)): projects = db.list_projects() return [{"id": p.id, "name": p.name, "description": p.description} for p in projects] - @app.post("/api/v1/projects/{project_id}/upload", response_model=AnalysisResult, tags=["Projects"]) async def upload_audio(project_id: str, file: UploadFile = File(...), _=Depends(verify_api_key)): """上传音频到指定项目 - Phase 3: 支持多文件融合""" @@ -1455,10 +1379,8 @@ async def upload_audio(project_id: str, file: UploadFile = File(...), _=Depends( created_at=datetime.now().isoformat(), ) - # Phase 3: Document Upload API - @app.post("/api/v1/projects/{project_id}/upload-document") async def upload_document(project_id: str, file: UploadFile = File(...), _=Depends(verify_api_key)): """上传 PDF/DOCX 文档到指定项目""" @@ -1578,10 +1500,8 @@ async def upload_document(project_id: str, file: UploadFile = File(...), _=Depen "created_at": datetime.now().isoformat(), } - # Phase 3: Knowledge Base API - @app.get("/api/v1/projects/{project_id}/knowledge-base") async def get_knowledge_base(project_id: str, _=Depends(verify_api_key)): """获取项目知识库 - 包含所有实体、关系、术语表""" @@ -1674,10 +1594,8 @@ async def get_knowledge_base(project_id: str, _=Depends(verify_api_key)): ], } - # Phase 3: Glossary API - @app.post("/api/v1/projects/{project_id}/glossary") async def add_glossary_term(project_id: str, term: GlossaryTermCreate, _=Depends(verify_api_key)): """添加术语到项目术语表""" @@ -1697,7 +1615,6 @@ async def add_glossary_term(project_id: str, term: GlossaryTermCreate, _=Depends return {"id": term_id, "term": term.term, "pronunciation": term.pronunciation, "success": True} - @app.get("/api/v1/projects/{project_id}/glossary") async def get_glossary(project_id: str, _=Depends(verify_api_key)): """获取项目术语表""" @@ -1708,7 +1625,6 @@ async def get_glossary(project_id: str, _=Depends(verify_api_key)): glossary = db.list_glossary(project_id) return glossary - @app.delete("/api/v1/glossary/{term_id}") async def delete_glossary_term(term_id: str, _=Depends(verify_api_key)): """删除术语""" @@ -1719,10 +1635,8 @@ async def delete_glossary_term(term_id: str, _=Depends(verify_api_key)): db.delete_glossary_term(term_id) return {"success": True} - # Phase 3: Entity Alignment API - @app.post("/api/v1/projects/{project_id}/align-entities") async def align_project_entities( project_id: str, @@ -1766,7 +1680,6 @@ async def align_project_entities( return {"success": True, "merged_count": merged_count, "merged_pairs": merged_pairs} - @app.get("/api/v1/projects/{project_id}/entities") async def get_project_entities(project_id: str, _=Depends(verify_api_key)): """获取项目的全局实体列表""" @@ -1786,7 +1699,6 @@ async def get_project_entities(project_id: str, _=Depends(verify_api_key)): for e in entities ] - @app.get("/api/v1/projects/{project_id}/relations") async def get_project_relations(project_id: str, _=Depends(verify_api_key)): """获取项目的实体关系列表""" @@ -1813,7 +1725,6 @@ async def get_project_relations(project_id: str, _=Depends(verify_api_key)): for r in relations ] - @app.get("/api/v1/projects/{project_id}/transcripts") async def get_project_transcripts(project_id: str, _=Depends(verify_api_key)): """获取项目的转录列表""" @@ -1835,7 +1746,6 @@ async def get_project_transcripts(project_id: str, _=Depends(verify_api_key)): for t in transcripts ] - @app.get("/api/v1/entities/{entity_id}/mentions") async def get_entity_mentions(entity_id: str, _=Depends(verify_api_key)): """获取实体的所有提及位置""" @@ -1856,10 +1766,8 @@ async def get_entity_mentions(entity_id: str, _=Depends(verify_api_key)): for m in mentions ] - # Health check - Legacy endpoint (deprecated, use /api/v1/health) - @app.get("/health") async def legacy_health_check(): return { @@ -1879,10 +1787,8 @@ async def legacy_health_check(): "plugin_manager_available": PLUGIN_MANAGER_AVAILABLE, } - # ==================== Phase 4: Agent 助手 API ==================== - @app.post("/api/v1/projects/{project_id}/agent/query") async def agent_query(project_id: str, query: AgentQuery, _=Depends(verify_api_key)): """Agent RAG 问答""" @@ -1940,7 +1846,6 @@ async def agent_query(project_id: str, query: AgentQuery, _=Depends(verify_api_k answer = await llm.rag_query(query.query, context, project_context) return {"answer": answer, "project_id": project_id} - @app.post("/api/v1/projects/{project_id}/agent/command") async def agent_command(project_id: str, command: AgentCommand, _=Depends(verify_api_key)): """Agent 指令执行 - 解析并执行自然语言指令""" @@ -2033,7 +1938,6 @@ async def agent_command(project_id: str, command: AgentCommand, _=Depends(verify return result - @app.get("/api/v1/projects/{project_id}/agent/suggest") async def agent_suggest(project_id: str, _=Depends(verify_api_key)): """获取 Agent 建议 - 基于项目数据提供洞察""" @@ -2071,10 +1975,8 @@ async def agent_suggest(project_id: str, _=Depends(verify_api_key)): return {"suggestions": []} - # ==================== Phase 4: 知识溯源 API ==================== - @app.get("/api/v1/relations/{relation_id}/provenance") async def get_relation_provenance(relation_id: str, _=Depends(verify_api_key)): """获取关系的知识溯源信息""" @@ -2103,7 +2005,6 @@ async def get_relation_provenance(relation_id: str, _=Depends(verify_api_key)): ), } - @app.get("/api/v1/entities/{entity_id}/details") async def get_entity_details(entity_id: str, _=Depends(verify_api_key)): """获取实体详情，包含所有提及位置""" @@ -2118,7 +2019,6 @@ async def get_entity_details(entity_id: str, _=Depends(verify_api_key)): return entity - @app.get("/api/v1/entities/{entity_id}/evolution") async def get_entity_evolution(entity_id: str, _=Depends(verify_api_key)): """分析实体的演变和态度变化""" @@ -2151,10 +2051,8 @@ async def get_entity_evolution(entity_id: str, _=Depends(verify_api_key)): ], } - # ==================== Phase 4: 实体管理增强 API ==================== - @app.get("/api/v1/projects/{project_id}/entities/search") async def search_entities(project_id: str, q: str, _=Depends(verify_api_key)): """搜索实体""" @@ -2167,10 +2065,8 @@ async def search_entities(project_id: str, q: str, _=Depends(verify_api_key)): {"id": e.id, "name": e.name, "type": e.type, "definition": e.definition} for e in entities ] - # ==================== Phase 5: 时间线视图 API ==================== - @app.get("/api/v1/projects/{project_id}/timeline") async def get_project_timeline( project_id: str, @@ -2192,7 +2088,6 @@ async def get_project_timeline( return {"project_id": project_id, "events": timeline, "total_count": len(timeline)} - @app.get("/api/v1/projects/{project_id}/timeline/summary") async def get_timeline_summary(project_id: str, _=Depends(verify_api_key)): """获取项目时间线摘要统计""" @@ -2208,7 +2103,6 @@ async def get_timeline_summary(project_id: str, _=Depends(verify_api_key)): return {"project_id": project_id, "project_name": project.name, **summary} - @app.get("/api/v1/entities/{entity_id}/timeline") async def get_entity_timeline(entity_id: str, _=Depends(verify_api_key)): """获取单个实体的时间线""" @@ -2230,16 +2124,13 @@ async def get_entity_timeline(entity_id: str, _=Depends(verify_api_key)): "total_count": len(timeline), } - # ==================== Phase 5: 知识推理与问答增强 API ==================== - class ReasoningQuery(BaseModel): query: str reasoning_depth: str = "medium" # shallow/medium/deep stream: bool = False - @app.post("/api/v1/projects/{project_id}/reasoning/query") async def reasoning_query(project_id: str, query: ReasoningQuery, _=Depends(verify_api_key)): """ @@ -2293,7 +2184,6 @@ async def reasoning_query(project_id: str, query: ReasoningQuery, _=Depends(veri "project_id": project_id, } - @app.post("/api/v1/projects/{project_id}/reasoning/inference-path") async def find_inference_path( project_id: str, @@ -2342,11 +2232,9 @@ async def find_inference_path( "total_paths": len(paths), } - class SummaryRequest(BaseModel): summary_type: str = "comprehensive" # comprehensive/executive/technical/risk - @app.post("/api/v1/projects/{project_id}/reasoning/summary") async def project_summary(project_id: str, req: SummaryRequest, _=Depends(verify_api_key)): """ @@ -2389,10 +2277,8 @@ async def project_summary(project_id: str, req: SummaryRequest, _=Depends(verify return {"project_id": project_id, "summary_type": req.summary_type, **summary**summary} - # ==================== Phase 5: 实体属性扩展 API ==================== - class AttributeTemplateCreate(BaseModel): name: str type: str # text, number, date, select, multiselect, boolean @@ -2402,7 +2288,6 @@ class AttributeTemplateCreate(BaseModel): is_required: bool = False sort_order: int = 0 - class AttributeTemplateUpdate(BaseModel): name: str | None = None type: str | None = None @@ -2412,7 +2297,6 @@ class AttributeTemplateUpdate(BaseModel): is_required: bool | None = None sort_order: int | None = None - class EntityAttributeSet(BaseModel): name: str type: str @@ -2421,15 +2305,12 @@ class EntityAttributeSet(BaseModel): options: list[str] | None = None change_reason: str | None = "" - class EntityAttributeBatchSet(BaseModel): attributes: list[EntityAttributeSet] change_reason: str | None = "" - # 属性模板管理 API - @app.post("/api/v1/projects/{project_id}/attribute-templates") async def create_attribute_template_endpoint( project_id: str, @@ -2466,7 +2347,6 @@ async def create_attribute_template_endpoint( "success": True, } - @app.get("/api/v1/projects/{project_id}/attribute-templates") async def list_attribute_templates_endpoint(project_id: str, _=Depends(verify_api_key)): """列出项目的所有属性模板""" @@ -2490,7 +2370,6 @@ async def list_attribute_templates_endpoint(project_id: str, _=Depends(verify_ap for t in templates ] - @app.get("/api/v1/attribute-templates/{template_id}") async def get_attribute_template_endpoint(template_id: str, _=Depends(verify_api_key)): """获取属性模板详情""" @@ -2514,7 +2393,6 @@ async def get_attribute_template_endpoint(template_id: str, _=Depends(verify_api "sort_order": template.sort_order, } - @app.put("/api/v1/attribute-templates/{template_id}") async def update_attribute_template_endpoint( template_id: str, @@ -2535,7 +2413,6 @@ async def update_attribute_template_endpoint( return {"id": updated.id, "name": updated.name, "type": updated.type, "success": True} - @app.delete("/api/v1/attribute-templates/{template_id}") async def delete_attribute_template_endpoint(template_id: str, _=Depends(verify_api_key)): """删除属性模板""" @@ -2547,10 +2424,8 @@ async def delete_attribute_template_endpoint(template_id: str, _=Depends(verify_ return {"success": True, "message": f"Template {template_id} deleted"} - # 实体属性值管理 API - @app.post("/api/v1/entities/{entity_id}/attributes") async def set_entity_attribute_endpoint( entity_id: str, @@ -2659,7 +2534,6 @@ async def set_entity_attribute_endpoint( "success": True, } - @app.post("/api/v1/entities/{entity_id}/attributes/batch") async def batch_set_entity_attributes_endpoint( entity_id: str, @@ -2705,7 +2579,6 @@ async def batch_set_entity_attributes_endpoint( "success": True, } - @app.get("/api/v1/entities/{entity_id}/attributes") async def get_entity_attributes_endpoint(entity_id: str, _=Depends(verify_api_key)): """获取实体的所有属性值""" @@ -2730,7 +2603,6 @@ async def get_entity_attributes_endpoint(entity_id: str, _=Depends(verify_api_ke for a in attrs ] - @app.delete("/api/v1/entities/{entity_id}/attributes/{template_id}") async def delete_entity_attribute_endpoint( entity_id: str, @@ -2747,10 +2619,8 @@ async def delete_entity_attribute_endpoint( return {"success": True, "message": "Attribute deleted"} - # 属性历史 API - @app.get("/api/v1/entities/{entity_id}/attributes/history") async def get_entity_attribute_history_endpoint( entity_id: str, @@ -2777,7 +2647,6 @@ async def get_entity_attribute_history_endpoint( for h in history ] - @app.get("/api/v1/attribute-templates/{template_id}/history") async def get_template_history_endpoint( template_id: str, @@ -2805,10 +2674,8 @@ async def get_template_history_endpoint( for h in history ] - # 属性筛选搜索 API - @app.get("/api/v1/projects/{project_id}/entities/search-by-attributes") async def search_entities_by_attributes_endpoint( project_id: str, @@ -2844,10 +2711,8 @@ async def search_entities_by_attributes_endpoint( for e in entities ] - # ==================== 导出功能 API ==================== - @app.get("/api/v1/projects/{project_id}/export/graph-svg") async def export_graph_svg_endpoint(project_id: str, _=Depends(verify_api_key)): """导出知识图谱为 SVG""" @@ -2901,7 +2766,6 @@ async def export_graph_svg_endpoint(project_id: str, _=Depends(verify_api_key)): headers={"Content-Disposition": f"attachment; filename=insightflow-graph-{project_id}.svg"}, ) - @app.get("/api/v1/projects/{project_id}/export/graph-png") async def export_graph_png_endpoint(project_id: str, _=Depends(verify_api_key)): """导出知识图谱为 PNG""" @@ -2955,7 +2819,6 @@ async def export_graph_png_endpoint(project_id: str, _=Depends(verify_api_key)): headers={"Content-Disposition": f"attachment; filename=insightflow-graph-{project_id}.png"}, ) - @app.get("/api/v1/projects/{project_id}/export/entities-excel") async def export_entities_excel_endpoint(project_id: str, _=Depends(verify_api_key)): """导出实体数据为 Excel""" @@ -2996,7 +2859,6 @@ async def export_entities_excel_endpoint(project_id: str, _=Depends(verify_api_k }, ) - @app.get("/api/v1/projects/{project_id}/export/entities-csv") async def export_entities_csv_endpoint(project_id: str, _=Depends(verify_api_key)): """导出实体数据为 CSV""" @@ -3037,7 +2899,6 @@ async def export_entities_csv_endpoint(project_id: str, _=Depends(verify_api_key }, ) - @app.get("/api/v1/projects/{project_id}/export/relations-csv") async def export_relations_csv_endpoint(project_id: str, _=Depends(verify_api_key)): """导出关系数据为 CSV""" @@ -3076,7 +2937,6 @@ async def export_relations_csv_endpoint(project_id: str, _=Depends(verify_api_ke }, ) - @app.get("/api/v1/projects/{project_id}/export/report-pdf") async def export_report_pdf_endpoint(project_id: str, _=Depends(verify_api_key)): """导出项目报告为 PDF""" @@ -3164,7 +3024,6 @@ async def export_report_pdf_endpoint(project_id: str, _=Depends(verify_api_key)) }, ) - @app.get("/api/v1/projects/{project_id}/export/project-json") async def export_project_json_endpoint(project_id: str, _=Depends(verify_api_key)): """导出完整项目数据为 JSON""" @@ -3241,7 +3100,6 @@ async def export_project_json_endpoint(project_id: str, _=Depends(verify_api_key }, ) - @app.get("/api/v1/transcripts/{transcript_id}/export/markdown") async def export_transcript_markdown_endpoint(transcript_id: str, _=Depends(verify_api_key)): """导出转录文本为 Markdown""" @@ -3303,25 +3161,20 @@ async def export_transcript_markdown_endpoint(transcript_id: str, _=Depends(veri }, ) - # ==================== Neo4j Graph Database API ==================== - class Neo4jSyncRequest(BaseModel): project_id: str - class PathQueryRequest(BaseModel): source_entity_id: str target_entity_id: str max_depth: int = 10 - class GraphQueryRequest(BaseModel): entity_ids: list[str] depth: int = 1 - @app.get("/api/v1/neo4j/status") async def neo4j_status(_=Depends(verify_api_key)): """获取 Neo4j 连接状态""" @@ -3340,7 +3193,6 @@ async def neo4j_status(_=Depends(verify_api_key)): except (RuntimeError, ValueError, TypeError, ConnectionError) as e: return {"available": True, "connected": False, "message": str(e)} - @app.post("/api/v1/neo4j/sync") async def neo4j_sync_project(request: Neo4jSyncRequest, _=Depends(verify_api_key)): """同步项目数据到 Neo4j""" @@ -3407,7 +3259,6 @@ async def neo4j_sync_project(request: Neo4jSyncRequest, _=Depends(verify_api_key ), } - @app.get("/api/v1/projects/{project_id}/graph/stats") async def get_graph_stats(project_id: str, _=Depends(verify_api_key)): """获取项目图统计信息""" @@ -3421,7 +3272,6 @@ async def get_graph_stats(project_id: str, _=Depends(verify_api_key)): stats = manager.get_graph_stats(project_id) return stats - @app.post("/api/v1/graph/shortest-path") async def find_shortest_path(request: PathQueryRequest, _=Depends(verify_api_key)): """查找两个实体之间的最短路径""" @@ -3446,7 +3296,6 @@ async def find_shortest_path(request: PathQueryRequest, _=Depends(verify_api_key "path": {"nodes": path.nodes, "relationships": path.relationships, "length": path.length}, } - @app.post("/api/v1/graph/paths") async def find_all_paths(request: PathQueryRequest, _=Depends(verify_api_key)): """查找两个实体之间的所有路径""" @@ -3470,7 +3319,6 @@ async def find_all_paths(request: PathQueryRequest, _=Depends(verify_api_key)): ], } - @app.get("/api/v1/entities/{entity_id}/neighbors") async def get_entity_neighbors( entity_id: str, @@ -3489,7 +3337,6 @@ async def get_entity_neighbors( neighbors = manager.find_neighbors(entity_id, relation_type, limit) return {"entity_id": entity_id, "count": len(neighbors), "neighbors": neighbors} - @app.get("/api/v1/entities/{entity_id1}/common-neighbors/{entity_id2}") async def get_common_neighbors(entity_id1: str, entity_id2: str, _=Depends(verify_api_key)): """获取两个实体的共同邻居""" @@ -3508,7 +3355,6 @@ async def get_common_neighbors(entity_id1: str, entity_id2: str, _=Depends(verif "common_neighbors": common, } - @app.get("/api/v1/projects/{project_id}/graph/centrality") async def get_centrality_analysis( project_id: str, @@ -3538,7 +3384,6 @@ async def get_centrality_analysis( ], } - @app.get("/api/v1/projects/{project_id}/graph/communities") async def get_communities(project_id: str, _=Depends(verify_api_key)): """获取社区发现结果""" @@ -3558,7 +3403,6 @@ async def get_communities(project_id: str, _=Depends(verify_api_key)): ], } - @app.post("/api/v1/graph/subgraph") async def get_subgraph(request: GraphQueryRequest, _=Depends(verify_api_key)): """获取子图""" @@ -3572,10 +3416,8 @@ async def get_subgraph(request: GraphQueryRequest, _=Depends(verify_api_key)): subgraph = manager.get_subgraph(request.entity_ids, request.depth) return subgraph - # ==================== Phase 6: API Key Management Endpoints ==================== - @app.post("/api/v1/api-keys", response_model=ApiKeyCreateResponse, tags=["API Keys"]) async def create_api_key(request: ApiKeyCreate, _=Depends(verify_api_key)): """ @@ -3613,7 +3455,6 @@ async def create_api_key(request: ApiKeyCreate, _=Depends(verify_api_key)): ), ) - @app.get("/api/v1/api-keys", response_model=ApiKeyListResponse, tags=["API Keys"]) async def list_api_keys( status: str | None = None, @@ -3653,7 +3494,6 @@ async def list_api_keys( total=len(keys), ) - @app.get("/api/v1/api-keys/{key_id}", response_model=ApiKeyResponse, tags=["API Keys"]) async def get_api_key(key_id: str, _=Depends(verify_api_key)): """获取单个 API Key 详情""" @@ -3679,7 +3519,6 @@ async def get_api_key(key_id: str, _=Depends(verify_api_key)): total_calls=key.total_calls, ) - @app.patch("/api/v1/api-keys/{key_id}", response_model=ApiKeyResponse, tags=["API Keys"]) async def update_api_key(key_id: str, request: ApiKeyUpdate, _=Depends(verify_api_key)): """ @@ -3724,7 +3563,6 @@ async def update_api_key(key_id: str, request: ApiKeyUpdate, _=Depends(verify_ap total_calls=key.total_calls, ) - @app.delete("/api/v1/api-keys/{key_id}", tags=["API Keys"]) async def revoke_api_key(key_id: str, reason: str = "", _=Depends(verify_api_key)): """ @@ -3743,7 +3581,6 @@ async def revoke_api_key(key_id: str, reason: str = "", _=Depends(verify_api_key return {"success": True, "message": f"API Key {key_id} revoked"} - @app.get("/api/v1/api-keys/{key_id}/stats", response_model=ApiStatsResponse, tags=["API Keys"]) async def get_api_key_stats(key_id: str, days: int = 30, _=Depends(verify_api_key)): """ @@ -3769,7 +3606,6 @@ async def get_api_key_stats(key_id: str, days: int = 30, _=Depends(verify_api_ke daily=stats["daily"], ) - @app.get("/api/v1/api-keys/{key_id}/logs", response_model=ApiLogsResponse, tags=["API Keys"]) async def get_api_key_logs( key_id: str, @@ -3813,7 +3649,6 @@ async def get_api_key_logs( total=len(logs), ) - @app.get("/api/v1/rate-limit/status", response_model=RateLimitStatus, tags=["API Keys"]) async def get_rate_limit_status(request: Request, _=Depends(verify_api_key)): """获取当前请求的限流状态""" @@ -3846,16 +3681,13 @@ async def get_rate_limit_status(request: Request, _=Depends(verify_api_key)): window="minute", ) - # ==================== Phase 6: System Endpoints ==================== - @app.get("/api/v1/health", tags=["System"]) async def api_health_check(): """健康检查端点""" return {"status": "healthy", "version": "0.7.0", "timestamp": datetime.now().isoformat()} - @app.get("/api/v1/status", tags=["System"]) async def system_status(): """系统状态信息""" @@ -3885,24 +3717,20 @@ async def system_status(): return status - # ==================== Phase 7: Workflow Automation Endpoints ==================== # Workflow Manager singleton _workflow_manager: Any | None = None - def get_workflow_manager_instance() -> Any: global _workflow_manager if _workflow_manager is None and WORKFLOW_AVAILABLE and DB_AVAILABLE: - from workflow_manager import WorkflowManager db = get_db_manager() _workflow_manager = WorkflowManager(db) _workflow_manager.start() return _workflow_manager - @app.post("/api/v1/workflows", response_model=WorkflowResponse, tags=["Workflows"]) async def create_workflow_endpoint(request: WorkflowCreate, _=Depends(verify_api_key)): """ @@ -3967,7 +3795,6 @@ async def create_workflow_endpoint(request: WorkflowCreate, _=Depends(verify_api except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/workflows", response_model=WorkflowListResponse, tags=["Workflows"]) async def list_workflows_endpoint( project_id: str | None = None, @@ -4009,7 +3836,6 @@ async def list_workflows_endpoint( total=len(workflows), ) - @app.get("/api/v1/workflows/{workflow_id}", response_model=WorkflowResponse, tags=["Workflows"]) async def get_workflow_endpoint(workflow_id: str, _=Depends(verify_api_key)): """获取单个工作流详情""" @@ -4043,7 +3869,6 @@ async def get_workflow_endpoint(workflow_id: str, _=Depends(verify_api_key)): fail_count=workflow.fail_count, ) - @app.patch("/api/v1/workflows/{workflow_id}", response_model=WorkflowResponse, tags=["Workflows"]) async def update_workflow_endpoint( workflow_id: str, @@ -4083,7 +3908,6 @@ async def update_workflow_endpoint( fail_count=updated.fail_count, ) - @app.delete("/api/v1/workflows/{workflow_id}", tags=["Workflows"]) async def delete_workflow_endpoint(workflow_id: str, _=Depends(verify_api_key)): """删除工作流""" @@ -4098,7 +3922,6 @@ async def delete_workflow_endpoint(workflow_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "Workflow deleted successfully"} - @app.post( "/api/v1/workflows/{workflow_id}/trigger", response_model=WorkflowTriggerResponse, @@ -4133,7 +3956,6 @@ async def trigger_workflow_endpoint( except (RuntimeError, TypeError, ConnectionError) as e: raise HTTPException(status_code=500, detail=str(e)) - @app.get( "/api/v1/workflows/{workflow_id}/logs", response_model=WorkflowLogListResponse, @@ -4173,7 +3995,6 @@ async def get_workflow_logs_endpoint( total=len(logs), ) - @app.get( "/api/v1/workflows/{workflow_id}/stats", response_model=WorkflowStatsResponse, @@ -4189,10 +4010,8 @@ async def get_workflow_stats_endpoint(workflow_id: str, days: int = 30, _=Depend return WorkflowStatsResponse(**stats) - # ==================== Phase 7: Webhook Endpoints ==================== - @app.post("/api/v1/webhooks", response_model=WebhookResponse, tags=["Webhooks"]) async def create_webhook_endpoint(request: WebhookCreate, _=Depends(verify_api_key)): """ @@ -4239,7 +4058,6 @@ async def create_webhook_endpoint(request: WebhookCreate, _=Depends(verify_api_k except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/webhooks", response_model=WebhookListResponse, tags=["Webhooks"]) async def list_webhooks_endpoint(_=Depends(verify_api_key)): """获取 Webhook 列表""" @@ -4270,7 +4088,6 @@ async def list_webhooks_endpoint(_=Depends(verify_api_key)): total=len(webhooks), ) - @app.get("/api/v1/webhooks/{webhook_id}", response_model=WebhookResponse, tags=["Webhooks"]) async def get_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): """获取单个 Webhook 详情""" @@ -4298,7 +4115,6 @@ async def get_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): fail_count=webhook.fail_count, ) - @app.patch("/api/v1/webhooks/{webhook_id}", response_model=WebhookResponse, tags=["Webhooks"]) async def update_webhook_endpoint( webhook_id: str, @@ -4332,7 +4148,6 @@ async def update_webhook_endpoint( fail_count=updated.fail_count, ) - @app.delete("/api/v1/webhooks/{webhook_id}", tags=["Webhooks"]) async def delete_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): """删除 Webhook 配置""" @@ -4347,7 +4162,6 @@ async def delete_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "Webhook deleted successfully"} - @app.post("/api/v1/webhooks/{webhook_id}/test", tags=["Webhooks"]) async def test_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): """测试 Webhook 配置""" @@ -4378,12 +4192,10 @@ async def test_webhook_endpoint(webhook_id: str, _=Depends(verify_api_key)): else: raise HTTPException(status_code=400, detail="Webhook test failed") - # ==================== Phase 7: Multimodal Support Endpoints ==================== # Pydantic Models for Multimodal API - class VideoUploadResponse(BaseModel): video_id: str project_id: str @@ -4394,7 +4206,6 @@ class VideoUploadResponse(BaseModel): ocr_text_preview: str message: str - class ImageUploadResponse(BaseModel): image_id: str project_id: str @@ -4405,7 +4216,6 @@ class ImageUploadResponse(BaseModel): entity_count: int status: str - class MultimodalEntityLinkResponse(BaseModel): link_id: str source_entity_id: str @@ -4416,19 +4226,16 @@ class MultimodalEntityLinkResponse(BaseModel): confidence: float evidence: str - class MultimodalAlignmentRequest(BaseModel): project_id: str threshold: float = 0.85 - class MultimodalAlignmentResponse(BaseModel): project_id: str aligned_count: int links: list[MultimodalEntityLinkResponse] message: str - class MultimodalStatsResponse(BaseModel): project_id: str video_count: int @@ -4437,7 +4244,6 @@ class MultimodalStatsResponse(BaseModel): cross_modal_links: int modality_distribution: dict[str, int] - @app.post( "/api/v1/projects/{project_id}/upload-video", response_model=VideoUploadResponse, @@ -4623,7 +4429,6 @@ async def upload_video_endpoint( message="Video processed successfully", ) - @app.post( "/api/v1/projects/{project_id}/upload-image", response_model=ImageUploadResponse, @@ -4774,7 +4579,6 @@ async def upload_image_endpoint( status="completed", ) - @app.post("/api/v1/projects/{project_id}/upload-images-batch", tags=["Multimodal"]) async def upload_images_batch_endpoint( project_id: str, @@ -4861,7 +4665,6 @@ async def upload_images_batch_endpoint( "results": results, } - @app.post( "/api/v1/projects/{project_id}/multimodal/align", response_model=MultimodalAlignmentResponse, @@ -4974,7 +4777,6 @@ async def align_multimodal_entities_endpoint( message=f"Successfully aligned {len(saved_links)} cross-modal entity pairs", ) - @app.get( "/api/v1/projects/{project_id}/multimodal/stats", response_model=MultimodalStatsResponse, @@ -5042,7 +4844,6 @@ async def get_multimodal_stats_endpoint(project_id: str, _=Depends(verify_api_ke modality_distribution=modality_dist, ) - @app.get("/api/v1/projects/{project_id}/videos", tags=["Multimodal"]) async def list_project_videos_endpoint(project_id: str, _=Depends(verify_api_key)): """获取项目的视频列表""" @@ -5079,7 +4880,6 @@ async def list_project_videos_endpoint(project_id: str, _=Depends(verify_api_key for v in videos ] - @app.get("/api/v1/projects/{project_id}/images", tags=["Multimodal"]) async def list_project_images_endpoint(project_id: str, _=Depends(verify_api_key)): """获取项目的图片列表""" @@ -5117,7 +4917,6 @@ async def list_project_images_endpoint(project_id: str, _=Depends(verify_api_key for img in images ] - @app.get("/api/v1/videos/{video_id}/frames", tags=["Multimodal"]) async def get_video_frames_endpoint(video_id: str, _=Depends(verify_api_key)): """获取视频的关键帧列表""" @@ -5147,7 +4946,6 @@ async def get_video_frames_endpoint(video_id: str, _=Depends(verify_api_key)): for f in frames ] - @app.get("/api/v1/entities/{entity_id}/multimodal-mentions", tags=["Multimodal"]) async def get_entity_multimodal_mentions_endpoint(entity_id: str, _=Depends(verify_api_key)): """获取实体的多模态提及信息""" @@ -5182,7 +4980,6 @@ async def get_entity_multimodal_mentions_endpoint(entity_id: str, _=Depends(veri for m in mentions ] - @app.get("/api/v1/projects/{project_id}/multimodal/suggest-merges", tags=["Multimodal"]) async def suggest_multimodal_merges_endpoint(project_id: str, _=Depends(verify_api_key)): """ @@ -5266,10 +5063,8 @@ async def suggest_multimodal_merges_endpoint(project_id: str, _=Depends(verify_a ], } - # ==================== Phase 7: Multimodal Support API ==================== - class VideoUploadResponse(BaseModel): video_id: str filename: str @@ -5282,7 +5077,6 @@ class VideoUploadResponse(BaseModel): status: str message: str - class ImageUploadResponse(BaseModel): image_id: str filename: str @@ -5291,7 +5085,6 @@ class ImageUploadResponse(BaseModel): status: str message: str - class MultimodalEntityLinkResponse(BaseModel): link_id: str entity_id: str @@ -5301,15 +5094,12 @@ class MultimodalEntityLinkResponse(BaseModel): evidence: str modalities: list[str] - class MultimodalProfileResponse(BaseModel): entity_id: str entity_name: str - # ==================== Phase 7 Task 7: Plugin Management Pydantic Models ==================== - class PluginCreate(BaseModel): name: str = Field(..., description="插件名称") plugin_type: str = Field( @@ -5322,13 +5112,11 @@ class PluginCreate(BaseModel): project_id: str = Field(..., description="关联项目ID") config: dict = Field(default_factory=dict, description="插件配置") - class PluginUpdate(BaseModel): name: str | None = None status: str | None = None # active, inactive, error, pending config: dict | None = None - class PluginResponse(BaseModel): id: str name: str @@ -5341,19 +5129,16 @@ class PluginResponse(BaseModel): last_used_at: str | None use_count: int - class PluginListResponse(BaseModel): plugins: list[PluginResponse] total: int - class ChromeExtensionTokenCreate(BaseModel): name: str = Field(..., description="令牌名称") project_id: str | None = Field(default=None, description="关联项目ID") permissions: list[str] = Field(default=["read"], description="权限列表: read, write, delete") expires_days: int | None = Field(default=None, description="过期天数") - class ChromeExtensionTokenResponse(BaseModel): id: str token: str = Field(..., description="令牌（仅显示一次）") @@ -5363,7 +5148,6 @@ class ChromeExtensionTokenResponse(BaseModel): expires_at: str | None created_at: str - class ChromeExtensionImportRequest(BaseModel): token: str = Field(..., description="Chrome扩展令牌") url: str = Field(..., description="网页URL") @@ -5371,7 +5155,6 @@ class ChromeExtensionImportRequest(BaseModel): content: str = Field(..., description="网页正文内容") html_content: str | None = Field(default=None, description="HTML内容（可选）") - class BotSessionCreate(BaseModel): session_id: str = Field(..., description="群ID或会话ID") session_name: str = Field(..., description="会话名称") @@ -5379,7 +5162,6 @@ class BotSessionCreate(BaseModel): webhook_url: str = Field(default="", description="Webhook URL") secret: str = Field(default="", description="签名密钥") - class BotSessionResponse(BaseModel): id: str bot_type: str @@ -5392,19 +5174,16 @@ class BotSessionResponse(BaseModel): last_message_at: str | None message_count: int - class BotMessageRequest(BaseModel): session_id: str = Field(..., description="会话ID") msg_type: str = Field(default="text", description="消息类型: text, audio, file") content: dict = Field(default_factory=dict, description="消息内容") - class BotMessageResponse(BaseModel): success: bool response: str error: str | None = None - class WebhookEndpointCreate(BaseModel): name: str = Field(..., description="端点名称") endpoint_type: str = Field(..., description="端点类型: zapier, make, custom") @@ -5414,7 +5193,6 @@ class WebhookEndpointCreate(BaseModel): auth_config: dict = Field(default_factory=dict, description="认证配置") trigger_events: list[str] = Field(default_factory=list, description="触发事件列表") - class WebhookEndpointResponse(BaseModel): id: str name: str @@ -5428,13 +5206,11 @@ class WebhookEndpointResponse(BaseModel): last_triggered_at: str | None trigger_count: int - class WebhookTestResponse(BaseModel): success: bool endpoint_id: str message: str - class WebDAVSyncCreate(BaseModel): name: str = Field(..., description="同步配置名称") project_id: str = Field(..., description="关联项目ID") @@ -5448,7 +5224,6 @@ class WebDAVSyncCreate(BaseModel): ) sync_interval: int = Field(default=3600, description="同步间隔（秒）") - class WebDAVSyncResponse(BaseModel): id: str name: str @@ -5464,12 +5239,10 @@ class WebDAVSyncResponse(BaseModel): created_at: str sync_count: int - class WebDAVTestResponse(BaseModel): success: bool message: str - class WebDAVSyncResult(BaseModel): success: bool message: str @@ -5478,11 +5251,9 @@ class WebDAVSyncResult(BaseModel): remote_path: str | None = None error: str | None = None - # Plugin Manager singleton _plugin_manager_instance: "PluginManager | None" = None - def get_plugin_manager_instance() -> "PluginManager | None": global _plugin_manager_instance if _plugin_manager_instance is None and PLUGIN_MANAGER_AVAILABLE and DB_AVAILABLE: @@ -5490,10 +5261,8 @@ def get_plugin_manager_instance() -> "PluginManager | None": _plugin_manager_instance = get_plugin_manager(db) return _plugin_manager_instance - # ==================== Phase 7 Task 7: Plugin Management Endpoints ==================== - @app.post("/api/v1/plugins", response_model=PluginResponse, tags=["Plugins"]) async def create_plugin_endpoint(request: PluginCreate, _=Depends(verify_api_key)): """ @@ -5536,7 +5305,6 @@ async def create_plugin_endpoint(request: PluginCreate, _=Depends(verify_api_key use_count=created.use_count, ) - @app.get("/api/v1/plugins", response_model=PluginListResponse, tags=["Plugins"]) async def list_plugins_endpoint( project_id: str | None = None, @@ -5570,7 +5338,6 @@ async def list_plugins_endpoint( total=len(plugins), ) - @app.get("/api/v1/plugins/{plugin_id}", response_model=PluginResponse, tags=["Plugins"]) async def get_plugin_endpoint(plugin_id: str, _=Depends(verify_api_key)): """获取插件详情""" @@ -5596,7 +5363,6 @@ async def get_plugin_endpoint(plugin_id: str, _=Depends(verify_api_key)): use_count=plugin.use_count, ) - @app.patch("/api/v1/plugins/{plugin_id}", response_model=PluginResponse, tags=["Plugins"]) async def update_plugin_endpoint(plugin_id: str, request: PluginUpdate, _=Depends(verify_api_key)): """更新插件""" @@ -5624,7 +5390,6 @@ async def update_plugin_endpoint(plugin_id: str, request: PluginUpdate, _=Depend use_count=updated.use_count, ) - @app.delete("/api/v1/plugins/{plugin_id}", tags=["Plugins"]) async def delete_plugin_endpoint(plugin_id: str, _=Depends(verify_api_key)): """删除插件""" @@ -5639,10 +5404,8 @@ async def delete_plugin_endpoint(plugin_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "Plugin deleted successfully"} - # ==================== Phase 7 Task 7: Chrome Extension Endpoints ==================== - @app.post( "/api/v1/plugins/chrome/tokens", response_model=ChromeExtensionTokenResponse, @@ -5683,7 +5446,6 @@ async def create_chrome_token_endpoint( created_at=token.created_at, ) - @app.get("/api/v1/plugins/chrome/tokens", tags=["Chrome Extension"]) async def list_chrome_tokens_endpoint(project_id: str | None = None, _=Depends(verify_api_key)): """列出 Chrome 扩展令牌""" @@ -5716,7 +5478,6 @@ async def list_chrome_tokens_endpoint(project_id: str | None = None, _=Depends(v "total": len(tokens), } - @app.delete("/api/v1/plugins/chrome/tokens/{token_id}", tags=["Chrome Extension"]) async def revoke_chrome_token_endpoint(token_id: str, _=Depends(verify_api_key)): """撤销 Chrome 扩展令牌""" @@ -5736,7 +5497,6 @@ async def revoke_chrome_token_endpoint(token_id: str, _=Depends(verify_api_key)) return {"success": True, "message": "Token revoked successfully"} - @app.post("/api/v1/plugins/chrome/import", tags=["Chrome Extension"]) async def chrome_import_webpage_endpoint(request: ChromeExtensionImportRequest): """ @@ -5772,10 +5532,8 @@ async def chrome_import_webpage_endpoint(request: ChromeExtensionImportRequest): return result - # ==================== Phase 7 Task 7: Bot Endpoints ==================== - @app.post("/api/v1/plugins/bot/feishu/sessions", response_model=BotSessionResponse, tags=["Bot"]) async def create_feishu_session_endpoint(request: BotSessionCreate, _=Depends(verify_api_key)): """创建飞书机器人会话""" @@ -5809,7 +5567,6 @@ async def create_feishu_session_endpoint(request: BotSessionCreate, _=Depends(ve message_count=session.message_count, ) - @app.post("/api/v1/plugins/bot/dingtalk/sessions", response_model=BotSessionResponse, tags=["Bot"]) async def create_dingtalk_session_endpoint(request: BotSessionCreate, _=Depends(verify_api_key)): """创建钉钉机器人会话""" @@ -5843,7 +5600,6 @@ async def create_dingtalk_session_endpoint(request: BotSessionCreate, _=Depends( message_count=session.message_count, ) - @app.get("/api/v1/plugins/bot/{bot_type}/sessions", tags=["Bot"]) async def list_bot_sessions_endpoint( bot_type: str, @@ -5886,7 +5642,6 @@ async def list_bot_sessions_endpoint( "total": len(sessions), } - @app.post("/api/v1/plugins/bot/{bot_type}/webhook", tags=["Bot"]) async def bot_webhook_endpoint(bot_type: str, request: Request): """ @@ -5940,7 +5695,6 @@ async def bot_webhook_endpoint(bot_type: str, request: Request): return result - @app.post("/api/v1/plugins/bot/{bot_type}/sessions/{session_id}/send", tags=["Bot"]) async def send_bot_message_endpoint( bot_type: str, @@ -5972,10 +5726,8 @@ async def send_bot_message_endpoint( return {"success": success, "message": "Message sent" if success else "Failed to send message"} - # ==================== Phase 7 Task 7: Integration Endpoints ==================== - @app.post( "/api/v1/plugins/integrations/zapier", response_model=WebhookEndpointResponse, @@ -6015,7 +5767,6 @@ async def create_zapier_endpoint(request: WebhookEndpointCreate, _=Depends(verif trigger_count=endpoint.trigger_count, ) - @app.post( "/api/v1/plugins/integrations/make", response_model=WebhookEndpointResponse, @@ -6055,7 +5806,6 @@ async def create_make_endpoint(request: WebhookEndpointCreate, _=Depends(verify_ trigger_count=endpoint.trigger_count, ) - @app.get("/api/v1/plugins/integrations/{endpoint_type}", tags=["Integrations"]) async def list_integration_endpoints_endpoint( endpoint_type: str, @@ -6100,7 +5850,6 @@ async def list_integration_endpoints_endpoint( "total": len(endpoints), } - @app.post( "/api/v1/plugins/integrations/{endpoint_id}/test", response_model=WebhookTestResponse, @@ -6132,7 +5881,6 @@ async def test_integration_endpoint(endpoint_id: str, _=Depends(verify_api_key)) message=result["message"], ) - @app.post("/api/v1/plugins/integrations/{endpoint_id}/trigger", tags=["Integrations"]) async def trigger_integration_endpoint( endpoint_id: str, @@ -6164,10 +5912,8 @@ async def trigger_integration_endpoint( "message": "Triggered successfully" if success else "Trigger failed", } - # ==================== Phase 7 Task 7: WebDAV Endpoints ==================== - @app.post("/api/v1/plugins/webdav", response_model=WebDAVSyncResponse, tags=["WebDAV"]) async def create_webdav_sync_endpoint(request: WebDAVSyncCreate, _=Depends(verify_api_key)): """ @@ -6211,7 +5957,6 @@ async def create_webdav_sync_endpoint(request: WebDAVSyncCreate, _=Depends(verif sync_count=sync.sync_count, ) - @app.get("/api/v1/plugins/webdav", tags=["WebDAV"]) async def list_webdav_syncs_endpoint(project_id: str | None = None, _=Depends(verify_api_key)): """列出 WebDAV 同步配置""" @@ -6248,7 +5993,6 @@ async def list_webdav_syncs_endpoint(project_id: str | None = None, _=Depends(ve "total": len(syncs), } - @app.post( "/api/v1/plugins/webdav/{sync_id}/test", response_model=WebDAVTestResponse, @@ -6276,7 +6020,6 @@ async def test_webdav_connection_endpoint(sync_id: str, _=Depends(verify_api_key message=result.get("message") or result.get("error", "Unknown result"), ) - @app.post("/api/v1/plugins/webdav/{sync_id}/sync", response_model=WebDAVSyncResult, tags=["WebDAV"]) async def sync_webdav_endpoint(sync_id: str, _=Depends(verify_api_key)): """执行 WebDAV 同步""" @@ -6304,7 +6047,6 @@ async def sync_webdav_endpoint(sync_id: str, _=Depends(verify_api_key)): error=result.get("error"), ) - @app.delete("/api/v1/plugins/webdav/{sync_id}", tags=["WebDAV"]) async def delete_webdav_sync_endpoint(sync_id: str, _=Depends(verify_api_key)): """删除 WebDAV 同步配置""" @@ -6324,7 +6066,6 @@ async def delete_webdav_sync_endpoint(sync_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "WebDAV sync configuration deleted"} - @app.get("/api/v1/openapi.json", include_in_schema=False) async def get_openapi(): """获取 OpenAPI 规范""" @@ -6338,7 +6079,6 @@ async def get_openapi(): tags=app.openapi_tags, ) - # Serve frontend - MUST be last to not override API routes app.mount("/", StaticFiles(directory="frontend", html=True), name="frontend") @@ -6347,14 +6087,12 @@ if __name__ == "__main__": uvicorn.run(app, host="0.0.0.0", port=8000) - class PluginCreateRequest(BaseModel): name: str plugin_type: str project_id: str | None = None config: dict | None = {} - class PluginResponse(BaseModel): id: str name: str @@ -6364,7 +6102,6 @@ class PluginResponse(BaseModel): api_key: str created_at: str - class BotSessionResponse(BaseModel): id: str plugin_id: str @@ -6377,7 +6114,6 @@ class BotSessionResponse(BaseModel): created_at: str last_message_at: str | None - class WebhookEndpointResponse(BaseModel): id: str plugin_id: str @@ -6389,7 +6125,6 @@ class WebhookEndpointResponse(BaseModel): trigger_count: int created_at: str - class WebDAVSyncResponse(BaseModel): id: str plugin_id: str @@ -6405,7 +6140,6 @@ class WebDAVSyncResponse(BaseModel): last_sync_at: str | None created_at: str - class ChromeClipRequest(BaseModel): url: str title: str @@ -6414,7 +6148,6 @@ class ChromeClipRequest(BaseModel): meta: dict | None = {} project_id: str | None = None - class ChromeClipResponse(BaseModel): clip_id: str project_id: str @@ -6423,7 +6156,6 @@ class ChromeClipResponse(BaseModel): status: str message: str - class BotMessagePayload(BaseModel): platform: str session_id: str @@ -6433,19 +6165,16 @@ class BotMessagePayload(BaseModel): content: str project_id: str | None = None - class BotMessageResult(BaseModel): success: bool reply: str | None = None session_id: str action: str | None = None - class WebhookPayload(BaseModel): event: str data: dict - @app.post("/api/v1/plugins", response_model=PluginResponse, tags=["Plugins"]) async def create_plugin(request: PluginCreateRequest, api_key: str = Depends(verify_api_key)): """创建插件""" @@ -6470,7 +6199,6 @@ async def create_plugin(request: PluginCreateRequest, api_key: str = Depends(ver created_at=plugin.created_at, ) - @app.get("/api/v1/plugins", tags=["Plugins"]) async def list_plugins( project_id: str | None = None, @@ -6499,7 +6227,6 @@ async def list_plugins( ], } - @app.get("/api/v1/plugins/{plugin_id}", response_model=PluginResponse, tags=["Plugins"]) async def get_plugin(plugin_id: str, api_key: str = Depends(verify_api_key)): """获取插件详情""" @@ -6522,7 +6249,6 @@ async def get_plugin(plugin_id: str, api_key: str = Depends(verify_api_key)): created_at=plugin.created_at, ) - @app.delete("/api/v1/plugins/{plugin_id}", tags=["Plugins"]) async def delete_plugin(plugin_id: str, api_key: str = Depends(verify_api_key)): """删除插件""" @@ -6534,7 +6260,6 @@ async def delete_plugin(plugin_id: str, api_key: str = Depends(verify_api_key)): return {"success": True, "message": "Plugin deleted"} - @app.post("/api/v1/plugins/{plugin_id}/regenerate-key", tags=["Plugins"]) async def regenerate_plugin_key(plugin_id: str, api_key: str = Depends(verify_api_key)): """重新生成插件 API Key""" @@ -6546,10 +6271,8 @@ async def regenerate_plugin_key(plugin_id: str, api_key: str = Depends(verify_ap return {"success": True, "api_key": new_key} - # ==================== Chrome Extension API ==================== - @app.post( "/api/v1/plugins/chrome/clip", response_model=ChromeClipResponse, @@ -6624,10 +6347,8 @@ URL: {request.url} message="Content saved successfully", ) - # ==================== Bot API ==================== - @app.post("/api/v1/bots/webhook/{platform}", response_model=BotMessageResponse, tags=["Bot"]) async def bot_webhook( platform: str, @@ -6665,7 +6386,6 @@ async def bot_webhook( action="reply", ) - @app.get("/api/v1/bots/sessions", response_model=list[BotSessionResponse], tags=["Bot"]) async def list_bot_sessions( plugin_id: str | None = None, @@ -6695,10 +6415,8 @@ async def list_bot_sessions( for s in sessions ] - # ==================== Webhook Integration API ==================== - @app.post( "/api/v1/webhook-endpoints", response_model=WebhookEndpointResponse, @@ -6737,7 +6455,6 @@ async def create_integration_webhook_endpoint( created_at=endpoint.created_at, ) - @app.get( "/api/v1/webhook-endpoints", response_model=list[WebhookEndpointResponse], @@ -6769,7 +6486,6 @@ async def list_webhook_endpoints( for e in endpoints ] - @app.post("/webhook/{endpoint_type}/{token}", tags=["Integrations"]) async def receive_webhook( endpoint_type: str, @@ -6820,10 +6536,8 @@ async def receive_webhook( return {"success": True, "endpoint_id": endpoint.id, "received_at": datetime.now().isoformat()} - # ==================== WebDAV API ==================== - @app.post("/api/v1/webdav-syncs", response_model=WebDAVSyncResponse, tags=["WebDAV"]) async def create_webdav_sync( plugin_id: str, @@ -6872,7 +6586,6 @@ async def create_webdav_sync( created_at=sync.created_at, ) - @app.get("/api/v1/webdav-syncs", response_model=list[WebDAVSyncResponse], tags=["WebDAV"]) async def list_webdav_syncs(plugin_id: str | None = None, api_key: str = Depends(verify_api_key)): """列出 WebDAV 同步配置""" @@ -6901,7 +6614,6 @@ async def list_webdav_syncs(plugin_id: str | None = None, api_key: str = Depends for s in syncs ] - @app.post("/api/v1/webdav-syncs/{sync_id}/test", tags=["WebDAV"]) async def test_webdav_connection(sync_id: str, api_key: str = Depends(verify_api_key)): """测试 WebDAV 连接""" @@ -6914,7 +6626,6 @@ async def test_webdav_connection(sync_id: str, api_key: str = Depends(verify_api if not sync: raise HTTPException(status_code=404, detail="WebDAV sync not found") - from plugin_manager import WebDAVSync as WebDAVSyncHandler handler = WebDAVSyncHandler(manager) @@ -6922,7 +6633,6 @@ async def test_webdav_connection(sync_id: str, api_key: str = Depends(verify_api return {"success": success, "message": message} - @app.post("/api/v1/webdav-syncs/{sync_id}/sync", tags=["WebDAV"]) async def trigger_webdav_sync(sync_id: str, api_key: str = Depends(verify_api_key)): """手动触发 WebDAV 同步""" @@ -6946,10 +6656,8 @@ async def trigger_webdav_sync(sync_id: str, api_key: str = Depends(verify_api_ke return {"success": True, "sync_id": sync_id, "status": "running", "message": "Sync started"} - # ==================== Plugin Activity Logs ==================== - @app.get("/api/v1/plugins/{plugin_id}/logs", tags=["Plugins"]) async def get_plugin_logs( plugin_id: str, @@ -6977,12 +6685,10 @@ async def get_plugin_logs( ], } - # ==================== Phase 7 Task 3: Security & Compliance API ==================== # Pydantic models for security API - class AuditLogResponse(BaseModel): id: str action_type: str @@ -6995,18 +6701,15 @@ class AuditLogResponse(BaseModel): error_message: str | None = None created_at: str - class AuditStatsResponse(BaseModel): total_actions: int success_count: int failure_count: int action_breakdown: dict[str, dict[str, int]] - class EncryptionEnableRequest(BaseModel): master_password: str - class EncryptionConfigResponse(BaseModel): id: str project_id: str @@ -7015,7 +6718,6 @@ class EncryptionConfigResponse(BaseModel): created_at: str updated_at: str - class MaskingRuleCreateRequest(BaseModel): name: str rule_type: str # phone, email, id_card, bank_card, name, address, custom @@ -7024,7 +6726,6 @@ class MaskingRuleCreateRequest(BaseModel): description: str | None = None priority: int = 0 - class MaskingRuleResponse(BaseModel): id: str project_id: str @@ -7038,18 +6739,15 @@ class MaskingRuleResponse(BaseModel): created_at: str updated_at: str - class MaskingApplyRequest(BaseModel): text: str rule_types: list[str] | None = None - class MaskingApplyResponse(BaseModel): original_text: str masked_text: str applied_rules: list[str] - class AccessPolicyCreateRequest(BaseModel): name: str description: str | None = None @@ -7060,7 +6758,6 @@ class AccessPolicyCreateRequest(BaseModel): max_access_count: int | None = None require_approval: bool = False - class AccessPolicyResponse(BaseModel): id: str project_id: str @@ -7076,13 +6773,11 @@ class AccessPolicyResponse(BaseModel): created_at: str updated_at: str - class AccessRequestCreateRequest(BaseModel): policy_id: str request_reason: str | None = None expires_hours: int = 24 - class AccessRequestResponse(BaseModel): id: str policy_id: str @@ -7094,10 +6789,8 @@ class AccessRequestResponse(BaseModel): expires_at: str | None = None created_at: str - # ==================== Audit Logs API ==================== - @app.get("/api/v1/audit-logs", response_model=list[AuditLogResponse], tags=["Security"]) async def get_audit_logs( user_id: str | None = None, @@ -7144,7 +6837,6 @@ async def get_audit_logs( for log in logs ] - @app.get("/api/v1/audit-logs/stats", response_model=AuditStatsResponse, tags=["Security"]) async def get_audit_stats( start_time: str | None = None, @@ -7160,10 +6852,8 @@ async def get_audit_stats( return AuditStatsResponse(**stats) - # ==================== Encryption API ==================== - @app.post( "/api/v1/projects/{project_id}/encryption/enable", response_model=EncryptionConfigResponse, @@ -7193,7 +6883,6 @@ async def enable_project_encryption( except RuntimeError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/projects/{project_id}/encryption/disable", tags=["Security"]) async def disable_project_encryption( project_id: str, @@ -7212,7 +6901,6 @@ async def disable_project_encryption( return {"success": True, "message": "Encryption disabled successfully"} - @app.post("/api/v1/projects/{project_id}/encryption/verify", tags=["Security"]) async def verify_encryption_password( project_id: str, @@ -7228,7 +6916,6 @@ async def verify_encryption_password( return {"valid": is_valid} - @app.get( "/api/v1/projects/{project_id}/encryption", response_model=EncryptionConfigResponse | None, @@ -7254,10 +6941,8 @@ async def get_encryption_config(project_id: str, api_key: str = Depends(verify_a updated_at=config.updated_at, ) - # ==================== Data Masking API ==================== - @app.post( "/api/v1/projects/{project_id}/masking-rules", response_model=MaskingRuleResponse, @@ -7303,7 +6988,6 @@ async def create_masking_rule( updated_at=rule.updated_at, ) - @app.get( "/api/v1/projects/{project_id}/masking-rules", response_model=list[MaskingRuleResponse], @@ -7338,7 +7022,6 @@ async def get_masking_rules( for rule in rules ] - @app.put("/api/v1/masking-rules/{rule_id}", response_model=MaskingRuleResponse, tags=["Security"]) async def update_masking_rule( rule_id: str, @@ -7389,7 +7072,6 @@ async def update_masking_rule( updated_at=rule.updated_at, ) - @app.delete("/api/v1/masking-rules/{rule_id}", tags=["Security"]) async def delete_masking_rule(rule_id: str, api_key: str = Depends(verify_api_key)): """删除脱敏规则""" @@ -7404,7 +7086,6 @@ async def delete_masking_rule(rule_id: str, api_key: str = Depends(verify_api_ke return {"success": True, "message": "Masking rule deleted"} - @app.post( "/api/v1/projects/{project_id}/masking/apply", response_model=MaskingApplyResponse, @@ -7438,10 +7119,8 @@ async def apply_masking( applied_rules=applied_rules, ) - # ==================== Data Access Policy API ==================== - @app.post( "/api/v1/projects/{project_id}/access-policies", response_model=AccessPolicyResponse, @@ -7488,7 +7167,6 @@ async def create_access_policy( updated_at=policy.updated_at, ) - @app.get( "/api/v1/projects/{project_id}/access-policies", response_model=list[AccessPolicyResponse], @@ -7527,7 +7205,6 @@ async def get_access_policies( for policy in policies ] - @app.post("/api/v1/access-policies/{policy_id}/check", tags=["Security"]) async def check_access_permission( policy_id: str, @@ -7544,10 +7221,8 @@ async def check_access_permission( return {"allowed": allowed, "reason": reason if not allowed else None} - # ==================== Access Request API ==================== - @app.post("/api/v1/access-requests", response_model=AccessRequestResponse, tags=["Security"]) async def create_access_request( request: AccessRequestCreateRequest, @@ -7579,7 +7254,6 @@ async def create_access_request( created_at=access_request.created_at, ) - @app.post( "/api/v1/access-requests/{request_id}/approve", response_model=AccessRequestResponse, @@ -7613,7 +7287,6 @@ async def approve_access_request( created_at=access_request.created_at, ) - @app.post( "/api/v1/access-requests/{request_id}/reject", response_model=AccessRequestResponse, @@ -7646,14 +7319,12 @@ async def reject_access_request( created_at=access_request.created_at, ) - # ========================================== # Phase 7 Task 4: 协作与共享 API # ========================================== # ----- 请求模型 ----- - class ShareLinkCreate(BaseModel): permission: str = "read_only" # read_only, comment, edit, admin expires_in_days: int | None = None @@ -7662,12 +7333,10 @@ class ShareLinkCreate(BaseModel): allow_download: bool = False allow_export: bool = False - class ShareLinkVerify(BaseModel): token: str password: str | None = None - class CommentCreate(BaseModel): target_type: str # entity, relation, transcript, project target_id: str @@ -7675,29 +7344,23 @@ class CommentCreate(BaseModel): content: str mentions: list[str] | None = None - class CommentUpdate(BaseModel): content: str - class CommentResolve(BaseModel): resolved: bool - class TeamMemberInvite(BaseModel): user_id: str user_name: str user_email: str role: str = "viewer" # owner, admin, editor, viewer, commenter - class TeamMemberRoleUpdate(BaseModel): role: str - # ----- 项目分享 ----- - @app.post("/api/v1/projects/{project_id}/shares") async def create_share_link( project_id: str, @@ -7730,7 +7393,6 @@ async def create_share_link( "share_url": f"/share/{share.token}", } - @app.get("/api/v1/projects/{project_id}/shares") async def list_project_shares(project_id: str): """列出项目的所有分享链接""" @@ -7759,7 +7421,6 @@ async def list_project_shares(project_id: str): ], } - @app.post("/api/v1/shares/verify") async def verify_share_link(request: ShareLinkVerify): """验证分享链接""" @@ -7783,7 +7444,6 @@ async def verify_share_link(request: ShareLinkVerify): "allow_export": share.allow_export, } - @app.get("/api/v1/shares/{token}/access") async def access_shared_project(token: str, password: str | None = None): """通过分享链接访问项目""" @@ -7821,7 +7481,6 @@ async def access_shared_project(token: str, password: str | None = None): "allow_export": share.allow_export, } - @app.delete("/api/v1/shares/{share_id}") async def revoke_share_link(share_id: str, revoked_by: str = "current_user"): """撤销分享链接""" @@ -7836,10 +7495,8 @@ async def revoke_share_link(share_id: str, revoked_by: str = "current_user"): return {"success": True, "message": "Share link revoked"} - # ----- 评论和批注 ----- - @app.post("/api/v1/projects/{project_id}/comments") async def add_comment( project_id: str, @@ -7875,7 +7532,6 @@ async def add_comment( "resolved": comment.resolved, } - @app.get("/api/v1/{target_type}/{target_id}/comments") async def get_comments(target_type: str, target_id: str, include_resolved: bool = True): """获取评论列表""" @@ -7904,7 +7560,6 @@ async def get_comments(target_type: str, target_id: str, include_resolved: bool ], } - @app.get("/api/v1/projects/{project_id}/comments") async def get_project_comments(project_id: str, limit: int = 50, offset: int = 0): """获取项目下的所有评论""" @@ -7932,7 +7587,6 @@ async def get_project_comments(project_id: str, limit: int = 50, offset: int = 0 ], } - @app.put("/api/v1/comments/{comment_id}") async def update_comment(comment_id: str, request: CommentUpdate, updated_by: str = "current_user"): """更新评论""" @@ -7947,7 +7601,6 @@ async def update_comment(comment_id: str, request: CommentUpdate, updated_by: st return {"id": comment.id, "content": comment.content, "updated_at": comment.updated_at} - @app.post("/api/v1/comments/{comment_id}/resolve") async def resolve_comment(comment_id: str, resolved_by: str = "current_user"): """标记评论为已解决""" @@ -7962,7 +7615,6 @@ async def resolve_comment(comment_id: str, resolved_by: str = "current_user"): return {"success": True, "message": "Comment resolved"} - @app.delete("/api/v1/comments/{comment_id}") async def delete_comment(comment_id: str, deleted_by: str = "current_user"): """删除评论""" @@ -7977,10 +7629,8 @@ async def delete_comment(comment_id: str, deleted_by: str = "current_user"): return {"success": True, "message": "Comment deleted"} - # ----- 变更历史 ----- - @app.get("/api/v1/projects/{project_id}/history") async def get_change_history( project_id: str, @@ -8017,7 +7667,6 @@ async def get_change_history( ], } - @app.get("/api/v1/projects/{project_id}/history/stats") async def get_change_history_stats(project_id: str): """获取变更统计""" @@ -8029,7 +7678,6 @@ async def get_change_history_stats(project_id: str): return stats - @app.get("/api/v1/{entity_type}/{entity_id}/versions") async def get_entity_versions(entity_type: str, entity_id: str): """获取实体版本历史""" @@ -8056,7 +7704,6 @@ async def get_entity_versions(entity_type: str, entity_id: str): ], } - @app.post("/api/v1/history/{record_id}/revert") async def revert_change(record_id: str, reverted_by: str = "current_user"): """回滚变更""" @@ -8071,10 +7718,8 @@ async def revert_change(record_id: str, reverted_by: str = "current_user"): return {"success": True, "message": "Change reverted"} - # ----- 团队成员 ----- - @app.post("/api/v1/projects/{project_id}/members") async def invite_team_member( project_id: str, @@ -8105,7 +7750,6 @@ async def invite_team_member( "permissions": member.permissions, } - @app.get("/api/v1/projects/{project_id}/members") async def list_team_members(project_id: str): """列出团队成员""" @@ -8132,7 +7776,6 @@ async def list_team_members(project_id: str): ], } - @app.put("/api/v1/members/{member_id}/role") async def update_member_role( member_id: str, @@ -8151,7 +7794,6 @@ async def update_member_role( return {"success": True, "message": "Member role updated"} - @app.delete("/api/v1/members/{member_id}") async def remove_team_member(member_id: str, removed_by: str = "current_user"): """移除团队成员""" @@ -8166,7 +7808,6 @@ async def remove_team_member(member_id: str, removed_by: str = "current_user"): return {"success": True, "message": "Member removed"} - @app.get("/api/v1/projects/{project_id}/permissions") async def check_project_permissions(project_id: str, user_id: str = "current_user"): """检查用户权限""" @@ -8187,10 +7828,8 @@ async def check_project_permissions(project_id: str, user_id: str = "current_use return {"has_access": True, "role": user_member.role, "permissions": user_member.permissions} - # ==================== Phase 7 Task 6: Advanced Search & Discovery ==================== - class FullTextSearchRequest(BaseModel): """全文搜索请求""" @@ -8199,7 +7838,6 @@ class FullTextSearchRequest(BaseModel): operator: str = "AND" # AND, OR, NOT limit: int = 20 - class SemanticSearchRequest(BaseModel): """语义搜索请求""" @@ -8208,7 +7846,6 @@ class SemanticSearchRequest(BaseModel): threshold: float = 0.7 limit: int = 20 - @app.post("/api/v1/search/fulltext", tags=["Search"]) async def fulltext_search( project_id: str, @@ -8251,7 +7888,6 @@ async def fulltext_search( ], } - @app.post("/api/v1/search/semantic", tags=["Search"]) async def semantic_search( project_id: str, @@ -8282,7 +7918,6 @@ async def semantic_search( ], } - @app.get("/api/v1/entities/{entity_id}/paths/{target_entity_id}", tags=["Search"]) async def find_entity_paths( entity_id: str, @@ -8327,7 +7962,6 @@ async def find_entity_paths( ], } - @app.get("/api/v1/entities/{entity_id}/network", tags=["Search"]) async def get_entity_network(entity_id: str, depth: int = 2, _=Depends(verify_api_key)): """获取实体关系网络""" @@ -8339,7 +7973,6 @@ async def get_entity_network(entity_id: str, depth: int = 2, _=Depends(verify_ap return network - @app.get("/api/v1/projects/{project_id}/knowledge-gaps", tags=["Search"]) async def detect_knowledge_gaps(project_id: str, _=Depends(verify_api_key)): """检测知识缺口""" @@ -8369,7 +8002,6 @@ async def detect_knowledge_gaps(project_id: str, _=Depends(verify_api_key)): ], } - @app.post("/api/v1/projects/{project_id}/search/index", tags=["Search"]) async def index_project_for_search(project_id: str, _=Depends(verify_api_key)): """为项目创建搜索索引""" @@ -8384,10 +8016,8 @@ async def index_project_for_search(project_id: str, _=Depends(verify_api_key)): else: raise HTTPException(status_code=500, detail="Failed to index project") - # ==================== Phase 7 Task 8: Performance & Scaling ==================== - @app.get("/api/v1/cache/stats", tags=["Performance"]) async def get_cache_stats(_=Depends(verify_api_key)): """获取缓存统计""" @@ -8407,7 +8037,6 @@ async def get_cache_stats(_=Depends(verify_api_key)): "expired_count": stats.expired_count, } - @app.post("/api/v1/cache/clear", tags=["Performance"]) async def clear_cache(pattern: str | None = None, _=Depends(verify_api_key)): """清除缓存""" @@ -8422,7 +8051,6 @@ async def clear_cache(pattern: str | None = None, _=Depends(verify_api_key)): else: raise HTTPException(status_code=500, detail="Failed to clear cache") - @app.get("/api/v1/performance/metrics", tags=["Performance"]) async def get_performance_metrics( metric_type: str | None = None, @@ -8462,7 +8090,6 @@ async def get_performance_metrics( ], } - @app.get("/api/v1/performance/summary", tags=["Performance"]) async def get_performance_summary(hours: int = 24, _=Depends(verify_api_key)): """获取性能汇总统计""" @@ -8474,7 +8101,6 @@ async def get_performance_summary(hours: int = 24, _=Depends(verify_api_key)): return summary - @app.get("/api/v1/tasks/{task_id}/status", tags=["Performance"]) async def get_task_status(task_id: str, _=Depends(verify_api_key)): """获取任务状态""" @@ -8502,7 +8128,6 @@ async def get_task_status(task_id: str, _=Depends(verify_api_key)): "priority": task.priority, } - @app.get("/api/v1/tasks", tags=["Performance"]) async def list_tasks( project_id: str | None = None, @@ -8533,7 +8158,6 @@ async def list_tasks( ], } - @app.post("/api/v1/tasks/{task_id}/cancel", tags=["Performance"]) async def cancel_task(task_id: str, _=Depends(verify_api_key)): """取消任务""" @@ -8551,7 +8175,6 @@ async def cancel_task(task_id: str, _=Depends(verify_api_key)): detail="Failed to cancel task or task already completed", ) - @app.get("/api/v1/shards", tags=["Performance"]) async def list_shards(_=Depends(verify_api_key)): """列出数据库分片""" @@ -8574,30 +8197,25 @@ async def list_shards(_=Depends(verify_api_key)): ], } - # ============================================ # Phase 8: Multi-Tenant SaaS APIs # ============================================ - class CreateTenantRequest(BaseModel): name: str description: str | None = None tier: str = "free" - class UpdateTenantRequest(BaseModel): name: str | None = None description: str | None = None tier: str | None = None status: str | None = None - class AddDomainRequest(BaseModel): domain: str is_primary: bool = False - class UpdateBrandingRequest(BaseModel): logo_url: str | None = None favicon_url: str | None = None @@ -8607,19 +8225,15 @@ class UpdateBrandingRequest(BaseModel): custom_js: str | None = None login_page_bg: str | None = None - class InviteMemberRequest(BaseModel): email: str role: str = "member" - class UpdateMemberRequest(BaseModel): role: str | None = None - # Tenant Management APIs - @app.post("/api/v1/tenants", tags=["Tenants"]) async def create_tenant( request: CreateTenantRequest, @@ -8649,7 +8263,6 @@ async def create_tenant( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants", tags=["Tenants"]) async def list_my_tenants( user_id: str = Header(..., description="当前用户ID"), @@ -8663,7 +8276,6 @@ async def list_my_tenants( tenants = manager.get_user_tenants(user_id) return {"tenants": tenants} - @app.get("/api/v1/tenants/{tenant_id}", tags=["Tenants"]) async def get_tenant(tenant_id: str, _=Depends(verify_api_key)): """获取租户详情""" @@ -8689,7 +8301,6 @@ async def get_tenant(tenant_id: str, _=Depends(verify_api_key)): "resource_limits": tenant.resource_limits, } - @app.put("/api/v1/tenants/{tenant_id}", tags=["Tenants"]) async def update_tenant(tenant_id: str, request: UpdateTenantRequest, _=Depends(verify_api_key)): """更新租户信息""" @@ -8717,7 +8328,6 @@ async def update_tenant(tenant_id: str, request: UpdateTenantRequest, _=Depends( "updated_at": tenant.updated_at.isoformat(), } - @app.delete("/api/v1/tenants/{tenant_id}", tags=["Tenants"]) async def delete_tenant(tenant_id: str, _=Depends(verify_api_key)): """删除租户""" @@ -8732,10 +8342,8 @@ async def delete_tenant(tenant_id: str, _=Depends(verify_api_key)): return {"message": "Tenant deleted successfully"} - # Domain Management APIs - @app.post("/api/v1/tenants/{tenant_id}/domains", tags=["Tenants"]) async def add_domain(tenant_id: str, request: AddDomainRequest, _=Depends(verify_api_key)): """为租户添加自定义域名""" @@ -8765,7 +8373,6 @@ async def add_domain(tenant_id: str, request: AddDomainRequest, _=Depends(verify except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/domains", tags=["Tenants"]) async def list_domains(tenant_id: str, _=Depends(verify_api_key)): """列出租户的所有域名""" @@ -8790,7 +8397,6 @@ async def list_domains(tenant_id: str, _=Depends(verify_api_key)): ], } - @app.post("/api/v1/tenants/{tenant_id}/domains/{domain_id}/verify", tags=["Tenants"]) async def verify_domain(tenant_id: str, domain_id: str, _=Depends(verify_api_key)): """验证域名所有权""" @@ -8805,7 +8411,6 @@ async def verify_domain(tenant_id: str, domain_id: str, _=Depends(verify_api_key "message": "Domain verified successfully" if success else "Domain verification failed", } - @app.delete("/api/v1/tenants/{tenant_id}/domains/{domain_id}", tags=["Tenants"]) async def remove_domain(tenant_id: str, domain_id: str, _=Depends(verify_api_key)): """移除域名绑定""" @@ -8820,10 +8425,8 @@ async def remove_domain(tenant_id: str, domain_id: str, _=Depends(verify_api_key return {"message": "Domain removed successfully"} - # Branding APIs - @app.get("/api/v1/tenants/{tenant_id}/branding", tags=["Tenants"]) async def get_branding(tenant_id: str, _=Depends(verify_api_key)): """获取租户品牌配置""" @@ -8854,7 +8457,6 @@ async def get_branding(tenant_id: str, _=Depends(verify_api_key)): "login_page_bg": branding.login_page_bg, } - @app.put("/api/v1/tenants/{tenant_id}/branding", tags=["Tenants"]) async def update_branding( tenant_id: str, @@ -8886,7 +8488,6 @@ async def update_branding( "updated_at": branding.updated_at.isoformat(), } - @app.get("/api/v1/tenants/{tenant_id}/branding.css", tags=["Tenants"]) async def get_branding_css(tenant_id: str): """获取租户品牌 CSS（公开端点，无需认证）""" @@ -8898,10 +8499,8 @@ async def get_branding_css(tenant_id: str): return PlainTextResponse(content=css, media_type="text/css") - # Member Management APIs - @app.post("/api/v1/tenants/{tenant_id}/members", tags=["Tenants"]) async def invite_member( tenant_id: str, @@ -8932,7 +8531,6 @@ async def invite_member( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/members", tags=["Tenants"]) async def list_members(tenant_id: str, status: str | None = None, _=Depends(verify_api_key)): """列出租户成员""" @@ -8959,7 +8557,6 @@ async def list_members(tenant_id: str, status: str | None = None, _=Depends(veri ], } - @app.put("/api/v1/tenants/{tenant_id}/members/{member_id}", tags=["Tenants"]) async def update_member( tenant_id: str, @@ -8979,7 +8576,6 @@ async def update_member( return {"message": "Member updated successfully"} - @app.delete("/api/v1/tenants/{tenant_id}/members/{member_id}", tags=["Tenants"]) async def remove_member(tenant_id: str, member_id: str, _=Depends(verify_api_key)): """移除成员""" @@ -8994,10 +8590,8 @@ async def remove_member(tenant_id: str, member_id: str, _=Depends(verify_api_key return {"message": "Member removed successfully"} - # Usage & Limits APIs - @app.get("/api/v1/tenants/{tenant_id}/usage", tags=["Tenants"]) async def get_tenant_usage(tenant_id: str, _=Depends(verify_api_key)): """获取租户资源使用统计""" @@ -9009,7 +8603,6 @@ async def get_tenant_usage(tenant_id: str, _=Depends(verify_api_key)): return stats - @app.get("/api/v1/tenants/{tenant_id}/limits/{resource_type}", tags=["Tenants"]) async def check_resource_limit(tenant_id: str, resource_type: str, _=Depends(verify_api_key)): """检查特定资源是否超限""" @@ -9027,10 +8620,8 @@ async def check_resource_limit(tenant_id: str, resource_type: str, _=Depends(ver "usage_percentage": round(current / limit * 100, 2) if limit > 0 else 0, } - # Public tenant resolution API (for custom domains) - @app.get("/api/v1/resolve-tenant", tags=["Tenants"]) async def resolve_tenant_by_domain(domain: str): """通过域名解析租户（用于自定义域名路由）""" @@ -9057,7 +8648,6 @@ async def resolve_tenant_by_domain(domain: str): }, } - @app.get("/api/v1/health", tags=["System"]) async def detailed_health_check(): """健康检查""" @@ -9103,12 +8693,10 @@ async def detailed_health_check(): return health - # ==================== Phase 8: Multi-Tenant SaaS API ==================== # Pydantic Models for Tenant API - class TenantCreate(BaseModel): name: str = Field(..., description="租户名称") slug: str = Field(..., description="URL 友好的唯一标识（小写字母、数字、连字符）") @@ -9119,7 +8707,6 @@ class TenantCreate(BaseModel): ) billing_email: str = Field(default="", description="计费邮箱") - class TenantUpdate(BaseModel): name: str | None = None description: str | None = None @@ -9129,7 +8716,6 @@ class TenantUpdate(BaseModel): max_projects: int | None = None max_members: int | None = None - class TenantResponse(BaseModel): id: str name: str @@ -9145,11 +8731,9 @@ class TenantResponse(BaseModel): created_at: str updated_at: str - class TenantDomainCreate(BaseModel): domain: str = Field(..., description="自定义域名") - class TenantDomainResponse(BaseModel): id: str tenant_id: str @@ -9161,7 +8745,6 @@ class TenantDomainResponse(BaseModel): created_at: str verified_at: str | None - class TenantBrandingUpdate(BaseModel): logo_url: str | None = None logo_dark_url: str | None = None @@ -9182,13 +8765,11 @@ class TenantBrandingUpdate(BaseModel): login_page_description: str | None = None footer_text: str | None = None - class TenantMemberInvite(BaseModel): email: str = Field(..., description="被邀请者邮箱") name: str = Field(default="", description="被邀请者姓名") role: str = Field(default="viewer", description="角色: owner, admin, editor, viewer, guest") - class TenantMemberResponse(BaseModel): id: str tenant_id: str @@ -9203,13 +8784,11 @@ class TenantMemberResponse(BaseModel): last_active_at: str | None created_at: str - class TenantRoleCreate(BaseModel): name: str = Field(..., description="角色名称") description: str = Field(default="", description="角色描述") permissions: list[str] = Field(default_factory=list, description="权限列表") - class TenantRoleResponse(BaseModel): id: str tenant_id: str @@ -9219,7 +8798,6 @@ class TenantRoleResponse(BaseModel): is_system: bool created_at: str - class TenantStatsResponse(BaseModel): tenant_id: str project_count: int @@ -9228,10 +8806,8 @@ class TenantStatsResponse(BaseModel): api_calls_today: int api_calls_month: int - # Tenant API Endpoints - @app.post("/api/v1/tenants", response_model=TenantResponse, tags=["Tenants"]) async def create_tenant_endpoint(tenant: TenantCreate, request: Request, _=Depends(verify_api_key)): """创建新租户""" @@ -9258,7 +8834,6 @@ async def create_tenant_endpoint(tenant: TenantCreate, request: Request, _=Depen except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants", response_model=list[TenantResponse], tags=["Tenants"]) async def list_tenants_endpoint( status: str | None = None, @@ -9284,7 +8859,6 @@ async def list_tenants_endpoint( ) return [t.to_dict() for t in tenants] - @app.get("/api/v1/tenants/{tenant_id}", response_model=TenantResponse, tags=["Tenants"]) async def get_tenant_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取租户详情""" @@ -9299,7 +8873,6 @@ async def get_tenant_endpoint(tenant_id: str, _=Depends(verify_api_key)): return tenant.to_dict() - @app.get("/api/v1/tenants/slug/{slug}", response_model=TenantResponse, tags=["Tenants"]) async def get_tenant_by_slug_endpoint(slug: str, _=Depends(verify_api_key)): """根据 slug 获取租户""" @@ -9314,7 +8887,6 @@ async def get_tenant_by_slug_endpoint(slug: str, _=Depends(verify_api_key)): return tenant.to_dict() - @app.put("/api/v1/tenants/{tenant_id}", response_model=TenantResponse, tags=["Tenants"]) async def update_tenant_endpoint(tenant_id: str, update: TenantUpdate, _=Depends(verify_api_key)): """更新租户信息""" @@ -9334,7 +8906,6 @@ async def update_tenant_endpoint(tenant_id: str, update: TenantUpdate, _=Depends except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.delete("/api/v1/tenants/{tenant_id}", tags=["Tenants"]) async def delete_tenant_endpoint(tenant_id: str, _=Depends(verify_api_key)): """删除租户（标记为过期）""" @@ -9349,10 +8920,8 @@ async def delete_tenant_endpoint(tenant_id: str, _=Depends(verify_api_key)): return {"success": True, "message": f"Tenant {tenant_id} deleted"} - # Tenant Domain API - @app.post( "/api/v1/tenants/{tenant_id}/domains", response_model=TenantDomainResponse, @@ -9380,7 +8949,6 @@ async def add_tenant_domain_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get( "/api/v1/tenants/{tenant_id}/domains", response_model=list[TenantDomainResponse], @@ -9395,7 +8963,6 @@ async def list_tenant_domains_endpoint(tenant_id: str, _=Depends(verify_api_key) domains = tenant_manager.get_tenant_domains(tenant_id) return [d.to_dict() for d in domains] - @app.post("/api/v1/tenants/{tenant_id}/domains/{domain_id}/verify", tags=["Tenants"]) async def verify_tenant_domain_endpoint(tenant_id: str, domain_id: str, _=Depends(verify_api_key)): """验证域名 DNS 记录""" @@ -9410,7 +8977,6 @@ async def verify_tenant_domain_endpoint(tenant_id: str, domain_id: str, _=Depend return {"success": True, "message": "Domain verified successfully"} - @app.post("/api/v1/tenants/{tenant_id}/domains/{domain_id}/activate", tags=["Tenants"]) async def activate_tenant_domain_endpoint( tenant_id: str, @@ -9429,7 +8995,6 @@ async def activate_tenant_domain_endpoint( return {"success": True, "message": "Domain activated successfully"} - @app.delete("/api/v1/tenants/{tenant_id}/domains/{domain_id}", tags=["Tenants"]) async def remove_tenant_domain_endpoint(tenant_id: str, domain_id: str, _=Depends(verify_api_key)): """移除域名绑定""" @@ -9444,10 +9009,8 @@ async def remove_tenant_domain_endpoint(tenant_id: str, domain_id: str, _=Depend return {"success": True, "message": "Domain removed successfully"} - # Tenant Branding API - @app.get("/api/v1/tenants/{tenant_id}/branding", tags=["Tenants"]) async def get_tenant_branding_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取租户品牌配置""" @@ -9462,7 +9025,6 @@ async def get_tenant_branding_endpoint(tenant_id: str, _=Depends(verify_api_key) return branding.to_dict() - @app.put("/api/v1/tenants/{tenant_id}/branding", tags=["Tenants"]) async def update_tenant_branding_endpoint( tenant_id: str, @@ -9484,7 +9046,6 @@ async def update_tenant_branding_endpoint( return updated.to_dict() - @app.get("/api/v1/tenants/{tenant_id}/branding/theme.css", tags=["Tenants"]) async def get_tenant_theme_css_endpoint(tenant_id: str): """获取租户主题 CSS（公开访问）""" @@ -9499,10 +9060,8 @@ async def get_tenant_theme_css_endpoint(tenant_id: str): return PlainTextResponse(content=branding.get_theme_css(), media_type="text/css") - # Tenant Member API - @app.post( "/api/v1/tenants/{tenant_id}/members/invite", response_model=TenantMemberResponse, @@ -9537,7 +9096,6 @@ async def invite_tenant_member_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/tenants/members/accept-invitation", tags=["Tenants"]) async def accept_invitation_endpoint(token: str, user_id: str): """接受邀请加入租户""" @@ -9552,7 +9110,6 @@ async def accept_invitation_endpoint(token: str, user_id: str): return member.to_dict() - @app.get( "/api/v1/tenants/{tenant_id}/members", response_model=list[TenantMemberResponse], @@ -9576,7 +9133,6 @@ async def list_tenant_members_endpoint( members = tenant_manager.list_members(tenant_id, status=status_enum, role=role_enum) return [m.to_dict() for m in members] - @app.put("/api/v1/tenants/{tenant_id}/members/{member_id}/role", tags=["Tenants"]) async def update_member_role_endpoint( tenant_id: str, @@ -9609,7 +9165,6 @@ async def update_member_role_endpoint( except ValueError as e: raise HTTPException(status_code=403, detail=str(e)) - @app.delete("/api/v1/tenants/{tenant_id}/members/{member_id}", tags=["Tenants"]) async def remove_tenant_member_endpoint( tenant_id: str, @@ -9636,10 +9191,8 @@ async def remove_tenant_member_endpoint( except ValueError as e: raise HTTPException(status_code=403, detail=str(e)) - # Tenant Role API - @app.get( "/api/v1/tenants/{tenant_id}/roles", response_model=list[TenantRoleResponse], @@ -9654,7 +9207,6 @@ async def list_tenant_roles_endpoint(tenant_id: str, _=Depends(verify_api_key)): roles = tenant_manager.list_roles(tenant_id) return [r.to_dict() for r in roles] - @app.post("/api/v1/tenants/{tenant_id}/roles", response_model=TenantRoleResponse, tags=["Tenants"]) async def create_tenant_role_endpoint( tenant_id: str, @@ -9678,7 +9230,6 @@ async def create_tenant_role_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.put("/api/v1/tenants/{tenant_id}/roles/{role_id}/permissions", tags=["Tenants"]) async def update_role_permissions_endpoint( tenant_id: str, @@ -9700,7 +9251,6 @@ async def update_role_permissions_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.delete("/api/v1/tenants/{tenant_id}/roles/{role_id}", tags=["Tenants"]) async def delete_tenant_role_endpoint(tenant_id: str, role_id: str, _=Depends(verify_api_key)): """删除自定义角色""" @@ -9717,7 +9267,6 @@ async def delete_tenant_role_endpoint(tenant_id: str, role_id: str, _=Depends(ve except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/permissions", tags=["Tenants"]) async def list_tenant_permissions_endpoint(_=Depends(verify_api_key)): """获取所有可用的租户权限列表""" @@ -9729,10 +9278,8 @@ async def list_tenant_permissions_endpoint(_=Depends(verify_api_key)): "permissions": [{"id": k, "name": v} for k, v in tenant_manager.PERMISSION_NAMES.items()], } - # Tenant Resolution API - @app.get("/api/v1/tenants/resolve", tags=["Tenants"]) async def resolve_tenant_endpoint( host: str | None = None, @@ -9752,7 +9299,6 @@ async def resolve_tenant_endpoint( return tenant.to_dict() - @app.get("/api/v1/tenants/{tenant_id}/context", tags=["Tenants"]) async def get_tenant_context_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取租户完整上下文""" @@ -9767,14 +9313,12 @@ async def get_tenant_context_endpoint(tenant_id: str, _=Depends(verify_api_key)) return context - # ============================================ # Phase 8 Task 2: Subscription & Billing APIs # ============================================ # Pydantic Models for Subscription API - class CreateSubscriptionRequest(BaseModel): plan_id: str = Field(..., description="订阅计划ID") billing_cycle: str = Field(default="monthly", description="计费周期: monthly/yearly") @@ -9784,51 +9328,42 @@ class CreateSubscriptionRequest(BaseModel): ) trial_days: int = Field(default=0, description="试用天数") - class ChangePlanRequest(BaseModel): new_plan_id: str = Field(..., description="新计划ID") prorate: bool = Field(default=True, description="是否按比例计算差价") - class CancelSubscriptionRequest(BaseModel): at_period_end: bool = Field(default=True, description="是否在周期结束时取消") - class CreatePaymentRequest(BaseModel): amount: float = Field(..., description="支付金额") currency: str = Field(default="CNY", description="货币") provider: str = Field(..., description="支付提供商: stripe/alipay/wechat") payment_method: str | None = Field(default=None, description="支付方式") - class RequestRefundRequest(BaseModel): payment_id: str = Field(..., description="支付记录ID") amount: float = Field(..., description="退款金额") reason: str = Field(..., description="退款原因") - class ProcessRefundRequest(BaseModel): action: str = Field(..., description="操作: approve/reject") reason: str | None = Field(default=None, description="拒绝原因（拒绝时必填）") - class RecordUsageRequest(BaseModel): resource_type: str = Field(..., description="资源类型: transcription/storage/api_call/export") quantity: float = Field(..., description="使用量") unit: str = Field(..., description="单位: minutes/mb/count/page") description: str | None = Field(default=None, description="描述") - class CreateCheckoutSessionRequest(BaseModel): plan_id: str = Field(..., description="计划ID") billing_cycle: str = Field(default="monthly", description="计费周期") success_url: str = Field(..., description="支付成功回调URL") cancel_url: str = Field(..., description="支付取消回调URL") - # Subscription Plan APIs - @app.get("/api/v1/subscription-plans", tags=["Subscriptions"]) async def list_subscription_plans( include_inactive: bool = Query(default=False, description="包含已停用计划"), @@ -9859,7 +9394,6 @@ async def list_subscription_plans( ], } - @app.get("/api/v1/subscription-plans/{plan_id}", tags=["Subscriptions"]) async def get_subscription_plan(plan_id: str, _=Depends(verify_api_key)): """获取订阅计划详情""" @@ -9886,10 +9420,8 @@ async def get_subscription_plan(plan_id: str, _=Depends(verify_api_key)): "created_at": plan.created_at.isoformat(), } - # Subscription APIs - @app.post("/api/v1/tenants/{tenant_id}/subscription", tags=["Subscriptions"]) async def create_subscription( tenant_id: str, @@ -9927,7 +9459,6 @@ async def create_subscription( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/subscription", tags=["Subscriptions"]) async def get_tenant_subscription(tenant_id: str, _=Depends(verify_api_key)): """获取租户当前订阅""" @@ -9964,7 +9495,6 @@ async def get_tenant_subscription(tenant_id: str, _=Depends(verify_api_key)): }, } - @app.put("/api/v1/tenants/{tenant_id}/subscription/change-plan", tags=["Subscriptions"]) async def change_subscription_plan( tenant_id: str, @@ -9997,7 +9527,6 @@ async def change_subscription_plan( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/tenants/{tenant_id}/subscription/cancel", tags=["Subscriptions"]) async def cancel_subscription( tenant_id: str, @@ -10030,10 +9559,8 @@ async def cancel_subscription( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - # Usage APIs - @app.post("/api/v1/tenants/{tenant_id}/usage", tags=["Subscriptions"]) async def record_usage(tenant_id: str, request: RecordUsageRequest, _=Depends(verify_api_key)): """记录用量""" @@ -10059,7 +9586,6 @@ async def record_usage(tenant_id: str, request: RecordUsageRequest, _=Depends(ve "recorded_at": record.recorded_at.isoformat(), } - @app.get("/api/v1/tenants/{tenant_id}/usage", tags=["Subscriptions"]) async def get_usage_summary( tenant_id: str, @@ -10080,10 +9606,8 @@ async def get_usage_summary( return summary - # Payment APIs - @app.get("/api/v1/tenants/{tenant_id}/payments", tags=["Subscriptions"]) async def list_payments( tenant_id: str, @@ -10117,7 +9641,6 @@ async def list_payments( "total": len(payments), } - @app.get("/api/v1/tenants/{tenant_id}/payments/{payment_id}", tags=["Subscriptions"]) async def get_payment(tenant_id: str, payment_id: str, _=Depends(verify_api_key)): """获取支付记录详情""" @@ -10147,10 +9670,8 @@ async def get_payment(tenant_id: str, payment_id: str, _=Depends(verify_api_key) "created_at": payment.created_at.isoformat(), } - # Invoice APIs - @app.get("/api/v1/tenants/{tenant_id}/invoices", tags=["Subscriptions"]) async def list_invoices( tenant_id: str, @@ -10187,7 +9708,6 @@ async def list_invoices( "total": len(invoices), } - @app.get("/api/v1/tenants/{tenant_id}/invoices/{invoice_id}", tags=["Subscriptions"]) async def get_invoice(tenant_id: str, invoice_id: str, _=Depends(verify_api_key)): """获取发票详情""" @@ -10218,10 +9738,8 @@ async def get_invoice(tenant_id: str, invoice_id: str, _=Depends(verify_api_key) "created_at": invoice.created_at.isoformat(), } - # Refund APIs - @app.post("/api/v1/tenants/{tenant_id}/refunds", tags=["Subscriptions"]) async def request_refund( tenant_id: str, @@ -10255,7 +9773,6 @@ async def request_refund( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/refunds", tags=["Subscriptions"]) async def list_refunds( tenant_id: str, @@ -10291,7 +9808,6 @@ async def list_refunds( "total": len(refunds), } - @app.post("/api/v1/tenants/{tenant_id}/refunds/{refund_id}/process", tags=["Subscriptions"]) async def process_refund( tenant_id: str, @@ -10333,10 +9849,8 @@ async def process_refund( else: raise HTTPException(status_code=400, detail="Invalid action") - # Billing History API - @app.get("/api/v1/tenants/{tenant_id}/billing-history", tags=["Subscriptions"]) async def get_billing_history( tenant_id: str, @@ -10374,10 +9888,8 @@ async def get_billing_history( "total": len(history), } - # Payment Provider Integration APIs - @app.post("/api/v1/tenants/{tenant_id}/checkout/stripe", tags=["Subscriptions"]) async def create_stripe_checkout( tenant_id: str, @@ -10403,7 +9915,6 @@ async def create_stripe_checkout( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/tenants/{tenant_id}/checkout/alipay", tags=["Subscriptions"]) async def create_alipay_order( tenant_id: str, @@ -10428,7 +9939,6 @@ async def create_alipay_order( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/tenants/{tenant_id}/checkout/wechat", tags=["Subscriptions"]) async def create_wechat_order( tenant_id: str, @@ -10453,10 +9963,8 @@ async def create_wechat_order( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - # Webhook Handlers - @app.post("/webhooks/stripe", tags=["Subscriptions"]) async def stripe_webhook(request: Request): """Stripe Webhook 处理""" @@ -10473,7 +9981,6 @@ async def stripe_webhook(request: Request): else: raise HTTPException(status_code=400, detail="Webhook processing failed") - @app.post("/webhooks/alipay", tags=["Subscriptions"]) async def alipay_webhook(request: Request): """支付宝 Webhook 处理""" @@ -10490,7 +9997,6 @@ async def alipay_webhook(request: Request): else: raise HTTPException(status_code=400, detail="Webhook processing failed") - @app.post("/webhooks/wechat", tags=["Subscriptions"]) async def wechat_webhook(request: Request): """微信支付 Webhook 处理""" @@ -10507,12 +10013,10 @@ async def wechat_webhook(request: Request): else: raise HTTPException(status_code=400, detail="Webhook processing failed") - # ==================== Phase 8: Enterprise Features API ==================== # Pydantic Models for Enterprise - class SSOConfigCreate(BaseModel): provider: str = Field( ..., @@ -10535,7 +10039,6 @@ class SSOConfigCreate(BaseModel): default_role: str = Field(default="member", description="默认角色") domain_restriction: list[str] = Field(default_factory=list, description="允许的邮箱域名") - class SSOConfigUpdate(BaseModel): entity_id: str | None = None sso_url: str | None = None @@ -10555,7 +10058,6 @@ class SSOConfigUpdate(BaseModel): domain_restriction: list[str] | None = None status: str | None = None - class SCIMConfigCreate(BaseModel): provider: str = Field(..., description="身份提供商") scim_base_url: str = Field(..., description="SCIM 服务端地址") @@ -10564,7 +10066,6 @@ class SCIMConfigCreate(BaseModel): attribute_mapping: dict[str, str] | None = Field(default=None, description="属性映射") sync_rules: dict[str, Any] | None = Field(default=None, description="同步规则") - class SCIMConfigUpdate(BaseModel): scim_base_url: str | None = None scim_token: str | None = None @@ -10573,7 +10074,6 @@ class SCIMConfigUpdate(BaseModel): sync_rules: dict[str, Any] | None = None status: str | None = None - class AuditExportCreate(BaseModel): export_format: str = Field(..., description="导出格式: json/csv/pdf/xlsx") start_date: str = Field(..., description="开始日期 (ISO 格式)") @@ -10584,7 +10084,6 @@ class AuditExportCreate(BaseModel): description="合规标准: soc2/iso27001/gdpr/hipaa/pci_dss", ) - class RetentionPolicyCreate(BaseModel): name: str = Field(..., description="策略名称") description: str | None = Field(default=None, description="策略描述") @@ -10601,7 +10100,6 @@ class RetentionPolicyCreate(BaseModel): archive_location: str | None = Field(default=None, description="归档位置") archive_encryption: bool = Field(default=True, description="归档加密") - class RetentionPolicyUpdate(BaseModel): name: str | None = None description: str | None = None @@ -10615,10 +10113,8 @@ class RetentionPolicyUpdate(BaseModel): archive_encryption: bool | None = None is_active: bool | None = None - # SSO/SAML APIs - @app.post("/api/v1/tenants/{tenant_id}/sso-configs", tags=["Enterprise"]) async def create_sso_config_endpoint( tenant_id: str, @@ -10669,7 +10165,6 @@ async def create_sso_config_endpoint( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/sso-configs", tags=["Enterprise"]) async def list_sso_configs_endpoint(tenant_id: str, _=Depends(verify_api_key)): """列出租户的所有 SSO 配置""" @@ -10697,7 +10192,6 @@ async def list_sso_configs_endpoint(tenant_id: str, _=Depends(verify_api_key)): "total": len(configs), } - @app.get("/api/v1/tenants/{tenant_id}/sso-configs/{config_id}", tags=["Enterprise"]) async def get_sso_config_endpoint(tenant_id: str, config_id: str, _=Depends(verify_api_key)): """获取 SSO 配置详情""" @@ -10731,7 +10225,6 @@ async def get_sso_config_endpoint(tenant_id: str, config_id: str, _=Depends(veri "updated_at": config.updated_at.isoformat(), } - @app.put("/api/v1/tenants/{tenant_id}/sso-configs/{config_id}", tags=["Enterprise"]) async def update_sso_config_endpoint( tenant_id: str, @@ -10760,7 +10253,6 @@ async def update_sso_config_endpoint( "updated_at": updated.updated_at.isoformat(), } - @app.delete("/api/v1/tenants/{tenant_id}/sso-configs/{config_id}", tags=["Enterprise"]) async def delete_sso_config_endpoint(tenant_id: str, config_id: str, _=Depends(verify_api_key)): """删除 SSO 配置""" @@ -10776,7 +10268,6 @@ async def delete_sso_config_endpoint(tenant_id: str, config_id: str, _=Depends(v manager.delete_sso_config(config_id) return {"success": True} - @app.get("/api/v1/tenants/{tenant_id}/sso-configs/{config_id}/metadata", tags=["Enterprise"]) async def get_sso_metadata_endpoint( tenant_id: str, @@ -10803,10 +10294,8 @@ async def get_sso_metadata_endpoint( "slo_url": f"{base_url}/api/v1/sso/saml/{tenant_id}/slo", } - # SCIM APIs - @app.post("/api/v1/tenants/{tenant_id}/scim-configs", tags=["Enterprise"]) async def create_scim_config_endpoint( tenant_id: str, @@ -10842,7 +10331,6 @@ async def create_scim_config_endpoint( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/scim-configs", tags=["Enterprise"]) async def get_scim_config_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取租户的 SCIM 配置""" @@ -10868,7 +10356,6 @@ async def get_scim_config_endpoint(tenant_id: str, _=Depends(verify_api_key)): "created_at": config.created_at.isoformat(), } - @app.put("/api/v1/tenants/{tenant_id}/scim-configs/{config_id}", tags=["Enterprise"]) async def update_scim_config_endpoint( tenant_id: str, @@ -10897,7 +10384,6 @@ async def update_scim_config_endpoint( "updated_at": updated.updated_at.isoformat(), } - @app.post("/api/v1/tenants/{tenant_id}/scim-configs/{config_id}/sync", tags=["Enterprise"]) async def sync_scim_users_endpoint(tenant_id: str, config_id: str, _=Depends(verify_api_key)): """执行 SCIM 用户同步""" @@ -10914,7 +10400,6 @@ async def sync_scim_users_endpoint(tenant_id: str, config_id: str, _=Depends(ver return result - @app.get("/api/v1/tenants/{tenant_id}/scim-users", tags=["Enterprise"]) async def list_scim_users_endpoint( tenant_id: str, @@ -10945,10 +10430,8 @@ async def list_scim_users_endpoint( "total": len(users), } - # Audit Log Export APIs - @app.post("/api/v1/tenants/{tenant_id}/audit-exports", tags=["Enterprise"]) async def create_audit_export_endpoint( tenant_id: str, @@ -10990,7 +10473,6 @@ async def create_audit_export_endpoint( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/audit-exports", tags=["Enterprise"]) async def list_audit_exports_endpoint( tenant_id: str, @@ -11024,7 +10506,6 @@ async def list_audit_exports_endpoint( "total": len(exports), } - @app.get("/api/v1/tenants/{tenant_id}/audit-exports/{export_id}", tags=["Enterprise"]) async def get_audit_export_endpoint(tenant_id: str, export_id: str, _=Depends(verify_api_key)): """获取审计日志导出详情""" @@ -11056,7 +10537,6 @@ async def get_audit_export_endpoint(tenant_id: str, export_id: str, _=Depends(ve "error_message": export.error_message, } - @app.post("/api/v1/tenants/{tenant_id}/audit-exports/{export_id}/download", tags=["Enterprise"]) async def download_audit_export_endpoint( tenant_id: str, @@ -11086,10 +10566,8 @@ async def download_audit_export_endpoint( "expires_at": export.expires_at.isoformat() if export.expires_at else None, } - # Data Retention Policy APIs - @app.post("/api/v1/tenants/{tenant_id}/retention-policies", tags=["Enterprise"]) async def create_retention_policy_endpoint( tenant_id: str, @@ -11132,7 +10610,6 @@ async def create_retention_policy_endpoint( except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/retention-policies", tags=["Enterprise"]) async def list_retention_policies_endpoint( tenant_id: str, @@ -11163,7 +10640,6 @@ async def list_retention_policies_endpoint( "total": len(policies), } - @app.get("/api/v1/tenants/{tenant_id}/retention-policies/{policy_id}", tags=["Enterprise"]) async def get_retention_policy_endpoint(tenant_id: str, policy_id: str, _=Depends(verify_api_key)): """获取数据保留策略详情""" @@ -11198,7 +10674,6 @@ async def get_retention_policy_endpoint(tenant_id: str, policy_id: str, _=Depend "created_at": policy.created_at.isoformat(), } - @app.put("/api/v1/tenants/{tenant_id}/retention-policies/{policy_id}", tags=["Enterprise"]) async def update_retention_policy_endpoint( tenant_id: str, @@ -11223,7 +10698,6 @@ async def update_retention_policy_endpoint( return {"id": updated.id, "updated_at": updated.updated_at.isoformat()} - @app.delete("/api/v1/tenants/{tenant_id}/retention-policies/{policy_id}", tags=["Enterprise"]) async def delete_retention_policy_endpoint( tenant_id: str, @@ -11243,7 +10717,6 @@ async def delete_retention_policy_endpoint( manager.delete_retention_policy(policy_id) return {"success": True} - @app.post("/api/v1/tenants/{tenant_id}/retention-policies/{policy_id}/execute", tags=["Enterprise"]) async def execute_retention_policy_endpoint( tenant_id: str, @@ -11270,7 +10743,6 @@ async def execute_retention_policy_endpoint( "created_at": job.created_at.isoformat(), } - @app.get("/api/v1/tenants/{tenant_id}/retention-policies/{policy_id}/jobs", tags=["Enterprise"]) async def list_retention_jobs_endpoint( tenant_id: str, @@ -11307,26 +10779,22 @@ async def list_retention_jobs_endpoint( "total": len(jobs), } - # ============================================ # Phase 8 Task 7: Globalization & Localization API # ============================================ # Pydantic Models for Localization API - class TranslationCreate(BaseModel): key: str = Field(..., description="翻译键") value: str = Field(..., description="翻译值") namespace: str = Field(default="common", description="命名空间") context: str | None = Field(default=None, description="上下文说明") - class TranslationUpdate(BaseModel): value: str = Field(..., description="翻译值") context: str | None = Field(default=None, description="上下文说明") - class LocalizationSettingsCreate(BaseModel): default_language: str = Field(default="en", description="默认语言") supported_languages: list[str] = Field(default=["en"], description="支持的语言列表") @@ -11336,7 +10804,6 @@ class LocalizationSettingsCreate(BaseModel): region_code: str = Field(default="global", description="区域代码") data_residency: str = Field(default="regional", description="数据驻留策略") - class LocalizationSettingsUpdate(BaseModel): default_language: str | None = None supported_languages: list[str] | None = None @@ -11346,37 +10813,30 @@ class LocalizationSettingsUpdate(BaseModel): region_code: str | None = None data_residency: str | None = None - class DataCenterMappingRequest(BaseModel): region_code: str = Field(..., description="区域代码") data_residency: str = Field(default="regional", description="数据驻留策略") - class FormatDateTimeRequest(BaseModel): timestamp: str = Field(..., description="ISO格式时间戳") timezone: str | None = Field(default=None, description="目标时区") format_type: str = Field(default="datetime", description="格式类型: date/time/datetime") - class FormatNumberRequest(BaseModel): number: float = Field(..., description="数字") decimal_places: int | None = Field(default=None, description="小数位数") - class FormatCurrencyRequest(BaseModel): amount: float = Field(..., description="金额") currency: str = Field(..., description="货币代码") - class ConvertTimezoneRequest(BaseModel): timestamp: str = Field(..., description="ISO格式时间戳") from_tz: str = Field(..., description="源时区") to_tz: str = Field(..., description="目标时区") - # Translation APIs - @app.get("/api/v1/translations/{language}/{key}", tags=["Localization"]) async def get_translation( language: str, @@ -11396,7 +10856,6 @@ async def get_translation( return {"key": key, "language": language, "namespace": namespace, "value": value} - @app.post("/api/v1/translations/{language}", tags=["Localization"]) async def create_translation(language: str, request: TranslationCreate, _=Depends(verify_api_key)): """创建/更新翻译""" @@ -11421,7 +10880,6 @@ async def create_translation(language: str, request: TranslationCreate, _=Depend "created_at": translation.created_at.isoformat(), } - @app.put("/api/v1/translations/{language}/{key}", tags=["Localization"]) async def update_translation( language: str, @@ -11452,7 +10910,6 @@ async def update_translation( "updated_at": translation.updated_at.isoformat(), } - @app.delete("/api/v1/translations/{language}/{key}", tags=["Localization"]) async def delete_translation( language: str, @@ -11472,7 +10929,6 @@ async def delete_translation( return {"success": True, "message": "Translation deleted"} - @app.get("/api/v1/translations", tags=["Localization"]) async def list_translations( language: str | None = Query(default=None, description="语言代码"), @@ -11504,10 +10960,8 @@ async def list_translations( "total": len(translations), } - # Language APIs - @app.get("/api/v1/languages", tags=["Localization"]) async def list_languages(active_only: bool = Query(default=True, description="仅返回激活的语言")): """列出支持的语言""" @@ -11535,7 +10989,6 @@ async def list_languages(active_only: bool = Query(default=True, description=" "total": len(languages), } - @app.get("/api/v1/languages/{code}", tags=["Localization"]) async def get_language(code: str): """获取语言详情""" @@ -11565,10 +11018,8 @@ async def get_language(code: str): "calendar_type": lang.calendar_type, } - # Data Center APIs - @app.get("/api/v1/data-centers", tags=["Localization"]) async def list_data_centers( status: str | None = Query(default=None, description="状态过滤"), @@ -11598,7 +11049,6 @@ async def list_data_centers( "total": len(data_centers), } - @app.get("/api/v1/data-centers/{dc_id}", tags=["Localization"]) async def get_data_center(dc_id: str): """获取数据中心详情""" @@ -11623,7 +11073,6 @@ async def get_data_center(dc_id: str): "capabilities": dc.capabilities, } - @app.get("/api/v1/tenants/{tenant_id}/data-center", tags=["Localization"]) async def get_tenant_data_center(tenant_id: str, _=Depends(verify_api_key)): """获取租户数据中心配置""" @@ -11670,7 +11119,6 @@ async def get_tenant_data_center(tenant_id: str, _=Depends(verify_api_key)): "created_at": mapping.created_at.isoformat(), } - @app.post("/api/v1/tenants/{tenant_id}/data-center", tags=["Localization"]) async def set_tenant_data_center( tenant_id: str, @@ -11696,10 +11144,8 @@ async def set_tenant_data_center( "created_at": mapping.created_at.isoformat(), } - # Payment Method APIs - @app.get("/api/v1/payment-methods", tags=["Localization"]) async def list_payment_methods( country_code: str | None = Query(default=None, description="国家代码"), @@ -11732,7 +11178,6 @@ async def list_payment_methods( "total": len(methods), } - @app.get("/api/v1/payment-methods/localized", tags=["Localization"]) async def get_localized_payment_methods( country_code: str = Query(..., description="国家代码"), @@ -11747,10 +11192,8 @@ async def get_localized_payment_methods( return {"country_code": country_code, "language": language, "payment_methods": methods} - # Country APIs - @app.get("/api/v1/countries", tags=["Localization"]) async def list_countries( region: str | None = Query(default=None, description="区域过滤"), @@ -11781,7 +11224,6 @@ async def list_countries( "total": len(countries), } - @app.get("/api/v1/countries/{code}", tags=["Localization"]) async def get_country(code: str): """获取国家详情""" @@ -11809,10 +11251,8 @@ async def get_country(code: str): "vat_rate": country.vat_rate, } - # Localization Settings APIs - @app.get("/api/v1/tenants/{tenant_id}/localization", tags=["Localization"]) async def get_localization_settings(tenant_id: str, _=Depends(verify_api_key)): """获取租户本地化设置""" @@ -11842,7 +11282,6 @@ async def get_localization_settings(tenant_id: str, _=Depends(verify_api_key)): "updated_at": settings.updated_at.isoformat(), } - @app.post("/api/v1/tenants/{tenant_id}/localization", tags=["Localization"]) async def create_localization_settings( tenant_id: str, @@ -11878,7 +11317,6 @@ async def create_localization_settings( "created_at": settings.created_at.isoformat(), } - @app.put("/api/v1/tenants/{tenant_id}/localization", tags=["Localization"]) async def update_localization_settings( tenant_id: str, @@ -11910,10 +11348,8 @@ async def update_localization_settings( "updated_at": settings.updated_at.isoformat(), } - # Formatting APIs - @app.post("/api/v1/format/datetime", tags=["Localization"]) async def format_datetime_endpoint( request: FormatDateTimeRequest, @@ -11945,7 +11381,6 @@ async def format_datetime_endpoint( "format_type": request.format_type, } - @app.post("/api/v1/format/number", tags=["Localization"]) async def format_number_endpoint( request: FormatNumberRequest, @@ -11964,7 +11399,6 @@ async def format_number_endpoint( return {"original": request.number, "formatted": formatted, "language": language} - @app.post("/api/v1/format/currency", tags=["Localization"]) async def format_currency_endpoint( request: FormatCurrencyRequest, @@ -11988,7 +11422,6 @@ async def format_currency_endpoint( "language": language, } - @app.post("/api/v1/convert/timezone", tags=["Localization"]) async def convert_timezone_endpoint(request: ConvertTimezoneRequest): """转换时区""" @@ -12011,7 +11444,6 @@ async def convert_timezone_endpoint(request: ConvertTimezoneRequest): "converted": converted.isoformat(), } - @app.get("/api/v1/detect/locale", tags=["Localization"]) async def detect_locale( accept_language: str | None = Header(default=None, description="Accept-Language 头"), @@ -12029,7 +11461,6 @@ async def detect_locale( return preferences - @app.get("/api/v1/calendar/{calendar_type}", tags=["Localization"]) async def get_calendar_info( calendar_type: str, @@ -12045,12 +11476,10 @@ async def get_calendar_info( return info - # ============================================ # Phase 8 Task 4: AI 能力增强 API # ============================================ - class CreateCustomModelRequest(BaseModel): name: str description: str @@ -12058,29 +11487,24 @@ class CreateCustomModelRequest(BaseModel): training_data: dict hyperparameters: dict = Field(default_factory=lambda: {"epochs": 10, "learning_rate": 0.001}) - class AddTrainingSampleRequest(BaseModel): text: str entities: list[dict] metadata: dict = Field(default_factory=dict) - class TrainModelRequest(BaseModel): model_id: str - class PredictRequest(BaseModel): model_id: str text: str - class MultimodalAnalysisRequest(BaseModel): provider: str input_type: str input_urls: list[str] prompt: str - class CreateKGRAGRequest(BaseModel): name: str description: str @@ -12088,19 +11512,16 @@ class CreateKGRAGRequest(BaseModel): retrieval_config: dict generation_config: dict - class KGRAGQueryRequest(BaseModel): rag_id: str query: str - class SmartSummaryRequest(BaseModel): source_type: str source_id: str summary_type: str content_data: dict - class CreatePredictionModelRequest(BaseModel): name: str prediction_type: str @@ -12108,21 +11529,17 @@ class CreatePredictionModelRequest(BaseModel): features: list[str] model_config: dict - class PredictDataRequest(BaseModel): model_id: str input_data: dict - class PredictionFeedbackRequest(BaseModel): prediction_id: str actual_value: str is_correct: bool - # 自定义模型管理 API - @app.post("/api/v1/tenants/{tenant_id}/ai/custom-models", tags=["AI Enhancement"]) async def create_custom_model( tenant_id: str, @@ -12155,7 +11572,6 @@ async def create_custom_model( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/ai/custom-models", tags=["AI Enhancement"]) async def list_custom_models( tenant_id: str, @@ -12187,7 +11603,6 @@ async def list_custom_models( ], } - @app.get("/api/v1/ai/custom-models/{model_id}", tags=["AI Enhancement"]) async def get_custom_model(model_id: str): """获取自定义模型详情""" @@ -12216,7 +11631,6 @@ async def get_custom_model(model_id: str): "created_by": model.created_by, } - @app.post("/api/v1/ai/custom-models/{model_id}/samples", tags=["AI Enhancement"]) async def add_training_sample(model_id: str, request: AddTrainingSampleRequest): """添加训练样本""" @@ -12240,7 +11654,6 @@ async def add_training_sample(model_id: str, request: AddTrainingSampleRequest): "created_at": sample.created_at, } - @app.get("/api/v1/ai/custom-models/{model_id}/samples", tags=["AI Enhancement"]) async def get_training_samples(model_id: str): """获取训练样本""" @@ -12263,7 +11676,6 @@ async def get_training_samples(model_id: str): ], } - @app.post("/api/v1/ai/custom-models/{model_id}/train", tags=["AI Enhancement"]) async def train_custom_model(model_id: str): """训练自定义模型""" @@ -12283,7 +11695,6 @@ async def train_custom_model(model_id: str): except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/ai/custom-models/predict", tags=["AI Enhancement"]) async def predict_with_custom_model(request: PredictRequest): """使用自定义模型预测""" @@ -12298,10 +11709,8 @@ async def predict_with_custom_model(request: PredictRequest): except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - # 多模态分析 API - @app.post( "/api/v1/tenants/{tenant_id}/projects/{project_id}/ai/multimodal", tags=["AI Enhancement"], @@ -12335,7 +11744,6 @@ async def analyze_multimodal(tenant_id: str, project_id: str, request: Multimoda except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/ai/multimodal", tags=["AI Enhancement"]) async def list_multimodal_analyses( tenant_id: str, @@ -12365,10 +11773,8 @@ async def list_multimodal_analyses( ], } - # 知识图谱 RAG API - @app.post("/api/v1/tenants/{tenant_id}/projects/{project_id}/ai/kg-rag", tags=["AI Enhancement"]) async def create_kg_rag(tenant_id: str, project_id: str, request: CreateKGRAGRequest): """创建知识图谱 RAG 配置""" @@ -12395,7 +11801,6 @@ async def create_kg_rag(tenant_id: str, project_id: str, request: CreateKGRAGReq "created_at": rag.created_at, } - @app.get("/api/v1/tenants/{tenant_id}/ai/kg-rag", tags=["AI Enhancement"]) async def list_kg_rags( tenant_id: str, @@ -12422,7 +11827,6 @@ async def list_kg_rags( ], } - @app.post("/api/v1/ai/kg-rag/query", tags=["AI Enhancement"]) async def query_kg_rag( request: KGRAGQueryRequest, @@ -12457,10 +11861,8 @@ async def query_kg_rag( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - # 智能摘要 API - @app.post("/api/v1/tenants/{tenant_id}/projects/{project_id}/ai/summarize", tags=["AI Enhancement"]) async def generate_smart_summary(tenant_id: str, project_id: str, request: SmartSummaryRequest): """生成智能摘要""" @@ -12491,7 +11893,6 @@ async def generate_smart_summary(tenant_id: str, project_id: str, request: Smart "created_at": summary.created_at, } - @app.get("/api/v1/tenants/{tenant_id}/projects/{project_id}/ai/summaries", tags=["AI Enhancement"]) async def list_smart_summaries( tenant_id: str, @@ -12508,10 +11909,8 @@ async def list_smart_summaries( # 这里需要从数据库查询，暂时返回空列表 return {"summaries": []} - # 预测模型 API - @app.post( "/api/v1/tenants/{tenant_id}/projects/{project_id}/ai/prediction-models", tags=["AI Enhancement"], @@ -12550,7 +11949,6 @@ async def create_prediction_model( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/tenants/{tenant_id}/ai/prediction-models", tags=["AI Enhancement"]) async def list_prediction_models( tenant_id: str, @@ -12581,7 +11979,6 @@ async def list_prediction_models( ], } - @app.get("/api/v1/ai/prediction-models/{model_id}", tags=["AI Enhancement"]) async def get_prediction_model(model_id: str): """获取预测模型详情""" @@ -12610,7 +12007,6 @@ async def get_prediction_model(model_id: str): "created_at": model.created_at, } - @app.post("/api/v1/ai/prediction-models/{model_id}/train", tags=["AI Enhancement"]) async def train_prediction_model( model_id: str, @@ -12632,7 +12028,6 @@ async def train_prediction_model( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/ai/prediction-models/predict", tags=["AI Enhancement"]) async def predict(request: PredictDataRequest): """进行预测""" @@ -12657,7 +12052,6 @@ async def predict(request: PredictDataRequest): except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ai/prediction-models/{model_id}/results", tags=["AI Enhancement"]) async def get_prediction_results( model_id: str, @@ -12687,7 +12081,6 @@ async def get_prediction_results( ], } - @app.post("/api/v1/ai/prediction-results/feedback", tags=["AI Enhancement"]) async def update_prediction_feedback(request: PredictionFeedbackRequest): """更新预测反馈""" @@ -12703,12 +12096,10 @@ async def update_prediction_feedback(request: PredictionFeedbackRequest): return {"status": "success", "message": "Feedback updated"} - # ==================== Phase 8 Task 5: Growth & Analytics Endpoints ==================== # Pydantic Models for Growth API - class TrackEventRequest(BaseModel): tenant_id: str user_id: str @@ -12722,13 +12113,11 @@ class TrackEventRequest(BaseModel): utm_medium: str | None = None utm_campaign: str | None = None - class CreateFunnelRequest(BaseModel): name: str description: str = "" steps: list[dict] # [{"name": "", "event_name": ""}] - class CreateExperimentRequest(BaseModel): name: str description: str = "" @@ -12742,19 +12131,16 @@ class CreateExperimentRequest(BaseModel): min_sample_size: int = 100 confidence_level: float = 0.95 - class AssignVariantRequest(BaseModel): user_id: str user_attributes: dict = Field(default_factory=dict) - class RecordMetricRequest(BaseModel): variant_id: str user_id: str metric_name: str metric_value: float - class CreateEmailTemplateRequest(BaseModel): name: str template_type: str # welcome, onboarding, feature_announcement, churn_recovery, etc. @@ -12766,14 +12152,12 @@ class CreateEmailTemplateRequest(BaseModel): from_email: str = "noreply@insightflow.io" reply_to: str | None = None - class CreateCampaignRequest(BaseModel): name: str template_id: str recipients: list[dict] # [{"user_id": "", "email": ""}] scheduled_at: str | None = None - class CreateAutomationWorkflowRequest(BaseModel): name: str description: str = "" @@ -12781,7 +12165,6 @@ class CreateAutomationWorkflowRequest(BaseModel): trigger_conditions: dict = Field(default_factory=dict) actions: list[dict] # [{"type": "send_email", "template_id": ""}] - class CreateReferralProgramRequest(BaseModel): name: str description: str = "" @@ -12793,12 +12176,10 @@ class CreateReferralProgramRequest(BaseModel): referral_code_length: int = 8 expiry_days: int = 30 - class ApplyReferralCodeRequest(BaseModel): referral_code: str referee_id: str - class CreateTeamIncentiveRequest(BaseModel): name: str description: str = "" @@ -12809,21 +12190,17 @@ class CreateTeamIncentiveRequest(BaseModel): valid_from: str valid_until: str - # Growth Manager singleton _growth_manager: "GrowthManager | None" = None - def get_growth_manager_instance() -> "GrowthManager | None": global _growth_manager if _growth_manager is None and GROWTH_MANAGER_AVAILABLE: _growth_manager = GrowthManager() return _growth_manager - # ==================== 用户行为分析 API ==================== - @app.post("/api/v1/analytics/track", tags=["Growth & Analytics"]) async def track_event_endpoint(request: TrackEventRequest): """ @@ -12861,7 +12238,6 @@ async def track_event_endpoint(request: TrackEventRequest): except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=500, detail=str(e)) - @app.get("/api/v1/analytics/dashboard/{tenant_id}", tags=["Growth & Analytics"]) async def get_analytics_dashboard(tenant_id: str): """获取实时分析仪表板数据""" @@ -12873,7 +12249,6 @@ async def get_analytics_dashboard(tenant_id: str): return dashboard - @app.get("/api/v1/analytics/summary/{tenant_id}", tags=["Growth & Analytics"]) async def get_analytics_summary( tenant_id: str, @@ -12893,7 +12268,6 @@ async def get_analytics_summary( return summary - @app.get("/api/v1/analytics/user-profile/{tenant_id}/{user_id}", tags=["Growth & Analytics"]) async def get_user_profile(tenant_id: str, user_id: str): """获取用户画像""" @@ -12919,10 +12293,8 @@ async def get_user_profile(tenant_id: str, user_id: str): "engagement_score": profile.engagement_score, } - # ==================== 转化漏斗 API ==================== - @app.post("/api/v1/analytics/funnels", tags=["Growth & Analytics"]) async def create_funnel_endpoint(request: CreateFunnelRequest, created_by: str = "system"): """创建转化漏斗""" @@ -12949,7 +12321,6 @@ async def create_funnel_endpoint(request: CreateFunnelRequest, created_by: str = "created_at": funnel.created_at, } - @app.get("/api/v1/analytics/funnels/{funnel_id}/analyze", tags=["Growth & Analytics"]) async def analyze_funnel_endpoint( funnel_id: str, @@ -12980,7 +12351,6 @@ async def analyze_funnel_endpoint( "drop_off_points": analysis.drop_off_points, } - @app.get("/api/v1/analytics/retention/{tenant_id}", tags=["Growth & Analytics"]) async def calculate_retention( tenant_id: str, @@ -13000,10 +12370,8 @@ async def calculate_retention( return retention - # ==================== A/B 测试 API ==================== - @app.post("/api/v1/experiments", tags=["Growth & Analytics"]) async def create_experiment_endpoint(request: CreateExperimentRequest, created_by: str = "system"): """创建 A/B 测试实验""" @@ -13041,7 +12409,6 @@ async def create_experiment_endpoint(request: CreateExperimentRequest, created_b except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/experiments", tags=["Growth & Analytics"]) async def list_experiments(status: str | None = None): """列出实验""" @@ -13069,7 +12436,6 @@ async def list_experiments(status: str | None = None): ], } - @app.get("/api/v1/experiments/{experiment_id}", tags=["Growth & Analytics"]) async def get_experiment_endpoint(experiment_id: str): """获取实验详情""" @@ -13096,7 +12462,6 @@ async def get_experiment_endpoint(experiment_id: str): "end_date": experiment.end_date.isoformat() if experiment.end_date else None, } - @app.post("/api/v1/experiments/{experiment_id}/assign", tags=["Growth & Analytics"]) async def assign_variant_endpoint(experiment_id: str, request: AssignVariantRequest): """为用户分配实验变体""" @@ -13116,7 +12481,6 @@ async def assign_variant_endpoint(experiment_id: str, request: AssignVariantRequ return {"experiment_id": experiment_id, "user_id": request.user_id, "variant_id": variant_id} - @app.post("/api/v1/experiments/{experiment_id}/metrics", tags=["Growth & Analytics"]) async def record_experiment_metric_endpoint(experiment_id: str, request: RecordMetricRequest): """记录实验指标""" @@ -13135,7 +12499,6 @@ async def record_experiment_metric_endpoint(experiment_id: str, request: RecordM return {"success": True} - @app.get("/api/v1/experiments/{experiment_id}/analyze", tags=["Growth & Analytics"]) async def analyze_experiment_endpoint(experiment_id: str): """分析实验结果""" @@ -13151,7 +12514,6 @@ async def analyze_experiment_endpoint(experiment_id: str): return result - @app.post("/api/v1/experiments/{experiment_id}/start", tags=["Growth & Analytics"]) async def start_experiment_endpoint(experiment_id: str): """启动实验""" @@ -13171,7 +12533,6 @@ async def start_experiment_endpoint(experiment_id: str): "start_date": experiment.start_date.isoformat() if experiment.start_date else None, } - @app.post("/api/v1/experiments/{experiment_id}/stop", tags=["Growth & Analytics"]) async def stop_experiment_endpoint(experiment_id: str): """停止实验""" @@ -13191,10 +12552,8 @@ async def stop_experiment_endpoint(experiment_id: str): "end_date": experiment.end_date.isoformat() if experiment.end_date else None, } - # ==================== 邮件营销 API ==================== - @app.post("/api/v1/email/templates", tags=["Growth & Analytics"]) async def create_email_template_endpoint(request: CreateEmailTemplateRequest): """创建邮件模板""" @@ -13229,7 +12588,6 @@ async def create_email_template_endpoint(request: CreateEmailTemplateRequest): except (RuntimeError, ValueError, TypeError) as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/email/templates", tags=["Growth & Analytics"]) async def list_email_templates(template_type: str | None = None): """列出邮件模板""" @@ -13256,7 +12614,6 @@ async def list_email_templates(template_type: str | None = None): ], } - @app.get("/api/v1/email/templates/{template_id}", tags=["Growth & Analytics"]) async def get_email_template_endpoint(template_id: str): """获取邮件模板详情""" @@ -13281,7 +12638,6 @@ async def get_email_template_endpoint(template_id: str): "from_email": template.from_email, } - @app.post("/api/v1/email/templates/{template_id}/render", tags=["Growth & Analytics"]) async def render_template_endpoint(template_id: str, variables: dict): """渲染邮件模板""" @@ -13297,7 +12653,6 @@ async def render_template_endpoint(template_id: str, variables: dict): return rendered - @app.post("/api/v1/email/campaigns", tags=["Growth & Analytics"]) async def create_email_campaign_endpoint(request: CreateCampaignRequest): """创建邮件营销活动""" @@ -13326,7 +12681,6 @@ async def create_email_campaign_endpoint(request: CreateCampaignRequest): "scheduled_at": campaign.scheduled_at, } - @app.post("/api/v1/email/campaigns/{campaign_id}/send", tags=["Growth & Analytics"]) async def send_campaign_endpoint(campaign_id: str): """发送邮件营销活动""" @@ -13342,7 +12696,6 @@ async def send_campaign_endpoint(campaign_id: str): return result - @app.post("/api/v1/email/workflows", tags=["Growth & Analytics"]) async def create_automation_workflow_endpoint(request: CreateAutomationWorkflowRequest): """创建自动化工作流""" @@ -13369,10 +12722,8 @@ async def create_automation_workflow_endpoint(request: CreateAutomationWorkflowR "created_at": workflow.created_at, } - # ==================== 推荐系统 API ==================== - @app.post("/api/v1/referral/programs", tags=["Growth & Analytics"]) async def create_referral_program_endpoint(request: CreateReferralProgramRequest): """创建推荐计划""" @@ -13405,7 +12756,6 @@ async def create_referral_program_endpoint(request: CreateReferralProgramRequest "is_active": program.is_active, } - @app.post("/api/v1/referral/programs/{program_id}/generate-code", tags=["Growth & Analytics"]) async def generate_referral_code_endpoint(program_id: str, referrer_id: str): """生成推荐码""" @@ -13427,7 +12777,6 @@ async def generate_referral_code_endpoint(program_id: str, referrer_id: str): "expires_at": referral.expires_at.isoformat(), } - @app.post("/api/v1/referral/apply", tags=["Growth & Analytics"]) async def apply_referral_code_endpoint(request: ApplyReferralCodeRequest): """应用推荐码""" @@ -13443,7 +12792,6 @@ async def apply_referral_code_endpoint(request: ApplyReferralCodeRequest): return {"success": True, "message": "Referral code applied successfully"} - @app.get("/api/v1/referral/programs/{program_id}/stats", tags=["Growth & Analytics"]) async def get_referral_stats_endpoint(program_id: str): """获取推荐统计""" @@ -13456,7 +12804,6 @@ async def get_referral_stats_endpoint(program_id: str): return stats - @app.post("/api/v1/team-incentives", tags=["Growth & Analytics"]) async def create_team_incentive_endpoint(request: CreateTeamIncentiveRequest): """创建团队升级激励""" @@ -13489,7 +12836,6 @@ async def create_team_incentive_endpoint(request: CreateTeamIncentiveRequest): "valid_until": incentive.valid_until.isoformat(), } - @app.get("/api/v1/team-incentives/check", tags=["Growth & Analytics"]) async def check_team_incentive_eligibility(tenant_id: str, current_tier: str, team_size: int): """检查团队激励资格""" @@ -13512,7 +12858,6 @@ async def check_team_incentive_eligibility(tenant_id: str, current_tier: str, te ], } - # Serve frontend - MUST be last to not override API routes # ============================================ @@ -13538,7 +12883,6 @@ except ImportError as e: # Pydantic Models for Developer Ecosystem API - class SDKReleaseCreate(BaseModel): name: str language: str @@ -13554,7 +12898,6 @@ class SDKReleaseCreate(BaseModel): file_size: int = 0 checksum: str = "" - class SDKReleaseUpdate(BaseModel): name: str | None = None description: str | None = None @@ -13564,7 +12907,6 @@ class SDKReleaseUpdate(BaseModel): repository_url: str | None = None status: str | None = None - class SDKVersionCreate(BaseModel): version: str is_lts: bool = False @@ -13573,7 +12915,6 @@ class SDKVersionCreate(BaseModel): checksum: str = "" file_size: int = 0 - class TemplateCreate(BaseModel): name: str description: str @@ -13591,13 +12932,11 @@ class TemplateCreate(BaseModel): file_size: int = 0 checksum: str = "" - class TemplateReviewCreate(BaseModel): rating: int = Field(..., ge=1, le=5) comment: str = "" is_verified_purchase: bool = False - class PluginCreate(BaseModel): name: str description: str @@ -13618,13 +12957,11 @@ class PluginCreate(BaseModel): file_size: int = 0 checksum: str = "" - class PluginReviewCreate(BaseModel): rating: int = Field(..., ge=1, le=5) comment: str = "" is_verified_purchase: bool = False - class DeveloperProfileCreate(BaseModel): display_name: str email: str @@ -13633,7 +12970,6 @@ class DeveloperProfileCreate(BaseModel): github_url: str | None = None avatar_url: str | None = None - class DeveloperProfileUpdate(BaseModel): display_name: str | None = None bio: str | None = None @@ -13641,7 +12977,6 @@ class DeveloperProfileUpdate(BaseModel): github_url: str | None = None avatar_url: str | None = None - class CodeExampleCreate(BaseModel): title: str description: str = "" @@ -13653,7 +12988,6 @@ class CodeExampleCreate(BaseModel): sdk_id: str | None = None api_endpoints: list[str] = Field(default_factory=list) - class PortalConfigCreate(BaseModel): name: str description: str = "" @@ -13670,21 +13004,17 @@ class PortalConfigCreate(BaseModel): discord_url: str | None = None api_base_url: str = "https://api.insightflow.io" - # Developer Ecosystem Manager singleton _developer_ecosystem_manager: "DeveloperEcosystemManager | None" = None - def get_developer_ecosystem_manager_instance() -> "DeveloperEcosystemManager | None": global _developer_ecosystem_manager if _developer_ecosystem_manager is None and DEVELOPER_ECOSYSTEM_AVAILABLE: _developer_ecosystem_manager = DeveloperEcosystemManager() return _developer_ecosystem_manager - # ==================== SDK Release & Management API ==================== - @app.post("/api/v1/developer/sdks", tags=["Developer Ecosystem"]) async def create_sdk_release_endpoint( request: SDKReleaseCreate, @@ -13726,7 +13056,6 @@ async def create_sdk_release_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/developer/sdks", tags=["Developer Ecosystem"]) async def list_sdk_releases_endpoint( language: str | None = Query(default=None, description="SDK语言过滤"), @@ -13761,7 +13090,6 @@ async def list_sdk_releases_endpoint( ], } - @app.get("/api/v1/developer/sdks/{sdk_id}", tags=["Developer Ecosystem"]) async def get_sdk_release_endpoint(sdk_id: str): """获取 SDK 发布详情""" @@ -13795,7 +13123,6 @@ async def get_sdk_release_endpoint(sdk_id: str): "published_at": sdk.published_at, } - @app.put("/api/v1/developer/sdks/{sdk_id}", tags=["Developer Ecosystem"]) async def update_sdk_release_endpoint(sdk_id: str, request: SDKReleaseUpdate): """更新 SDK 发布""" @@ -13817,7 +13144,6 @@ async def update_sdk_release_endpoint(sdk_id: str, request: SDKReleaseUpdate): "updated_at": sdk.updated_at, } - @app.post("/api/v1/developer/sdks/{sdk_id}/publish", tags=["Developer Ecosystem"]) async def publish_sdk_release_endpoint(sdk_id: str): """发布 SDK""" @@ -13832,7 +13158,6 @@ async def publish_sdk_release_endpoint(sdk_id: str): return {"id": sdk.id, "status": sdk.status.value, "published_at": sdk.published_at} - @app.post("/api/v1/developer/sdks/{sdk_id}/download", tags=["Developer Ecosystem"]) async def increment_sdk_download_endpoint(sdk_id: str): """记录 SDK 下载""" @@ -13844,7 +13169,6 @@ async def increment_sdk_download_endpoint(sdk_id: str): return {"success": True, "message": "Download counted"} - @app.get("/api/v1/developer/sdks/{sdk_id}/versions", tags=["Developer Ecosystem"]) async def get_sdk_versions_endpoint(sdk_id: str): """获取 SDK 版本历史""" @@ -13868,7 +13192,6 @@ async def get_sdk_versions_endpoint(sdk_id: str): ], } - @app.post("/api/v1/developer/sdks/{sdk_id}/versions", tags=["Developer Ecosystem"]) async def add_sdk_version_endpoint(sdk_id: str, request: SDKVersionCreate): """添加 SDK 版本""" @@ -13895,10 +13218,8 @@ async def add_sdk_version_endpoint(sdk_id: str, request: SDKVersionCreate): "created_at": version.created_at, } - # ==================== Template Market API ==================== - @app.post("/api/v1/developer/templates", tags=["Developer Ecosystem"]) async def create_template_endpoint( request: TemplateCreate, @@ -13943,7 +13264,6 @@ async def create_template_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/developer/templates", tags=["Developer Ecosystem"]) async def list_templates_endpoint( category: str | None = Query(default=None, description="分类过滤"), @@ -13993,7 +13313,6 @@ async def list_templates_endpoint( ], } - @app.get("/api/v1/developer/templates/{template_id}", tags=["Developer Ecosystem"]) async def get_template_endpoint(template_id: str): """获取模板详情""" @@ -14030,7 +13349,6 @@ async def get_template_endpoint(template_id: str): "created_at": template.created_at, } - @app.post("/api/v1/developer/templates/{template_id}/approve", tags=["Developer Ecosystem"]) async def approve_template_endpoint(template_id: str, reviewed_by: str = Header(default="system")): """审核通过模板""" @@ -14045,7 +13363,6 @@ async def approve_template_endpoint(template_id: str, reviewed_by: str = Header( return {"id": template.id, "status": template.status.value} - @app.post("/api/v1/developer/templates/{template_id}/publish", tags=["Developer Ecosystem"]) async def publish_template_endpoint(template_id: str): """发布模板""" @@ -14064,7 +13381,6 @@ async def publish_template_endpoint(template_id: str): "published_at": template.published_at, } - @app.post("/api/v1/developer/templates/{template_id}/reject", tags=["Developer Ecosystem"]) async def reject_template_endpoint(template_id: str, reason: str = ""): """拒绝模板""" @@ -14079,7 +13395,6 @@ async def reject_template_endpoint(template_id: str, reason: str = ""): return {"id": template.id, "status": template.status.value} - @app.post("/api/v1/developer/templates/{template_id}/install", tags=["Developer Ecosystem"]) async def install_template_endpoint(template_id: str): """安装模板""" @@ -14091,7 +13406,6 @@ async def install_template_endpoint(template_id: str): return {"success": True, "message": "Template installed"} - @app.post("/api/v1/developer/templates/{template_id}/reviews", tags=["Developer Ecosystem"]) async def add_template_review_endpoint( template_id: str, @@ -14121,7 +13435,6 @@ async def add_template_review_endpoint( "created_at": review.created_at, } - @app.get("/api/v1/developer/templates/{template_id}/reviews", tags=["Developer Ecosystem"]) async def get_template_reviews_endpoint( template_id: str, @@ -14149,10 +13462,8 @@ async def get_template_reviews_endpoint( ], } - # ==================== Plugin Market API ==================== - @app.post("/api/v1/developer/plugins", tags=["Developer Ecosystem"]) async def create_developer_plugin_endpoint( request: PluginCreate, @@ -14201,7 +13512,6 @@ async def create_developer_plugin_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/developer/plugins", tags=["Developer Ecosystem"]) async def list_developer_plugins_endpoint( category: str | None = Query(default=None, description="分类过滤"), @@ -14248,7 +13558,6 @@ async def list_developer_plugins_endpoint( ], } - @app.get("/api/v1/developer/plugins/{plugin_id}", tags=["Developer Ecosystem"]) async def get_developer_plugin_endpoint(plugin_id: str): """获取插件详情""" @@ -14288,7 +13597,6 @@ async def get_developer_plugin_endpoint(plugin_id: str): "created_at": plugin.created_at, } - @app.post("/api/v1/developer/plugins/{plugin_id}/review", tags=["Developer Ecosystem"]) async def review_plugin_endpoint( plugin_id: str, @@ -14318,7 +13626,6 @@ async def review_plugin_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/developer/plugins/{plugin_id}/publish", tags=["Developer Ecosystem"]) async def publish_plugin_endpoint(plugin_id: str): """发布插件""" @@ -14333,7 +13640,6 @@ async def publish_plugin_endpoint(plugin_id: str): return {"id": plugin.id, "status": plugin.status.value, "published_at": plugin.published_at} - @app.post("/api/v1/developer/plugins/{plugin_id}/install", tags=["Developer Ecosystem"]) async def install_plugin_endpoint(plugin_id: str, active: bool = True): """安装插件""" @@ -14345,7 +13651,6 @@ async def install_plugin_endpoint(plugin_id: str, active: bool = True): return {"success": True, "message": "Plugin installed"} - @app.post("/api/v1/developer/plugins/{plugin_id}/reviews", tags=["Developer Ecosystem"]) async def add_plugin_review_endpoint( plugin_id: str, @@ -14375,7 +13680,6 @@ async def add_plugin_review_endpoint( "created_at": review.created_at, } - @app.get("/api/v1/developer/plugins/{plugin_id}/reviews", tags=["Developer Ecosystem"]) async def get_plugin_reviews_endpoint( plugin_id: str, @@ -14403,10 +13707,8 @@ async def get_plugin_reviews_endpoint( ], } - # ==================== Developer Revenue Sharing API ==================== - @app.get("/api/v1/developer/revenues/{developer_id}", tags=["Developer Ecosystem"]) async def get_developer_revenues_endpoint( developer_id: str, @@ -14440,7 +13742,6 @@ async def get_developer_revenues_endpoint( ], } - @app.get("/api/v1/developer/revenues/{developer_id}/summary", tags=["Developer Ecosystem"]) async def get_developer_revenue_summary_endpoint(developer_id: str): """获取开发者收益汇总""" @@ -14452,10 +13753,8 @@ async def get_developer_revenue_summary_endpoint(developer_id: str): return summary - # ==================== Developer Profile & Management API ==================== - @app.post("/api/v1/developer/profiles", tags=["Developer Ecosystem"]) async def create_developer_profile_endpoint(request: DeveloperProfileCreate): """创建开发者档案""" @@ -14485,7 +13784,6 @@ async def create_developer_profile_endpoint(request: DeveloperProfileCreate): "created_at": profile.created_at, } - @app.get("/api/v1/developer/profiles/{developer_id}", tags=["Developer Ecosystem"]) async def get_developer_profile_endpoint(developer_id: str): """获取开发者档案""" @@ -14517,7 +13815,6 @@ async def get_developer_profile_endpoint(developer_id: str): "verified_at": profile.verified_at, } - @app.get("/api/v1/developer/profiles/user/{user_id}", tags=["Developer Ecosystem"]) async def get_developer_profile_by_user_endpoint(user_id: str): """通过用户ID获取开发者档案""" @@ -14539,7 +13836,6 @@ async def get_developer_profile_by_user_endpoint(user_id: str): "total_downloads": profile.total_downloads, } - @app.put("/api/v1/developer/profiles/{developer_id}", tags=["Developer Ecosystem"]) async def update_developer_profile_endpoint(developer_id: str, request: DeveloperProfileUpdate): """更新开发者档案""" @@ -14548,7 +13844,6 @@ async def update_developer_profile_endpoint(developer_id: str, request: Develope return {"message": "Profile update endpoint - to be implemented"} - @app.post("/api/v1/developer/profiles/{developer_id}/verify", tags=["Developer Ecosystem"]) async def verify_developer_endpoint( developer_id: str, @@ -14575,7 +13870,6 @@ async def verify_developer_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.post("/api/v1/developer/profiles/{developer_id}/update-stats", tags=["Developer Ecosystem"]) async def update_developer_stats_endpoint(developer_id: str): """更新开发者统计信息""" @@ -14587,10 +13881,8 @@ async def update_developer_stats_endpoint(developer_id: str): return {"success": True, "message": "Developer stats updated"} - # ==================== Code Examples API ==================== - @app.post("/api/v1/developer/code-examples", tags=["Developer Ecosystem"]) async def create_code_example_endpoint( request: CodeExampleCreate, @@ -14626,7 +13918,6 @@ async def create_code_example_endpoint( "created_at": example.created_at, } - @app.get("/api/v1/developer/code-examples", tags=["Developer Ecosystem"]) async def list_code_examples_endpoint( language: str | None = Query(default=None, description="编程语言过滤"), @@ -14660,7 +13951,6 @@ async def list_code_examples_endpoint( ], } - @app.get("/api/v1/developer/code-examples/{example_id}", tags=["Developer Ecosystem"]) async def get_code_example_endpoint(example_id: str): """获取代码示例详情""" @@ -14693,7 +13983,6 @@ async def get_code_example_endpoint(example_id: str): "created_at": example.created_at, } - @app.post("/api/v1/developer/code-examples/{example_id}/copy", tags=["Developer Ecosystem"]) async def copy_code_example_endpoint(example_id: str): """复制代码示例""" @@ -14705,10 +13994,8 @@ async def copy_code_example_endpoint(example_id: str): return {"success": True, "message": "Code copied"} - # ==================== API Documentation API ==================== - @app.get("/api/v1/developer/api-docs", tags=["Developer Ecosystem"]) async def get_latest_api_documentation_endpoint(): """获取最新 API 文档""" @@ -14729,7 +14016,6 @@ async def get_latest_api_documentation_endpoint(): "generated_by": doc.generated_by, } - @app.get("/api/v1/developer/api-docs/{doc_id}", tags=["Developer Ecosystem"]) async def get_api_documentation_endpoint(doc_id: str): """获取 API 文档详情""" @@ -14753,10 +14039,8 @@ async def get_api_documentation_endpoint(doc_id: str): "generated_by": doc.generated_by, } - # ==================== Developer Portal API ==================== - @app.post("/api/v1/developer/portal-configs", tags=["Developer Ecosystem"]) async def create_portal_config_endpoint(request: PortalConfigCreate): """创建开发者门户配置""" @@ -14790,7 +14074,6 @@ async def create_portal_config_endpoint(request: PortalConfigCreate): "created_at": config.created_at, } - @app.get("/api/v1/developer/portal-configs", tags=["Developer Ecosystem"]) async def get_active_portal_config_endpoint(): """获取活跃的开发者门户配置""" @@ -14820,7 +14103,6 @@ async def get_active_portal_config_endpoint(): "is_active": config.is_active, } - @app.get("/api/v1/developer/portal-configs/{config_id}", tags=["Developer Ecosystem"]) async def get_portal_config_endpoint(config_id: str): """获取开发者门户配置""" @@ -14845,23 +14127,19 @@ async def get_portal_config_endpoint(config_id: str): "is_active": config.is_active, } - # ==================== Phase 8 Task 8: Operations & Monitoring Endpoints ==================== # Ops Manager singleton _ops_manager: "OpsManager | None" = None - def get_ops_manager_instance() -> "OpsManager | None": global _ops_manager if _ops_manager is None and OPS_MANAGER_AVAILABLE: _ops_manager = get_ops_manager() return _ops_manager - # Pydantic Models for Ops API - class AlertRuleCreate(BaseModel): name: str = Field(..., description="告警规则名称") description: str = Field(default="", description="告警规则描述") @@ -14876,7 +14154,6 @@ class AlertRuleCreate(BaseModel): labels: dict = Field(default_factory=dict, description="标签") annotations: dict = Field(default_factory=dict, description="注释") - class AlertRuleResponse(BaseModel): id: str name: str @@ -14895,7 +14172,6 @@ class AlertRuleResponse(BaseModel): created_at: str updated_at: str - class AlertChannelCreate(BaseModel): name: str = Field(..., description="渠道名称") channel_type: str = Field( @@ -14908,7 +14184,6 @@ class AlertChannelCreate(BaseModel): description="过滤的告警级别", ) - class AlertChannelResponse(BaseModel): id: str name: str @@ -14921,7 +14196,6 @@ class AlertChannelResponse(BaseModel): last_used_at: str | None created_at: str - class AlertResponse(BaseModel): id: str rule_id: str @@ -14938,7 +14212,6 @@ class AlertResponse(BaseModel): acknowledged_by: str | None suppression_count: int - class HealthCheckCreate(BaseModel): name: str = Field(..., description="健康检查名称") target_type: str = Field(..., description="目标类型: service, database, api") @@ -14949,7 +14222,6 @@ class HealthCheckCreate(BaseModel): timeout: int = Field(default=10, description="超时时间（秒）") retry_count: int = Field(default=3, description="重试次数") - class HealthCheckResponse(BaseModel): id: str name: str @@ -14961,7 +14233,6 @@ class HealthCheckResponse(BaseModel): is_enabled: bool created_at: str - class AutoScalingPolicyCreate(BaseModel): name: str = Field(..., description="策略名称") resource_type: str = Field( @@ -14977,7 +14248,6 @@ class AutoScalingPolicyCreate(BaseModel): scale_down_step: int = Field(default=1, description="缩容步长") cooldown_period: int = Field(default=300, description="冷却时间（秒）") - class BackupJobCreate(BaseModel): name: str = Field(..., description="备份任务名称") backup_type: str = Field(..., description="备份类型: full, incremental, differential") @@ -14989,10 +14259,8 @@ class BackupJobCreate(BaseModel): compression_enabled: bool = Field(default=True, description="是否压缩") storage_location: str | None = Field(default=None, description="存储位置") - # Alert Rules API - @app.post( "/api/v1/ops/alert-rules", response_model=AlertRuleResponse, @@ -15049,7 +14317,6 @@ async def create_alert_rule_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ops/alert-rules", tags=["Operations & Monitoring"]) async def list_alert_rules_endpoint( tenant_id: str, @@ -15085,7 +14352,6 @@ async def list_alert_rules_endpoint( for rule in rules ] - @app.get( "/api/v1/ops/alert-rules/{rule_id}", response_model=AlertRuleResponse, @@ -15121,7 +14387,6 @@ async def get_alert_rule_endpoint(rule_id: str, _=Depends(verify_api_key)): updated_at=rule.updated_at, ) - @app.patch( "/api/v1/ops/alert-rules/{rule_id}", response_model=AlertRuleResponse, @@ -15157,7 +14422,6 @@ async def update_alert_rule_endpoint(rule_id: str, updates: dict, _=Depends(veri updated_at=rule.updated_at, ) - @app.delete("/api/v1/ops/alert-rules/{rule_id}", tags=["Operations & Monitoring"]) async def delete_alert_rule_endpoint(rule_id: str, _=Depends(verify_api_key)): """删除告警规则""" @@ -15172,10 +14436,8 @@ async def delete_alert_rule_endpoint(rule_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "Alert rule deleted"} - # Alert Channels API - @app.post( "/api/v1/ops/alert-channels", response_model=AlertChannelResponse, @@ -15216,7 +14478,6 @@ async def create_alert_channel_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ops/alert-channels", tags=["Operations & Monitoring"]) async def list_alert_channels_endpoint(tenant_id: str, _=Depends(verify_api_key)): """列出租户的告警渠道""" @@ -15242,7 +14503,6 @@ async def list_alert_channels_endpoint(tenant_id: str, _=Depends(verify_api_key) for channel in channels ] - @app.post("/api/v1/ops/alert-channels/{channel_id}/test", tags=["Operations & Monitoring"]) async def test_alert_channel_endpoint(channel_id: str, _=Depends(verify_api_key)): """测试告警渠道""" @@ -15257,10 +14517,8 @@ async def test_alert_channel_endpoint(channel_id: str, _=Depends(verify_api_key) else: raise HTTPException(status_code=400, detail="Failed to send test alert") - # Alerts API - @app.get("/api/v1/ops/alerts", tags=["Operations & Monitoring"]) async def list_alerts_endpoint( tenant_id: str, @@ -15300,7 +14558,6 @@ async def list_alerts_endpoint( for alert in alerts ] - @app.post("/api/v1/ops/alerts/{alert_id}/acknowledge", tags=["Operations & Monitoring"]) async def acknowledge_alert_endpoint( alert_id: str, @@ -15319,7 +14576,6 @@ async def acknowledge_alert_endpoint( return {"success": True, "message": "Alert acknowledged"} - @app.post("/api/v1/ops/alerts/{alert_id}/resolve", tags=["Operations & Monitoring"]) async def resolve_alert_endpoint(alert_id: str, _=Depends(verify_api_key)): """解决告警""" @@ -15334,10 +14590,8 @@ async def resolve_alert_endpoint(alert_id: str, _=Depends(verify_api_key)): return {"success": True, "message": "Alert resolved"} - # Resource Metrics API - @app.post("/api/v1/ops/resource-metrics", tags=["Operations & Monitoring"]) async def record_resource_metric_endpoint( tenant_id: str, @@ -15377,7 +14631,6 @@ async def record_resource_metric_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ops/resource-metrics", tags=["Operations & Monitoring"]) async def get_resource_metrics_endpoint( tenant_id: str, @@ -15405,10 +14658,8 @@ async def get_resource_metrics_endpoint( for m in metrics ] - # Capacity Planning API - @app.post("/api/v1/ops/capacity-plans", tags=["Operations & Monitoring"]) async def create_capacity_plan_endpoint( tenant_id: str, @@ -15447,7 +14698,6 @@ async def create_capacity_plan_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ops/capacity-plans", tags=["Operations & Monitoring"]) async def list_capacity_plans_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取容量规划列表""" @@ -15472,10 +14722,8 @@ async def list_capacity_plans_endpoint(tenant_id: str, _=Depends(verify_api_key) for plan in plans ] - # Auto Scaling API - @app.post("/api/v1/ops/auto-scaling-policies", tags=["Operations & Monitoring"]) async def create_auto_scaling_policy_endpoint( tenant_id: str, @@ -15518,7 +14766,6 @@ async def create_auto_scaling_policy_endpoint( except ValueError as e: raise HTTPException(status_code=400, detail=str(e)) - @app.get("/api/v1/ops/auto-scaling-policies", tags=["Operations & Monitoring"]) async def list_auto_scaling_policies_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取自动扩缩容策略列表""" @@ -15542,7 +14789,6 @@ async def list_auto_scaling_policies_endpoint(tenant_id: str, _=Depends(verify_a for policy in policies ] - @app.get("/api/v1/ops/scaling-events", tags=["Operations & Monitoring"]) async def list_scaling_events_endpoint( tenant_id: str, @@ -15572,10 +14818,8 @@ async def list_scaling_events_endpoint( for event in events ] - # Health Check API - @app.post( "/api/v1/ops/health-checks", response_model=HealthCheckResponse, @@ -15616,7 +14860,6 @@ async def create_health_check_endpoint( created_at=check.created_at, ) - @app.get("/api/v1/ops/health-checks", tags=["Operations & Monitoring"]) async def list_health_checks_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取健康检查列表""" @@ -15641,7 +14884,6 @@ async def list_health_checks_endpoint(tenant_id: str, _=Depends(verify_api_key)) for check in checks ] - @app.post("/api/v1/ops/health-checks/{check_id}/execute", tags=["Operations & Monitoring"]) async def execute_health_check_endpoint(check_id: str, _=Depends(verify_api_key)): """执行健康检查""" @@ -15660,10 +14902,8 @@ async def execute_health_check_endpoint(check_id: str, _=Depends(verify_api_key) "checked_at": result.checked_at, } - # Backup API - @app.post("/api/v1/ops/backup-jobs", tags=["Operations & Monitoring"]) async def create_backup_job_endpoint( tenant_id: str, @@ -15699,7 +14939,6 @@ async def create_backup_job_endpoint( "created_at": job.created_at, } - @app.get("/api/v1/ops/backup-jobs", tags=["Operations & Monitoring"]) async def list_backup_jobs_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取备份任务列表""" @@ -15722,7 +14961,6 @@ async def list_backup_jobs_endpoint(tenant_id: str, _=Depends(verify_api_key)): for job in jobs ] - @app.post("/api/v1/ops/backup-jobs/{job_id}/execute", tags=["Operations & Monitoring"]) async def execute_backup_endpoint(job_id: str, _=Depends(verify_api_key)): """执行备份""" @@ -15743,7 +14981,6 @@ async def execute_backup_endpoint(job_id: str, _=Depends(verify_api_key)): "storage_path": record.storage_path, } - @app.get("/api/v1/ops/backup-records", tags=["Operations & Monitoring"]) async def list_backup_records_endpoint( tenant_id: str, @@ -15772,10 +15009,8 @@ async def list_backup_records_endpoint( for record in records ] - # Cost Optimization API - @app.post("/api/v1/ops/cost-reports", tags=["Operations & Monitoring"]) async def generate_cost_report_endpoint( tenant_id: str, @@ -15801,7 +15036,6 @@ async def generate_cost_report_endpoint( "created_at": report.created_at, } - @app.get("/api/v1/ops/idle-resources", tags=["Operations & Monitoring"]) async def get_idle_resources_endpoint(tenant_id: str, _=Depends(verify_api_key)): """获取闲置资源列表""" @@ -15826,7 +15060,6 @@ async def get_idle_resources_endpoint(tenant_id: str, _=Depends(verify_api_key)) for resource in idle_resources ] - @app.post("/api/v1/ops/cost-optimization-suggestions", tags=["Operations & Monitoring"]) async def generate_cost_optimization_suggestions_endpoint( tenant_id: str, @@ -15856,7 +15089,6 @@ async def generate_cost_optimization_suggestions_endpoint( for suggestion in suggestions ] - @app.get("/api/v1/ops/cost-optimization-suggestions", tags=["Operations & Monitoring"]) async def list_cost_optimization_suggestions_endpoint( tenant_id: str, @@ -15886,7 +15118,6 @@ async def list_cost_optimization_suggestions_endpoint( for suggestion in suggestions ] - @app.post( "/api/v1/ops/cost-optimization-suggestions/{suggestion_id}/apply", tags=["Operations & Monitoring"], @@ -15916,6 +15147,5 @@ async def apply_cost_optimization_suggestion_endpoint( }, } - if __name__ == "__main__": uvicorn.run(app, host="0.0.0.0", port=8000) diff --git a/backend/multimodal_entity_linker.py b/backend/multimodal_entity_linker.py index 022ff54..89437d5 100644 --- a/backend/multimodal_entity_linker.py +++ b/backend/multimodal_entity_linker.py @@ -17,7 +17,6 @@ try: except ImportError: NUMPY_AVAILABLE = False - @dataclass class MultimodalEntity: """多模态实体""" @@ -36,7 +35,6 @@ class MultimodalEntity: if self.modality_features is None: self.modality_features = {} - @dataclass class EntityLink: """实体关联""" @@ -51,7 +49,6 @@ class EntityLink: confidence: float evidence: str - @dataclass class AlignmentResult: """对齐结果""" @@ -62,7 +59,6 @@ class AlignmentResult: match_type: str # exact, fuzzy, embedding confidence: float - @dataclass class FusionResult: """知识融合结果""" @@ -73,7 +69,6 @@ class FusionResult: source_modalities: list[str] confidence: float - class MultimodalEntityLinker: """多模态实体关联器 - 跨模态实体对齐和知识融合""" @@ -525,11 +520,9 @@ class MultimodalEntityLinker: ), } - # Singleton instance _multimodal_entity_linker = None - def get_multimodal_entity_linker(similarity_threshold: float = 0.85) -> MultimodalEntityLinker: """获取多模态实体关联器单例""" global _multimodal_entity_linker diff --git a/backend/multimodal_processor.py b/backend/multimodal_processor.py index 8ddf881..6b6e43b 100644 --- a/backend/multimodal_processor.py +++ b/backend/multimodal_processor.py @@ -38,7 +38,6 @@ try: except ImportError: FFMPEG_AVAILABLE = False - @dataclass class VideoFrame: """视频关键帧数据类""" @@ -56,7 +55,6 @@ class VideoFrame: if self.entities_detected is None: self.entities_detected = [] - @dataclass class VideoInfo: """视频信息数据类""" @@ -80,7 +78,6 @@ class VideoInfo: if self.metadata is None: self.metadata = {} - @dataclass class VideoProcessingResult: """视频处理结果""" @@ -93,7 +90,6 @@ class VideoProcessingResult: success: bool error_message: str = "" - class MultimodalProcessor: """多模态处理器 - 处理视频文件""" @@ -461,11 +457,9 @@ class MultimodalProcessor: shutil.rmtree(dir_path) os.makedirs(dir_path, exist_ok=True) - # Singleton instance _multimodal_processor = None - def get_multimodal_processor( temp_dir: str | None = None, frame_interval: int = 5 ) -> MultimodalProcessor: diff --git a/backend/neo4j_manager.py b/backend/neo4j_manager.py index 1e60024..bc7a228 100644 --- a/backend/neo4j_manager.py +++ b/backend/neo4j_manager.py @@ -26,7 +26,6 @@ except ImportError: NEO4J_AVAILABLE = False logger.warning("Neo4j driver not installed. Neo4j features will be disabled.") - @dataclass class GraphEntity: """图数据库中的实体节点""" @@ -45,7 +44,6 @@ class GraphEntity: if self.properties is None: self.properties = {} - @dataclass class GraphRelation: """图数据库中的关系边""" @@ -61,7 +59,6 @@ class GraphRelation: if self.properties is None: self.properties = {} - @dataclass class PathResult: """路径查询结果""" @@ -71,7 +68,6 @@ class PathResult: length: int total_weight: float = 0.0 - @dataclass class CommunityResult: """社区发现结果""" @@ -81,7 +77,6 @@ class CommunityResult: size: int density: float = 0.0 - @dataclass class CentralityResult: """中心性分析结果""" @@ -91,7 +86,6 @@ class CentralityResult: score: float rank: int = 0 - class Neo4jManager: """Neo4j 图数据库管理器""" @@ -998,11 +992,9 @@ class Neo4jManager: return {"nodes": nodes, "relationships": relationships} - # 全局单例 _neo4j_manager = None - def get_neo4j_manager() -> Neo4jManager: """获取 Neo4j 管理器单例""" global _neo4j_manager @@ -1010,7 +1002,6 @@ def get_neo4j_manager() -> Neo4jManager: _neo4j_manager = Neo4jManager() return _neo4j_manager - def close_neo4j_manager() -> None: """关闭 Neo4j 连接""" global _neo4j_manager @@ -1018,10 +1009,8 @@ def close_neo4j_manager() -> None: _neo4j_manager.close() _neo4j_manager = None - # 便捷函数 - def sync_project_to_neo4j( project_id: str, project_name: str, @@ -1079,7 +1068,6 @@ def sync_project_to_neo4j( f"{len(relations)} relations", ) - if __name__ == "__main__": # 测试代码 logging.basicConfig(level=logging.INFO) diff --git a/backend/ops_manager.py b/backend/ops_manager.py index 4a381a4..d711df9 100644 --- a/backend/ops_manager.py +++ b/backend/ops_manager.py @@ -29,7 +29,6 @@ import httpx # Database path DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db") - class AlertSeverity(StrEnum): """告警严重级别 P0-P3""" @@ -38,7 +37,6 @@ class AlertSeverity(StrEnum): P2 = "p2" # 一般 - 部分功能受影响，需要4小时内处理 P3 = "p3" # 轻微 - 非核心功能问题，24小时内处理 - class AlertStatus(StrEnum): """告警状态""" @@ -47,7 +45,6 @@ class AlertStatus(StrEnum): ACKNOWLEDGED = "acknowledged" # 已确认 SUPPRESSED = "suppressed" # 已抑制 - class AlertChannelType(StrEnum): """告警渠道类型""" @@ -60,7 +57,6 @@ class AlertChannelType(StrEnum): SMS = "sms" WEBHOOK = "webhook" - class AlertRuleType(StrEnum): """告警规则类型""" @@ -69,7 +65,6 @@ class AlertRuleType(StrEnum): PREDICTIVE = "predictive" # 预测性告警 COMPOSITE = "composite" # 复合告警 - class ResourceType(StrEnum): """资源类型""" @@ -82,7 +77,6 @@ class ResourceType(StrEnum): CACHE = "cache" QUEUE = "queue" - class ScalingAction(StrEnum): """扩缩容动作""" @@ -90,7 +84,6 @@ class ScalingAction(StrEnum): SCALE_DOWN = "scale_down" # 缩容 MAINTAIN = "maintain" # 保持 - class HealthStatus(StrEnum): """健康状态""" @@ -99,7 +92,6 @@ class HealthStatus(StrEnum): UNHEALTHY = "unhealthy" UNKNOWN = "unknown" - class BackupStatus(StrEnum): """备份状态""" @@ -109,7 +101,6 @@ class BackupStatus(StrEnum): FAILED = "failed" VERIFIED = "verified" - @dataclass class AlertRule: """告警规则""" @@ -133,7 +124,6 @@ class AlertRule: updated_at: str created_by: str - @dataclass class AlertChannel: """告警渠道配置""" @@ -151,7 +141,6 @@ class AlertChannel: created_at: str updated_at: str - @dataclass class Alert: """告警实例""" @@ -175,7 +164,6 @@ class Alert: notification_sent: dict[str, bool] # 渠道发送状态 suppression_count: int # 抑制计数 - @dataclass class AlertSuppressionRule: """告警抑制规则""" @@ -189,7 +177,6 @@ class AlertSuppressionRule: created_at: str expires_at: str | None - @dataclass class AlertGroup: """告警聚合组""" @@ -201,7 +188,6 @@ class AlertGroup: created_at: str updated_at: str - @dataclass class ResourceMetric: """资源指标""" @@ -216,7 +202,6 @@ class ResourceMetric: timestamp: str metadata: dict - @dataclass class CapacityPlan: """容量规划""" @@ -232,7 +217,6 @@ class CapacityPlan: estimated_cost: float created_at: str - @dataclass class AutoScalingPolicy: """自动扩缩容策略""" @@ -253,7 +237,6 @@ class AutoScalingPolicy: created_at: str updated_at: str - @dataclass class ScalingEvent: """扩缩容事件""" @@ -271,7 +254,6 @@ class ScalingEvent: completed_at: str | None error_message: str | None - @dataclass class HealthCheck: """健康检查配置""" @@ -292,7 +274,6 @@ class HealthCheck: created_at: str updated_at: str - @dataclass class HealthCheckResult: """健康检查结果""" @@ -306,7 +287,6 @@ class HealthCheckResult: details: dict checked_at: str - @dataclass class FailoverConfig: """故障转移配置""" @@ -324,7 +304,6 @@ class FailoverConfig: created_at: str updated_at: str - @dataclass class FailoverEvent: """故障转移事件""" @@ -340,7 +319,6 @@ class FailoverEvent: completed_at: str | None rolled_back_at: str | None - @dataclass class BackupJob: """备份任务""" @@ -360,7 +338,6 @@ class BackupJob: created_at: str updated_at: str - @dataclass class BackupRecord: """备份记录""" @@ -377,7 +354,6 @@ class BackupRecord: error_message: str | None storage_path: str - @dataclass class CostReport: """成本报告""" @@ -392,7 +368,6 @@ class CostReport: anomalies: list[dict] # 异常检测 created_at: str - @dataclass class ResourceUtilization: """资源利用率""" @@ -408,7 +383,6 @@ class ResourceUtilization: report_date: str recommendations: list[str] - @dataclass class IdleResource: """闲置资源""" @@ -425,7 +399,6 @@ class IdleResource: recommendation: str detected_at: str - @dataclass class CostOptimizationSuggestion: """成本优化建议""" @@ -445,7 +418,6 @@ class CostOptimizationSuggestion: created_at: str applied_at: str | None - class OpsManager: """运维与监控管理主类""" @@ -3148,11 +3120,9 @@ class OpsManager: applied_at=row["applied_at"], ) - # Singleton instance _ops_manager = None - def get_ops_manager() -> OpsManager: global _ops_manager if _ops_manager is None: diff --git a/backend/oss_uploader.py b/backend/oss_uploader.py index 743f82d..bd25a81 100644 --- a/backend/oss_uploader.py +++ b/backend/oss_uploader.py @@ -9,7 +9,6 @@ from datetime import datetime import oss2 - class OSSUploader: def __init__(self) -> None: self.access_key = os.getenv("ALI_ACCESS_KEY") @@ -41,11 +40,9 @@ class OSSUploader: """删除 OSS 对象""" self.bucket.delete_object(object_name) - # 单例 _oss_uploader = None - def get_oss_uploader() -> OSSUploader: global _oss_uploader if _oss_uploader is None: diff --git a/backend/performance_manager.py b/backend/performance_manager.py index 689acd8..3037132 100644 --- a/backend/performance_manager.py +++ b/backend/performance_manager.py @@ -42,7 +42,6 @@ except ImportError: # ==================== 数据模型 ==================== - @dataclass class CacheStats: """缓存统计数据模型""" @@ -59,7 +58,6 @@ class CacheStats: if self.total_requests > 0: self.hit_rate = round(self.hits / self.total_requests, 4) - @dataclass class CacheEntry: """缓存条目数据模型""" @@ -72,7 +70,6 @@ class CacheEntry: last_accessed: float = 0 size_bytes: int = 0 - @dataclass class PerformanceMetric: """性能指标数据模型""" @@ -94,7 +91,6 @@ class PerformanceMetric: "metadata": self.metadata, } - @dataclass class TaskInfo: """任务信息数据模型""" @@ -126,7 +122,6 @@ class TaskInfo: "max_retries": self.max_retries, } - @dataclass class ShardInfo: """分片信息数据模型""" @@ -139,10 +134,8 @@ class ShardInfo: created_at: str = "" last_accessed: str = "" - # ==================== Redis 缓存层 ==================== - class CacheManager: """ 缓存管理器 @@ -603,10 +596,8 @@ class CacheManager: return count - # ==================== 数据库分片 ==================== - class DatabaseSharding: """ 数据库分片管理器 @@ -909,10 +900,8 @@ class DatabaseSharding: "message": "Rebalancing analysis completed", } - # ==================== 异步任务队列 ==================== - class TaskQueue: """ 异步任务队列管理器 @@ -1299,10 +1288,8 @@ class TaskQueue: "backend": "celery" if self.use_celery else "memory", } - # ==================== 性能监控 ==================== - class PerformanceMonitor: """ 性能监控器 @@ -1622,10 +1609,8 @@ class PerformanceMonitor: return deleted - # ==================== 性能装饰器 ==================== - def cached( cache_manager: CacheManager, key_prefix: str = "", @@ -1670,7 +1655,6 @@ def cached( return decorator - def monitored(monitor: PerformanceMonitor, metric_type: str, endpoint: str | None = None) -> None: """ 性能监控装饰器 @@ -1698,10 +1682,8 @@ def monitored(monitor: PerformanceMonitor, metric_type: str, endpoint: str | Non return decorator - # ==================== 性能管理器 ==================== - class PerformanceManager: """ 性能管理器 - 统一入口 @@ -1763,11 +1745,9 @@ class PerformanceManager: return stats - # 单例模式 _performance_manager = None - def get_performance_manager( db_path: str = "insightflow.db", redis_url: str | None = None, diff --git a/backend/plugin_manager.py b/backend/plugin_manager.py index 80832a6..5b4e920 100644 --- a/backend/plugin_manager.py +++ b/backend/plugin_manager.py @@ -11,7 +11,6 @@ import json import os import sqlite3 import time -import urllib.parse import uuid from dataclasses import dataclass, field from datetime import datetime @@ -31,7 +30,6 @@ try: except ImportError: WEBDAV_AVAILABLE = False - class PluginType(Enum): """插件类型""" @@ -43,7 +41,6 @@ class PluginType(Enum): WEBDAV = "webdav" CUSTOM = "custom" - class PluginStatus(Enum): """插件状态""" @@ -52,7 +49,6 @@ class PluginStatus(Enum): ERROR = "error" PENDING = "pending" - @dataclass class Plugin: """插件配置""" @@ -68,7 +64,6 @@ class Plugin: last_used_at: str | None = None use_count: int = 0 - @dataclass class PluginConfig: """插件详细配置""" @@ -81,7 +76,6 @@ class PluginConfig: created_at: str = "" updated_at: str = "" - @dataclass class BotSession: """机器人会话""" @@ -99,7 +93,6 @@ class BotSession: last_message_at: str | None = None message_count: int = 0 - @dataclass class WebhookEndpoint: """Webhook 端点配置（Zapier/Make集成）""" @@ -118,7 +111,6 @@ class WebhookEndpoint: last_triggered_at: str | None = None trigger_count: int = 0 - @dataclass class WebDAVSync: """WebDAV 同步配置""" @@ -140,7 +132,6 @@ class WebDAVSync: updated_at: str = "" sync_count: int = 0 - @dataclass class ChromeExtensionToken: """Chrome 扩展令牌""" @@ -157,7 +148,6 @@ class ChromeExtensionToken: use_count: int = 0 is_revoked: bool = False - class PluginManager: """插件管理主类""" @@ -409,7 +399,6 @@ class PluginManager: conn.commit() conn.close() - class ChromeExtensionHandler: """Chrome 扩展处理器""" @@ -438,7 +427,6 @@ class ChromeExtensionHandler: now = datetime.now().isoformat() expires_at = None if expires_days: - from datetime import timedelta expires_at = (datetime.now() + timedelta(days=expires_days)).isoformat() @@ -618,7 +606,6 @@ class ChromeExtensionHandler: "content_length": len(content), } - class BotHandler: """飞书/钉钉机器人处理器""" @@ -953,7 +940,6 @@ class BotHandler: ) return response.status_code == 200 - class WebhookIntegration: """Zapier/Make Webhook 集成""" @@ -1179,7 +1165,6 @@ class WebhookIntegration: "message": "Test event sent successfully" if success else "Failed to send test event", } - class WebDAVSyncManager: """WebDAV 同步管理""" @@ -1441,11 +1426,9 @@ class WebDAVSyncManager: return {"success": False, "error": str(e)} - # Singleton instance _plugin_manager = None - def get_plugin_manager(db_manager=None) -> None: """获取 PluginManager 单例""" global _plugin_manager diff --git a/backend/rate_limiter.py b/backend/rate_limiter.py index 75d2bb6..4fb9bfa 100644 --- a/backend/rate_limiter.py +++ b/backend/rate_limiter.py @@ -12,7 +12,6 @@ from collections.abc import Callable from dataclasses import dataclass from functools import wraps - @dataclass class RateLimitConfig: """限流配置""" @@ -21,7 +20,6 @@ class RateLimitConfig: burst_size: int = 10 # 突发请求数 window_size: int = 60 # 窗口大小（秒） - @dataclass class RateLimitInfo: """限流信息""" @@ -31,7 +29,6 @@ class RateLimitInfo: reset_time: int # 重置时间戳 retry_after: int # 需要等待的秒数 - class SlidingWindowCounter: """滑动窗口计数器""" @@ -63,7 +60,6 @@ class SlidingWindowCounter: for k in old_keys: self.requests.pop(k, None) - class RateLimiter: """API 限流器""" @@ -162,11 +158,9 @@ class RateLimiter: self.counters.clear() self.configs.clear() - # 全局限流器实例 _rate_limiter: RateLimiter | None = None - def get_rate_limiter() -> RateLimiter: """获取限流器实例""" global _rate_limiter @@ -174,10 +168,8 @@ def get_rate_limiter() -> RateLimiter: _rate_limiter = RateLimiter() return _rate_limiter - # 限流装饰器（用于函数级别限流） - def rate_limit(requests_per_minute: int = 60, key_func: Callable | None = None) -> None: """ 限流装饰器 @@ -220,6 +212,5 @@ def rate_limit(requests_per_minute: int = 60, key_func: Callable | None = None) return decorator - class RateLimitExceeded(Exception): """限流异常""" diff --git a/backend/search_manager.py b/backend/search_manager.py index 1f6db0f..5b245f4 100644 --- a/backend/search_manager.py +++ b/backend/search_manager.py @@ -19,7 +19,6 @@ from dataclasses import dataclass, field from datetime import datetime from enum import Enum - class SearchOperator(Enum): """搜索操作符""" @@ -27,7 +26,6 @@ class SearchOperator(Enum): OR = "OR" NOT = "NOT" - # 尝试导入 sentence-transformers 用于语义搜索 try: from sentence_transformers import SentenceTransformer @@ -39,7 +37,6 @@ except ImportError: # ==================== 数据模型 ==================== - @dataclass class SearchResult: """搜索结果数据模型""" @@ -63,7 +60,6 @@ class SearchResult: "metadata": self.metadata, } - @dataclass class SemanticSearchResult: """语义搜索结果数据模型""" @@ -89,7 +85,6 @@ class SemanticSearchResult: result["embedding_dim"] = len(self.embedding) return result - @dataclass class EntityPath: """实体关系路径数据模型""" @@ -119,7 +114,6 @@ class EntityPath: "path_description": self.path_description, } - @dataclass class KnowledgeGap: """知识缺口数据模型""" @@ -147,7 +141,6 @@ class KnowledgeGap: "metadata": self.metadata, } - @dataclass class SearchIndex: """搜索索引数据模型""" @@ -161,7 +154,6 @@ class SearchIndex: created_at: str updated_at: str - @dataclass class TextEmbedding: """文本 Embedding 数据模型""" @@ -174,10 +166,8 @@ class TextEmbedding: model_name: str created_at: str - # ==================== 全文搜索 ==================== - class FullTextSearch: """ 全文搜索模块 @@ -805,10 +795,8 @@ class FullTextSearch: conn.close() return stats - # ==================== 语义搜索 ==================== - class SemanticSearch: """ 语义搜索模块 @@ -1180,10 +1168,8 @@ class SemanticSearch: print(f"删除 embedding 失败: {e}") return False - # ==================== 实体关系路径发现 ==================== - class EntityPathDiscovery: """ 实体关系路径发现模块 @@ -1680,10 +1666,8 @@ class EntityPathDiscovery: bridge_scores.sort(key=lambda x: x["bridge_score"], reverse=True) return bridge_scores[:20] # 返回前20 - # ==================== 知识缺口识别 ==================== - class KnowledgeGapDetection: """ 知识缺口识别模块 @@ -2092,10 +2076,8 @@ class KnowledgeGapDetection: return recommendations - # ==================== 搜索管理器 ==================== - class SearchManager: """ 搜索管理器 - 统一入口 @@ -2273,11 +2255,9 @@ class SearchManager: "semantic_search_available": self.semantic_search.is_available(), } - # 单例模式 _search_manager = None - def get_search_manager(db_path: str = "insightflow.db") -> SearchManager: """获取搜索管理器单例""" global _search_manager @@ -2285,10 +2265,8 @@ def get_search_manager(db_path: str = "insightflow.db") -> SearchManager: _search_manager = SearchManager(db_path) return _search_manager - # 便捷函数 - def fulltext_search( query: str, project_id: str | None = None, @@ -2298,7 +2276,6 @@ def fulltext_search( manager = get_search_manager() return manager.fulltext_search.search(query, project_id, limit=limit) - def semantic_search( query: str, project_id: str | None = None, @@ -2308,13 +2285,11 @@ def semantic_search( manager = get_search_manager() return manager.semantic_search.search(query, project_id, top_k=top_k) - def find_entity_path(source_id: str, target_id: str, max_depth: int = 5) -> EntityPath | None: """查找实体路径便捷函数""" manager = get_search_manager() return manager.path_discovery.find_shortest_path(source_id, target_id, max_depth) - def detect_knowledge_gaps(project_id: str) -> list[KnowledgeGap]: """知识缺口检测便捷函数""" manager = get_search_manager() diff --git a/backend/security_manager.py b/backend/security_manager.py index 8f7a274..d54a3dc 100644 --- a/backend/security_manager.py +++ b/backend/security_manager.py @@ -25,7 +25,6 @@ except ImportError: CRYPTO_AVAILABLE = False print("Warning: cryptography not available, encryption features disabled") - class AuditActionType(Enum): """审计动作类型""" @@ -48,7 +47,6 @@ class AuditActionType(Enum): WEBHOOK_SEND = "webhook_send" BOT_MESSAGE = "bot_message" - class DataSensitivityLevel(Enum): """数据敏感度级别""" @@ -57,7 +55,6 @@ class DataSensitivityLevel(Enum): CONFIDENTIAL = "confidential" # 机密 SECRET = "secret" # 绝密 - class MaskingRuleType(Enum): """脱敏规则类型""" @@ -69,7 +66,6 @@ class MaskingRuleType(Enum): ADDRESS = "address" # 地址 CUSTOM = "custom" # 自定义 - @dataclass class AuditLog: """审计日志条目""" @@ -91,7 +87,6 @@ class AuditLog: def to_dict(self) -> dict[str, Any]: return asdict(self) - @dataclass class EncryptionConfig: """加密配置""" @@ -109,7 +104,6 @@ class EncryptionConfig: def to_dict(self) -> dict[str, Any]: return asdict(self) - @dataclass class MaskingRule: """脱敏规则""" @@ -129,7 +123,6 @@ class MaskingRule: def to_dict(self) -> dict[str, Any]: return asdict(self) - @dataclass class DataAccessPolicy: """数据访问策略""" @@ -151,7 +144,6 @@ class DataAccessPolicy: def to_dict(self) -> dict[str, Any]: return asdict(self) - @dataclass class AccessRequest: """访问请求（用于需要审批的访问）""" @@ -169,7 +161,6 @@ class AccessRequest: def to_dict(self) -> dict[str, Any]: return asdict(self) - class SecurityManager: """安全管理器""" @@ -1255,11 +1246,9 @@ class SecurityManager: created_at=row[8], ) - # 全局安全管理器实例 _security_manager = None - def get_security_manager(db_path: str = "insightflow.db") -> SecurityManager: """获取安全管理器实例""" global _security_manager diff --git a/backend/subscription_manager.py b/backend/subscription_manager.py index f517e69..14fec03 100644 --- a/backend/subscription_manager.py +++ b/backend/subscription_manager.py @@ -21,7 +21,6 @@ from typing import Any logger = logging.getLogger(__name__) - class SubscriptionStatus(StrEnum): """订阅状态""" @@ -32,7 +31,6 @@ class SubscriptionStatus(StrEnum): TRIAL = "trial" # 试用中 PENDING = "pending" # 待支付 - class PaymentProvider(StrEnum): """支付提供商""" @@ -41,7 +39,6 @@ class PaymentProvider(StrEnum): WECHAT = "wechat" # 微信支付 BANK_TRANSFER = "bank_transfer" # 银行转账 - class PaymentStatus(StrEnum): """支付状态""" @@ -52,7 +49,6 @@ class PaymentStatus(StrEnum): REFUNDED = "refunded" # 已退款 PARTIAL_REFUNDED = "partial_refunded" # 部分退款 - class InvoiceStatus(StrEnum): """发票状态""" @@ -63,7 +59,6 @@ class InvoiceStatus(StrEnum): VOID = "void" # 作废 CREDIT_NOTE = "credit_note" # 贷项通知单 - class RefundStatus(StrEnum): """退款状态""" @@ -73,7 +68,6 @@ class RefundStatus(StrEnum): COMPLETED = "completed" # 已完成 FAILED = "failed" # 失败 - @dataclass class SubscriptionPlan: """订阅计划数据类""" @@ -92,7 +86,6 @@ class SubscriptionPlan: updated_at: datetime metadata: dict[str, Any] - @dataclass class Subscription: """订阅数据类""" @@ -113,7 +106,6 @@ class Subscription: updated_at: datetime metadata: dict[str, Any] - @dataclass class UsageRecord: """用量记录数据类""" @@ -128,7 +120,6 @@ class UsageRecord: description: str | None metadata: dict[str, Any] - @dataclass class Payment: """支付记录数据类""" @@ -150,7 +141,6 @@ class Payment: created_at: datetime updated_at: datetime - @dataclass class Invoice: """发票数据类""" @@ -174,7 +164,6 @@ class Invoice: created_at: datetime updated_at: datetime - @dataclass class Refund: """退款数据类""" @@ -197,7 +186,6 @@ class Refund: created_at: datetime updated_at: datetime - @dataclass class BillingHistory: """账单历史数据类""" @@ -213,7 +201,6 @@ class BillingHistory: created_at: datetime metadata: dict[str, Any] - class SubscriptionManager: """订阅与计费管理器""" @@ -2242,11 +2229,9 @@ class SubscriptionManager: metadata=json.loads(row["metadata"] or "{}"), ) - # 全局订阅管理器实例 subscription_manager = None - def get_subscription_manager(db_path: str = "insightflow.db") -> SubscriptionManager: """获取订阅管理器实例（单例模式）""" global subscription_manager diff --git a/backend/tenant_manager.py b/backend/tenant_manager.py index 174eee6..2e06700 100644 --- a/backend/tenant_manager.py +++ b/backend/tenant_manager.py @@ -23,7 +23,6 @@ from typing import Any logger = logging.getLogger(__name__) - class TenantLimits: """租户资源限制常量""" @@ -43,7 +42,6 @@ class TenantLimits: UNLIMITED = -1 - class TenantStatus(StrEnum): """租户状态""" @@ -53,7 +51,6 @@ class TenantStatus(StrEnum): EXPIRED = "expired" # 过期 PENDING = "pending" # 待激活 - class TenantTier(StrEnum): """租户订阅层级""" @@ -61,7 +58,6 @@ class TenantTier(StrEnum): PRO = "pro" # 专业版 ENTERPRISE = "enterprise" # 企业版 - class TenantRole(StrEnum): """租户角色""" @@ -70,7 +66,6 @@ class TenantRole(StrEnum): MEMBER = "member" # 成员 VIEWER = "viewer" # 查看者 - class DomainStatus(StrEnum): """域名状态""" @@ -79,7 +74,6 @@ class DomainStatus(StrEnum): FAILED = "failed" # 验证失败 EXPIRED = "expired" # 已过期 - @dataclass class Tenant: """租户数据类""" @@ -98,7 +92,6 @@ class Tenant: resource_limits: dict[str, Any] # 资源限制 metadata: dict[str, Any] # 元数据 - @dataclass class TenantDomain: """租户域名数据类""" @@ -116,7 +109,6 @@ class TenantDomain: ssl_enabled: bool # SSL 是否启用 ssl_expires_at: datetime | None - @dataclass class TenantBranding: """租户品牌配置数据类""" @@ -134,7 +126,6 @@ class TenantBranding: created_at: datetime updated_at: datetime - @dataclass class TenantMember: """租户成员数据类""" @@ -151,7 +142,6 @@ class TenantMember: last_active_at: datetime | None status: str # active/pending/suspended - @dataclass class TenantPermission: """租户权限定义数据类""" @@ -166,7 +156,6 @@ class TenantPermission: conditions: dict | None # 条件限制 created_at: datetime - class TenantManager: """租户管理器 - 多租户 SaaS 架构核心""" @@ -1634,10 +1623,8 @@ class TenantManager: status=row["status"], ) - # ==================== 租户上下文管理 ==================== - class TenantContext: """租户上下文管理器 - 用于请求级别的租户隔离""" @@ -1670,11 +1657,9 @@ class TenantContext: cls._current_tenant_id = None cls._current_user_id = None - # 全局租户管理器实例 tenant_manager = None - def get_tenant_manager(db_path: str = "insightflow.db") -> TenantManager: """获取租户管理器实例（单例模式）""" global tenant_manager diff --git a/backend/test_phase7_task6_8.py b/backend/test_phase7_task6_8.py index a786940..428cc6c 100644 --- a/backend/test_phase7_task6_8.py +++ b/backend/test_phase7_task6_8.py @@ -20,7 +20,6 @@ from search_manager import ( # 添加 backend 到路径 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) - def test_fulltext_search() -> None: """测试全文搜索""" print("\n" + " = " * 60) @@ -63,7 +62,6 @@ def test_fulltext_search() -> None: print("\n✓ 全文搜索测试完成") return True - def test_semantic_search() -> None: """测试语义搜索""" print("\n" + " = " * 60) @@ -99,7 +97,6 @@ def test_semantic_search() -> None: print("\n✓ 语义搜索测试完成") return True - def test_entity_path_discovery() -> None: """测试实体路径发现""" print("\n" + " = " * 60) @@ -118,7 +115,6 @@ def test_entity_path_discovery() -> None: print("\n✓ 实体路径发现测试完成") return True - def test_knowledge_gap_detection() -> None: """测试知识缺口识别""" print("\n" + " = " * 60) @@ -137,7 +133,6 @@ def test_knowledge_gap_detection() -> None: print("\n✓ 知识缺口识别测试完成") return True - def test_cache_manager() -> None: """测试缓存管理器""" print("\n" + " = " * 60) @@ -186,7 +181,6 @@ def test_cache_manager() -> None: print("\n✓ 缓存管理器测试完成") return True - def test_task_queue() -> None: """测试任务队列""" print("\n" + " = " * 60) @@ -228,7 +222,6 @@ def test_task_queue() -> None: print("\n✓ 任务队列测试完成") return True - def test_performance_monitor() -> None: """测试性能监控""" print("\n" + " = " * 60) @@ -275,7 +268,6 @@ def test_performance_monitor() -> None: print("\n✓ 性能监控测试完成") return True - def test_search_manager() -> None: """测试搜索管理器""" print("\n" + " = " * 60) @@ -296,7 +288,6 @@ def test_search_manager() -> None: print("\n✓ 搜索管理器测试完成") return True - def test_performance_manager() -> None: """测试性能管理器""" print("\n" + " = " * 60) @@ -321,7 +312,6 @@ def test_performance_manager() -> None: print("\n✓ 性能管理器测试完成") return True - def run_all_tests() -> None: """运行所有测试""" print("\n" + " = " * 60) @@ -408,7 +398,6 @@ def run_all_tests() -> None: return passed == total - if __name__ == "__main__": success = run_all_tests() sys.exit(0 if success else 1) diff --git a/backend/test_phase8_task1.py b/backend/test_phase8_task1.py index 2abc7f5..28f4f3b 100644 --- a/backend/test_phase8_task1.py +++ b/backend/test_phase8_task1.py @@ -17,7 +17,6 @@ from tenant_manager import get_tenant_manager sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) - def test_tenant_management() -> None: """测试租户管理功能""" print(" = " * 60) @@ -70,7 +69,6 @@ def test_tenant_management() -> None: return tenant.id - def test_domain_management(tenant_id: str) -> None: """测试域名管理功能""" print("\n" + " = " * 60) @@ -116,7 +114,6 @@ def test_domain_management(tenant_id: str) -> None: return domain.id - def test_branding_management(tenant_id: str) -> None: """测试品牌白标功能""" print("\n" + " = " * 60) @@ -156,7 +153,6 @@ def test_branding_management(tenant_id: str) -> None: return branding.id - def test_member_management(tenant_id: str) -> None: """测试成员管理功能""" print("\n" + " = " * 60) @@ -217,7 +213,6 @@ def test_member_management(tenant_id: str) -> None: return member1.id, member2.id - def test_usage_tracking(tenant_id: str) -> None: """测试资源使用统计功能""" print("\n" + " = " * 60) @@ -259,7 +254,6 @@ def test_usage_tracking(tenant_id: str) -> None: return stats - def cleanup(tenant_id: str, domain_id: str, member_ids: list) -> None: """清理测试数据""" print("\n" + " = " * 60) @@ -283,7 +277,6 @@ def cleanup(tenant_id: str, domain_id: str, member_ids: list) -> None: manager.delete_tenant(tenant_id) print(f"✅ 租户已删除: {tenant_id}") - def main() -> None: """主测试函数""" print("\n" + " = " * 60) @@ -321,6 +314,5 @@ def main() -> None: except Exception as e: print(f"⚠️ 清理失败: {e}") - if __name__ == "__main__": main() diff --git a/backend/test_phase8_task2.py b/backend/test_phase8_task2.py index d5261c4..f283e03 100644 --- a/backend/test_phase8_task2.py +++ b/backend/test_phase8_task2.py @@ -11,7 +11,6 @@ from subscription_manager import PaymentProvider, SubscriptionManager sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) - def test_subscription_manager() -> None: """测试订阅管理器""" print(" = " * 60) @@ -225,7 +224,6 @@ def test_subscription_manager() -> None: os.remove(db_path) print(f"\n清理临时数据库: {db_path}") - if __name__ == "__main__": try: test_subscription_manager() diff --git a/backend/test_phase8_task4.py b/backend/test_phase8_task4.py index 73a04ff..14732f7 100644 --- a/backend/test_phase8_task4.py +++ b/backend/test_phase8_task4.py @@ -13,7 +13,6 @@ from ai_manager import ModelType, PredictionType, get_ai_manager # Add backend directory to path sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) - def test_custom_model() -> None: """测试自定义模型功能""" print("\n=== 测试自定义模型 ===") @@ -88,7 +87,6 @@ def test_custom_model() -> None: return model.id - async def test_train_and_predict(model_id: str) -> None: """测试训练和预测""" print("\n=== 测试模型训练和预测 ===") @@ -115,7 +113,6 @@ async def test_train_and_predict(model_id: str) -> None: except Exception as e: print(f" 预测失败: {e}") - def test_prediction_models() -> None: """测试预测模型""" print("\n=== 测试预测模型 ===") @@ -157,7 +154,6 @@ def test_prediction_models() -> None: return trend_model.id, anomaly_model.id - async def test_predictions(trend_model_id: str, anomaly_model_id: str) -> None: """测试预测功能""" print("\n=== 测试预测功能 ===") @@ -194,7 +190,6 @@ async def test_predictions(trend_model_id: str, anomaly_model_id: str) -> None: ) print(f" 检测结果: {anomaly_result.prediction_data}") - def test_kg_rag() -> None: """测试知识图谱 RAG""" print("\n=== 测试知识图谱 RAG ===") @@ -224,7 +219,6 @@ def test_kg_rag() -> None: return rag.id - async def test_kg_rag_query(rag_id: str) -> None: """测试 RAG 查询""" print("\n=== 测试知识图谱 RAG 查询 ===") @@ -295,7 +289,6 @@ async def test_kg_rag_query(rag_id: str) -> None: except Exception as e: print(f" 查询失败: {e}") - async def test_smart_summary() -> None: """测试智能摘要""" print("\n=== 测试智能摘要 ===") @@ -343,7 +336,6 @@ async def test_smart_summary() -> None: except Exception as e: print(f" 生成失败: {e}") - async def main() -> None: """主测试函数""" print(" = " * 60) @@ -382,6 +374,5 @@ async def main() -> None: traceback.print_exc() - if __name__ == "__main__": asyncio.run(main()) diff --git a/backend/test_phase8_task5.py b/backend/test_phase8_task5.py index 5cf5a7b..dde6d9a 100644 --- a/backend/test_phase8_task5.py +++ b/backend/test_phase8_task5.py @@ -32,7 +32,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__)) if backend_dir not in sys.path: sys.path.insert(0, backend_dir) - class TestGrowthManager: """测试 Growth Manager 功能""" @@ -739,12 +738,10 @@ class TestGrowthManager: print("✨ 测试完成!") print(" = " * 60) - async def main() -> None: """主函数""" tester = TestGrowthManager() await tester.run_all_tests() - if __name__ == "__main__": asyncio.run(main()) diff --git a/backend/test_phase8_task6.py b/backend/test_phase8_task6.py index 0c6d65a..18e64be 100644 --- a/backend/test_phase8_task6.py +++ b/backend/test_phase8_task6.py @@ -29,7 +29,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__)) if backend_dir not in sys.path: sys.path.insert(0, backend_dir) - class TestDeveloperEcosystem: """开发者生态系统测试类""" @@ -692,12 +691,10 @@ console.log('Upload complete:', result.id); print(" = " * 60) - def main() -> None: """主函数""" test = TestDeveloperEcosystem() test.run_all_tests() - if __name__ == "__main__": main() diff --git a/backend/test_phase8_task8.py b/backend/test_phase8_task8.py index 039ae70..0152034 100644 --- a/backend/test_phase8_task8.py +++ b/backend/test_phase8_task8.py @@ -30,7 +30,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__)) if backend_dir not in sys.path: sys.path.insert(0, backend_dir) - class TestOpsManager: """测试运维与监控管理器""" @@ -244,7 +243,6 @@ class TestOpsManager: self.log("Recorded 10 resource metrics") # 手动创建告警 - from ops_manager import Alert alert_id = f"test_alert_{datetime.now().strftime('%Y%m%d%H%M%S')}" now = datetime.now().isoformat() @@ -733,12 +731,10 @@ class TestOpsManager: print(" = " * 60) - def main() -> None: """主函数""" test = TestOpsManager() test.run_all_tests() - if __name__ == "__main__": main() diff --git a/backend/tingwu_client.py b/backend/tingwu_client.py index 1fbe97a..aee33e3 100644 --- a/backend/tingwu_client.py +++ b/backend/tingwu_client.py @@ -8,7 +8,6 @@ import time from datetime import datetime from typing import Any - class TingwuClient: def __init__(self) -> None: self.access_key = os.getenv("ALI_ACCESS_KEY", "") @@ -89,8 +88,6 @@ class TingwuClient: try: # 导入移到文件顶部会导致循环导入，保持在这里 from alibabacloud_openapi_util import models as open_api_models - from alibabacloud_tingwu20230930 import models as tingwu_models - from alibabacloud_tingwu20230930.client import Client as TingwuSDKClient config = open_api_models.Config( access_key_id=self.access_key, diff --git a/backend/workflow_manager.py b/backend/workflow_manager.py index e067b8f..482a916 100644 --- a/backend/workflow_manager.py +++ b/backend/workflow_manager.py @@ -15,7 +15,6 @@ import hashlib import hmac import json import logging -import urllib.parse import uuid from collections.abc import Callable from dataclasses import dataclass, field @@ -39,7 +38,6 @@ DEFAULT_RETRY_DELAY = 5 # 默认重试延迟(秒) logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) - class WorkflowStatus(Enum): """工作流状态""" @@ -48,7 +46,6 @@ class WorkflowStatus(Enum): ERROR = "error" COMPLETED = "completed" - class WorkflowType(Enum): """工作流类型""" @@ -58,7 +55,6 @@ class WorkflowType(Enum): SCHEDULED_REPORT = "scheduled_report" # 定时报告 CUSTOM = "custom" # 自定义工作流 - class WebhookType(Enum): """Webhook 类型""" @@ -67,7 +63,6 @@ class WebhookType(Enum): SLACK = "slack" CUSTOM = "custom" - class TaskStatus(Enum): """任务执行状态""" @@ -77,7 +72,6 @@ class TaskStatus(Enum): FAILED = "failed" CANCELLED = "cancelled" - @dataclass class WorkflowTask: """工作流任务定义""" @@ -101,7 +95,6 @@ class WorkflowTask: if not self.updated_at: self.updated_at = self.created_at - @dataclass class WebhookConfig: """Webhook 配置""" @@ -126,7 +119,6 @@ class WebhookConfig: if not self.updated_at: self.updated_at = self.created_at - @dataclass class Workflow: """工作流定义""" @@ -156,7 +148,6 @@ class Workflow: if not self.updated_at: self.updated_at = self.created_at - @dataclass class WorkflowLog: """工作流执行日志""" @@ -177,7 +168,6 @@ class WorkflowLog: if not self.created_at: self.created_at = datetime.now().isoformat() - class WebhookNotifier: """Webhook 通知器 - 支持飞书、钉钉、Slack""" @@ -335,7 +325,6 @@ class WebhookNotifier: """关闭 HTTP 客户端""" await self.http_client.aclose() - class WorkflowManager: """工作流管理器 - 核心管理类""" @@ -1520,11 +1509,9 @@ class WorkflowManager: ], } - # Singleton instance _workflow_manager = None - def get_workflow_manager(db_manager=None) -> WorkflowManager: """获取 WorkflowManager 单例""" global _workflow_manager diff --git a/code_review_fixer.py b/code_review_fixer.py index 9556ae9..b734d11 100644 --- a/code_review_fixer.py +++ b/code_review_fixer.py @@ -15,7 +15,6 @@ PROJECT_PATH = Path("/root/.openclaw/workspace/projects/insightflow") # 修复报告 report = {"fixed": [], "manual_review": [], "errors": []} - def find_python_files() -> list[Path]: """查找所有 Python 文件""" py_files = [] @@ -24,7 +23,6 @@ def find_python_files() -> list[Path]: py_files.append(py_file) return py_files - def check_duplicate_imports(content: str, file_path: Path) -> list[dict]: """检查重复导入""" issues = [] @@ -47,7 +45,6 @@ def check_duplicate_imports(content: str, file_path: Path) -> list[dict]: imports[line_stripped] = i return issues - def check_bare_excepts(content: str, file_path: Path) -> list[dict]: """检查裸异常捕获""" issues = [] @@ -60,7 +57,6 @@ def check_bare_excepts(content: str, file_path: Path) -> list[dict]: issues.append({"line": i, "type": "bare_except", "content": stripped}) return issues - def check_line_length(content: str, file_path: Path) -> list[dict]: """检查行长度（PEP8: 79字符，这里放宽到 100）""" issues = [] @@ -78,7 +74,6 @@ def check_line_length(content: str, file_path: Path) -> list[dict]: ) return issues - def check_unused_imports(content: str, file_path: Path) -> list[dict]: """检查未使用的导入""" issues = [] @@ -108,7 +103,6 @@ def check_unused_imports(content: str, file_path: Path) -> list[dict]: pass return issues - def check_string_formatting(content: str, file_path: Path) -> list[dict]: """检查混合字符串格式化（建议使用 f-string）""" issues = [] @@ -133,7 +127,6 @@ def check_string_formatting(content: str, file_path: Path) -> list[dict]: ) return issues - def check_magic_numbers(content: str, file_path: Path) -> list[dict]: """检查魔法数字""" issues = [] @@ -176,7 +169,6 @@ def check_magic_numbers(content: str, file_path: Path) -> list[dict]: ) return issues - def check_sql_injection(content: str, file_path: Path) -> list[dict]: """检查 SQL 注入风险""" issues = [] @@ -203,7 +195,6 @@ def check_sql_injection(content: str, file_path: Path) -> list[dict]: ) return issues - def check_cors_config(content: str, file_path: Path) -> list[dict]: """检查 CORS 配置""" issues = [] @@ -221,7 +212,6 @@ def check_cors_config(content: str, file_path: Path) -> list[dict]: ) return issues - def fix_bare_excepts(content: str) -> str: """修复裸异常捕获""" lines = content.split("\n") @@ -239,7 +229,6 @@ def fix_bare_excepts(content: str) -> str: return "\n".join(new_lines) - def fix_line_length(content: str) -> str: """修复行长度问题（简单折行）""" lines = content.split("\n") @@ -258,7 +247,6 @@ def fix_line_length(content: str) -> str: return "\n".join(new_lines) - def analyze_file(file_path: Path) -> dict: """分析单个文件""" try: @@ -279,7 +267,6 @@ def analyze_file(file_path: Path) -> dict: return issues - def fix_file(file_path: Path, issues: dict) -> bool: """自动修复文件问题""" try: @@ -299,7 +286,6 @@ def fix_file(file_path: Path, issues: dict) -> bool: report["errors"].append(f"{file_path}: {e}") return False - def generate_report(all_issues: dict) -> str: """生成修复报告""" lines = [] @@ -368,7 +354,6 @@ def generate_report(all_issues: dict) -> str: return "\n".join(lines) - def git_commit_and_push() -> None: """提交并推送代码""" try: @@ -410,7 +395,6 @@ def git_commit_and_push() -> None: except Exception as e: return f"❌ 错误: {e}" - def main() -> None: """主函数""" print("🔍 开始代码审查...") @@ -448,6 +432,5 @@ def main() -> None: print("\n✅ 代码审查完成!") return report_content - if __name__ == "__main__": main() diff --git a/code_reviewer.py b/code_reviewer.py index 9638c64..357b6f9 100644 --- a/code_reviewer.py +++ b/code_reviewer.py @@ -7,7 +7,6 @@ import ast import re from pathlib import Path - class CodeIssue: def __init__( self, @@ -27,7 +26,6 @@ class CodeIssue: def __repr__(self) -> str: return f"{self.severity.upper()}: {self.file_path}:{self.line_no} - {self.issue_type}: {self.message}" - class CodeReviewer: def __init__(self, base_path: str) -> None: self.base_path = Path(base_path) @@ -422,7 +420,6 @@ class CodeReviewer: return "\n".join(report) - def main() -> None: base_path = "/root/.openclaw/workspace/projects/insightflow/backend" reviewer = CodeReviewer(base_path) @@ -447,6 +444,5 @@ def main() -> None: return reviewer - if __name__ == "__main__": main()