From 92ba9562b3802e8a3ca7a637515c66c6f3343373 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 31 Jan 2024 15:20:10 -0300 Subject: [PATCH] Refactor MonitorService class and add delete_vertex_builds method --- src/backend/langflow/services/monitor/service.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/monitor/service.py b/src/backend/langflow/services/monitor/service.py index 397a938ee..b081aded9 100644 --- a/src/backend/langflow/services/monitor/service.py +++ b/src/backend/langflow/services/monitor/service.py @@ -3,12 +3,11 @@ from pathlib import Path from typing import TYPE_CHECKING, Optional import duckdb -from loguru import logger -from platformdirs import user_cache_dir - from langflow.services.base import Service from langflow.services.monitor.schema import MessageModel, TransactionModel, VertexBuildModel from langflow.services.monitor.utils import add_row_to_table, drop_and_create_table_if_schema_mismatch +from loguru import logger +from platformdirs import user_cache_dir if TYPE_CHECKING: from langflow.services.settings.manager import SettingsService @@ -85,6 +84,14 @@ class MonitorService(Service): return df.to_dict(orient="records") + def delete_vertex_builds(self, flow_id: Optional[str] = None): + query = "DELETE FROM vertex_builds" + if flow_id: + query += f" WHERE flow_id = '{flow_id}'" + + with duckdb.connect(str(self.db_path)) as conn: + conn.execute(query) + def get_messages( self, sender_type: Optional[str] = None,