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

Update ChatOpenAIComponent parameters

Browse source
This commit is contained in:
anovazzi1 2024-01-15 16:53:22 -03:00
commit 1afaf587af
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -13,13 +13,13 @@ class ChatOpenAIComponent(CustomComponent):
return {
"max_tokens": {
"display_name": "Max Tokens",
"field_type": "NestedDict",
"field_type": "int",
"advanced": False,
"required": False,
},
"model_kwargs": {
"display_name": "Model Kwargs",
"field_type": "dict",
"field_type": "NestedDict",
"advanced": True,
"required": False,
},
@ -63,13 +63,15 @@ class ChatOpenAIComponent(CustomComponent):
def build(
self,
max_tokens: Optional[int] = None,
model_kwargs: Optional[NestedDict] = None,
max_tokens: Optional[int] = 256,
model_kwargs: Optional[NestedDict] = {},
model_name: Optional[str] = "gpt-4-1106-preview",
openai_api_base: Optional[str] = None,
openai_api_key: Optional[str] = None,
temperature: float = 0.7,
) -> Union[BaseLanguageModel, BaseLLM]:
if(not openai_api_base):
openai_api_base = "https://api.openai.com/v1"
return ChatOpenAI(
max_tokens=max_tokens,
model_kwargs=model_kwargs,