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

refactor: Update output names in TextInput, TextOutput, RecordsOutput, ChatInput, and ChatOutput

Browse source
This commit is contained in:
ogabrielluiz 2024-06-07 14:40:03 -03:00
commit 834570df8b
3 changed files with 15 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/field_typing/constants.py
View file

@ -6,7 +6,7 @@ from langchain.memory.chat_memory import BaseChatMemory
from langchain_core.document_loaders import BaseLoader from langchain_core.document_loaders import BaseLoader
from langchain_core.documents import Document from langchain_core.documents import Document
from langchain_core.embeddings import Embeddings from langchain_core.embeddings import Embeddings
from langchain_core.language_models import BaseLLM, BaseLanguageModel from langchain_core.language_models import BaseLanguageModel, BaseLLM
from langchain_core.memory import BaseMemory from langchain_core.memory import BaseMemory
from langchain_core.output_parsers import BaseOutputParser from langchain_core.output_parsers import BaseOutputParser
from langchain_core.prompts import BasePromptTemplate, ChatPromptTemplate, PromptTemplate from langchain_core.prompts import BasePromptTemplate, ChatPromptTemplate, PromptTemplate

6
src/backend/base/langflow/graph/vertex/types.py
View file

@ -156,7 +156,7 @@ class InterfaceVertex(ComponentVertex):
if isinstance(message, str): if isinstance(message, str):
message = unescape_string(message) message = unescape_string(message)
stream_url = None stream_url = None
text_output = self.results["Message"] text_output = self.results["message"]
if isinstance(text_output, (AIMessage, AIMessageChunk)): if isinstance(text_output, (AIMessage, AIMessageChunk)):
artifacts = ChatOutputResponse.from_message( artifacts = ChatOutputResponse.from_message(
text_output, text_output,
@ -173,8 +173,8 @@ class InterfaceVertex(ComponentVertex):
elif isinstance(message, (AsyncIterator, Iterator)): elif isinstance(message, (AsyncIterator, Iterator)):
stream_url = self.build_stream_url() stream_url = self.build_stream_url()
message = "" message = ""
self.results["Message"] = message self.results["message"] = message
self.results["Record"].message = message self.results["record"].message = message
self._built_object = self.results self._built_object = self.results
elif not isinstance(text_output, str): elif not isinstance(text_output, str):
message = str(text_output) message = str(text_output)

19
src/backend/base/langflow/template/field/base.py
View file

@ -84,7 +84,7 @@ class Input(BaseModel):
def serialize_model(self, handler): def serialize_model(self, handler):
result = handler(self) result = handler(self)
# If the field is str, we add the Text input type # If the field is str, we add the Text input type
if self.field_type in [str, Text]: if self.field_type in ["str", "Text"]:
if "input_types" not in result: if "input_types" not in result:
result["input_types"] = ["Text"] result["input_types"] = ["Text"]
if self.field_type == Text: if self.field_type == Text:
@ -110,7 +110,10 @@ class Input(BaseModel):
# this should be done for all types # this should be done for all types
# How to check if v is a type? # How to check if v is a type?
if isinstance(v, (type, GenericAlias)): if isinstance(v, (type, GenericAlias)):
return str(v) if isinstance(v, type):
v = v.__name__
else:
v = str(v)
elif not isinstance(v, str): elif not isinstance(v, str):
raise ValueError(f"type must be a string or a type, not {type(v)}") raise ValueError(f"type must be a string or a type, not {type(v)}")
return v return v
@ -144,19 +147,19 @@ class Input(BaseModel):
class Output(BaseModel): class Output(BaseModel):
types: Optional[list[str]] = Field(default=[], serialization_alias="types") types: Optional[list[str]] = Field(default=[])
"""List of output types for the field.""" """List of output types for the field."""
selected: Optional[str] = Field(default=None, serialization_alias="selected") selected: Optional[str] = Field(default=None)
"""The selected output type for the field.""" """The selected output type for the field."""
name: str = Field(default=None)
"""The name of the field."""
display_name: Optional[str] = Field(default=None) display_name: Optional[str] = Field(default=None)
"""The display name of the field.""" """The display name of the field."""
name: str = Field(default=None, serialization_alias="name") method: Optional[str] = Field(default=None)
"""The name of the field."""
method: Optional[str] = Field(default=None, serialization_alias="method")
"""The method to use for the output.""" """The method to use for the output."""
def to_dict(self): def to_dict(self):