Fix: fix pg query for group error string (#465)

* fix:get_grouped_counts

* fix: code format

Author: fridayL

src/memos/api/config.py

@@ -861,9 +861,9 @@ def create_user_config(user_name: str, user_id: str) -> tuple[MOSConfig, General
                             "reorganize": os.getenv("MOS_ENABLE_REORGANIZE", "false").lower()
                             == "true",
                             "memory_size": {
-                                "WorkingMemory": os.getenv("NEBULAR_WORKING_MEMORY", 20),
-                                "LongTermMemory": os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6),
-                                "UserMemory": os.getenv("NEBULAR_USER_MEMORY", 1e6),
+                                "WorkingMemory": int(os.getenv("NEBULAR_WORKING_MEMORY", 20)),
+                                "LongTermMemory": int(os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6)),
+                                "UserMemory": int(os.getenv("NEBULAR_USER_MEMORY", 1e6)),
                             },
                             "search_strategy": {
                                 "fast_graph": bool(os.getenv("FAST_GRAPH", "false") == "true"),
@@ -933,9 +933,9 @@ def get_default_cube_config() -> GeneralMemCubeConfig | None:
                             == "true",
                             "internet_retriever": internet_config,
                             "memory_size": {
-                                "WorkingMemory": os.getenv("NEBULAR_WORKING_MEMORY", 20),
-                                "LongTermMemory": os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6),
-                                "UserMemory": os.getenv("NEBULAR_USER_MEMORY", 1e6),
+                                "WorkingMemory": int(os.getenv("NEBULAR_WORKING_MEMORY", 20)),
+                                "LongTermMemory": int(os.getenv("NEBULAR_LONGTERM_MEMORY", 1e6)),
+                                "UserMemory": int(os.getenv("NEBULAR_USER_MEMORY", 1e6)),
                             },
                             "search_strategy": {
                                 "fast_graph": bool(os.getenv("FAST_GRAPH", "false") == "true"),

src/memos/graph_dbs/polardb.py

@@ -1740,9 +1740,11 @@ def get_grouped_counts(
         for field in group_fields:
             alias = field.replace(".", "_")
             return_fields.append(
-                f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype) AS {alias}"
+                f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype)::text AS {alias}"
+            )
+            group_by_fields.append(
+                f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype)::text"
             )
-            group_by_fields.append(alias)
 
         # Full SQL query construction
         query = f"""
@@ -1751,7 +1753,6 @@ def get_grouped_counts(
             {where_clause}
             GROUP BY {", ".join(group_by_fields)}
         """
-
         conn = self._get_connection()
         try:
             with conn.cursor() as cursor:
@@ -1772,7 +1773,7 @@ def get_grouped_counts(
                         else:
                             group_values[field] = str(value)
                     count_value = row[-1]  # Last column is count
-                    output.append({**group_values, "count": count_value})
+                    output.append({**group_values, "count": int(count_value)})
 
                 return output
 

src/memos/memories/textual/tree_text_memory/organize/manager.py

@@ -156,7 +156,9 @@ def _refresh_memory_size(self, user_name: str | None = None) -> None:
         results = self.graph_store.get_grouped_counts(
             group_fields=["memory_type"], user_name=user_name
         )
-        self.current_memory_size = {record["memory_type"]: record["count"] for record in results}
+        self.current_memory_size = {
+            record["memory_type"]: int(record["count"]) for record in results
+        }
         logger.info(f"[MemoryManager] Refreshed memory sizes: {self.current_memory_size}")
 
     def _process_memory(self, memory: TextualMemoryItem, user_name: str | None = None):