update

Author: kausmeows

libs/agno/tests/integration/models/aimlapi/test_tool_use.py

@@ -41,7 +41,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.get("tool_name") for tc in chunk.tools):
@@ -87,7 +86,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.get("tool_name") for tc in chunk.tools):

libs/agno/tests/integration/models/anthropic/test_tool_use.py

@@ -41,7 +41,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -88,7 +87,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/aws/bedrock/test_tool_use.py

@@ -38,7 +38,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/aws/claude/test_tool_use.py

@@ -40,7 +40,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -88,7 +87,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/azure/ai_foundry/test_tool_use.py

@@ -40,7 +40,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -85,7 +84,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/azure/openai/test_tool_use.py

@@ -40,7 +40,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -85,7 +84,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/cerebras/test_tool_use.py

@@ -35,15 +35,16 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
-        if chunk.tools:
-            if any(tc.tool_name for tc in chunk.tools):
+
+        # Check for ToolCallStartedEvent or ToolCallCompletedEvent
+        if chunk.event in ["ToolCallStarted", "ToolCallCompleted"] and hasattr(chunk, "tool") and chunk.tool:
+            if chunk.tool.tool_name:
                 tool_call_seen = True
 
     assert len(responses) > 0
     assert tool_call_seen, "No tool calls observed in stream"
-    assert any("France" in r.content for r in responses if r.content)
+    assert any("France" in r.content for r in responses if hasattr(r, "content") and r.content)
 
 
 @pytest.mark.asyncio
@@ -78,15 +79,16 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
-        if chunk.tools:
-            if any(tc.tool_name for tc in chunk.tools):
+
+        # Check for ToolCallStartedEvent or ToolCallCompletedEvent
+        if chunk.event in ["ToolCallStarted", "ToolCallCompleted"] and hasattr(chunk, "tool") and chunk.tool:
+            if chunk.tool.tool_name:
                 tool_call_seen = True
 
     assert len(responses) > 0
     assert tool_call_seen, "No tool calls observed in stream"
-    assert any("France" in r.content for r in responses if r.content)
+    assert any("France" in r.content for r in responses if hasattr(r, "content") and r.content)
 
 
 def test_tool_use_with_content():

libs/agno/tests/integration/models/cerebras_openai/test_tool_use.py

@@ -35,7 +35,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -78,7 +77,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/cohere/test_tool_use.py

@@ -40,7 +40,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -88,7 +87,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):

libs/agno/tests/integration/models/deepinfra/test_tool_use.py

@@ -41,7 +41,6 @@ def test_tool_use_stream():
     tool_call_seen = False
 
     for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):
@@ -91,7 +90,6 @@ async def test_async_tool_use_stream():
     tool_call_seen = False
 
     async for chunk in response_stream:
-        
         responses.append(chunk)
         if chunk.tools:
             if any(tc.tool_name for tc in chunk.tools):