From b4a1a15aa9d5f2fac5184983069d96d0ca08f26b Mon Sep 17 00:00:00 2001
From: Matthew Farrellee <matt@cs.wisc.edu>
Date: Thu, 2 Oct 2025 13:20:50 -0400
Subject: [PATCH] chore: fix agents tests for non-ollama providers, provide
 max_tokens

---
 tests/integration/agents/test_agents.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/integration/agents/test_agents.py b/tests/integration/agents/test_agents.py
index 23529f91e..07ba7bb01 100644
--- a/tests/integration/agents/test_agents.py
+++ b/tests/integration/agents/test_agents.py
@@ -68,6 +68,7 @@ def agent_config(llama_stack_client, text_model_id):
                 "temperature": 0.0001,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         tools=[],
         input_shields=available_shields,
@@ -88,6 +89,7 @@ def agent_config_without_safety(text_model_id):
                 "temperature": 0.0001,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         tools=[],
         enable_session_persistence=False,
@@ -198,6 +200,7 @@ def test_tool_config(agent_config):
                 "temperature": 1.0,
                 "top_p": 0.9,
             },
+            "max_tokens": 512,
         },
         toolgroups=[],
         enable_session_persistence=False,