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

Add STREAM_INFO_TEXT constant to model components

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 11:19:11 -03:00
commit 1f28774be3
11 changed files with 21 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/base/constants.py Normal file
View file

@ -0,0 +1 @@
STREAM_INFO_TEXT = "Stream the response from the model. Streaming works only in Chat."

3
src/backend/base/langflow/components/models/AmazonBedrockModel.py
View file

@ -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,
},
}

4
src/backend/base/langflow/components/models/AnthropicModel.py
View file

@ -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",

4
src/backend/base/langflow/components/models/AzureOpenAIModel.py
View file

@ -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": {

4
src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
View file

@ -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": {

4
src/backend/base/langflow/components/models/CohereModel.py
View file

@ -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": {

4
src/backend/base/langflow/components/models/GoogleGenerativeAIModel.py
View file

@ -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": {

4
src/backend/base/langflow/components/models/HuggingFaceModel.py
View file

@ -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": {

3
src/backend/base/langflow/components/models/OllamaModel.py
View file

@ -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",

4
src/backend/base/langflow/components/models/OpenAIModel.py
View file

@ -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": {

4
src/backend/base/langflow/components/models/VertexAiModel.py
View file

@ -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": {