Move test helper functions to the main library

src/apps/slackapp/tests/test_get_response.py

@@ -7,9 +7,9 @@
 from slackapp.bolt_app import get_response
 
 import sherpa_ai.config as cfg
+from sherpa_ai.test_utils.llms import get_llm
 from sherpa_ai.verbose_loggers import DummyVerboseLogger
 from tests.conftest import external_api
-from tests.fixtures.llms import get_llm
 
 
 @pytest.mark.external_api

src/tests/integration_tests/test_orchestrator.py

@@ -6,7 +6,7 @@
 from sherpa_ai.events import EventType
 from sherpa_ai.memory import SharedMemory
 from sherpa_ai.orchestrator import Orchestrator, OrchestratorConfig
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 logger.remove()
 logger.add(sys.stderr, level="INFO")

src/tests/integration_tests/test_plan_critics.py

@@ -13,7 +13,7 @@
 )
 from sherpa_ai.memory import SharedMemory
 from sherpa_ai.orchestrator import Orchestrator, OrchestratorConfig
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 logger.remove()
 logger.add(sys.stderr, level="DEBUG")

src/tests/integration_tests/test_task_agent.py

@@ -10,8 +10,8 @@
 import sherpa_ai.config as cfg
 from sherpa_ai.connectors.vectorstores import get_vectordb
 from sherpa_ai.task_agent import TaskAgent
+from sherpa_ai.test_utils.llms import get_llm
 from sherpa_ai.tools import ContextTool, SearchTool
-from tests.fixtures.llms import get_llm
 
 
 def config_task_agent(

src/tests/unit_tests/actions/test_context_search.py

@@ -1,7 +1,7 @@
 import pytest
 
 from sherpa_ai.actions.context_search import ContextSearch
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 
 @pytest.mark.external_api

src/tests/unit_tests/actions/test_planning.py

@@ -1,7 +1,7 @@
 import pytest
 
 from sherpa_ai.actions.planning import TaskPlanning
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 
 @pytest.mark.external_api

src/tests/unit_tests/agents/test_critic_agent.py

@@ -1,7 +1,7 @@
 from unittest.mock import MagicMock
 
 from sherpa_ai.agents.critic import Critic
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 task = "Write a hello world program"
 plan = """

src/tests/unit_tests/agents/test_ml_engineer.py

@@ -6,8 +6,8 @@
 from sherpa_ai.agents import MLEngineer
 from sherpa_ai.events import EventType
 from sherpa_ai.memory import SharedMemory
-from tests.fixtures.llms import get_llm
-from tests.fixtures.loggers import config_logger_level
+from sherpa_ai.test_utils.llms import get_llm
+from sherpa_ai.test_utils.loggers import config_logger_level
 
 
 @pytest.mark.external_api

src/tests/unit_tests/agents/test_physicist.py

@@ -6,8 +6,8 @@
 from sherpa_ai.agents import Physicist
 from sherpa_ai.events import EventType
 from sherpa_ai.memory import SharedMemory
-from tests.fixtures.llms import get_llm
-from tests.fixtures.loggers import config_logger_level
+from sherpa_ai.test_utils.llms import get_llm
+from sherpa_ai.test_utils.loggers import config_logger_level
 
 
 @pytest.mark.external_api

src/tests/unit_tests/agents/test_planner.py

@@ -4,7 +4,7 @@
 
 # from sherpa_ai.agents.programmer import Programmer
 from sherpa_ai.memory.shared_memory import SharedMemory
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 
 def test_planner_succeeds(get_llm):  # noqa: F811

src/tests/unit_tests/agents/test_qa_agent.py

@@ -6,7 +6,7 @@
 from sherpa_ai.agents import QAAgent
 from sherpa_ai.events import EventType
 from sherpa_ai.memory import SharedMemory
-from tests.fixtures.llms import get_llm
+from sherpa_ai.test_utils.llms import get_llm
 
 
 def test_task_agent_succeeds(get_llm):  # noqa: F811

src/tests/unit_tests/test_citation_validation.py

@@ -8,8 +8,8 @@
 from sherpa_ai.events import EventType
 from sherpa_ai.memory import SharedMemory
 from sherpa_ai.output_parsers.citation_validation import CitationValidation
+from sherpa_ai.test_utils.llms import get_llm, get_real_llm
 from sherpa_ai.utils import extract_urls
-from tests.fixtures.llms import get_llm, get_real_llm
 
 
 def test_citation_validation():