tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 16:24:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): clear _async_success_callback in vertex fine-tune mocked tests to prevent Datadog interference (#21825)

Browse Source
This commit is contained in:
Ishaan Jaff
2026-02-21 14:16:18 -08:00
committed by GitHub
parent 494aad4a68
commit a1ead765ed
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/batches_tests/test_fine_tuning_api.py
+16 -4
View File
@@ -210,9 +210,13 @@ async def test_create_vertex_fine_tune_jobs_mocked():
# Save original callbacks to restore later
original_callbacks = litellm.callbacks
# Disable callbacks to avoid Datadog logging interfering with the mock
original_success_callback = litellm.success_callback
original_async_success_callback = litellm._async_success_callback
# Disable all callbacks to avoid Datadog/other loggers interfering with the mock
litellm.callbacks = []
litellm.success_callback = []
litellm._async_success_callback = []
try:
with patch(
"litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler.post",
@@ -259,6 +263,8 @@ async def test_create_vertex_fine_tune_jobs_mocked():
finally:
# Restore original callbacks
litellm.callbacks = original_callbacks
litellm.success_callback = original_success_callback
litellm._async_success_callback = original_async_success_callback
@pytest.mark.asyncio()
@@ -291,9 +297,13 @@ async def test_create_vertex_fine_tune_jobs_mocked_with_hyperparameters():
# Save original callbacks to restore later
original_callbacks = litellm.callbacks
# Disable callbacks to avoid Datadog logging interfering with the mock
original_success_callback = litellm.success_callback
original_async_success_callback = litellm._async_success_callback
# Disable all callbacks to avoid Datadog/other loggers interfering with the mock
litellm.callbacks = []
litellm.success_callback = []
litellm._async_success_callback = []
try:
with patch(
"litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler.post",
@@ -352,6 +362,8 @@ async def test_create_vertex_fine_tune_jobs_mocked_with_hyperparameters():
finally:
# Restore original callbacks
litellm.callbacks = original_callbacks
litellm.success_callback = original_success_callback
litellm._async_success_callback = original_async_success_callback
# Testing OpenAI -> Vertex AI param mapping