fix: re-add feature to read multiple notes

Author: phernandez

src/basic_memory/api/routers/resource_router.py

@@ -21,7 +21,7 @@
 router = APIRouter(prefix="/resource", tags=["resources"])
 
 
-def get_entity_ids(item: SearchIndexRow) -> list[int]:
+def get_entity_ids(item: SearchIndexRow) -> list[int]: # pyright: ignore [reportReturnType]
     match item.type:
         case SearchItemType.ENTITY:
             return [item.id]
@@ -31,6 +31,8 @@ def get_entity_ids(item: SearchIndexRow) -> list[int]:
             from_entity = item.from_id
             to_entity = item.to_id
             return [from_entity, to_entity] if to_entity else [from_entity]
+        case _:
+            raise ValueError(f"Unexpected type: {item.type}")
 
 
 @router.get("/{identifier:path}")

src/basic_memory/cli/commands/import_chatgpt.py

@@ -69,7 +69,11 @@ def traverse_messages(
 
 
 def format_chat_markdown(
-    title: str, mapping: Dict[str, Any], root_id: Optional[str], created_at: float, modified_at: float
+    title: str,
+    mapping: Dict[str, Any],
+    root_id: Optional[str],
+    created_at: float,
+    modified_at: float,
 ) -> str:
     """Format chat as clean markdown."""
 

src/basic_memory/db.py

@@ -140,11 +140,15 @@ async def run_migrations(app_config: ProjectConfig, database_type=DatabaseType.F
 
         # Set required Alembic config options programmatically
         config.set_main_option("script_location", str(alembic_dir))
-        config.set_main_option("file_template",
-                               "%%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s")
+        config.set_main_option(
+            "file_template",
+            "%%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s",
+        )
         config.set_main_option("timezone", "UTC")
         config.set_main_option("revision_environment", "false")
-        config.set_main_option("sqlalchemy.url", DatabaseType.get_db_url(app_config.database_path, database_type))
+        config.set_main_option(
+            "sqlalchemy.url", DatabaseType.get_db_url(app_config.database_path, database_type)
+        )
 
         command.upgrade(config, "head")
         logger.info("Migrations completed successfully")
@@ -153,4 +157,4 @@ async def run_migrations(app_config: ProjectConfig, database_type=DatabaseType.F
         await SearchRepository(session_maker).init_search_index()
     except Exception as e:  # pragma: no cover
         logger.error(f"Error running migrations: {e}")
-        raise
+        raise

src/basic_memory/mcp/tools/utils.py

@@ -151,4 +151,4 @@ async def call_delete(
         return response
     except HTTPStatusError as e:
         logger.error(f"Error calling DELETE {url}: {e}")
-        raise ToolError(f"Error calling tool: {e}") from e
+        raise ToolError(f"Error calling tool: {e}") from e

src/basic_memory/models/search.py

@@ -31,4 +31,4 @@
     tokenize='unicode61 tokenchars 0x2F',  -- Hex code for /
     prefix='1,2,3,4'                    -- Support longer prefixes for paths
 );
-""")
+""")

src/basic_memory/repository/search_repository.py

@@ -73,35 +73,35 @@ async def init_search_index(self):
             async with db.scoped_session(self.session_maker) as session:
                 await session.execute(CREATE_SEARCH_INDEX)
                 await session.commit()
-        except Exception as e: # pragma: no cover
+        except Exception as e:  # pragma: no cover
             logger.error(f"Error initializing search index: {e}")
             raise e
-        
+
     def _prepare_search_term(self, term: str, is_prefix: bool = True) -> str:
         """Prepare a search term for FTS5 query.
-        
+
         Args:
             term: The search term to prepare
             is_prefix: Whether to add prefix search capability (* suffix)
-            
+
         For FTS5:
         - Special characters and phrases need to be quoted
         - Terms with spaces or special chars need quotes
         """
         if "*" in term:
             return term
-        
+
         # List of special characters that need quoting (excluding *)
         special_chars = ["/", "-", ".", " ", "(", ")", "[", "]", '"', "'"]
-                    
+
         # Check if term contains any special characters
         needs_quotes = any(c in term for c in special_chars)
-        
+
         if needs_quotes:
             # If the term already contains quotes, escape them and add a wildcard
             term = term.replace('"', '""')
             term = f'"{term}"*'
-                        
+
         return term
 
     async def search(
@@ -140,14 +140,15 @@ async def search(
         # Handle permalink match search, supports *
         if permalink_match:
             # Clean and prepare permalink for FTS5 GLOB match
-            permalink_text = self._prepare_search_term(permalink_match.lower().strip(), is_prefix=False)
+            permalink_text = self._prepare_search_term(
+                permalink_match.lower().strip(), is_prefix=False
+            )
             params["permalink"] = permalink_text
             if "*" in permalink_match:
                 conditions.append("permalink GLOB :permalink")
-            else: 
+            else:
                 conditions.append("permalink MATCH :permalink")
 
-
         # Handle type filter
         if types:
             type_list = ", ".join(f"'{t.value}'" for t in types)
@@ -223,7 +224,9 @@ async def search(
 
         logger.debug(f"Found {len(results)} search results")
         for r in results:
-            logger.debug(f"Search result: type:{r.type} title: {r.title} permalink: {r.permalink} score: {r.score}")
+            logger.debug(
+                f"Search result: type:{r.type} title: {r.title} permalink: {r.permalink} score: {r.score}"
+            )
 
         return results
 

src/basic_memory/schemas/response.py

@@ -10,6 +10,7 @@
 3. Responses include everything needed for next operations
 4. Bulk operations return all affected items
 """
+
 from datetime import datetime
 from typing import List, Optional, Dict
 
@@ -42,6 +43,7 @@ class ObservationResponse(Observation, SQLAlchemyModel):
         "context": "Initial database design meeting"
     }
     """
+
     permalink: Permalink
 
 
@@ -59,6 +61,7 @@ class RelationResponse(Relation, SQLAlchemyModel):
         "context": "Comprehensive test suite"
     }
     """
+
     permalink: Permalink
 
     from_id: Permalink = Field(

src/basic_memory/services/entity_service.py

@@ -184,12 +184,12 @@ async def get_by_permalink(self, permalink: str) -> EntityModel:
         if not db_entity:
             raise EntityNotFoundError(f"Entity not found: {permalink}")
         return db_entity
-    
+
     async def get_entities_by_id(self, ids: List[int]) -> Sequence[EntityModel]:
         """Get specific entities and their relationships."""
         logger.debug(f"Getting entities: {ids}")
         return await self.repository.find_by_ids(ids)
-    
+
     async def get_entities_by_permalinks(self, permalinks: List[str]) -> Sequence[EntityModel]:
         """Get specific nodes and their relationships."""
         logger.debug(f"Getting entities permalinks: {permalinks}")

src/basic_memory/services/link_resolver.py

@@ -46,7 +46,7 @@ async def resolve_link(self, link_text: str, use_search: bool = True) -> Optiona
             logger.debug(f"Found title match: {entity.title}")
             return entity
 
-        if use_search and '*' not in clean_text:
+        if use_search and "*" not in clean_text:
             # 3. Fall back to search for fuzzy matching on title
             results = await self.search_service.search(
                 query=SearchQuery(title=clean_text, types=[SearchItemType.ENTITY]),

tests/api/test_resource_router.py

@@ -165,7 +165,7 @@ async def test_get_resource_entities(client, test_config, entity_repository):
     assert response.status_code == 200
     assert response.headers["content-type"] == "text/markdown; charset=utf-8"
     assert (
-            f"""
+        f"""
 --- memory://test/test-entity {entity1.updated_at.isoformat()} {entity1.checksum[:8]}
 
 # Test Content
@@ -176,11 +176,10 @@ async def test_get_resource_entities(client, test_config, entity_repository):
 - links to [[Test Entity]]
 
     """.strip()
-            in response.text
+        in response.text
     )
 
 
-
 @pytest.mark.asyncio
 async def test_get_resource_relation(client, test_config, entity_repository):
     """Test getting content by relation permalink."""