M10: harden knowledge trust and telemetry

This commit is contained in:
NuklearRabbit
2026-08-10 02:51:07 +02:00
parent e577c16db5
commit de151914b6
5 changed files with 151 additions and 5 deletions
+1 -1
View File
@@ -110,7 +110,7 @@ def demo_reset(
status_code=status.HTTP_403_FORBIDDEN, status_code=status.HTTP_403_FORBIDDEN,
detail="Demo reset is disabled on this deployment.", detail="Demo reset is disabled on this deployment.",
) )
result = reset_and_seed(db) result = reset_and_seed(db, preserve_integration_telemetry=True)
integrity = scenario_integrity_report(db) integrity = scenario_integrity_report(db)
record_audit_event( record_audit_event(
db, db,
+18 -4
View File
@@ -80,9 +80,17 @@ def _read_csv(name: str) -> list[dict[str, str]]:
return list(csv.DictReader(handle)) return list(csv.DictReader(handle))
def clear_all(db: Session) -> None: _PERSISTENT_TELEMETRY_ACTIONS = (
"mcp_tool_request",
"n8n_return_followup_recorded",
"n8n_workflow_failure_registered",
"n8n_procedures_synced",
"knowledge_question_asked",
)
def clear_all(db: Session, *, preserve_integration_telemetry: bool = False) -> None:
for model in ( for model in (
AuditEvent,
OutboxEvent, OutboxEvent,
IdempotencyRecord, IdempotencyRecord,
DataQualityIssue, DataQualityIssue,
@@ -94,6 +102,10 @@ def clear_all(db: Session) -> None:
User, User,
): ):
db.execute(delete(model)) db.execute(delete(model))
if preserve_integration_telemetry:
db.execute(delete(AuditEvent).where(AuditEvent.action.not_in(_PERSISTENT_TELEMETRY_ACTIONS)))
else:
db.execute(delete(AuditEvent))
def load_seed(db: Session) -> SeedResult: def load_seed(db: Session) -> SeedResult:
@@ -428,10 +440,12 @@ def load_seed(db: Session) -> SeedResult:
return SeedResult(counts=counts, anchor_date=today, seeded_at=seeded_at) return SeedResult(counts=counts, anchor_date=today, seeded_at=seeded_at)
def reset_and_seed(db: Session) -> SeedResult: def reset_and_seed(
db: Session, *, preserve_integration_telemetry: bool = False
) -> SeedResult:
from app.services.data_quality import run_scan from app.services.data_quality import run_scan
clear_all(db) clear_all(db, preserve_integration_telemetry=preserve_integration_telemetry)
result = load_seed(db) result = load_seed(db)
db.commit() db.commit()
scan = run_scan(db) scan = run_scan(db)
+60
View File
@@ -21,6 +21,54 @@ _LEAD_ANSWER_TEMPLATE = {
} }
_DEFAULT_LANGUAGE = "en-GB" _DEFAULT_LANGUAGE = "en-GB"
_DOMAIN_CONCEPTS: dict[str, tuple[str, ...]] = {
"damage": ("damage", "damaged", "schade", "beschadigd", "dommage", "endommagé"),
"vehicle": ("vehicle", "car", "voertuig", "wagen", "véhicule", "voiture"),
"return": ("return", "returned", "retour", "terugbrengen", "restitution"),
"fuel": ("fuel", "brandstof", "carburant"),
"booking": ("booking", "reservation", "boeking", "réservation"),
"odometer": ("odometer", "mileage", "kilometer", "kilométrage", "compteur"),
"customer": ("customer", "client", "klant"),
"cleaning": ("cleaning", "reiniging", "poetsen", "nettoyage"),
"maintenance": ("maintenance", "service", "onderhoud", "entretien"),
"conflict": ("conflict", "overlap", "overlapping", "conflit", "chevauchement"),
"checkout": ("checkout", "departure", "vertrek", "départ"),
"availability": ("available", "availability", "beschikbaar", "disponible", "disponibilité"),
"technical": ("technical", "warning", "technisch", "waarschuwing", "technique", "alerte"),
}
def _question_concepts(question: str) -> set[str]:
normalized = question.casefold()
return {
concept
for concept, terms in _DOMAIN_CONCEPTS.items()
if any(term in normalized for term in terms)
}
def _deduplicate_sources(sources: list[SourceCard]) -> list[SourceCard]:
seen: set[tuple[str, str]] = set()
unique: list[SourceCard] = []
for source in sources:
key = (source.document_id, source.section)
if key in seen:
continue
seen.add(key)
unique.append(source)
return unique
def _rank_sources_for_concepts(sources: list[SourceCard], concepts: set[str]) -> list[SourceCard]:
if "damage" not in concepts:
return sources
return sorted(
sources,
key=lambda source: (
0 if "damage" in f"{source.document_id} {source.title}".casefold() else 1
),
)
class RAGcoreKnowledgeProvider: class RAGcoreKnowledgeProvider:
"""Adapter for the central RAGcore service, against its real `/v1/*` contract """Adapter for the central RAGcore service, against its real `/v1/*` contract
@@ -199,6 +247,9 @@ class RAGcoreKnowledgeProvider:
except (TypeError, KeyError, ValueError): except (TypeError, KeyError, ValueError):
return unavailable return unavailable
sources = _deduplicate_sources(sources)
concepts = _question_concepts(question)
sources = _rank_sources_for_concepts(sources, concepts)
if not sources: if not sources:
return GroundedAnswer( return GroundedAnswer(
answer="", answer="",
@@ -208,6 +259,15 @@ class RAGcoreKnowledgeProvider:
correlation_id=correlation_id, correlation_id=correlation_id,
) )
if not concepts:
return GroundedAnswer(
answer="",
evidence_state="insufficient",
sources=sources,
provider=self.name,
correlation_id=correlation_id,
)
template = _LEAD_ANSWER_TEMPLATE.get(language, _LEAD_ANSWER_TEMPLATE[_DEFAULT_LANGUAGE]) template = _LEAD_ANSWER_TEMPLATE.get(language, _LEAD_ANSWER_TEMPLATE[_DEFAULT_LANGUAGE])
lead = sources[0] lead = sources[0]
answer = template.format(title=lead.title, excerpt=lead.excerpt) answer = template.format(title=lead.title, excerpt=lead.excerpt)
+19
View File
@@ -65,3 +65,22 @@ def test_logout_invalidates_session(ops_client):
def test_logout_without_a_session_is_safe(client): def test_logout_without_a_session_is_safe(client):
response = client.post("/api/v1/demo/logout") response = client.post("/api/v1/demo/logout")
assert response.status_code == 200 assert response.status_code == 200
def test_demo_reset_preserves_integration_telemetry(ops_client, client):
from app.core.config import get_settings
settings = get_settings()
probe = client.get(
"/api/v1/integrations/mcp/operations-summary",
headers={
"X-Service-Token": settings.mcp_hub_service_token,
"X-Client-Id": "reset-probe",
},
)
assert probe.status_code == 200
assert ops_client.post("/api/v1/demo/reset").status_code == 200
assert client.post("/api/v1/demo/login", json={"role": "operations_manager"}).status_code == 200
events = client.get("/api/v1/audit", params={"action": "mcp_tool_request"}).json()
assert any(event["actor_label"] == "reset-probe" for event in events)
+53
View File
@@ -450,3 +450,56 @@ def test_ragcore_provider_fallback_with_no_search_results_is_insufficient(monkey
assert answer.evidence_state == "insufficient" assert answer.evidence_state == "insufficient"
assert answer.answer == "" assert answer.answer == ""
assert answer.sources == [] assert answer.sources == []
def test_ragcore_search_fallback_rejects_out_of_domain_question(monkeypatch):
provider = RAGcoreKnowledgeProvider()
monkeypatch.setattr(provider._settings, "ragcore_space_id", "space-1")
monkeypatch.setattr(
provider,
"_client",
lambda: _FakeClient(
post_responses={
"/v1/answers": _FakeResponse(503, {}),
"/v1/search": _FakeResponse(200, _search_body()),
}
),
)
answer = provider.ask(
"Who won the football world cup in 1998?",
"test-correlation-out-of-domain",
)
assert answer.evidence_state == "insufficient"
assert answer.answer == ""
def test_ragcore_search_fallback_prefers_damage_procedure(monkeypatch):
provider = RAGcoreKnowledgeProvider()
monkeypatch.setattr(provider._settings, "ragcore_space_id", "space-1")
search = _search_body()
search["results"].append(
{
"citation": {
"document_id": "damage-procedure",
"document_version_id": "version-2",
"title": "damage-procedure.md",
"section": "Damage",
"excerpt": "Record damage and keep the vehicle blocked.",
},
"rank": 2,
"scores": {"fused": 0.01},
}
)
monkeypatch.setattr(
provider,
"_client",
lambda: _FakeClient(
post_responses={
"/v1/answers": _FakeResponse(503, {}),
"/v1/search": _FakeResponse(200, search),
}
),
)
answer = provider.ask("Wat moet ik doen bij schade?", "test-correlation-damage", "nl-BE")
assert answer.evidence_state == "grounded"
assert answer.sources[0].document_id == "damage-procedure"