From bd2978a26bdece7a87075e8bae3b72362d332ada Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 31 Jan 2025 11:08:10 -0500 Subject: [PATCH] add additional params to LLM calls --- src/crewai/llm.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/crewai/llm.py b/src/crewai/llm.py index ef8746fd5..bbf8e35d9 100644 --- a/src/crewai/llm.py +++ b/src/crewai/llm.py @@ -137,6 +137,7 @@ class LLM: api_version: Optional[str] = None, api_key: Optional[str] = None, callbacks: List[Any] = [], + **kwargs, ): self.model = model self.timeout = timeout @@ -158,6 +159,7 @@ class LLM: self.api_key = api_key self.callbacks = callbacks self.context_window_size = 0 + self.additional_params = kwargs litellm.drop_params = True @@ -240,6 +242,7 @@ class LLM: "api_key": self.api_key, "stream": False, "tools": tools, + **self.additional_params, } # Remove None values from params