feat: add governed public knowledge retrieval

This commit is contained in:
张胜宇
2026-09-10 18:42:44 +08:00
parent ba22a2220f
commit 511a8ca18f
20 changed files with 729 additions and 7 deletions
+11
View File
@@ -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(