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

Update ChatOpenAIComponent with NestedDict support

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

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

@ -1,8 +1,8 @@
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, Union
from langchain_community.chat_models.openai import ChatOpenAI from langchain_community.chat_models.openai import ChatOpenAI
from langflow.field_typing import BaseLanguageModel from langflow.field_typing import BaseLanguageModel, NestedDict
class ChatOpenAIComponent(CustomComponent): class ChatOpenAIComponent(CustomComponent):
@ -13,7 +13,7 @@ class ChatOpenAIComponent(CustomComponent):
return { return {
"max_tokens": { "max_tokens": {
"display_name": "Max Tokens", "display_name": "Max Tokens",
"field_type": "int", "field_type": "NestedDict",
"advanced": False, "advanced": False,
"required": False, "required": False,
}, },
@ -64,14 +64,12 @@ class ChatOpenAIComponent(CustomComponent):
def build( def build(
self, self,
max_tokens: Optional[int] = None, max_tokens: Optional[int] = None,
model_kwargs: Optional[Dict[str, Any]] = None, model_kwargs: Optional[NestedDict] = None,
model_name: Optional[str] = "gpt-4-1106-preview", model_name: Optional[str] = "gpt-4-1106-preview",
openai_api_base: Optional[str] = None, openai_api_base: Optional[str] = None,
openai_api_key: Optional[str] = None, openai_api_key: Optional[str] = None,
temperature: float = 0.7, temperature: float = 0.7,
) -> Union[BaseLanguageModel, BaseLLM]: ) -> Union[BaseLanguageModel, BaseLLM]:
# Assuming there is a class `ChatOpenAI` that takes these parameters
# The `ChatOpenAI` class must be imported or defined elsewhere in the actual implementation
return ChatOpenAI( return ChatOpenAI(
max_tokens=max_tokens, max_tokens=max_tokens,
model_kwargs=model_kwargs, model_kwargs=model_kwargs,