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