From ac627e831f4081d3e3334bad482a45999032db0e Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Thu, 20 Jun 2024 18:18:04 -0300
Subject: [PATCH] refactor: Update OpenAIModelComponent to include seed in
 model_kwargs

---
 .../base/langflow/components/models/OpenAIModel.py       | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/backend/base/langflow/components/models/OpenAIModel.py b/src/backend/base/langflow/components/models/OpenAIModel.py
index db8e8f60d..6545cf3dc 100644
--- a/src/backend/base/langflow/components/models/OpenAIModel.py
+++ b/src/backend/base/langflow/components/models/OpenAIModel.py
@@ -82,25 +82,26 @@ class OpenAIModelComponent(LCModelComponent):
         temperature = self.temperature
         model_name: str = self.model_name
         max_tokens = self.max_tokens
-        model_kwargs = self.model_kwargs
+        model_kwargs = self.model_kwargs or {}
         openai_api_base = self.openai_api_base or "https://api.openai.com/v1"
         json_mode = bool(output_schema_dict)
         seed = self.seed
+        model_kwargs["seed"] = seed
+
         if openai_api_key:
             api_key = SecretStr(openai_api_key)
         else:
             api_key = None
         output = ChatOpenAI(
             max_tokens=max_tokens or None,
-            model_kwargs=model_kwargs or {},
+            model_kwargs=model_kwargs,
             model=model_name,
             base_url=openai_api_base,
             api_key=api_key,
             temperature=temperature or 0.1,
-            seed=seed,
         )
         if json_mode:
-            output = output.with_structured_output(schema=output_schema_dict, method="json_mode")
+            output = output.with_structured_output(schema=output_schema_dict, method="json_mode")  # type: ignore
 
         return output