diff --git a/src/backend/base/langflow/base/constants.py b/src/backend/base/langflow/base/constants.py new file mode 100644 index 000000000..86bb461ed --- /dev/null +++ b/src/backend/base/langflow/base/constants.py @@ -0,0 +1 @@ +STREAM_INFO_TEXT = "Stream the response from the model. Streaming works only in Chat." diff --git a/src/backend/base/langflow/components/models/AmazonBedrockModel.py b/src/backend/base/langflow/components/models/AmazonBedrockModel.py index 2fe12bbce..1015f1684 100644 --- a/src/backend/base/langflow/components/models/AmazonBedrockModel.py +++ b/src/backend/base/langflow/components/models/AmazonBedrockModel.py @@ -2,6 +2,7 @@ from typing import Optional from langchain_community.chat_models.bedrock import BedrockChat +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent from langflow.field_typing import Text @@ -65,7 +66,7 @@ class AmazonBedrockComponent(LCModelComponent): }, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, } diff --git a/src/backend/base/langflow/components/models/AnthropicModel.py b/src/backend/base/langflow/components/models/AnthropicModel.py index 837db9318..29b2a1fe4 100644 --- a/src/backend/base/langflow/components/models/AnthropicModel.py +++ b/src/backend/base/langflow/components/models/AnthropicModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain_anthropic.chat_models import ChatAnthropic from pydantic.v1 import SecretStr +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class AnthropicLLM(LCModelComponent): @@ -67,7 +67,7 @@ class AnthropicLLM(LCModelComponent): "stream": { "display_name": "Stream", "advanced": True, - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, }, "system_message": { "display_name": "System Message", diff --git a/src/backend/base/langflow/components/models/AzureOpenAIModel.py b/src/backend/base/langflow/components/models/AzureOpenAIModel.py index 7b5a73b9e..51117eb56 100644 --- a/src/backend/base/langflow/components/models/AzureOpenAIModel.py +++ b/src/backend/base/langflow/components/models/AzureOpenAIModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain.llms.base import BaseLanguageModel from langchain_openai import AzureChatOpenAI +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class AzureChatOpenAIComponent(LCModelComponent): @@ -81,7 +81,7 @@ class AzureChatOpenAIComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py index a3aff1b6e..f4a7415a4 100644 --- a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py +++ b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint from pydantic.v1 import SecretStr +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class QianfanChatEndpointComponent(LCModelComponent): @@ -83,7 +83,7 @@ class QianfanChatEndpointComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/CohereModel.py b/src/backend/base/langflow/components/models/CohereModel.py index d4eeaa186..21406f793 100644 --- a/src/backend/base/langflow/components/models/CohereModel.py +++ b/src/backend/base/langflow/components/models/CohereModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain_community.chat_models.cohere import ChatCohere from pydantic.v1 import SecretStr +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class CohereComponent(LCModelComponent): @@ -47,7 +47,7 @@ class CohereComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py b/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py index 10aad490a..412ff2d4e 100644 --- a/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py +++ b/src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain_google_genai import ChatGoogleGenerativeAI from pydantic.v1 import SecretStr +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import RangeSpec, Text class GoogleGenerativeAIComponent(LCModelComponent): @@ -67,7 +67,7 @@ class GoogleGenerativeAIComponent(LCModelComponent): "input_value": {"display_name": "Input", "info": "The input to the model."}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/HuggingFaceModel.py b/src/backend/base/langflow/components/models/HuggingFaceModel.py index 4230a29e4..4be24ab13 100644 --- a/src/backend/base/langflow/components/models/HuggingFaceModel.py +++ b/src/backend/base/langflow/components/models/HuggingFaceModel.py @@ -3,8 +3,8 @@ from typing import Optional from langchain_community.chat_models.huggingface import ChatHuggingFace from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class HuggingFaceEndpointsComponent(LCModelComponent): @@ -39,7 +39,7 @@ class HuggingFaceEndpointsComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/OllamaModel.py b/src/backend/base/langflow/components/models/OllamaModel.py index cd8296cc7..41d75ba3e 100644 --- a/src/backend/base/langflow/components/models/OllamaModel.py +++ b/src/backend/base/langflow/components/models/OllamaModel.py @@ -3,6 +3,7 @@ from typing import Any, Dict, List, Optional # from langchain_community.chat_models import ChatOllama from langchain_community.chat_models import ChatOllama +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent # from langchain.chat_models import ChatOllama @@ -201,7 +202,7 @@ class ChatOllamaComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, }, "system_message": { "display_name": "System Message", diff --git a/src/backend/base/langflow/components/models/OpenAIModel.py b/src/backend/base/langflow/components/models/OpenAIModel.py index 4be804671..f16e49144 100644 --- a/src/backend/base/langflow/components/models/OpenAIModel.py +++ b/src/backend/base/langflow/components/models/OpenAIModel.py @@ -2,8 +2,8 @@ from typing import Optional from langchain_openai import ChatOpenAI +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import NestedDict, Text class OpenAIModelComponent(LCModelComponent): @@ -69,7 +69,7 @@ class OpenAIModelComponent(LCModelComponent): }, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": { diff --git a/src/backend/base/langflow/components/models/VertexAiModel.py b/src/backend/base/langflow/components/models/VertexAiModel.py index 5f778bbbc..69387c872 100644 --- a/src/backend/base/langflow/components/models/VertexAiModel.py +++ b/src/backend/base/langflow/components/models/VertexAiModel.py @@ -2,8 +2,8 @@ from typing import List, Optional from langchain_core.messages.base import BaseMessage +from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing import Text class ChatVertexAIComponent(LCModelComponent): @@ -77,7 +77,7 @@ class ChatVertexAIComponent(LCModelComponent): "input_value": {"display_name": "Input"}, "stream": { "display_name": "Stream", - "info": "Stream the response from the model.", + "info": STREAM_INFO_TEXT, "advanced": True, }, "system_message": {