From 54316109d5eb0aa40e3140106d2532e33f3c645a Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Thu, 2 Jan 2025 14:34:18 -0500
Subject: [PATCH] more test changes

---
 pyproject.toml     | 2 +-
 tests/crew_test.py | 5 +++++
 uv.lock            | 9 +++++----
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 8df0c3668..0c608b5aa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ dependencies = [
     # Core Dependencies
     "pydantic>=2.4.2",
     "openai>=1.13.3",
-    "litellm>=1.44.22",
+    "litellm>=1.56.6",
     "instructor>=1.3.3",
     "crewai-tools>=0.17.0",
     
diff --git a/tests/crew_test.py b/tests/crew_test.py
index 41b6b9705..8354f6584 100644
--- a/tests/crew_test.py
+++ b/tests/crew_test.py
@@ -1051,6 +1051,7 @@ def test_three_task_with_async_execution():
 
 
 @pytest.mark.asyncio
+@pytest.mark.vcr(filter_headers=["authorization"])
 async def test_crew_async_kickoff():
     inputs = [
         {"topic": "dog"},
@@ -1098,6 +1099,7 @@ async def test_crew_async_kickoff():
 
 
 @pytest.mark.asyncio
+@pytest.mark.vcr(filter_headers=["authorization"])
 async def test_async_task_execution_call_count():
     from unittest.mock import MagicMock, patch
 
@@ -1153,6 +1155,7 @@ async def test_async_task_execution_call_count():
         assert mock_execute_sync.call_count == 1
 
 
+@pytest.mark.vcr(filter_headers=["authorization"])
 def test_kickoff_for_each_single_input():
     """Tests if kickoff_for_each works with a single input."""
 
@@ -1176,6 +1179,7 @@ def test_kickoff_for_each_single_input():
     assert len(results) == 1
 
 
+@pytest.mark.vcr(filter_headers=["authorization"])
 def test_kickoff_for_each_multiple_inputs():
     """Tests if kickoff_for_each works with multiple inputs."""
 
@@ -1203,6 +1207,7 @@ def test_kickoff_for_each_multiple_inputs():
     assert len(results) == len(inputs)
 
 
+@pytest.mark.vcr(filter_headers=["authorization"])
 def test_kickoff_for_each_empty_input():
     """Tests if kickoff_for_each handles an empty input list."""
     agent = Agent(
diff --git a/uv.lock b/uv.lock
index 8cafcc950..b0ecd95c7 100644
--- a/uv.lock
+++ b/uv.lock
@@ -7,6 +7,7 @@ resolution-markers = [
     "python_full_version >= '3.12.4'",
 ]
 
+
 [[package]]
 name = "agentops"
 version = "0.3.14"
@@ -677,7 +678,7 @@ requires-dist = [
     { name = "instructor", specifier = ">=1.3.3" },
     { name = "json-repair", specifier = ">=0.25.2" },
     { name = "jsonref", specifier = ">=1.1.0" },
-    { name = "litellm", specifier = ">=1.44.22" },
+    { name = "litellm", specifier = ">=1.56.6" },
     { name = "mem0ai", marker = "extra == 'mem0'", specifier = ">=0.1.29" },
     { name = "openai", specifier = ">=1.13.3" },
     { name = "openpyxl", specifier = ">=3.1.5" },
@@ -2223,7 +2224,7 @@ wheels = [
 
 [[package]]
 name = "litellm"
-version = "1.56.4"
+version = "1.56.6"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "aiohttp" },
@@ -2238,9 +2239,9 @@ dependencies = [
     { name = "tiktoken" },
     { name = "tokenizers" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/83/ea/2c51d16c244a64dd3f0bdb1757aef798cf943b92e5695da04e3e42ba09e0/litellm-1.56.4.tar.gz", hash = "sha256:2808ca21878d200f7676a3d11e5bf2b5e3349ae504628f279cd7297c7dbd2038", size = 6284983 }
+sdist = { url = "https://files.pythonhosted.org/packages/24/b2/659fdf0a6a351090e944aebffca65bb8c4097052154065e4f98ca1878fa9/litellm-1.56.6.tar.gz", hash = "sha256:24612fff40f31044257c16bc29aa086cbb084b830e427a19f4adb96deeea626d", size = 6289223 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/8f/25/2fd7b28a270b2963e8fa0ecf6aab4db47c54d932cc5aac8bc87e7ebc3755/litellm-1.56.4-py3-none-any.whl", hash = "sha256:699a8db46f7de045069a77c435e13244b5fdaf5df1c8cb5e6ad675ef7e104ccd", size = 6564370 },
+    { url = "https://files.pythonhosted.org/packages/bc/58/83e895c1367afcfc02a55c4eadb6cdd4e62b6a324b270b8be3b7eb8c8ffd/litellm-1.56.6-py3-none-any.whl", hash = "sha256:dc04becae6b09b401edfc13e9e648443e425a52c1d7217351c7841811dc8dbec", size = 6567472 },
 ]
 
 [[package]]