Skip to content

Commit 1bf3e2f

Browse files
committed
fix: code format
1 parent 99419fe commit 1bf3e2f

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

src/memos/graph_dbs/polardb.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1742,7 +1742,9 @@ def get_grouped_counts(
17421742
return_fields.append(
17431743
f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype)::text AS {alias}"
17441744
)
1745-
group_by_fields.append(f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype)::text")
1745+
group_by_fields.append(
1746+
f"ag_catalog.agtype_access_operator(properties, '\"{field}\"'::agtype)::text"
1747+
)
17461748

17471749
# Full SQL query construction
17481750
query = f"""

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,9 @@ def _refresh_memory_size(self, user_name: str | None = None) -> None:
156156
results = self.graph_store.get_grouped_counts(
157157
group_fields=["memory_type"], user_name=user_name
158158
)
159-
self.current_memory_size = {record["memory_type"]: int(record["count"]) for record in results}
159+
self.current_memory_size = {
160+
record["memory_type"]: int(record["count"]) for record in results
161+
}
160162
logger.info(f"[MemoryManager] Refreshed memory sizes: {self.current_memory_size}")
161163

162164
def _process_memory(self, memory: TextualMemoryItem, user_name: str | None = None):

0 commit comments

Comments
 (0)