mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
add db check at startup
This commit is contained in:
@@ -1,17 +1,15 @@
|
||||
"""Models package for basic-memory."""
|
||||
|
||||
import basic_memory
|
||||
from basic_memory.models.base import Base, SchemaVersion
|
||||
from basic_memory.models.base import Base
|
||||
from basic_memory.models.knowledge import Entity, Observation, Relation, ObservationCategory
|
||||
|
||||
SCHEMA_VERSION = basic_memory.__version__ + "-" + "003"
|
||||
|
||||
__all__ = [
|
||||
'Base',
|
||||
'Entity',
|
||||
'Observation',
|
||||
'ObservationCategory',
|
||||
'Relation',
|
||||
'SchemaVersion'
|
||||
"Base",
|
||||
"Entity",
|
||||
"Observation",
|
||||
"ObservationCategory",
|
||||
"Relation",
|
||||
]
|
||||
|
||||
|
||||
|
||||
@@ -8,14 +8,3 @@ class Base(AsyncAttrs, DeclarativeBase):
|
||||
"""Base class for all models"""
|
||||
pass
|
||||
|
||||
|
||||
class SchemaVersion(Base):
|
||||
"""Track database schema version."""
|
||||
|
||||
__tablename__ = "schema_version"
|
||||
|
||||
# Only one row will exist
|
||||
version: Mapped[str] = mapped_column(String, primary_key=True)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"SchemaVersion(version='{self.version}')"
|
||||
Reference in New Issue
Block a user