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):
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 ()
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments