Skip to content

Commit 7afe7b1

Browse files
committed
Fix problems with deleting logs
1 parent 4dd50fe commit 7afe7b1

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

modelcache/manager/scalar_data/sql_storage.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,20 @@ def mark_deleted(self, keys):
149149
def model_deleted(self, model_name):
150150
table_name = "cache_codegpt_answer"
151151
delete_sql = "Delete from {} WHERE model='{}'".format(table_name, model_name)
152+
153+
table_log_name = "modelcache_query_log"
154+
delete_log_sql = "Delete from {} WHERE model='{}'".format(table_log_name, model_name)
155+
152156
conn = self.pool.connection()
153157
# 使用连接执行删除数据操作
154158
try:
155159
with conn.cursor() as cursor:
156160
# 执行删除数据操作
157161
resp = cursor.execute(delete_sql)
158162
conn.commit()
163+
# 执行删除该模型对应日志操作 resp_log行数不返回
164+
resp_log = cursor.execute(delete_log_sql)
165+
conn.commit() # 分别提交事务
159166
finally:
160167
# 关闭连接,将连接返回给连接池
161168
conn.close()

modelcache/manager/scalar_data/sql_storage_sqlite.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,11 +260,16 @@ def mark_deleted(self, keys):
260260
def model_deleted(self, model_name):
261261
table_name = "modelcache_llm_answer"
262262
delete_sql = "Delete from {} WHERE model='{}'".format(table_name, model_name)
263+
264+
table_log_name = "modelcache_query_log"
265+
delete_log_sql = "Delete from {} WHERE model='{}'".format(table_log_name, model_name)
263266
conn = sqlite3.connect(self._url)
264267
try:
265268
cursor = conn.cursor()
266269
resp = cursor.execute(delete_sql)
267270
conn.commit()
271+
resp = cursor.execute(delete_log_sql)
272+
conn.commit()
268273
cursor.close()
269274
conn.close()
270275
finally:

0 commit comments

Comments
 (0)