phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_init_custom_logger_compatible_class_as_callback

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-24 21:27:22 -08:00
parent 3d8eb30a95
commit f77882948d
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/logging_callback_tests/test_unit_tests_init_callbacks.py
View file

@ -28,6 +28,7 @@ from litellm.integrations.prometheus import PrometheusLogger
from litellm.integrations.datadog.datadog import DataDogLogger
from litellm.integrations.datadog.datadog_llm_obs import DataDogLLMObsLogger
from litellm.integrations.gcs_bucket.gcs_bucket import GCSBucketLogger
from litellm.integrations.gcs_pubsub.pub_sub import GcsPubSubLogger
from litellm.integrations.opik.opik import OpikLogger
from litellm.integrations.opentelemetry import OpenTelemetry
from litellm.integrations.mlflow import MlflowLogger
@ -70,6 +71,7 @@ callback_class_str_to_classType = {
"langfuse": LangfusePromptManagement,
"otel": OpenTelemetry,
"pagerduty": PagerDutyAlerting,
"gcs_pubsub": GcsPubSubLogger,
}
expected_env_vars = {
@ -90,6 +92,8 @@ expected_env_vars = {
"ARIZE_API_KEY": "arize_api_key",
"ARGILLA_API_KEY": "argilla_api_key",
"PAGERDUTY_API_KEY": "pagerduty_api_key",
"GCS_PUBSUB_TOPIC_ID": "gcs_pubsub_topic_id",
"GCS_PUBSUB_PROJECT_ID": "gcs_pubsub_project_id",
}