feat(chat): 接入全局知识库上下文
This commit is contained in:
59
tests/test_chat_knowledge_context.py
Normal file
59
tests/test_chat_knowledge_context.py
Normal file
@@ -0,0 +1,59 @@
|
||||
import pytest
|
||||
|
||||
from review_agent.models import KnowledgeBaseDocument
|
||||
from review_agent.services import build_knowledge_context
|
||||
|
||||
|
||||
pytestmark = pytest.mark.django_db
|
||||
|
||||
|
||||
def test_build_knowledge_context_ignores_irrelevant_rag_chunks(monkeypatch):
|
||||
monkeypatch.setattr(
|
||||
"review_agent.services.search_knowledge_base",
|
||||
lambda query, n_results=5: {
|
||||
"query": query,
|
||||
"results": [
|
||||
{
|
||||
"source": "附件 4 体外诊断试剂注册申报资料要求及说明.doc",
|
||||
"text": "预期用途应明确产品用于检测的分析物和功能。",
|
||||
"score": 7.636,
|
||||
"metadata": {"source_type": "regulatory_document"},
|
||||
}
|
||||
],
|
||||
"error_message": "",
|
||||
},
|
||||
)
|
||||
|
||||
context = build_knowledge_context("孙之烨是谁")
|
||||
|
||||
assert context == ""
|
||||
|
||||
|
||||
def test_build_knowledge_context_uses_full_document_when_name_matches(settings, tmp_path, monkeypatch, django_user_model):
|
||||
settings.MEDIA_ROOT = tmp_path
|
||||
user = django_user_model.objects.create_user(username="owner", password="pass")
|
||||
document_path = tmp_path / "resume.txt"
|
||||
document_path.write_text(
|
||||
"孙之烨,负责审核智能体项目。\n完整经历:曾组织技术分享并带队参加竞赛。",
|
||||
encoding="utf-8",
|
||||
)
|
||||
KnowledgeBaseDocument.objects.create(
|
||||
user=user,
|
||||
display_name="孙之烨简历",
|
||||
original_name="孙之烨-260510.txt",
|
||||
storage_path=str(document_path),
|
||||
file_size=document_path.stat().st_size,
|
||||
status=KnowledgeBaseDocument.Status.ACTIVE,
|
||||
is_active=True,
|
||||
indexed_chunk_count=2,
|
||||
)
|
||||
monkeypatch.setattr(
|
||||
"review_agent.services.search_knowledge_base",
|
||||
lambda query, n_results=5: {"query": query, "results": [], "error_message": ""},
|
||||
)
|
||||
|
||||
context = build_knowledge_context("孙之烨是谁")
|
||||
|
||||
assert "全文材料" in context
|
||||
assert "来源:用户知识库/孙之烨-260510.txt" in context
|
||||
assert "完整经历:曾组织技术分享并带队参加竞赛" in context
|
||||
Reference in New Issue
Block a user