We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3ba1e6f commit 02706beCopy full SHA for 02706be
modelcache/manager/scalar_data/sql_storage_sqlite.py
@@ -168,13 +168,19 @@ def mark_deleted(self, keys):
168
def model_deleted(self, model_name):
169
table_name = "modelcache_llm_answer"
170
delete_sql = "Delete from {} WHERE model=?".format(table_name)
171
+
172
+ table_log_name = "modelcache_query_log"
173
+ delete_log_sql = "Delete from {} WHERE model=?".format(table_log_name)
174
conn = sqlite3.connect(self._url)
175
try:
176
cursor = conn.cursor()
177
cursor.execute(delete_sql, (model_name,))
178
conn.commit()
179
# get delete rows
180
deleted_rows_count = cursor.rowcount
181
182
+ cursor.execute(delete_log_sql, (model_name,))
183
+ conn.commit()
184
cursor.close()
185
except sqlite3.Error as e:
186
print(f"SQLite error: {e}")
0 commit comments