From 2afb0fb119b49cfdbed3cd874b225350402a2e46 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 17 Dec 2023 20:44:35 -0300 Subject: [PATCH] Add support for custom components in langchain_object --- src/backend/langflow/processing/process.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index 1a25965de..80846ec11 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -7,6 +7,7 @@ from langchain.agents import AgentExecutor from langchain.chains.base import Chain from langchain.schema import AgentAction, Document from langchain.vectorstores.base import VectorStore +from langflow.components.custom_components import CustomComponent from langflow.graph import Graph from langflow.interface.run import build_sorted_vertices, get_memory_key, update_memory_keys from langflow.services.deps import get_session_service @@ -135,6 +136,8 @@ def generate_result(langchain_object: Union[Chain, VectorStore], inputs: dict): elif isinstance(langchain_object, Runnable): result = langchain_object.invoke(inputs) result = result.content if hasattr(result, "content") else result + elif hasattr(langchain_object, "run") and isinstance(langchain_object, CustomComponent): + result = langchain_object.run(inputs) else: logger.warning(f"Unknown langchain_object type: {type(langchain_object)}")