diff --git a/memos/indexing.py b/memos/indexing.py index 1a0b805..beed369 100644 --- a/memos/indexing.py +++ b/memos/indexing.py @@ -174,6 +174,8 @@ def search_entities( folder_id: int = None, limit: int = 48, offset: int = 0, + start: int = None, + end: int = None, ) -> List[EntitySearchResult]: try: filter_by = [] @@ -181,6 +183,8 @@ def search_entities( filter_by.append(f"library_id:={library_id}") if folder_id is not None: filter_by.append(f"folder_id:={folder_id}") + if start is not None and end is not None: + filter_by.append(f"file_created_at:={start}..{end}") filter_by_str = " && ".join(filter_by) if filter_by else "" search_parameters = { diff --git a/memos/server.py b/memos/server.py index 54e0edd..359d14b 100644 --- a/memos/server.py +++ b/memos/server.py @@ -450,10 +450,14 @@ async def search_entities( folder_id: int = None, limit: Annotated[int, Query(ge=1, le=200)] = 48, offset: int = 0, + start: int = None, + end: int = None, db: Session = Depends(get_db), ): try: - return indexing.search_entities(client, q, library_id, folder_id, limit, offset) + return indexing.search_entities( + client, q, library_id, folder_id, limit, offset, start, end + ) except Exception as e: print(f"Error searching entities: {e}") raise HTTPException(