diff --git a/src/backend/base/langflow/components/models/OpenAIModel.py b/src/backend/base/langflow/components/models/OpenAIModel.py index 1960572c5..1b5dcbcd6 100644 --- a/src/backend/base/langflow/components/models/OpenAIModel.py +++ b/src/backend/base/langflow/components/models/OpenAIModel.py @@ -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",