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

Add ChatOpenAI model and update field types

Browse source
This commit is contained in:
anovazzi1 2024-01-10 11:47:54 -03:00
commit 11c5a3e001
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,7 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain.llms import BaseLLM from langchain.llms import BaseLLM
from typing import Optional, Dict, Union, Any from typing import Optional, Dict, Union, Any
from langchain_community.chat_models.openai import ChatOpenAI
from langflow.field_typing import BaseLanguageModel from langflow.field_typing import BaseLanguageModel
@ -12,19 +13,19 @@ class ChatOpenAIComponent(CustomComponent):
return { return {
"max_tokens": { "max_tokens": {
"display_name": "Max Tokens", "display_name": "Max Tokens",
"type": "int", "field_type": "int",
"advanced": False, "advanced": False,
"required": False, "required": False,
}, },
"model_kwargs": { "model_kwargs": {
"display_name": "Model Kwargs", "display_name": "Model Kwargs",
"type": "dict", "field_type": "dict",
"advanced": True, "advanced": True,
"required": False, "required": False,
}, },
"model_name": { "model_name": {
"display_name": "Model Name", "display_name": "Model Name",
"type": "str", "field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
"options": [ "options": [
@ -37,7 +38,7 @@ class ChatOpenAIComponent(CustomComponent):
}, },
"openai_api_base": { "openai_api_base": {
"display_name": "OpenAI API Base", "display_name": "OpenAI API Base",
"type": "str", "field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
"info": ( "info": (
@ -47,16 +48,16 @@ class ChatOpenAIComponent(CustomComponent):
}, },
"openai_api_key": { "openai_api_key": {
"display_name": "OpenAI API Key", "display_name": "OpenAI API Key",
"type": "str", "field_type": "str",
"advanced": False, "advanced": False,
"required": False, "required": False,
}, },
"temperature": { "temperature": {
"display_name": "Temperature", "display_name": "Temperature",
"type": "float", "field_type": "float",
"advanced": False, "advanced": False,
"required": False, "required": False,
"default": 0.7, "value": 0.7,
}, },
} }