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

Add support for custom components in langchain_object

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-17 20:44:35 -03:00
commit 2afb0fb119
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -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)}")