Initial curated RAGcore source from c16667f0c9397e2bfc8dd0255b114fba23c8d05e

tests/contract/sdk/python/test_answers.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+from collections.abc import Iterator
+from contextlib import contextmanager
+
+from ragcore_sdk import AnswerStreamEventType, RAGCoreClient
+
+from conftest import build_answer_app, run_app
+
+
+@contextmanager
+def _client(app: object) -> Iterator[RAGCoreClient]:
+    with run_app(app) as base_url, RAGCoreClient(base_url=base_url, token="sdk-test-token") as client:  # type: ignore[arg-type]  # noqa: E501
+        yield client
+
+
+def test_answer_sync_returns_typed_response() -> None:
+    fixture = build_answer_app()
+    with _client(fixture.app) as client:
+        response = client.answer(query="refund window", requested_space_ids=[fixture.space_id])
+
+    assert response.answer
+    assert response.citations
+    assert all(claim.citation_ids for claim in response.claims)
+
+
+def test_answer_stream_emits_metadata_then_done_in_order() -> None:
+    fixture = build_answer_app()
+    with _client(fixture.app) as client:
+        events = list(
+            client.answer_stream(query="refund window", requested_space_ids=[fixture.space_id])
+        )
+
+    assert [event.type for event in events] == [
+        AnswerStreamEventType.RETRIEVAL_METADATA,
+        AnswerStreamEventType.DONE,
+    ]
+    done_answer = events[1].as_done_answer()
+    assert done_answer is not None
+    assert done_answer.citations
+    assert "refund" in done_answer.answer.lower()
+    assert events[0].as_done_answer() is None
+
+
+def test_answer_stream_reports_terminal_error_event_on_invalid_generation() -> None:
+    fixture = build_answer_app(generator_responses=["not json", "still not json"])
+    with _client(fixture.app) as client:
+        events = list(
+            client.answer_stream(query="refund window", requested_space_ids=[fixture.space_id])
+        )
+
+    assert [event.type for event in events] == [
+        AnswerStreamEventType.RETRIEVAL_METADATA,
+        AnswerStreamEventType.ERROR,
+    ]
+    assert events[1].data["code"] == "validation_retries_exhausted"