feat: 迁移奶龙风控业务模块与演示文档
This commit is contained in:
@@ -7,12 +7,14 @@ from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from app.core.config import get_settings
|
||||
from app.core.errors import RecoverableAgentError
|
||||
from app.core.fund_contracts import FundQuoteQuery
|
||||
from app.core.risk_contracts import RiskAlertEvidenceQuery, RiskAlertQuery
|
||||
from app.infrastructure.fund_quote_cache import FundQuoteCache
|
||||
from app.infrastructure.memory_cache import MemoryCacheAdapter
|
||||
from app.infrastructure.vector_memory import VectorMemoryAdapter
|
||||
from app.service.agent.factory import AgentFactory
|
||||
from app.service.agent.governance import PlatformGovernance
|
||||
from app.service.agent.implementations.fund_query_demo import FundQueryDemoAgent
|
||||
from app.service.agent.implementations.risk_agent import RiskAgent
|
||||
from app.service.fund_quote_service import query_fund_quote_tool
|
||||
from app.service.intent_classifier import IntentClassifier
|
||||
from app.service.memory_recall_service import MemoryRecallService
|
||||
@@ -23,6 +25,11 @@ from app.service.model_gateway import (
|
||||
ModelEmbeddingService,
|
||||
ModelGenerationService,
|
||||
)
|
||||
from app.service.risk_tools import (
|
||||
get_alert_evidence_tool,
|
||||
get_risk_overview_tool,
|
||||
search_risk_alerts_tool,
|
||||
)
|
||||
from app.service.runtime_config_service import load_active_intent_configs
|
||||
from app.service.suitability_service import SuitabilityToolInput, suitability_tool_handler
|
||||
from app.service.tool_executor import ToolDefinition, ToolExecutor, ToolRegistry
|
||||
@@ -143,6 +150,27 @@ def get_agent_factory() -> AgentFactory:
|
||||
# (12s)与重试预算,多代码查询必然先撞工具超时。
|
||||
timeout_seconds=15,
|
||||
))
|
||||
registry.register(ToolDefinition(
|
||||
name="search_risk_alerts",
|
||||
input_model=RiskAlertQuery,
|
||||
handler=cast(Any, search_risk_alerts_tool),
|
||||
required_permission="risk:alert:read",
|
||||
allowed_roles=("risk_operator", "admin"),
|
||||
))
|
||||
registry.register(ToolDefinition(
|
||||
name="get_risk_overview",
|
||||
input_model=RiskAlertQuery,
|
||||
handler=cast(Any, get_risk_overview_tool),
|
||||
required_permission="risk:alert:read",
|
||||
allowed_roles=("risk_operator", "admin"),
|
||||
))
|
||||
registry.register(ToolDefinition(
|
||||
name="get_alert_evidence",
|
||||
input_model=RiskAlertEvidenceQuery,
|
||||
handler=cast(Any, get_alert_evidence_tool),
|
||||
required_permission="risk:alert:read",
|
||||
allowed_roles=("risk_operator", "admin"),
|
||||
))
|
||||
model_service = get_model_service()
|
||||
endpoint_resolver = DatabaseModelEndpointResolver()
|
||||
factory = AgentFactory(
|
||||
@@ -172,3 +200,7 @@ def register_business_agents(factory: AgentFactory) -> None:
|
||||
FundQueryDemoAgent.definition,
|
||||
lambda _context: FundQueryDemoAgent(FundQueryDemoAgent.definition),
|
||||
)
|
||||
factory.register(
|
||||
RiskAgent.definition,
|
||||
lambda _context: RiskAgent(RiskAgent.definition),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user