adding callbacks to llm

2026-01-08 15:48:29 +00:00 · 2024-09-23 00:54:01 -03:00
parent 6dad33f47c
commit 7d981ba8ce
1 changed files with 5 additions and 1 deletions
--- 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,