fix failing agent ops test

ishaan-jaff · ishaan-jaff · commit 44264ab6d696 · 2025-04-22T14:39:50.000-07:00
diff --git a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py
@@ -38,6 +38,7 @@
     LangfusePromptManagement,
 )
 from litellm.integrations.azure_storage.azure_storage import AzureBlobStorageLogger
+from litellm.integrations.agentops import AgentOps
 from litellm.integrations.humanloop import HumanloopLogger
 from litellm.proxy.hooks.dynamic_rate_limiter import _PROXY_DynamicRateLimitHandler
 from unittest.mock import patch
@@ -75,6 +76,7 @@
     "pagerduty": PagerDutyAlerting,
     "gcs_pubsub": GcsPubSubLogger,
     "anthropic_cache_control_hook": AnthropicCacheControlHook,
+    "agentops": AgentOps,
 }
 
 expected_env_vars = {

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@`
`38`	`38`	`LangfusePromptManagement,`
`39`	`39`	`)`
`40`	`40`	`from litellm.integrations.azure_storage.azure_storage import AzureBlobStorageLogger`
	`41`	`+from litellm.integrations.agentops import AgentOps`
`41`	`42`	`from litellm.integrations.humanloop import HumanloopLogger`
`42`	`43`	`from litellm.proxy.hooks.dynamic_rate_limiter import _PROXY_DynamicRateLimitHandler`
`43`	`44`	`from unittest.mock import patch`
`@@ -75,6 +76,7 @@`
`75`	`76`	`"pagerduty": PagerDutyAlerting,`
`76`	`77`	`"gcs_pubsub": GcsPubSubLogger,`
`77`	`78`	`"anthropic_cache_control_hook": AnthropicCacheControlHook,`
	`79`	`+ "agentops": AgentOps,`
`78`	`80`	`}`
`79`	`81`
`80`	`82`	`expected_env_vars = {`