fix failing agent ops test

This commit is contained in:
Ishaan Jaff 2025-04-22 14:39:50 -07:00
parent 66680c421d
commit 44264ab6d6

View file

@ -38,6 +38,7 @@ from litellm.integrations.langfuse.langfuse_prompt_management import (
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 @@ callback_class_str_to_classType = {
"pagerduty": PagerDutyAlerting,
"gcs_pubsub": GcsPubSubLogger,
"anthropic_cache_control_hook": AnthropicCacheControlHook,
"agentops": AgentOps,
}
expected_env_vars = {