mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
5b4f0eafcc
* configure logging * set mcp output logging also * fix type check errors * fix type check * rename Permalink schema type * fix type errors * add typechecks to ci workflow * pytest coverage setup * add tests for status cli * sync tests coverage * watch_service test coverage * tests for tool_utils.py * clean up imports * file_utils coverage * markdown plugins coverage * 99% test coverage * more test coverage, remove ObservationCategory * more tool coverage * fix type-check * format, upgrade deps --------- Co-authored-by: phernandez <phernandez@basicmachines.co>
29 lines
876 B
Python
29 lines
876 B
Python
"""Schemas for knowledge discovery and analytics endpoints."""
|
|
|
|
from typing import List, Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
from basic_memory.schemas.response import EntityResponse
|
|
|
|
|
|
class EntityTypeList(BaseModel):
|
|
"""List of unique entity types in the system."""
|
|
|
|
types: List[str]
|
|
|
|
|
|
class ObservationCategoryList(BaseModel):
|
|
"""List of unique observation categories in the system."""
|
|
|
|
categories: List[str]
|
|
|
|
|
|
class TypedEntityList(BaseModel):
|
|
"""List of entities of a specific type."""
|
|
|
|
entity_type: str = Field(..., description="Type of entities in the list")
|
|
entities: List[EntityResponse]
|
|
total: int = Field(..., description="Total number of entities")
|
|
sort_by: Optional[str] = Field(None, description="Field used for sorting")
|
|
include_related: bool = Field(False, description="Whether related entities are included")
|