From 3d4b9a81a3ba94662d1a4e20d22278e4ac2a2c0e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 22 Dec 2023 11:21:48 -0300 Subject: [PATCH] Refactor imports and update Result model --- src/backend/langflow/processing/process.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index 9432b7f96..3c7596774 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -7,12 +7,11 @@ from langchain.schema import AgentAction, Document from langchain.vectorstores.base import VectorStore from langchain_core.messages import AIMessage from langchain_core.runnables.base import Runnable -from loguru import logger -from pydantic import BaseModel - from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.run import build_sorted_vertices, get_memory_key, update_memory_keys from langflow.services.deps import get_session_service +from loguru import logger +from pydantic import BaseModel def fix_memory_inputs(langchain_object): @@ -201,7 +200,7 @@ async def generate_result(built_object: Union[Chain, VectorStore, Runnable], inp class Result(BaseModel): - result: Union[dict, List[dict], str, List[str], AIMessage] + result: Any session_id: str