test: fixed tests

ErikBjare · Nov 17, 2024 · b1d5408 · b1d5408
1 parent 43452b4
commit b1d5408
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_tools_rag.py b/tests/test_tools_rag.py
@@ -6,7 +6,8 @@
 import pytest
 from gptme import Message
 from gptme.tools.base import ToolSpec
-from gptme.tools.rag import _HAS_RAG, init_rag
+from gptme.tools.rag import _HAS_RAG
+from gptme.tools.rag import init as init_rag
 
 
 @pytest.fixture