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

Add support for Runnable objects in generate_result function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-17 17:32:03 -03:00
commit f36b6a5693
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -12,6 +12,7 @@ from langflow.interface.run import build_sorted_vertices, get_memory_key, update
from langflow.services.deps import get_session_service from langflow.services.deps import get_session_service
from loguru import logger from loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from langchain_core.runnables.base import Runnable
def fix_memory_inputs(langchain_object): def fix_memory_inputs(langchain_object):
@ -131,6 +132,10 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict):
result = langchain_object.search(**inputs) result = langchain_object.search(**inputs)
elif isinstance(langchain_object, Document): elif isinstance(langchain_object, Document):
result = langchain_object.dict() result = langchain_object.dict()
elif isinstance(langchain_object, Runnable):
result = langchain_object.invoke(inputs)
result = result.content if hasattr(result, "content") else result
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): if isinstance(langchain_object, Coroutine):