Fix tests

packages/ragbits-core/tests/unit/llms/factory/__init__.py

@@ -0,0 +1,5 @@
+import sys
+from pathlib import Path
+
+# Add "llms" to sys.path
+sys.path.append(str(Path(__file__).parent.parent))

packages/ragbits-core/tests/unit/llms/factory/test_get_default_llm.py

@@ -7,9 +7,8 @@ def test_get_default_llm(monkeypatch):
     """
     Test the get_llm_from_factory function.
     """
-    monkeypatch.setattr(
-        core_config, "default_llm_factory", "tests.unit.llms.factory.test_get_llm_from_factory.mock_llm_factory"
-    )
+
+    monkeypatch.setattr(core_config, "default_llm_factory", "factory.test_get_llm_from_factory.mock_llm_factory")
 
     llm = get_default_llm()
     assert isinstance(llm, LiteLLM)

packages/ragbits-core/tests/unit/llms/factory/test_get_llm_from_factory.py

@@ -16,7 +16,7 @@ def test_get_llm_from_factory():
     """
     Test the get_llm_from_factory function.
     """
-    llm = get_llm_from_factory("tests.unit.llms.factory.test_get_llm_from_factory.mock_llm_factory")
+    llm = get_llm_from_factory("factory.test_get_llm_from_factory.mock_llm_factory")
 
     assert isinstance(llm, LiteLLM)
     assert llm.model_name == "mock_model"