Merge pull request #40 from powerli2002/fix-mysql-delete

Fix problems with deleting logs

Author: peng3307165

modelcache/manager/scalar_data/sql_storage.py

@@ -149,13 +149,20 @@ def mark_deleted(self, keys):
     def model_deleted(self, model_name):
         table_name = "cache_codegpt_answer"
         delete_sql = "Delete from {} WHERE model='{}'".format(table_name, model_name)
+
+        table_log_name = "modelcache_query_log"
+        delete_log_sql = "Delete from {} WHERE model='{}'".format(table_log_name, model_name)
+
         conn = self.pool.connection()
         # 使用连接执行删除数据操作
         try:
             with conn.cursor() as cursor:
                 # 执行删除数据操作
                 resp = cursor.execute(delete_sql)
                 conn.commit()
+                # 执行删除该模型对应日志操作 resp_log行数不返回
+                resp_log = cursor.execute(delete_log_sql) 
+                conn.commit()  # 分别提交事务
         finally:
             # 关闭连接，将连接返回给连接池
             conn.close()

modelcache/manager/scalar_data/sql_storage_sqlite.py

@@ -168,13 +168,19 @@ def mark_deleted(self, keys):
     def model_deleted(self, model_name):
         table_name = "modelcache_llm_answer"
         delete_sql = "Delete from {} WHERE model=?".format(table_name)
+
+        table_log_name = "modelcache_query_log"
+        delete_log_sql = "Delete from {} WHERE model=?".format(table_log_name)
         conn = sqlite3.connect(self._url)
         try:
             cursor = conn.cursor()
             cursor.execute(delete_sql, (model_name,))
             conn.commit()
             # get delete rows
             deleted_rows_count = cursor.rowcount
+            
+            cursor.execute(delete_log_sql, (model_name,))
+            conn.commit()
             cursor.close()
         except sqlite3.Error as e:
             print(f"SQLite error: {e}")