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

Refactor content formatting in utils.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 22:39:14 -03:00
commit 2444a7dd09
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/langflow/interface/initialize/utils.py
View file

@ -1,13 +1,11 @@
import contextlib import contextlib
import json import json
from langflow.services.database.models.base import orjson_dumps
import orjson
from typing import Any, Dict, List from typing import Any, Dict, List
import orjson
from langchain.agents import ZeroShotAgent from langchain.agents import ZeroShotAgent
from langchain.schema import BaseOutputParser, Document
from langflow.services.database.models.base import orjson_dumps
from langchain.schema import Document, BaseOutputParser
def handle_node_type(node_type, class_object, params: Dict): def handle_node_type(node_type, class_object, params: Dict):
@ -85,8 +83,11 @@ def format_document(variable, input_variable: str, format_kwargs: Dict):
def format_content(variable): def format_content(variable):
if len(variable) > 1: if len(variable) > 1:
return "\n".join([item.page_content for item in variable if item.page_content]) return "\n".join([item.page_content for item in variable if item.page_content])
content = variable[0].page_content elif len(variable) == 1:
return try_to_load_json(content) content = variable[0].page_content
return try_to_load_json(content)
return ""
def try_to_load_json(content): def try_to_load_json(content):