mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
adding callbacks to llm
This commit is contained in:
@@ -48,9 +48,13 @@ class LLM:
|
|||||||
self.callbacks = callbacks
|
self.callbacks = callbacks
|
||||||
self.kwargs = kwargs
|
self.kwargs = kwargs
|
||||||
|
|
||||||
|
litellm.drop_params = True
|
||||||
litellm.callbacks = callbacks
|
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:
|
try:
|
||||||
params = {
|
params = {
|
||||||
"model": self.model,
|
"model": self.model,
|
||||||
|
|||||||
Reference in New Issue
Block a user