feat: add governed public knowledge retrieval
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
import pytest
|
||||
|
||||
from app.core.knowledge_contracts import KnowledgeHit, KnowledgeQuery
|
||||
from app.service.knowledge_authority import KnowledgeMysqlAuthority
|
||||
|
||||
|
||||
class FakeRow:
|
||||
id = 101
|
||||
title = "申购规则"
|
||||
content_text = '{"answer":"工作日确认"}'
|
||||
version = "v2"
|
||||
milvus_collection = "fin_policy_collection"
|
||||
|
||||
|
||||
class FakeSession:
|
||||
statement = None
|
||||
|
||||
async def scalars(self, statement):
|
||||
self.statement = statement
|
||||
return [FakeRow()]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_authority_filters_to_published_active_effective_knowledge() -> None:
|
||||
session = FakeSession()
|
||||
authority = KnowledgeMysqlAuthority(session)
|
||||
|
||||
hits = await authority.filter_published((
|
||||
KnowledgeHit(
|
||||
knowledge_id="101", collection="fin_policy_collection", snippet="摘要", score=0.91,
|
||||
),
|
||||
))
|
||||
|
||||
statement = str(session.statement)
|
||||
assert "review_status" in statement
|
||||
assert "status" in statement
|
||||
assert "effective_date" in statement
|
||||
assert "expire_date" in statement
|
||||
assert hits[0].answer == "工作日确认"
|
||||
assert hits[0].version == "v2"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_authority_keyword_fallback_only_returns_effective_public_knowledge() -> None:
|
||||
session = FakeSession()
|
||||
authority = KnowledgeMysqlAuthority(session)
|
||||
|
||||
hits = await authority.search_keyword(
|
||||
KnowledgeQuery(query="基金 申购确认", intents=("policy_explain",)),
|
||||
("fin_policy_collection",),
|
||||
5,
|
||||
)
|
||||
|
||||
statement = str(session.statement)
|
||||
assert "milvus_collection" in statement
|
||||
assert "content_text" in statement
|
||||
assert "review_status" in statement
|
||||
assert "effective_date" in statement
|
||||
assert hits[0].knowledge_id == "101"
|
||||
assert hits[0].collection == "fin_policy_collection"
|
||||
assert hits[0].answer == "工作日确认"
|
||||
Reference in New Issue
Block a user