jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor ChatOpenAIComponent configuration fields

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-07 21:00:35 -03:00
commit 7a4b2dfc7e
1 changed files with 0 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/components/llms/ChatOpenAI.py
View file

@ -2,7 +2,6 @@ from typing import Optional, Union
from langchain.llms import BaseLLM from langchain.llms import BaseLLM
from langchain_community.chat_models.openai import ChatOpenAI from langchain_community.chat_models.openai import ChatOpenAI
from langflow import CustomComponent from langflow import CustomComponent
from langflow.field_typing import BaseLanguageModel, NestedDict from langflow.field_typing import BaseLanguageModel, NestedDict
@ -15,19 +14,16 @@ class ChatOpenAIComponent(CustomComponent):
return { return {
"max_tokens": { "max_tokens": {
"display_name": "Max Tokens", "display_name": "Max Tokens",
"field_type": "int",
"advanced": False, "advanced": False,
"required": False, "required": False,
}, },
"model_kwargs": { "model_kwargs": {
"display_name": "Model Kwargs", "display_name": "Model Kwargs",
"field_type": "NestedDict",
"advanced": True, "advanced": True,
"required": False, "required": False,
}, },
"model_name": { "model_name": {
"display_name": "Model Name", "display_name": "Model Name",
"field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
"options": [ "options": [
@ -41,7 +37,6 @@ class ChatOpenAIComponent(CustomComponent):
}, },
"openai_api_base": { "openai_api_base": {
"display_name": "OpenAI API Base", "display_name": "OpenAI API Base",
"field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
"info": ( "info": (
@ -51,14 +46,12 @@ class ChatOpenAIComponent(CustomComponent):
}, },
"openai_api_key": { "openai_api_key": {
"display_name": "OpenAI API Key", "display_name": "OpenAI API Key",
"field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
"password": True, "password": True,
}, },
"temperature": { "temperature": {
"display_name": "Temperature", "display_name": "Temperature",
"field_type": "float",
"advanced": False, "advanced": False,
"required": False, "required": False,
"value": 0.7, "value": 0.7,