feat:新增投顾agent和nl2sqlagent

2026-09-13 16:19:24 +08:00
parent c80c6acac0
commit 163192bf55
122 changed files with 7488 additions and 362 deletions
@@ -0,0 +1,76 @@
+"""NL2SQL 管理接口请求模型。"""
+from __future__ import annotations
+
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class RoleCreateReq(BaseModel):
+    role_code: str = Field(min_length=1, max_length=64)
+    role_name: str = Field(min_length=1, max_length=128)
+    employee_role: str = Field(min_length=1, max_length=32)
+    can_query: bool = False
+    max_rows: int = Field(default=1000, gt=0, le=100000)
+    daily_quota: int = Field(default=0, ge=0, le=1000000)
+
+
+class RoleUpdateReq(BaseModel):
+    role_name: str | None = Field(default=None, min_length=1, max_length=128)
+    can_query: bool | None = None
+    max_rows: int | None = Field(default=None, gt=0, le=100000)
+    daily_quota: int | None = Field(default=None, ge=0, le=1000000)
+    status: Literal["active", "inactive"] | None = None
+
+
+class TablePermissionCreateReq(BaseModel):
+    table_name: str = Field(min_length=1, max_length=128)
+    row_scope_type: Literal["none", "customer_ids", "product_ids"] = "none"
+    row_scope_column: str | None = Field(default=None, max_length=128)
+
+
+class TablePermissionUpdateReq(BaseModel):
+    row_scope_type: Literal["none", "customer_ids", "product_ids"] | None = None
+    row_scope_column: str | None = Field(default=None, max_length=128)
+    status: Literal["active", "inactive"] | None = None
+
+
+class ColumnPermissionCreateReq(BaseModel):
+    table_name: str = Field(min_length=1, max_length=128)
+    column_name: str = Field(min_length=1, max_length=128)
+    access_mode: Literal["allow", "deny", "mask"] = "allow"
+    mask_type: Literal["partial", "hash"] | None = None
+
+
+class ColumnPermissionUpdateReq(BaseModel):
+    access_mode: Literal["allow", "deny", "mask"] | None = None
+    mask_type: Literal["partial", "hash"] | None = None
+    status: Literal["active", "inactive"] | None = None
+
+
+class SensitiveFieldCreateReq(BaseModel):
+    table_name: str = Field(min_length=1, max_length=128)
+    column_name: str = Field(min_length=1, max_length=128)
+    mask_type: Literal["partial", "hash"] = "partial"
+    description: str | None = Field(default=None, max_length=255)
+
+
+class SensitiveFieldUpdateReq(BaseModel):
+    mask_type: Literal["partial", "hash"] | None = None
+    description: str | None = Field(default=None, max_length=255)
+    status: Literal["active", "inactive"] | None = None
+
+
+class MaintenanceJobReq(BaseModel):
+    task: Literal["metadata_sync", "vector_cleanup", "consistency_check", "history_cleanup"]
+    before_days: int = Field(default=180, ge=1, le=3650)
+
+
+class RuntimeConfigUpdateReq(BaseModel):
+    """管理员动态调整 NL2SQL 非敏感运行参数。"""
+
+    cache_ttl: int | None = Field(default=None, ge=30, le=86400)
+    retrieval_top_k: int | None = Field(default=None, ge=1, le=50)
+    retrieval_threshold: float | None = Field(default=None, ge=0.0, le=1.0)
+    max_rows: int | None = Field(default=None, ge=1, le=100000)
+    max_join_depth: int | None = Field(default=None, ge=0, le=10)