phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-14 00:02:15 +05:30
parent a79211c270
commit 0bc7bdd550
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/logging_callback_tests/test_unit_tests_init_callbacks.py
View file

@ -31,6 +31,7 @@ from litellm.integrations.datadog.datadog_llm_obs import DataDogLLMObsLogger
from litellm.integrations.gcs_bucket.gcs_bucket import GCSBucketLogger from litellm.integrations.gcs_bucket.gcs_bucket import GCSBucketLogger
from litellm.integrations.opik.opik import OpikLogger from litellm.integrations.opik.opik import OpikLogger
from litellm.integrations.opentelemetry import OpenTelemetry from litellm.integrations.opentelemetry import OpenTelemetry
from litellm.integrations.mlflow import MlflowLogger
from litellm.integrations.argilla import ArgillaLogger from litellm.integrations.argilla import ArgillaLogger
from litellm.proxy.hooks.dynamic_rate_limiter import _PROXY_DynamicRateLimitHandler from litellm.proxy.hooks.dynamic_rate_limiter import _PROXY_DynamicRateLimitHandler
from unittest.mock import patch from unittest.mock import patch
@ -59,6 +60,7 @@ callback_class_str_to_classType = {
"logfire": OpenTelemetry, "logfire": OpenTelemetry,
"arize": OpenTelemetry, "arize": OpenTelemetry,
"langtrace": OpenTelemetry, "langtrace": OpenTelemetry,
"mlflow": MlflowLogger,
} }
expected_env_vars = { expected_env_vars = {