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

🐛 (chat.py): Call 'initialize_run()' method on the 'graph' object after retrieving it from cache to ensure proper initialization

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-21 21:59:51 -03:00
commit 2cac7b472f
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/v1/chat.py
View file

@ -25,6 +25,7 @@ from langflow.api.v1.schemas import (
VerticesOrderResponse, VerticesOrderResponse,
) )
from langflow.exceptions.component import ComponentBuildException from langflow.exceptions.component import ComponentBuildException
from langflow.graph.graph.base import Graph
from langflow.schema.schema import OutputLog from langflow.schema.schema import OutputLog
from langflow.services.auth.utils import get_current_active_user from langflow.services.auth.utils import get_current_active_user
from langflow.services.chat.service import ChatService from langflow.services.chat.service import ChatService
@ -161,7 +162,8 @@ async def build_vertex(
flow_id=flow_id_str, session=next(get_session()), chat_service=chat_service flow_id=flow_id_str, session=next(get_session()), chat_service=chat_service
) )
else: else:
graph = cache.get("result") graph: "Graph" = cache.get("result")
await graph.initialize_run()
vertex = graph.get_vertex(vertex_id) vertex = graph.get_vertex(vertex_id)
try: try: