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

Refactor OpenAIModelComponent configuration fields

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

11
src/backend/langflow/components/models/OpenAI.py
View file

@ -11,21 +11,19 @@ class OpenAIModelComponent(CustomComponent):
def build_config(self):
return {
"inputs": {"display_name": "Input"},
"max_tokens": {
"display_name": "Max Tokens",
"field_type": "int",
"advanced": False,
"required": False,
},
"model_kwargs": {
"display_name": "Model Kwargs",
"field_type": "NestedDict",
"advanced": True,
"required": False,
},
"model_name": {
"display_name": "Model Name",
"field_type": "str",
"advanced": False,
"required": False,
"options": [
@ -39,7 +37,6 @@ class OpenAIModelComponent(CustomComponent):
},
"openai_api_base": {
"display_name": "OpenAI API Base",
"field_type": "str",
"advanced": False,
"required": False,
"info": (
@ -49,14 +46,12 @@ class OpenAIModelComponent(CustomComponent):
},
"openai_api_key": {
"display_name": "OpenAI API Key",
"field_type": "str",
"advanced": False,
"required": False,
"password": True,
},
"temperature": {
"display_name": "Temperature",
"field_type": "float",
"advanced": False,
"required": False,
"value": 0.7,
@ -85,4 +80,6 @@ class OpenAIModelComponent(CustomComponent):
)
message = model.invoke(inputs)
return message.content if hasattr(message, "content") else message
result = message.content if hasattr(message, "content") else message
self.status = result
return result