From 888ec271473717e21c8bf97e66f036664b131f60 Mon Sep 17 00:00:00 2001 From: chentang Date: Mon, 9 Feb 2026 15:05:34 +0800 Subject: [PATCH] fix: allow get_user_names_by_memory_ids supports all types of db --- src/memos/api/routers/server_router.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/src/memos/api/routers/server_router.py b/src/memos/api/routers/server_router.py index 736c328ac..12e72c968 100644 --- a/src/memos/api/routers/server_router.py +++ b/src/memos/api/routers/server_router.py @@ -47,7 +47,6 @@ SuggestionResponse, TaskQueueResponse, ) -from memos.graph_dbs.polardb import PolarDBGraphDB from memos.log import get_logger from memos.mem_scheduler.base_scheduler import BaseScheduler from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker @@ -364,15 +363,6 @@ def feedback_memories(feedback_req: APIFeedbackRequest): ) def get_user_names_by_memory_ids(request: GetUserNamesByMemoryIdsRequest): """Get user names by memory ids.""" - if not isinstance(graph_db, PolarDBGraphDB): - raise HTTPException( - status_code=400, - detail=( - "graph_db must be an instance of PolarDBGraphDB to use " - "get_user_names_by_memory_ids" - f"current graph_db is: {graph_db.__class__.__name__}" - ), - ) result = graph_db.get_user_names_by_memory_ids(memory_ids=request.memory_ids) if vector_db: prefs = []