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

Refactor OpenAIModel.py: Remove unnecessary import and update api_key parameter

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-22 15:17:06 -03:00
commit af83e426bf
1 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/components/models/OpenAIModel.py
View file

@ -1,7 +1,6 @@
from typing import Optional from typing import Optional
from langchain_openai import ChatOpenAI from langchain_openai import ChatOpenAI
from pydantic.v1 import SecretStr
from langflow.components.models.base.model import LCModelComponent from langflow.components.models.base.model import LCModelComponent
from langflow.field_typing import NestedDict, Text from langflow.field_typing import NestedDict, Text
@ -78,16 +77,12 @@ class OpenAIModelComponent(LCModelComponent):
) -> Text: ) -> Text:
if not openai_api_base: if not openai_api_base:
openai_api_base = "https://api.openai.com/v1" openai_api_base = "https://api.openai.com/v1"
if openai_api_key:
secret_key = SecretStr(openai_api_key)
else:
secret_key = None
output = ChatOpenAI( output = ChatOpenAI(
max_tokens=max_tokens, max_tokens=max_tokens,
model_kwargs=model_kwargs, model_kwargs=model_kwargs,
model=model_name, model=model_name,
base_url=openai_api_base, base_url=openai_api_base,
api_key=secret_key, api_key=openai_api_key,
temperature=temperature, temperature=temperature,
) )