GuangYu/QuestionProject
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: expose knowledge type priority traces

Browse Source
This commit is contained in:
Nelson
2026-07-31 16:09:27 +08:00
parent 429636f3e8
commit 4e872febd5
5 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ai_knowledge_base_v2/apps/backend/app/services/knowledge_agent_service.py
View File

@@ -154,7 +154,7 @@ class KnowledgeAgentService:
candidate_limit=candidate_limit,
prefer_fixed_information=cls._is_fixed_information_query(retrieval_question),
)
trace.append(cls._trace("search_knowledge", len(trace) + 1, {"queryTerms": terms, "knowledgeIds": selected_ids, "candidateLimit": candidate_limit}, {"candidateCount": len(candidates), "candidates": [cls._candidate_trace(x) for x in candidates]}, started))
trace.append(cls._trace("search_knowledge", len(trace) + 1, {"queryTerms": terms, "knowledgeIds": selected_ids, "candidateLimit": candidate_limit, "preferFixedInformation": cls._is_fixed_information_query(retrieval_question)}, {"candidateCount": len(candidates), "candidates": [cls._candidate_trace(x) for x in candidates]}, started))
await cls._rerank(db, retrieval_question, candidates, trace, started)
selected = cls._select_sections(
candidates,
@@ -391,7 +391,7 @@ class KnowledgeAgentService:
@staticmethod
def _candidate_trace(item: Candidate) -> dict:
return {"knowledgeId": item.knowledge.id, "knowledgeName": item.knowledge.name, "versionId": item.version.id, "chunkId": item.chunk.id, "sectionId": item.section.id, "title": item.chunk.title, "lexicalScore": item.lexical_score, "rerankScore": item.rerank_score, "selected": item.selected, "discardReason": item.discard_reason}
return {"knowledgeId": item.knowledge.id, "knowledgeName": item.knowledge.name, "knowledgeType": item.knowledge.knowledge_type, "knowledgeTypePriority": _knowledge_type_priority_label(item.knowledge.knowledge_type), "versionId": item.version.id, "chunkId": item.chunk.id, "sectionId": item.section.id, "title": item.chunk.title, "lexicalScore": item.lexical_score, "rerankScore": item.rerank_score, "selected": item.selected, "discardReason": item.discard_reason}
@classmethod
def search_knowledge(
@@ -732,6 +732,15 @@ def _json_trace_value(value):
return str(value)
def _knowledge_type_priority_label(value: str) -> str:
return {
"fixed": "最高优先级：固定信息类冲突时优先",
"course": "课程内容优先于答疑和通用知识",
"qa": "过往答疑参考，不覆盖课程原文",
"general": "背景补充，不覆盖业务知识",
}.get(value, "背景补充")
def _heading_level(content: str) -> int | None:
match = re.match(r"^(#{1,6})\s+", content.lstrip())
return len(match.group(1)) if match else None