Skip to content

Commit 2b0e1f8

Browse files
authored
fix: allow get_user_names_by_memory_ids supports all types of db (#1063)
1 parent fc74b13 commit 2b0e1f8

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

src/memos/api/routers/server_router.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@
4747
SuggestionResponse,
4848
TaskQueueResponse,
4949
)
50-
from memos.graph_dbs.polardb import PolarDBGraphDB
5150
from memos.log import get_logger
5251
from memos.mem_scheduler.base_scheduler import BaseScheduler
5352
from memos.mem_scheduler.utils.status_tracker import TaskStatusTracker
@@ -364,15 +363,6 @@ def feedback_memories(feedback_req: APIFeedbackRequest):
364363
)
365364
def get_user_names_by_memory_ids(request: GetUserNamesByMemoryIdsRequest):
366365
"""Get user names by memory ids."""
367-
if not isinstance(graph_db, PolarDBGraphDB):
368-
raise HTTPException(
369-
status_code=400,
370-
detail=(
371-
"graph_db must be an instance of PolarDBGraphDB to use "
372-
"get_user_names_by_memory_ids"
373-
f"current graph_db is: {graph_db.__class__.__name__}"
374-
),
375-
)
376366
result = graph_db.get_user_names_by_memory_ids(memory_ids=request.memory_ids)
377367
if vector_db:
378368
prefs = []

0 commit comments

Comments
 (0)