From 7ebaf2914c03d33638e0a34d9ec751ddab08c33c Mon Sep 17 00:00:00 2001 From: ming Date: Sun, 9 Jun 2024 09:18:27 -0400 Subject: [PATCH] fix lint - type mismatch (#2117) * fix lint * empty commit to trigger ci flow --- src/backend/base/langflow/base/curl/parse.py | 2 +- .../base/langflow/services/monitor/service.py | 17 +++++++++++++---- 2 files changed, 14 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/base/curl/parse.py b/src/backend/base/langflow/base/curl/parse.py index 892abdde2..7a5eff92b 100644 --- a/src/backend/base/langflow/base/curl/parse.py +++ b/src/backend/base/langflow/base/curl/parse.py @@ -16,7 +16,7 @@ from collections import OrderedDict, namedtuple from http.cookies import SimpleCookie ParsedArgs = namedtuple( - "ParsedContext", + "ParsedArgs", [ "command", "url", diff --git a/src/backend/base/langflow/services/monitor/service.py b/src/backend/base/langflow/services/monitor/service.py index 02bc59cc2..805a5e6eb 100644 --- a/src/backend/base/langflow/services/monitor/service.py +++ b/src/backend/base/langflow/services/monitor/service.py @@ -1,6 +1,6 @@ from datetime import datetime from pathlib import Path -from typing import TYPE_CHECKING, Optional, Union +from typing import TYPE_CHECKING, Optional, Union, List import duckdb from loguru import logger @@ -109,12 +109,21 @@ class MonitorService(Service): return self.exec_query(query) - def delete_messages(self, message_ids: list[int]): - query = f"DELETE FROM messages WHERE index IN ({','.join(map(str, message_ids))})" + def delete_messages(self, message_ids: Union[List[int], str]): + if isinstance(message_ids, list): + # If message_ids is a list, join the string representations of the integers + ids_str = ','.join(map(str, message_ids)) + elif isinstance(message_ids, str): + # If message_ids is already a string, use it directly + ids_str = message_ids + else: + raise ValueError("message_ids must be a list of integers or a string") + + query = f"DELETE FROM messages WHERE index IN ({ids_str})" return self.exec_query(query) - def update_message(self, message_id: int, **kwargs): + def update_message(self, message_id: str, **kwargs): query = ( f"""UPDATE messages SET {', '.join(f"{k} = '{v}'" for k, v in kwargs.items())} WHERE index = {message_id}""" )