File tree Expand file tree Collapse file tree 2 files changed +12
-0
lines changed
modelcache/manager/scalar_data Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -149,13 +149,20 @@ def mark_deleted(self, keys):
149
149
def model_deleted (self , model_name ):
150
150
table_name = "cache_codegpt_answer"
151
151
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
+
152
156
conn = self .pool .connection ()
153
157
# 使用连接执行删除数据操作
154
158
try :
155
159
with conn .cursor () as cursor :
156
160
# 执行删除数据操作
157
161
resp = cursor .execute (delete_sql )
158
162
conn .commit ()
163
+ # 执行删除该模型对应日志操作 resp_log行数不返回
164
+ resp_log = cursor .execute (delete_log_sql )
165
+ conn .commit () # 分别提交事务
159
166
finally :
160
167
# 关闭连接,将连接返回给连接池
161
168
conn .close ()
Original file line number Diff line number Diff line change @@ -260,11 +260,16 @@ def mark_deleted(self, keys):
260
260
def model_deleted (self , model_name ):
261
261
table_name = "modelcache_llm_answer"
262
262
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 )
263
266
conn = sqlite3 .connect (self ._url )
264
267
try :
265
268
cursor = conn .cursor ()
266
269
resp = cursor .execute (delete_sql )
267
270
conn .commit ()
271
+ resp = cursor .execute (delete_log_sql )
272
+ conn .commit ()
268
273
cursor .close ()
269
274
conn .close ()
270
275
finally :
You can’t perform that action at this time.
0 commit comments