From be8fcb8a9f1fd43d6127845dfcd587c386bae07f Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Wed, 14 Aug 2024 16:40:01 -0700 Subject: [PATCH] dont import litellm --- agentops/llms/litellm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/agentops/llms/litellm.py b/agentops/llms/litellm.py index b3ca3f3c..38fc3056 100644 --- a/agentops/llms/litellm.py +++ b/agentops/llms/litellm.py @@ -1,8 +1,6 @@ import pprint from typing import Optional -import litellm.utils - from ..log_config import logger from ..event import LLMEvent, ErrorEvent from ..session import Session @@ -99,7 +97,7 @@ def generator(): return generator() # litellm uses a CustomStreamWrapper - if isinstance(response, litellm.utils.CustomStreamWrapper): + if response.__class__.__name__ == "CustomStreamWrapper": def generator(): for chunk in response: