admin/basicmachines-co-basic-memory
1
0
Fork 0
mirror of https://github.com/basicmachines-co/basic-memory synced 2026-06-21 13:47:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(mcp): use lightweight graph hydration lookup (#806)

Browse Source 
Signed-off-by: phernandez <paul@basicmachines.co>
This commit is contained in:
Paul Hernandez
2026-05-08 10:07:29 -05:00
committed by GitHub
parent 7918e5c6bf
commit 26381aeed1
4 changed files with 122 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/basic_memory/api/v2/utils.py
+5 -3
View File
@@ -18,7 +18,7 @@ from basic_memory.services.context_service import (
class EntityBatchLookup(Protocol):
async def find_by_ids(self, ids: List[int]) -> Sequence[Any]: ...
async def find_by_ids_for_hydration(self, ids: List[int]) -> Sequence[Any]: ...
class EntityServiceBatchLookup(Protocol):
@@ -76,7 +76,7 @@ async def to_graph_context(
if item.to_id:
entity_ids_needed.add(item.to_id)
# Batch fetch all entities at once - get both title and external_id
# Batch fetch just the entity fields needed to shape the response.
entity_title_lookup: dict[int, str] = {}
entity_external_id_lookup: dict[int, str] = {}
if entity_ids_needed:
@@ -87,7 +87,9 @@ async def to_graph_context(
phase="lookup_entities",
result_count=len(entity_ids_needed),
):
entities = await entity_repository.find_by_ids(list(entity_ids_needed))
entities = await entity_repository.find_by_ids_for_hydration(
list(entity_ids_needed)
)
for e in entities:
entity_title_lookup[e.id] = e.title
entity_external_id_lookup[e.id] = e.external_id