From ea49b3a045edc4aead95ffaad5000954750b4fc0 Mon Sep 17 00:00:00 2001
From: Bobby Lindsey <bobbywlindsey@users.noreply.github.com>
Date: Thu, 23 Jan 2025 16:07:13 -0700
Subject: [PATCH] Add support for custom parameters for LLM

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

diff --git a/src/crewai/llm.py b/src/crewai/llm.py
index 98b0bc855..06bb953ff 100644
--- a/src/crewai/llm.py
+++ b/src/crewai/llm.py
@@ -136,6 +136,7 @@ class LLM:
         api_version: Optional[str] = None,
         api_key: Optional[str] = None,
         callbacks: List[Any] = [],
+        **custom_params: Dict[str, Any],
     ):
         self.model = model
         self.timeout = timeout
@@ -156,6 +157,7 @@ class LLM:
         self.api_key = api_key
         self.callbacks = callbacks
         self.context_window_size = 0
+        self.custom_params = custom_params
 
         litellm.drop_params = True
 
@@ -237,6 +239,7 @@ class LLM:
                     "api_key": self.api_key,
                     "stream": False,
                     "tools": tools,
+                    **self.custom_params["custom_params"],
                 }
 
                 # Remove None values from params