from functools import lru_cache from typing import Any, cast from app.service.agent.factory import AgentFactory 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"), )) model_service = ModelGenerationService(ModelDispatchService(DatabaseModelGateway())) endpoint_resolver = DatabaseModelEndpointResolver() return AgentFactory( model_service=model_service, tool_executor=ToolExecutor(registry), intent_classifier=IntentClassifier(model_service), intent_endpoint_resolver=endpoint_resolver, )