add: add json parse for en

Author: fridayL

src/memos/mem_os/product.py

@@ -16,6 +16,7 @@
 from memos.mem_cube.general import GeneralMemCube
 from memos.mem_os.core import MOSCore
 from memos.mem_os.utils.format_utils import (
+    clean_json_response,
     convert_graph_to_tree_forworkmem,
     ensure_unique_tree_ids,
     filter_nodes_by_tree_ids,
@@ -657,7 +658,7 @@ def get_suggestion_query(self, user_id: str, language: str = "zh") -> list[str]:
             you should generate some suggestion query, the query should be user what to query,
             user recently memories is:
             {memories}
-            please generate 3 suggestion query in English,
+            if the user recently memories is empty, please generate 3 suggestion query in English,
             output should be a json format, the key is "query", the value is a list of suggestion query.
 
             example:
@@ -674,8 +675,8 @@ def get_suggestion_query(self, user_id: str, language: str = "zh") -> list[str]:
             memories = ""
         message_list = [{"role": "system", "content": suggestion_prompt.format(memories=memories)}]
         response = self.chat_llm.generate(message_list)
-        response_json = json.loads(response)
-
+        clean_response = clean_json_response(response)
+        response_json = json.loads(clean_response)
         return response_json["query"]
 
     def chat(

src/memos/mem_os/utils/format_utils.py

@@ -1340,3 +1340,16 @@ def ensure_unique_tree_ids(tree_result: dict[str, Any]) -> dict[str, Any]:
             print(f"   - {error}")
 
     return fixed_tree
+
+
+def clean_json_response(response: str) -> str:
+    """
+    Remove markdown JSON code block formatting from LLM response.
+
+    Args:
+        response: Raw response string that may contain ```json and ```
+
+    Returns:
+        str: Clean JSON string without markdown formatting
+    """
+    return response.replace("```json", "").replace("```", "").strip()