feat: add governed public knowledge retrieval
This commit is contained in:
@@ -7,6 +7,7 @@ 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.knowledge_contracts import KnowledgeQuery
|
||||
from app.core.nl2sql_contracts import FinancialNL2SQLInput
|
||||
from app.infrastructure.fund_quote_cache import FundQuoteCache
|
||||
from app.infrastructure.memory_cache import MemoryCacheAdapter
|
||||
@@ -18,6 +19,7 @@ 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.knowledge_tool_service import query_knowledge_tool
|
||||
from app.service.memory_recall_service import MemoryRecallService
|
||||
from app.service.model_gateway import (
|
||||
DatabaseModelEndpointResolver,
|
||||
@@ -154,6 +156,15 @@ def get_agent_factory() -> AgentFactory:
|
||||
allowed_roles=("advisor", "operator", "admin", "super_admin"),
|
||||
timeout_seconds=10,
|
||||
))
|
||||
registry.register(ToolDefinition(
|
||||
name="query_knowledge",
|
||||
input_model=KnowledgeQuery,
|
||||
handler=cast(Any, query_knowledge_tool),
|
||||
required_permission="knowledge:query",
|
||||
allowed_roles=("visitor", "customer"),
|
||||
# 15 秒向量端点预算外预留检索和权威回查时间,防止正常降级被工具层提前中断。
|
||||
timeout_seconds=20,
|
||||
))
|
||||
model_service = get_model_service()
|
||||
endpoint_resolver = DatabaseModelEndpointResolver()
|
||||
factory = AgentFactory(
|
||||
|
||||
Reference in New Issue
Block a user