feat: 更新 user_agent 服务,增强用户代理功能

This commit is contained in:
caoxiaozhu
2026-05-18 02:50:32 +00:00
parent 4414ffb34c
commit 35a3783481
2 changed files with 394 additions and 55 deletions

View File

@@ -202,21 +202,37 @@ def test_user_agent_knowledge_answer_generation_uses_fast_timeouts(monkeypatch)
assert captured["max_attempts"] == 1
def test_user_agent_prefers_structured_knowledge_hit_for_answer_generation() -> None:
def test_user_agent_prefers_structured_table_hit_for_standard_query() -> None:
selected = UserAgentService._select_knowledge_model_hits(
{
"hits": [
{"content": "raw hit 1"},
{"content": "raw hit 2"},
{"content": "# 问答线索补充\n\n- 第二章 报销时限:费用发生后 30 日内提交申请。"},
{"content": "# 结构化表格补充\n\n| 项目 | 金额 |"},
{"content": "# 结构化表格补充\n\n| 项目 | 餐补 |\n| 其他地区 | 55 |"},
]
}
},
question="餐补标准是多少?",
)
assert selected[0]["content"].startswith("# 问答线索补充")
assert selected[1]["content"].startswith("# 结构化表格补充")
assert selected[2]["content"] == "raw hit 1"
assert selected[0]["content"].startswith("# 结构化表格补充")
assert any(item["content"].startswith("# 结构化表格补充") for item in selected[:2])
def test_user_agent_prefers_relevant_raw_hit_over_generic_appendix() -> None:
selected = UserAgentService._select_knowledge_model_hits(
{
"hits": [
{"content": "# 章节导航\n\n- 第一章 总则\n- 第二章 职责分工"},
{"content": "# 问答线索补充\n\n- 第二章 职责分工:计划财务部负责财务审核。"},
{"content": "一般性说明文字,没有探亲差旅归口信息。"},
{"content": "附表3支出归口管理部门与归口业务范围\n组织人事部:探亲差旅、条件艰苦及安全风险较高区域补助等支出。"},
]
},
question="探亲差旅归哪个部门管理?",
)
assert "组织人事部" in selected[0]["content"]
def test_user_agent_uses_fast_knowledge_answer_without_model(monkeypatch) -> None:
@@ -266,6 +282,7 @@ def test_user_agent_uses_fast_knowledge_answer_without_model(monkeypatch) -> Non
assert response.answer.startswith("张三,您好。")
assert "当前能直接确认的是" in response.answer
assert "30 日内提交报销申请" in response.answer
assert "## 依据" not in response.answer
assert "答案整理阶段本轮没有及时返回" not in response.answer
@@ -314,6 +331,7 @@ def test_user_agent_fast_knowledge_answer_renders_relevant_table_preview() -> No
assert answer is not None
assert "| 项目 | 港澳台 | 其他地区 | 国外 |" in answer
assert "| 餐补 | 75 | 55 | 140 |" in answer
assert "## 依据" not in answer
def test_user_agent_fast_knowledge_answer_notes_missing_location_grounding() -> None:
@@ -360,6 +378,57 @@ def test_user_agent_fast_knowledge_answer_notes_missing_location_grounding() ->
assert answer is not None
assert "没有直接写出“北京”对应的地区档位或映射关系" in answer
assert "## 依据" not in answer
def test_user_agent_fast_knowledge_answer_expands_lead_in_list_items() -> None:
session_factory = build_session_factory()
with session_factory() as db:
ontology = SemanticOntologyService(db).parse(
OntologyParseRequest(
query="出差记录链条中断时,要提供哪些业务佐证材料?",
user_id="pytest",
context_json={"session_type": "knowledge"},
)
)
service = UserAgentService(db)
answer = service._build_fast_knowledge_answer(
UserAgentRequest(
run_id=ontology.run_id,
user_id="pytest",
message="出差记录链条中断时,要提供哪些业务佐证材料?",
ontology=ontology,
context_json={
"session_type": "knowledge",
"user_input_text": "出差记录链条中断时,要提供哪些业务佐证材料?",
},
tool_payload={
"result_type": "knowledge_search",
"hits": [
{
"title": "费用报销制度",
"content": (
"第十三条 差旅费\n\n"
"2出差记录链条中断时应提供业务佐证材料\n"
"① 登机牌、高速道路通行记录、其他道路通行记录、租车记录等。\n"
"② 支付记录。\n"
"③ 出差审批邮件、短信、微信等。"
),
}
],
},
),
citations=[],
)
assert answer is not None
assert "当前能直接确认的是" in answer
assert "登机牌、高速道路通行记录" in answer
assert "支付记录" in answer
assert "出差审批邮件、短信、微信等" in answer
assert "3" not in answer
assert "## 依据" not in answer
def test_user_agent_model_prompt_supports_contextual_personalization() -> None: