add db check at startup

This commit is contained in:
phernandez
2025-02-07 16:03:31 -06:00
parent e2882dbf84
commit ffbd091ad3
13 changed files with 360 additions and 269 deletions
+7 -9
View File
@@ -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",
]
-11
View File
@@ -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}')"