From 372f39bcadf593861e957461968b80f63caa9f02 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 13 Dec 2023 19:08:35 -0300 Subject: [PATCH] Fix build failure and optimize imports --- src/backend/langflow/api/v1/chat.py | 1 - src/backend/langflow/services/cache/factory.py | 3 +-- src/backend/langflow/services/database/service.py | 4 +--- 3 files changed, 2 insertions(+), 6 deletions(-) diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py index c034c2676..059ad23b6 100644 --- a/src/backend/langflow/api/v1/chat.py +++ b/src/backend/langflow/api/v1/chat.py @@ -178,7 +178,6 @@ async def stream_build( time_elapsed = format_elapsed_time(time.perf_counter() - start_time) update_build_status(cache_service, flow_id, BuildStatus.FAILURE) - vertex_id = vertex.parent_node_id if vertex.parent_is_top_level else vertex.id if vertex_id in graph.top_level_vertices: response = { diff --git a/src/backend/langflow/services/cache/factory.py b/src/backend/langflow/services/cache/factory.py index c74a82512..32f726c6b 100644 --- a/src/backend/langflow/services/cache/factory.py +++ b/src/backend/langflow/services/cache/factory.py @@ -1,7 +1,6 @@ from typing import TYPE_CHECKING -from langflow.services.cache.service import (BaseCacheService, InMemoryCache, - RedisCache) +from langflow.services.cache.service import BaseCacheService, InMemoryCache, RedisCache from langflow.services.factory import ServiceFactory from langflow.utils.logger import logger diff --git a/src/backend/langflow/services/database/service.py b/src/backend/langflow/services/database/service.py index 1216d5699..9765d858f 100644 --- a/src/backend/langflow/services/database/service.py +++ b/src/backend/langflow/services/database/service.py @@ -148,9 +148,7 @@ class DatabaseService(Service): try: command.check(alembic_cfg) except Exception as exc: - if isinstance( - exc, (util.exc.CommandError, util.exc.AutogenerateDiffsDetected) - ): + if isinstance(exc, (util.exc.CommandError, util.exc.AutogenerateDiffsDetected)): command.upgrade(alembic_cfg, "head") time.sleep(3)