From cfb9d55b0529275083919a45dea03940e7b9ae9f Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 22 Apr 2025 16:06:25 +0000
Subject: [PATCH] test: Fix o4-mini test to use mocking instead of real API
 calls

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 tests/llm_test.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/tests/llm_test.py b/tests/llm_test.py
index de55b98a5..1df6eaf45 100644
--- a/tests/llm_test.py
+++ b/tests/llm_test.py
@@ -535,7 +535,6 @@ def test_handle_streaming_tool_calls_no_tools(mock_emit):
     )
 
 
-@pytest.mark.vcr(filter_headers=["authorization"])
 def test_llm_o4_mini_stop_parameter():
     """Test that o4-mini model works correctly without stop parameter."""
     llm = LLM(model="o4-mini", stop=["STOP", "END"])
@@ -547,5 +546,26 @@ def test_llm_o4_mini_stop_parameter():
     
     assert "stop" not in params
     
-    response = llm.call(messages=[{"role": "user", "content": "Hello, world!"}])
-    assert response is not None
+    with patch("litellm.completion") as mock_completion:
+        # Create a mock response
+        mock_message = MagicMock()
+        mock_message.content = "Test response"
+        mock_choice = MagicMock()
+        mock_choice.message = mock_message
+        mock_response = MagicMock()
+        mock_response.choices = [mock_choice]
+        mock_response.usage = {
+            "prompt_tokens": 5,
+            "completion_tokens": 5,
+            "total_tokens": 10,
+        }
+        
+        # Set up the mock to return our response
+        mock_completion.return_value = mock_response
+        
+        response = llm.call(messages=[{"role": "user", "content": "Hello, world!"}])
+        
+        assert response == "Test response"
+        
+        call_args = mock_completion.call_args[1]
+        assert "stop" not in call_args