feat: 迁移奶龙风控业务模块与演示文档

2026-09-10 21:03:44 +08:00
parent 6516ccb385
commit a94d5c754d
62 changed files with 10815 additions and 0 deletions
@@ -0,0 +1,123 @@
+"""创建风控 Agent 本地验收所需的模型端点和发布配置。"""
+
+from __future__ import annotations
+
+import asyncio
+import hashlib
+import json
+from datetime import UTC, datetime
+
+from sqlalchemy import text
+
+from app.infrastructure.db import SessionFactory
+
+ADMIN_USER_ID = 9003
+ENDPOINT_CODE = "deepseek-flash"
+RELEASE_NO = "risk-agent-local-v1"
+TOOL_CONFIGS = (
+    ("risk:risk_overview", {"allowed_tools": ["get_risk_overview"]}),
+    ("risk:risk_search", {"allowed_tools": ["search_risk_alerts"]}),
+    ("risk:risk_evidence", {"allowed_tools": ["get_alert_evidence"]}),
+    ("risk:general", {"allowed_tools": []}),
+)
+INTENT_CONFIGS = (
+    ("risk_overview", "风险概览", ["请查看当前风险概览", "当前有多少高风险预警"], ["get_risk_overview"]),
+    ("risk_search", "风险查询", ["查询高风险预警", "查看规则 RW-007 命中的预警"], ["search_risk_alerts"]),
+    ("risk_evidence", "预警证据", ["查询预警编号 ALERT-001 的证据", "查看这条预警的证据链"], ["get_alert_evidence"]),
+    ("general", "通用风险咨询", ["奶龙风控智能助手能做什么", "说明你的功能边界"], []),
+)
+
+
+def checksum(value: dict) -> str:
+    payload = json.dumps(value, ensure_ascii=False, sort_keys=True, separators=(",", ":"))
+    return hashlib.sha256(payload.encode("utf-8")).hexdigest()
+
+
+async def seed() -> None:
+    now = datetime.now(UTC).replace(tzinfo=None)
+    async with SessionFactory() as session:
+        await session.execute(text("""
+            INSERT INTO model_endpoint_config
+            (endpoint_code, provider, model_name, base_url, secret_ref, capabilities,
+             allowed_data_levels, context_window, timeout_ms, status, created_by,
+             reviewer_id, reviewed_at, created_at, updated_at)
+            VALUES
+            (:endpoint_code, 'deepseek', 'deepseek-chat', 'https://api.deepseek.com',
+             'env:DEEPSEEK_API_KEY', :capabilities, :data_levels, 64000, 30000,
+             'active', :admin_id, :admin_id, :now, :now, :now)
+            ON DUPLICATE KEY UPDATE
+              provider=VALUES(provider), model_name=VALUES(model_name),
+              base_url=VALUES(base_url), secret_ref=VALUES(secret_ref),
+              capabilities=VALUES(capabilities), allowed_data_levels=VALUES(allowed_data_levels),
+              context_window=VALUES(context_window), timeout_ms=VALUES(timeout_ms),
+              status='active', reviewer_id=:admin_id, reviewed_at=:now, updated_at=:now
+        """), {
+            "endpoint_code": ENDPOINT_CODE,
+            "capabilities": json.dumps(["chat", "intent_classification", "risk_answer"]),
+            "data_levels": json.dumps(["internal"]),
+            "admin_id": ADMIN_USER_ID,
+            "now": now,
+        })
+
+        release_id = await session.scalar(
+            text("SELECT id FROM config_release WHERE status='active' LIMIT 1")
+        )
+        if release_id is None:
+            result = await session.execute(text("""
+                INSERT INTO config_release
+                (release_no, title, change_summary, status, created_by, reviewer_id,
+                 reviewed_at, activated_at, created_at, updated_at)
+                VALUES (:release_no, '奶龙风控智能助手本地配置',
+                        '发布 risk Agent 工具白名单和意图配置', 'active',
+                        :admin_id, :admin_id, :now, :now, :now, :now)
+            """), {"release_no": RELEASE_NO, "admin_id": ADMIN_USER_ID, "now": now})
+            release_id = int(result.lastrowid)
+
+        for config_key, value in TOOL_CONFIGS:
+            await session.execute(text("""
+                INSERT INTO platform_config_item
+                (release_id, namespace, config_key, value_json, schema_version, checksum, created_at)
+                VALUES (:release_id, 'agent_tools', :config_key, :value_json, '1', :checksum, :now)
+                ON DUPLICATE KEY UPDATE
+                  value_json=VALUES(value_json), schema_version=VALUES(schema_version),
+                  checksum=VALUES(checksum)
+            """), {
+                "release_id": release_id,
+                "config_key": config_key,
+                "value_json": json.dumps(value, ensure_ascii=False),
+                "checksum": checksum(value),
+                "now": now,
+            })
+
+        for intent_code, intent_name, examples, allowed_tools in INTENT_CONFIGS:
+            await session.execute(text("""
+                INSERT INTO agent_intent_config
+                (agent_type, intent_code, intent_name, description, examples,
+                 classifier_instruction, confidence_threshold, max_clarification_rounds,
+                 transfer_on_failure, allowed_tools, priority, version, status,
+                 effective_at, created_by, reviewer_id, reviewed_at, created_at, updated_at)
+                VALUES
+                ('risk', :intent_code, :intent_name, :description, :examples,
+                 :instruction, 0.6500, 2, 1, :allowed_tools, 100, 1, 'active',
+                 :now, :admin_id, :admin_id, :now, :now, :now)
+                ON DUPLICATE KEY UPDATE
+                  intent_name=VALUES(intent_name), description=VALUES(description),
+                  examples=VALUES(examples), classifier_instruction=VALUES(classifier_instruction),
+                  allowed_tools=VALUES(allowed_tools), status='active',
+                  effective_at=:now, reviewer_id=:admin_id, reviewed_at=:now, updated_at=:now
+            """), {
+                "intent_code": intent_code,
+                "intent_name": intent_name,
+                "description": f"奶龙风控智能助手：{intent_name}",
+                "examples": json.dumps(examples, ensure_ascii=False),
+                "instruction": "只处理风控只读查询、分析和边界说明，不执行人工处置。",
+                "allowed_tools": json.dumps(allowed_tools, ensure_ascii=False),
+                "admin_id": ADMIN_USER_ID,
+                "now": now,
+            })
+        await session.commit()
+    print(f"risk_agent_config_ready release_id={release_id} endpoint={ENDPOINT_CODE}")
+
+
+if __name__ == "__main__":
+    asyncio.run(seed())