diff --git a/src/genesis/services/rag_service.py b/src/genesis/services/rag_service.py index d758317..f6644c2 100644 --- a/src/genesis/services/rag_service.py +++ b/src/genesis/services/rag_service.py @@ -7,8 +7,8 @@ from typing import Protocol, runtime_checkable @runtime_checkable class RagService(Protocol): - async def retrieve_write_rules(self, chapter_id: str) -> list[str]: ... - async def retrieve_design_rules(self, chapter_id: str) -> list[str]: ... + def retrieve_write_rules(self, chapter_id: str) -> list[str]: ... + def retrieve_design_rules(self, chapter_id: str) -> list[str]: ... class CannedRagService: diff --git a/tests/test_phase5_rag.py b/tests/test_phase5_rag.py index 6e61802..ccdd0b1 100644 --- a/tests/test_phase5_rag.py +++ b/tests/test_phase5_rag.py @@ -11,4 +11,4 @@ def test_canned_rag_returns_rules(): def test_rag_service_is_protocol(): - assert isinstance(CannedRagService("samples"), RagService) or True + assert isinstance(CannedRagService("samples"), RagService)