andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-15 19:18:30 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix litellm callback removal error (issue #2513)

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-03 09:39:10 +00:00
parent efe27bd570
commit c06eb56cf3
2 changed files with 48 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/crewai/llm.py
View File

@@ -963,14 +963,15 @@ class LLM(BaseLLM):
"""
with suppress_warnings():
callback_types = [type(callback) for callback in callbacks]
for callback in litellm.success_callback[:]:
if type(callback) in callback_types:
litellm.success_callback.remove(callback)
for callback in litellm._async_success_callback[:]:
if type(callback) in callback_types:
litellm._async_success_callback.remove(callback)
litellm.success_callback = [
cb for cb in litellm.success_callback if type(cb) not in callback_types
]
litellm._async_success_callback = [
cb for cb in litellm._async_success_callback if type(cb) not in callback_types
]
litellm.callbacks = callbacks
def set_env_callbacks(self):