feat: Aggiungi gestione della memoria per Queen of Hearts e verifica …

…JSON valido

lib/activation.py

@@ -2,6 +2,7 @@
 from cat.looking_glass.cheshire_cat import CheshireCat
 from cat.db.crud import get_setting_by_name,upsert_setting_by_name
 from cat.db.models import Setting
+from .const import QUEEN_OF_HEARTS_INJESTION_SETTING
 
 ccat = CheshireCat()
 
@@ -91,7 +92,7 @@ def update_injested_ids(key, value):
 
 def activation(plugin):
 
-    setting_key=plugin.id + "_injested"
+    setting_key=QUEEN_OF_HEARTS_INJESTION_SETTING
 
     injested_ids = get_injested_ids(setting_key)
 

lib/const.py

@@ -0,0 +1 @@
+QUEEN_OF_HEARTS_INJESTION_SETTING = 'queen_of_hearts_injested'

lib/save_memory.py

@@ -2,17 +2,22 @@
 from .memory import Memory
 import json
 
+def is_valid_json(json_string):
+    try:
+        json.loads(json_string)
+        return True
+    except (ValueError, TypeError):
+        return False
 
 def save_memory(
     message,
     cat
 ):
-    log.info(message.content)
-    log.info(message.why.input)    
-    parsed_message = json.loads(message.content)
+    if (is_valid_json(message.content)):
+        parsed_message = json.loads(message.content)
 
-    value = parsed_message['value']
-    log.info(parsed_message)
-    memory = Memory(cat, value, message.why.input, parsed_message["why"])
-
-    memory.save()
+        value = parsed_message['value']
+        log.info(parsed_message)
+        memory = Memory(cat, value, message.why.input, parsed_message["why"])
+        
+        memory.save()

queen_of_hearts.py

@@ -1,5 +1,9 @@
 from cat.mad_hatter.decorators import hook
 from cat.mad_hatter.decorators import plugin
+from cat.mad_hatter.decorators import tool
+from cat.db.crud import get_setting_by_name,delete_setting_by_id
+
+from .lib.const import QUEEN_OF_HEARTS_INJESTION_SETTING
 from .lib.activation import activation
 from .lib.prefix import prompt_prefix
 from .lib.save_memory import save_memory
@@ -17,4 +21,18 @@ def activated(plugin):
 @hook
 def before_cat_sends_message(message, cat):
     save_memory( message,cat)
-    return message  
+    return message  
+
+
+@hook(priority=0)
+def agent_fast_reply(fast_reply, cat):
+    if (cat.working_memory.user_message_json.text == "/empty_qoh"):
+
+        setting = get_setting_by_name(QUEEN_OF_HEARTS_INJESTION_SETTING)
+        if setting is not None:
+            delete_setting_by_id(setting["setting_id"])
+            return {
+                "output": "Ok, la memoria di Queen of Hearts è stata svuotata."
+            }
+
+        return {"output":"Non c'è nulla da svuotare."}