mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
83 lines
1.8 KiB
Python
83 lines
1.8 KiB
Python
"""Knowledge graph schema exports.
|
|
|
|
This module exports all schema classes to simplify imports.
|
|
Rather than importing from individual schema files, you can
|
|
import everything from basic_memory.schemas.
|
|
"""
|
|
|
|
# Base types and models
|
|
from basic_memory.schemas.base import (
|
|
Observation,
|
|
EntityType,
|
|
RelationType,
|
|
Relation,
|
|
Entity,
|
|
)
|
|
|
|
# Delete operation models
|
|
from basic_memory.schemas.delete import (
|
|
DeleteEntityRequest,
|
|
DeleteRelationsRequest,
|
|
DeleteObservationsRequest,
|
|
)
|
|
|
|
# Request models
|
|
from basic_memory.schemas.request import (
|
|
AddObservationsRequest,
|
|
CreateEntityRequest,
|
|
SearchNodesRequest,
|
|
OpenNodesRequest,
|
|
CreateRelationsRequest,
|
|
)
|
|
|
|
# Response models
|
|
from basic_memory.schemas.response import (
|
|
SQLAlchemyModel,
|
|
ObservationResponse,
|
|
ObservationsResponse,
|
|
RelationResponse,
|
|
EntityResponse,
|
|
CreateEntityResponse,
|
|
SearchNodesResponse,
|
|
OpenNodesResponse,
|
|
AddObservationsResponse,
|
|
CreateRelationsResponse,
|
|
DeleteEntityResponse,
|
|
DeleteRelationsResponse,
|
|
DeleteObservationsResponse,
|
|
)
|
|
|
|
# For convenient imports, export all models
|
|
__all__ = [
|
|
# Base
|
|
"Observation",
|
|
"EntityType",
|
|
"RelationType",
|
|
"Relation",
|
|
"Entity",
|
|
# Requests
|
|
"AddObservationsRequest",
|
|
"CreateEntityRequest",
|
|
"SearchNodesRequest",
|
|
"OpenNodesRequest",
|
|
"CreateRelationsRequest",
|
|
# Responses
|
|
"SQLAlchemyModel",
|
|
"ObservationResponse",
|
|
"ObservationsResponse",
|
|
"RelationResponse",
|
|
"EntityResponse",
|
|
"CreateEntityResponse",
|
|
"SearchNodesResponse",
|
|
"OpenNodesResponse",
|
|
"AddObservationsResponse",
|
|
"CreateRelationsResponse",
|
|
"DeleteEntityResponse",
|
|
"DeleteRelationsResponse",
|
|
"DeleteObservationsResponse",
|
|
# Delete Operations
|
|
"DeleteEntityRequest",
|
|
"DeleteRelationsRequest",
|
|
"DeleteObservationsRequest",
|
|
]
|