From 2444a7dd094acc8bea6dd7ce404734ad8388e526 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 22:39:14 -0300 Subject: [PATCH] Refactor content formatting in utils.py --- .../langflow/interface/initialize/utils.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/src/backend/langflow/interface/initialize/utils.py b/src/backend/langflow/interface/initialize/utils.py index 4aa3fc52b..977e0dc74 100644 --- a/src/backend/langflow/interface/initialize/utils.py +++ b/src/backend/langflow/interface/initialize/utils.py @@ -1,13 +1,11 @@ import contextlib import json -from langflow.services.database.models.base import orjson_dumps -import orjson from typing import Any, Dict, List +import orjson from langchain.agents import ZeroShotAgent - - -from langchain.schema import Document, BaseOutputParser +from langchain.schema import BaseOutputParser, Document +from langflow.services.database.models.base import orjson_dumps 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): if len(variable) > 1: return "\n".join([item.page_content for item in variable if item.page_content]) - content = variable[0].page_content - return try_to_load_json(content) + elif len(variable) == 1: + content = variable[0].page_content + return try_to_load_json(content) + return "" + def try_to_load_json(content):