From 035929814e8ea46c4d31613ea44f7452d9d229ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Moura?= <joaomdmoura@gmail.com>
Date: Mon, 23 Sep 2024 00:54:01 -0300
Subject: [PATCH] adding callbacks to llm

---
 src/crewai/llm.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/crewai/llm.py b/src/crewai/llm.py
index 8c9f8ecfe..006733cd8 100644
--- a/src/crewai/llm.py
+++ b/src/crewai/llm.py
@@ -48,9 +48,13 @@ class LLM:
         self.callbacks = callbacks
         self.kwargs = kwargs
 
+        litellm.drop_params = True
         litellm.callbacks = callbacks
 
-    def call(self, messages: List[Dict[str, str]]) -> str:
+    def call(self, messages: List[Dict[str, str]], callbacks: List[Any] = []) -> str:
+        if callbacks and len(callbacks) > 0:
+            litellm.callbacks = callbacks
+
         try:
             params = {
                 "model": self.model,