From 44264ab6d696782f7d372a90c638c07f64cd5e19 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 22 Apr 2025 14:39:50 -0700
Subject: [PATCH] fix failing agent ops test

---
 tests/logging_callback_tests/test_unit_tests_init_callbacks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py
index 3ffa97be9fb..445c773d999 100644
--- 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 @@ 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 = {