diff --git a/pytest.ini b/pytest.ini index 32af169e..ae89a87c 100644 --- a/pytest.ini +++ b/pytest.ini @@ -1,6 +1,8 @@ [pytest] markers = live: real end-to-end tests that hit live integrations and the real LLM (opt-in: `pytest -m live`) +testpaths = + tests # By default, skip live tests so plain `pytest` stays fast and offline. addopts = -m "not live" diff --git a/tests/test_trigger_router_and_parking.py b/tests/test_trigger_router_and_parking.py index 052e5907..bcdfcf88 100644 --- a/tests/test_trigger_router_and_parking.py +++ b/tests/test_trigger_router_and_parking.py @@ -21,9 +21,13 @@ def __init__(self, response: str): self.response = response self.calls = 0 - async def generate_response_async(self, system_prompt: str, user_prompt: str): + async def generate_response_async( + self, system_prompt: str, user_prompt: str, prompt_name=None, **_kwargs + ): self.calls += 1 + self.last_system_prompt = system_prompt self.last_prompt = user_prompt + self.last_prompt_name = prompt_name return self.response @@ -39,6 +43,7 @@ def test_route_to_existing_session(self): router = SessionRouter(llm, ROUTING_PROMPT) result = run(router.route("message", "continue that task", "sessions")) assert result["session_id"] == "abc123" + assert llm.last_prompt_name == "ROUTE_TO_SESSION" assert llm.calls == 1 def test_new_session_decision(self):