feature:基本框架

2026-09-08 19:17:35 +08:00
commit 8698f1f53a
56 changed files with 3320 additions and 0 deletions
@@ -0,0 +1,37 @@
+"""统一响应格式：{code, message, data, trace_id}，错误码与需求文档 §6.1 对齐。"""
+from typing import Any
+
+from pydantic import BaseModel
+
+from utils.request_id import get_request_id
+
+
+class Code:
+    OK = 200
+    PARAM_ERROR = 400
+    UNAUTHORIZED = 401
+    FORBIDDEN = 403
+    NOT_FOUND = 404
+    SERVER_ERROR = 500
+    LLM_FAIL = 1001          # LLM 调用失败
+    KB_NO_RESULT = 1002      # 知识库检索无结果
+    SQL_GEN_FAIL = 1003      # NL2SQL 生成失败
+    RISK_TRIGGERED = 1004    # 风控规则触发 / 交易被拦截
+    NOT_SUITABLE = 1005      # 适当性不匹配
+
+
+class ApiResponse(BaseModel):
+    code: int = Code.OK
+    message: str = "success"
+    data: Any = None
+    trace_id: str | None = None
+
+
+def success(data: Any = None, message: str = "success") -> ApiResponse:
+    """成功响应。"""
+    return ApiResponse(code=Code.OK, message=message, data=data, trace_id=get_request_id())
+
+
+def fail(code: int, message: str, data: Any = None) -> ApiResponse:
+    """失败响应（业务错误码见 Code）。"""
+    return ApiResponse(code=code, message=message, data=data, trace_id=get_request_id())