fix integ tests

Author: pgrayy

tests_integ/models/providers.py

@@ -67,25 +67,12 @@ def __init__(self):
             "api_key": os.getenv("ANTHROPIC_API_KEY"),
         },
         model_id="claude-3-7-sonnet-20250219",
-        max_tokens=2048,
-        params={
-            "thinking": {
-                "type": "enabled",
-                "budget_tokens": 1024,
-            },
-        },
+        max_tokens=512,
     ),
 )
 bedrock = ProviderInfo(
     id="bedrock",
-    factory=lambda: BedrockModel(
-        additional_request_fields={
-            "thinking": {
-                "type": "enabled",
-                "budget_tokens": 1024,
-            },
-        },
-    ),
+    factory=lambda: BedrockModel(),
 )
 cohere = ProviderInfo(
     id="cohere",
@@ -101,15 +88,7 @@ def __init__(self):
 )
 litellm = ProviderInfo(
     id="litellm",
-    factory=lambda: LiteLLMModel(
-        model_id="bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0",
-        params={
-            "thinking": {
-                "budget_tokens": 1024,
-                "type": "enabled",
-            },
-        },
-    ),
+    factory=lambda: LiteLLMModel(model_id="bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0"),
 )
 llama = ProviderInfo(
     id="llama",
@@ -158,12 +137,7 @@ def __init__(self):
     factory=lambda: GeminiModel(
         client_args={"api_key": os.getenv("GOOGLE_API_KEY")},
         model_id="gemini-2.5-flash",
-        params={
-            "temperature": 0.7,
-            "thinking_config": {
-                "include_thoughts": True,
-            },
-        },
+        params={"temperature": 0.7},
     ),
 )
 

tests_integ/models/test_conformance.py

@@ -5,7 +5,7 @@
 
 from strands import Agent
 from strands.models import Model
-from tests_integ.models.providers import ProviderInfo, all_providers, cohere, gemini, llama, mistral, openai, writer
+from tests_integ.models.providers import ProviderInfo, all_providers, cohere, llama, mistral
 
 
 def get_models():
@@ -61,13 +61,3 @@ class Weather(BaseModel):
     assert len(result.time) > 0
     assert len(result.weather) > 0
     assert isinstance(result, Weather)
-
-
-def test_stream_reasoning(skip_for, model):
-    skip_for([cohere, gemini, llama, mistral, openai, writer], "reasoning is not supported")
-
-    agent = Agent(model)
-    result = agent("Please reason about the equation 2+2.")
-
-    assert "reasoningContent" in result.message["content"][0]
-    assert result.message["content"][0]["reasoningContent"]["reasoningText"]["text"]

tests_integ/models/test_model_litellm.py

@@ -121,6 +121,22 @@ async def test_agent_stream_async(agent):
     assert all(string in text for string in ["12:00", "sunny"])
 
 
+def test_agent_invoke_reasoning(agent, model):
+    model.update_config(
+        params={
+            "thinking": {
+                "budget_tokens": 1024,
+                "type": "enabled",
+            },
+        },
+    )
+
+    result = agent("Please reason about the equation 2+2.")
+
+    assert "reasoningContent" in result.message["content"][0]
+    assert result.message["content"][0]["reasoningContent"]["reasoningText"]["text"]
+
+
 def test_structured_output(agent, weather):
     tru_weather = agent.structured_output(type(weather), "The time is 12:00 and the weather is sunny")
     exp_weather = weather