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

🐛 fix(chat.py): replace direct assignment with update_build_status function to update build status in cache_service dictionary

Browse source 
🔧 chore(utils.py): add update_build_status function to update build status in cache_service dictionary
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-27 11:46:22 -03:00 • committed by anovazzi1
commit d6dcfd5266
2 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/api/v1/chat.py
View file

@ -13,6 +13,7 @@ from langflow.api.v1.schemas import BuildStatus, BuiltResponse, InitResponse, St
from langflow.graph.graph.base import Graph from langflow.graph.graph.base import Graph
from langflow.services.auth.utils import get_current_active_user, get_current_user from langflow.services.auth.utils import get_current_active_user, get_current_user
from langflow.services.cache.utils import update_build_status
from loguru import logger from loguru import logger
from langflow.services.getters import get_chat_service, get_session, get_cache_service from langflow.services.getters import get_chat_service, get_session, get_cache_service
from sqlmodel import Session from sqlmodel import Session
@ -157,7 +158,7 @@ async def stream_build(
graph = Graph.from_payload(graph_data) graph = Graph.from_payload(graph_data)
number_of_nodes = len(graph.nodes) number_of_nodes = len(graph.nodes)
cache_service[flow_id]["status"] = BuildStatus.IN_PROGRESS update_build_status(cache_service, flow_id, BuildStatus.IN_PROGRESS)
for i, vertex in enumerate(graph.generator_build(), 1): for i, vertex in enumerate(graph.generator_build(), 1):
try: try:
@ -184,7 +185,7 @@ async def stream_build(
logger.exception(exc) logger.exception(exc)
params = str(exc) params = str(exc)
valid = False valid = False
cache_service[flow_id]["status"] = BuildStatus.FAILURE update_build_status(cache_service, flow_id, BuildStatus.FAILURE)
response = { response = {
"valid": valid, "valid": valid,
@ -211,11 +212,12 @@ async def stream_build(
chat_service.set_cache(flow_id, langchain_object) chat_service.set_cache(flow_id, langchain_object)
# We need to reset the chat history # We need to reset the chat history
chat_service.chat_history.empty_history(flow_id) chat_service.chat_history.empty_history(flow_id)
cache_service[flow_id]["status"] = BuildStatus.SUCCESS update_build_status(cache_service, flow_id, BuildStatus.SUCCESS)
except Exception as exc: except Exception as exc:
logger.exception(exc) logger.exception(exc)
logger.error("Error while building the flow: %s", exc) logger.error("Error while building the flow: %s", exc)
cache_service[flow_id]["status"] = BuildStatus.FAILURE
update_build_status(cache_service, flow_id, BuildStatus.FAILURE)
yield str(StreamData(event="error", data={"error": str(exc)})) yield str(StreamData(event="error", data={"error": str(exc)}))
finally: finally:
yield str(StreamData(event="message", data=final_response)) yield str(StreamData(event="message", data=final_response))

7
src/backend/langflow/services/cache/utils.py vendored
View file

@ -9,6 +9,7 @@ from pathlib import Path
from typing import TYPE_CHECKING, Any, Dict from typing import TYPE_CHECKING, Any, Dict
from appdirs import user_cache_dir from appdirs import user_cache_dir
from fastapi import UploadFile from fastapi import UploadFile
from langflow.api.v1.schemas import BuildStatus
from langflow.services.database.models.base import orjson_dumps from langflow.services.database.models.base import orjson_dumps
if TYPE_CHECKING: if TYPE_CHECKING:
@ -202,3 +203,9 @@ def save_uploaded_file(file: UploadFile, folder_name):
new_file.write(chunk) new_file.write(chunk)
return file_path return file_path
def update_build_status(cache_service, flow_id: str, status: BuildStatus):
cached_flow = cache_service[flow_id]
cached_flow["status"] = status
cache_service[flow_id] = cached_flow