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

add get_observation_categories

Browse Source
This commit is contained in:
phernandez
2024-12-30 13:26:44 -06:00
parent 1dfcc0bc4b
commit 2bc35f34e3
7 changed files with 135 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/basic_memory/api/routers/discovery_router.py
+9 -1
View File
@@ -3,7 +3,7 @@
from fastapi import APIRouter
from loguru import logger
from basic_memory.deps import EntityServiceDep
from basic_memory.deps import EntityServiceDep, ObservationServiceDep
from basic_memory.schemas import EntityTypeList, ObservationCategoryList
router = APIRouter(prefix="/discovery", tags=["discovery"])
@@ -15,3 +15,11 @@ async def get_entity_types(entity_service: EntityServiceDep) -> EntityTypeList:
logger.debug("Getting all entity types")
types = await entity_service.get_entity_types()
return EntityTypeList(types=types)
@router.get("/observation-categories", response_model=ObservationCategoryList)
async def get_observation_categories(observation_service: ObservationServiceDep) -> ObservationCategoryList:
"""Get list of all unique observation categories in the system."""
logger.debug("Getting all observation categories")
categories = await observation_service.observation_categories()
return ObservationCategoryList(categories=categories)