From 4b54a9d234c6fe2ffc4480194b4a69882e6a20e2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 19:39:14 -0300 Subject: [PATCH] Fix typing error and handle Coroutine object in generate_result() function --- src/backend/langflow/processing/process.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index 8f0b4b043..ff1c1af79 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -1,7 +1,7 @@ import asyncio import json from pathlib import Path -from typing import Any, Dict, List, Optional, Tuple, Union +from typing import Any, Coroutine, Dict, List, Optional, Tuple, Union from langchain.chains.base import Chain from langchain.schema import AgentAction, Document @@ -138,6 +138,8 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict): result = langchain_object.dict() else: logger.warning(f"Unknown langchain_object type: {type(langchain_object)}") + if isinstance(langchain_object, Coroutine): + result = asyncio.run(langchain_object) result = langchain_object return result