group_fqcd_jr/app/service/agent/bootstrap.py

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