From 263e82f36a2101d6863636969857c70710a33168 Mon Sep 17 00:00:00 2001
From: phernandez <phernandez@basicmachines.co>
Date: Sat, 14 Dec 2024 18:40:49 -0600
Subject: [PATCH] add all tool endpoints to fastapi

---
 src/basic_memory/api/routers/knowledge.py | 54 ++++++++++++++++++++---
 1 file changed, 49 insertions(+), 5 deletions(-)

diff --git a/src/basic_memory/api/routers/knowledge.py b/src/basic_memory/api/routers/knowledge.py
index cac58fc8..e81ff45b 100644
--- a/src/basic_memory/api/routers/knowledge.py
+++ b/src/basic_memory/api/routers/knowledge.py
@@ -1,6 +1,5 @@
 """Router for knowledge graph operations."""
-from typing import Annotated
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter
 
 
 from basic_memory.deps import MemoryServiceDep
@@ -8,11 +7,15 @@ from basic_memory.schemas import (
     CreateEntitiesInput, CreateEntitiesResponse,
     SearchNodesInput, SearchNodesResponse,
     CreateRelationsInput, CreateRelationsResponse,
-    EntityOut, RelationOut, ObservationsIn, ObservationsOut, ObservationOut
+    EntityOut, RelationOut, ObservationsIn, ObservationsOut, ObservationOut,
+    OpenNodesInput, OpenNodesResponse,
+    DeleteEntitiesResponse,
+    DeleteObservationsInput, DeleteObservationsResponse
 )
 
 router = APIRouter(prefix="/knowledge", tags=["knowledge"])
 
+
 @router.post("/entities", response_model=CreateEntitiesResponse)
 async def create_entities(
     data: CreateEntitiesInput,
@@ -33,6 +36,26 @@ async def get_entity(
     return EntityOut.model_validate(entity)
 
 
+@router.delete("/entities/{entity_id}", response_model=DeleteEntitiesResponse)
+async def delete_entity(
+    entity_id: str,
+    memory_service: MemoryServiceDep
+) -> DeleteEntitiesResponse:
+    """Delete a specific entity by ID."""
+    deleted = await memory_service.delete_entities([entity_id])
+    return DeleteEntitiesResponse(deleted=deleted)  # pyright: ignore [reportArgumentType]
+
+
+@router.post("/nodes", response_model=OpenNodesResponse)
+async def open_nodes(
+    data: OpenNodesInput,
+    memory_service: MemoryServiceDep
+) -> OpenNodesResponse:
+    """Open specific nodes by their names."""
+    entities = await memory_service.open_nodes(data.names)
+    return OpenNodesResponse(entities=[EntityOut.model_validate(entity) for entity in entities])
+
+
 @router.post("/relations", response_model=CreateRelationsResponse)
 async def create_relations(
     data: CreateRelationsInput,
@@ -43,6 +66,16 @@ async def create_relations(
     return CreateRelationsResponse(relations=[RelationOut.model_validate(relation) for relation in relations])
 
 
+@router.delete("/relations/{relation_id}", response_model=DeleteEntitiesResponse)
+async def delete_relation(
+    relation_id: int,
+    memory_service: MemoryServiceDep
+) -> DeleteEntitiesResponse:
+    """Delete a specific relation by ID."""
+    # TODO: Implement delete_relation in memory service
+    raise NotImplementedError("Delete relation not implemented yet")
+
+
 @router.post("/observations", response_model=ObservationsOut)
 async def add_observations(
     data: ObservationsIn,
@@ -50,7 +83,18 @@ async def add_observations(
 ) -> ObservationsOut:
     """Add observations to an entity."""
     observations = await memory_service.add_observations(data)
-    return ObservationsOut(entity_id=data.entity_id, observations=[ObservationOut.model_validate(observation) for observation in observations])  # pyright: ignore [reportCallIssue]
+    return ObservationsOut(entity_id=data.entity_id, observations=[ObservationOut.model_validate(observation) for observation in observations])
+
+
+@router.delete("/observations", response_model=DeleteObservationsResponse)
+async def delete_observations(
+    data: DeleteObservationsInput,
+    memory_service: MemoryServiceDep
+) -> DeleteObservationsResponse:
+    """Delete observations from an entity."""
+    # TODO: Implement delete_observations in memory service
+    raise NotImplementedError("Delete observations not implemented yet")
+
 
 @router.post("/search", response_model=SearchNodesResponse)
 async def search_nodes(
@@ -59,4 +103,4 @@ async def search_nodes(
 ) -> SearchNodesResponse:
     """Search for entities in the knowledge graph."""
     matches = await memory_service.search_nodes(data.query)
-    return SearchNodesResponse(matches=[EntityOut.model_validate(entity) for entity in matches], query=data.query)
+    return SearchNodesResponse(matches=[EntityOut.model_validate(entity) for entity in matches], query=data.query)
\ No newline at end of file