more test changes

This commit is contained in:
Brandon Hancock
2025-01-02 14:34:18 -05:00
parent db61f5dd59
commit 54316109d5
3 changed files with 11 additions and 5 deletions

View File

@@ -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",

View File

@@ -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(

9
uv.lock generated
View File

@@ -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]]