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

Refactor code to use get_text() method instead of accessing the 'text' attribute directly.

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-29 22:29:47 -03:00
commit 617be2809f
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/base/prompts/utils.py
View file

@ -45,7 +45,7 @@ def record_to_string(record: Record) -> str:
Returns: Returns:
str: The record as a string. str: The record as a string.
""" """
return record.text return record.get_text()
def document_to_string(document: Document) -> str: def document_to_string(document: Document) -> str:

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

@ -125,7 +125,7 @@ class DocumentLoaderVertex(Vertex):
# show how many documents are in the list? # show how many documents are in the list?
if not isinstance(self._built_object, UnbuiltObject): if not isinstance(self._built_object, UnbuiltObject):
avg_length = sum(len(record.text) for record in self._built_object if hasattr(record, "text")) / len( avg_length = sum(len(record.get_text()) for record in self._built_object if hasattr(record, "text")) / len(
self._built_object self._built_object
) )
return f"""{self.display_name}({len(self._built_object)} records) return f"""{self.display_name}({len(self._built_object)} records)

2
src/backend/base/langflow/interface/custom/custom_component/custom_component.py
View file

@ -249,7 +249,7 @@ class CustomComponent(Component):
return "" return ""
markdown_string = "---\n" markdown_string = "---\n"
for record in records: for record in records:
markdown_string += f"- Text: {record.text}" markdown_string += f"- Text: {record.get_text()}"
if include_data: if include_data:
markdown_string += f" Data: {record.data}" markdown_string += f" Data: {record.data}"
markdown_string += "\n" markdown_string += "\n"