From 995dd44111b880fe3a83e656707bf8d1cd7a5652 Mon Sep 17 00:00:00 2001 From: harvey_xiang Date: Fri, 6 Feb 2026 17:35:20 +0800 Subject: [PATCH] fix: format error --- src/memos/api/middleware/__init__.py | 9 +++++---- src/memos/api/utils/api_keys.py | 2 +- src/memos/graph_dbs/neo4j_community.py | 2 +- src/memos/mem_feedback/feedback.py | 2 +- 4 files changed, 8 insertions(+), 7 deletions(-) diff --git a/src/memos/api/middleware/__init__.py b/src/memos/api/middleware/__init__.py index 64cbc5c60..fd39252f5 100644 --- a/src/memos/api/middleware/__init__.py +++ b/src/memos/api/middleware/__init__.py @@ -1,13 +1,14 @@ """Krolik middleware extensions for MemOS.""" -from .auth import verify_api_key, require_scope, require_admin, require_read, require_write +from .auth import require_admin, require_read, require_scope, require_write, verify_api_key from .rate_limit import RateLimitMiddleware + __all__ = [ - "verify_api_key", - "require_scope", + "RateLimitMiddleware", "require_admin", "require_read", + "require_scope", "require_write", - "RateLimitMiddleware", + "verify_api_key", ] diff --git a/src/memos/api/utils/api_keys.py b/src/memos/api/utils/api_keys.py index 559ddd355..29b493fd0 100644 --- a/src/memos/api/utils/api_keys.py +++ b/src/memos/api/utils/api_keys.py @@ -5,8 +5,8 @@ """ import hashlib -import os import secrets + from dataclasses import dataclass from datetime import datetime, timedelta diff --git a/src/memos/graph_dbs/neo4j_community.py b/src/memos/graph_dbs/neo4j_community.py index e34313fa2..da7297f31 100644 --- a/src/memos/graph_dbs/neo4j_community.py +++ b/src/memos/graph_dbs/neo4j_community.py @@ -1102,7 +1102,7 @@ def _parse_nodes(self, nodes_data: list[dict[str, Any]]) -> list[dict[str, Any]] # Merge embeddings into parsed nodes for parsed_node in parsed_nodes: node_id = parsed_node["id"] - parsed_node["metadata"]["embedding"] = vec_items_map.get(node_id, None) + parsed_node["metadata"]["embedding"] = vec_items_map.get(node_id) return parsed_nodes diff --git a/src/memos/mem_feedback/feedback.py b/src/memos/mem_feedback/feedback.py index cf67a9bc3..604bfc9bf 100644 --- a/src/memos/mem_feedback/feedback.py +++ b/src/memos/mem_feedback/feedback.py @@ -756,7 +756,7 @@ def filter_fault_update(self, operations: list[dict]): for judge in all_judge: valid_update = None if judge["judgement"] == "UPDATE_APPROVED": - valid_update = id2op.get(judge["id"], None) + valid_update = id2op.get(judge["id"]) if valid_update: valid_updates.append(valid_update)