diff --git a/haiku_rag_slim/haiku/rag/skills/_tools.py b/haiku_rag_slim/haiku/rag/skills/_tools.py index 1998797a..dc64ff21 100644 --- a/haiku_rag_slim/haiku/rag/skills/_tools.py +++ b/haiku_rag_slim/haiku/rag/skills/_tools.py @@ -255,7 +255,7 @@ def create_skill_extras( - 'db_path': path to the LanceDB used to configure the skill - 'config': config passed to (or derived for) the skill - 'list_documents': returns info for documents in the database - - 'list_documents': returns visulaizations for chunks in the database + - 'visualize_chunk': returns visualizations for chunks in the database """ async def visualize_chunk(chunk_id: str) -> list: diff --git a/tests/skills/test_rag.py b/tests/skills/test_rag.py index c7acca69..a82aff31 100644 --- a/tests/skills/test_rag.py +++ b/tests/skills/test_rag.py @@ -73,6 +73,7 @@ class TestRAGSkillCreation: "ask", "research", } + def test_create_skill_has_state(self, test_app_config, temp_db_path): from haiku.rag.skills.rag import RAGState, create_skill @@ -119,7 +120,9 @@ class TestSkillExtras: assert results[0]["title"] == "AI Overview" async def test_visualize_chunk_unknown_returns_empty( - self, test_app_config, rag_db, + self, + test_app_config, + rag_db, ): from haiku.rag.skills.rag import create_skill @@ -129,7 +132,10 @@ class TestSkillExtras: assert result == [] async def test_visualize_chunk_returns_images( - self, test_app_config, rag_db, monkeypatch, + self, + test_app_config, + rag_db, + monkeypatch, ): from haiku.rag.client import HaikuRAG from haiku.rag.skills.rag import create_skill