참고소스 수정본

참고/ontocast-main/test/test_agent_facts.py

@@ -0,0 +1,181 @@
+import importlib
+from types import SimpleNamespace
+from typing import cast
+
+import pytest
+from rdflib import URIRef
+
+from ontocast.onto.content_unit import ContentUnit
+from ontocast.onto.enum import FailureStage, Status
+from ontocast.onto.model import (
+    FactsCritiqueReport,
+    FactsRenderReport,
+    SemanticTriplesFactsReport,
+    TripleFix,
+)
+from ontocast.onto.ontology import Ontology
+from ontocast.onto.rdfgraph import RDFGraph
+from ontocast.onto.unit_states import UnitFactsState
+from ontocast.tool.atomic import AtomicToolBox
+
+criticise_facts_module = importlib.import_module("ontocast.agent.criticise_facts")
+render_facts_module = importlib.import_module("ontocast.agent.render_facts")
+
+
+def _build_content_unit(with_graph: bool = False) -> ContentUnit:
+    unit = ContentUnit(
+        text="Alice works for ACME.",
+        index=0,
+        doc_iri=URIRef("https://example.com/doc/d1"),
+    )
+    if with_graph:
+        unit.graph.parse(
+            data="""
+            @prefix ex: <https://example.com/ns#> .
+            ex:alice ex:worksFor ex:acme .
+            """,
+            format="turtle",
+        )
+    return unit
+
+
+def _build_ontology() -> Ontology:
+    ontology_graph = RDFGraph()
+    ontology_graph.parse(
+        data="""
+        @prefix onto: <https://example.com/onto#> .
+        @prefix owl: <http://www.w3.org/2002/07/owl#> .
+        onto:CompanyOntology a owl:Ontology .
+        """,
+        format="turtle",
+    )
+    return Ontology(graph=ontology_graph, iri="https://example.com/onto")
+
+
+def _build_tools() -> AtomicToolBox:
+    async def get_llm_tool(_budget_tracker):
+        return object()
+
+    return cast(AtomicToolBox, SimpleNamespace(get_llm_tool=get_llm_tool))
+
+
+@pytest.mark.anyio
+async def test_render_facts_routes_to_fresh_when_graph_is_empty(monkeypatch) -> None:
+    calls = {"fresh": 0, "update": 0}
+
+    async def fake_fresh(state: UnitFactsState, tools) -> UnitFactsState:
+        calls["fresh"] += 1
+        return state
+
+    async def fake_update(state: UnitFactsState, tools) -> UnitFactsState:
+        calls["update"] += 1
+        return state
+
+    monkeypatch.setattr(render_facts_module, "render_facts_fresh", fake_fresh)
+    monkeypatch.setattr(render_facts_module, "render_facts_update", fake_update)
+
+    state = UnitFactsState(
+        content_unit=_build_content_unit(with_graph=False),
+        ontology_snapshot=_build_ontology(),
+    )
+    result = await render_facts_module.render_facts(state, tools=_build_tools())
+
+    assert result is state
+    assert calls["fresh"] == 1
+    assert calls["update"] == 0
+
+
+@pytest.mark.anyio
+async def test_render_facts_fresh_sets_success_and_budget(monkeypatch) -> None:
+    async def fake_call_llm_with_retry(**kwargs):
+        rendered_graph = RDFGraph()
+        rendered_graph.parse(
+            data="""
+            @prefix ex: <https://example.com/ns#> .
+            ex:alice ex:worksFor ex:acme .
+            """,
+            format="turtle",
+        )
+        return FactsRenderReport(
+            facts_report=SemanticTriplesFactsReport(
+                semantic_graph=rendered_graph,
+                ontology_relevance_score=95,
+                triples_generation_score=94,
+            )
+        )
+
+    monkeypatch.setattr(
+        render_facts_module, "call_llm_with_retry", fake_call_llm_with_retry
+    )
+
+    state = UnitFactsState(
+        content_unit=_build_content_unit(with_graph=False),
+        ontology_snapshot=_build_ontology(),
+    )
+    result = await render_facts_module.render_facts_fresh(state, tools=_build_tools())
+
+    assert result.status == Status.SUCCESS
+    assert result.failure_stage is None
+    assert len(result.content_unit.graph) == 1
+    assert result.budget_tracker.facts_operations_count == 1
+    assert result.budget_tracker.facts_triples_generated == 1
+
+
+@pytest.mark.anyio
+async def test_criticise_facts_marks_failed_and_sets_suggestions(monkeypatch) -> None:
+    async def fake_call_llm_with_retry(**kwargs):
+        return FactsCritiqueReport(
+            success=False,
+            score=35,
+            actionable_triple_fixes=[
+                TripleFix(
+                    text_fragment="Alice works for ACME.",
+                    action="ADD",
+                    severity="important",
+                    explanation="Missing employment relation triple.",
+                    correct_value="ex:alice ex:worksFor ex:acme .",
+                )
+            ],
+            systemic_critique_summary="Misses key relations.",
+        )
+
+    monkeypatch.setattr(
+        criticise_facts_module, "call_llm_with_retry", fake_call_llm_with_retry
+    )
+
+    state = UnitFactsState(
+        content_unit=_build_content_unit(with_graph=True),
+        ontology_snapshot=_build_ontology(),
+    )
+    result = await criticise_facts_module.criticise_facts(state, tools=_build_tools())
+
+    assert result.status == Status.FAILED
+    assert result.failure_stage == FailureStage.FACTS_CRITIQUE
+    assert len(result.suggestions.actionable_fixes) == 1
+    assert result.failure_reason == "Facts Critic suggests improvements"
+
+
+@pytest.mark.anyio
+async def test_criticise_facts_accepts_high_score_even_when_success_false(
+    monkeypatch,
+) -> None:
+    async def fake_call_llm_with_retry(**kwargs):
+        return FactsCritiqueReport(
+            success=False,
+            score=95,
+            actionable_triple_fixes=[],
+            systemic_critique_summary="",
+        )
+
+    monkeypatch.setattr(
+        criticise_facts_module, "call_llm_with_retry", fake_call_llm_with_retry
+    )
+
+    state = UnitFactsState(
+        content_unit=_build_content_unit(with_graph=True),
+        ontology_snapshot=_build_ontology(),
+    )
+    result = await criticise_facts_module.criticise_facts(state, tools=_build_tools())
+
+    assert result.status == Status.SUCCESS
+    assert result.failure_stage is None