From 3f3125de300d2c1eb6d26ae65476934829d54754 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Thu, 13 Jun 2024 10:42:27 -0300 Subject: [PATCH] refactor: Update OpenAIModelComponent to handle optional model name and temperature Update the OpenAIModelComponent to handle an optional model name and temperature. This allows for more flexibility in configuring the component. The model name can now be set to None if not provided, and the temperature can be set to 0.1 if not specified. This improves the usability and customization options of the OpenAIModelComponent. --- src/backend/base/langflow/components/models/OpenAIModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/OpenAIModel.py b/src/backend/base/langflow/components/models/OpenAIModel.py index e9130d520..0e4b2ae46 100644 --- a/src/backend/base/langflow/components/models/OpenAIModel.py +++ b/src/backend/base/langflow/components/models/OpenAIModel.py @@ -79,9 +79,9 @@ class OpenAIModelComponent(LCModelComponent): output = ChatOpenAI( max_tokens=max_tokens or None, model_kwargs=model_kwargs or {}, - model=model_name, + model=model_name or None, base_url=openai_api_base, api_key=api_key, - temperature=temperature, + temperature=temperature or 0.1, ) return output