broadfield-dev commited on
Commit
3b79be2
·
verified ·
1 Parent(s): 14149fd

Update memory_logic.py

Browse files
Files changed (1) hide show
  1. 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
- _faiss_rules_index, _rules_items_list = _build_faiss_index(new_list, lambda r: r)
 
 
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: