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

Fix typing error and handle Coroutine object in

Browse source 
generate_result() function
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-28 19:39:14 -03:00
commit 4b54a9d234
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import asyncio import asyncio
import json import json
from pathlib import Path 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.chains.base import Chain
from langchain.schema import AgentAction, Document from langchain.schema import AgentAction, Document
@ -138,6 +138,8 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict):
result = langchain_object.dict() result = langchain_object.dict()
else: else:
logger.warning(f"Unknown langchain_object type: {type(langchain_object)}") logger.warning(f"Unknown langchain_object type: {type(langchain_object)}")
if isinstance(langchain_object, Coroutine):
result = asyncio.run(langchain_object)
result = langchain_object result = langchain_object
return result return result