Spaces:
Runtime error
Runtime error
Update memory_logic.py
Browse files- memory_logic.py +4 -2
memory_logic.py
CHANGED
@@ -259,11 +259,13 @@ def remove_rule_entry(rule_text_to_delete: str) -> bool:
|
|
259 |
if rule_text_to_delete not in _rules_items_list: return False
|
260 |
try:
|
261 |
new_list = [r for r in _rules_items_list if r != rule_text_to_delete]
|
262 |
-
|
|
|
|
|
263 |
|
264 |
if STORAGE_BACKEND == "SQLITE":
|
265 |
with _get_sqlite_connection() as conn: conn.execute("DELETE FROM rules WHERE rule_text = ?", (rule_text_to_delete,)); conn.commit()
|
266 |
-
elif STORAGE_BACKEND == "HF_DATASET":
|
267 |
Dataset.from_dict({"rule_text": list(_rules_items_list)}).push_to_hub(HF_RULES_DATASET_REPO, token=HF_TOKEN, private=True)
|
268 |
return True
|
269 |
except Exception as e:
|
|
|
259 |
if rule_text_to_delete not in _rules_items_list: return False
|
260 |
try:
|
261 |
new_list = [r for r in _rules_items_list if r != rule_text_to_delete]
|
262 |
+
new_index, valid_items = _build_faiss_index(new_list, lambda r: r)
|
263 |
+
_faiss_rules_index = new_index
|
264 |
+
_rules_items_list = valid_items
|
265 |
|
266 |
if STORAGE_BACKEND == "SQLITE":
|
267 |
with _get_sqlite_connection() as conn: conn.execute("DELETE FROM rules WHERE rule_text = ?", (rule_text_to_delete,)); conn.commit()
|
268 |
+
elif STORAGE_BACKEND == "HF_DATASET" and _rules_items_list:
|
269 |
Dataset.from_dict({"rule_text": list(_rules_items_list)}).push_to_hub(HF_RULES_DATASET_REPO, token=HF_TOKEN, private=True)
|
270 |
return True
|
271 |
except Exception as e:
|