refactor: Remove unused code and imports in OpenAIModel.py

This commit is contained in:
ogabrielluiz 2024-06-18 00:18:19 -03:00
commit 18e5766273

View file

@ -1,11 +1,10 @@
from langchain_openai import ChatOpenAI
from langflow.schema.message import Message
from pydantic.v1 import SecretStr
from langflow.base.constants import STREAM_INFO_TEXT
from langflow.base.models.model import LCModelComponent
from langflow.base.models.openai_constants import MODEL_NAMES
from langflow.field_typing import BaseLanguageModel, Text
from langflow.field_typing import BaseLanguageModel
from langflow.inputs import (
BoolInput,
DictInput,
@ -16,6 +15,7 @@ from langflow.inputs import (
SecretStrInput,
StrInput,
)
from langflow.schema.message import Message
from langflow.template import Output
@ -64,12 +64,6 @@ class OpenAIModelComponent(LCModelComponent):
info="System message to pass to the model.",
advanced=True,
),
BoolInput(
name="json_mode",
display_name="JSON Mode",
info="Enable JSON mode for the model output.",
advanced=True,
),
IntInput(
name="seed",
display_name="Seed",