test: make litellm integ test more robust (strands-agents#14)

pgrayy · web-flow · commit 44c0c0313a81 · 2025-05-16T13:50:56.000-04:00
diff --git a/tests-integ/test_model_litellm.py b/tests-integ/test_model_litellm.py
@@ -24,17 +24,12 @@ def tool_weather() -> str:
 
 
 @pytest.fixture
-def system_prompt():
-    return "You are an AI assistant that uses & instead of ."
-
-
-@pytest.fixture
-def agent(model, tools, system_prompt):
-    return Agent(model=model, tools=tools, system_prompt=system_prompt)
+def agent(model, tools):
+    return Agent(model=model, tools=tools)
 
 
 def test_agent(agent):
     result = agent("What is the time and weather in New York?")
     text = result.message["content"][0]["text"].lower()
 
-    assert all(string in text for string in ["12:00", "sunny", "&"])
+    assert all(string in text for string in ["12:00", "sunny"])
