61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
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 (
|
|
DatabaseModelEndpointResolver,
|
|
DatabaseModelGateway,
|
|
ModelDispatchService,
|
|
ModelGenerationService,
|
|
)
|
|
from app.service.suitability_service import SuitabilityToolInput, suitability_tool_handler
|
|
from app.service.tool_executor import ToolDefinition, ToolExecutor, ToolRegistry
|
|
|
|
|
|
@lru_cache(maxsize=1)
|
|
def get_agent_factory() -> AgentFactory:
|
|
"""HTTP 与 Worker 共用的唯一底座依赖组装入口。"""
|
|
registry = ToolRegistry()
|
|
registry.register(ToolDefinition(
|
|
name="check_suitability",
|
|
input_model=SuitabilityToolInput,
|
|
handler=cast(Any, suitability_tool_handler),
|
|
required_permission="suitability:read",
|
|
allowed_roles=("customer", "advisor", "operator", "admin"),
|
|
))
|
|
registry.register(ToolDefinition(
|
|
name="query_fund_quote",
|
|
input_model=FundQuoteQuery,
|
|
handler=cast(Any, query_fund_quote_tool),
|
|
required_permission="fund:quote:read",
|
|
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()
|
|
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
|