From f379d54686cf6ead10d5ee1d3cdd108bc80d9ee9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 9 Sep 2023 08:28:29 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(process.py):=20add=20support?= =?UTF-8?q?=20for=20processing=20Document=20objects=20in=20process=5Fgraph?= =?UTF-8?q?=5Fcached=20function?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/processing/process.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index 6d62cdd68..2f8451371 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -11,6 +11,7 @@ from langflow.graph import Graph from langchain.chains.base import Chain from langchain.vectorstores.base import VectorStore from typing import Any, Dict, List, Optional, Tuple, Union +from langchain.schema import Document def fix_memory_inputs(langchain_object): @@ -126,6 +127,8 @@ def process_graph_cached( logger.debug("Generated result and thought") elif isinstance(langchain_object, VectorStore): result = langchain_object.search(**inputs) + elif isinstance(langchain_object, Document): + result = langchain_object.dict() else: raise ValueError( f"Unknown langchain_object type: {type(langchain_object).__name__}"