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

Refactor imports and update Result model

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-22 11:21:48 -03:00
commit 3d4b9a81a3
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/processing/process.py
View file

@ -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