袁聪的第一次提交,包含nl2sql,行情数据,场外申购

This commit is contained in:
2026-09-10 09:23:22 +08:00
parent 46fc976b24
commit 5907fcd6d2
49 changed files with 6801 additions and 33 deletions
+17 -1
View File
@@ -2,7 +2,10 @@ from functools import lru_cache
from typing import Any, cast
from app.core.fund_contracts import FundQuoteQuery
from app.core.nl2sql_contracts import FinancialNL2SQLInput
from app.service.agent.factory import AgentFactory
from app.service.agent.offsite_fund_agent import OffsiteFundAgent
from app.service.financial_nl2sql_service import query_financial_data_tool
from app.service.fund_quote_service import query_fund_quote_tool
from app.service.intent_classifier import IntentClassifier
from app.service.model_gateway import (
@@ -34,11 +37,24 @@ def get_agent_factory() -> AgentFactory:
allowed_roles=("customer", "advisor", "operator", "risk_operator", "admin"),
timeout_seconds=5,
))
registry.register(ToolDefinition(
name="query_financial_data",
input_model=FinancialNL2SQLInput,
handler=cast(Any, query_financial_data_tool),
required_permission="financial:nl2sql:read",
allowed_roles=("advisor", "operator", "admin", "super_admin"),
timeout_seconds=10,
))
model_service = ModelGenerationService(ModelDispatchService(DatabaseModelGateway()))
endpoint_resolver = DatabaseModelEndpointResolver()
return AgentFactory(
factory = AgentFactory(
model_service=model_service,
tool_executor=ToolExecutor(registry),
intent_classifier=IntentClassifier(model_service),
intent_endpoint_resolver=endpoint_resolver,
)
factory.register(
OffsiteFundAgent.definition,
lambda _context: OffsiteFundAgent(OffsiteFundAgent.definition),
)
return factory