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

sanitize inputs, fix tests

Browse Source
This commit is contained in:
phernandez
2024-12-16 20:43:01 -06:00
parent 86329e0105
commit 6b684dcf9c
11 changed files with 137 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/basic_memory/api/routers/knowledge.py
+3 -4
View File
@@ -24,6 +24,7 @@ from basic_memory.schemas import (
AddObservationsResponse,
RelationResponse,
DeleteEntityRequest,
Entity,
)
router = APIRouter(prefix="/knowledge", tags=["knowledge"])
@@ -94,10 +95,8 @@ async def search_nodes(
@router.post("/nodes", response_model=OpenNodesResponse)
async def open_nodes(data: OpenNodesRequest, memory_service: MemoryServiceDep) -> OpenNodesResponse:
"""Open specific nodes by their names."""
entities = await memory_service.open_nodes(data.names)
return OpenNodesResponse(
entities=[EntityResponse.model_validate(entity) for entity in entities]
)
entities = await memory_service.open_nodes(data.entity_ids)
return OpenNodesResponse(entities=[Entity.model_validate(entity) for entity in entities])
## Delete endpoints