mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
0cb31a2f72
Co-authored-by: Shubham Naik <shub@memgpt.ai> Co-authored-by: Matt Zhou <mattzh1314@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Caren Thomas <caren@letta.com> Co-authored-by: cpacker <packercharles@gmail.com>
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy import ForeignKey, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from letta.orm.base import Base
|
|
|
|
|
|
def is_valid_uuid4(uuid_string: str) -> bool:
|
|
"""Check if a string is a valid UUID4."""
|
|
try:
|
|
uuid_obj = UUID(uuid_string)
|
|
return uuid_obj.version == 4
|
|
except ValueError:
|
|
return False
|
|
|
|
|
|
class OrganizationMixin(Base):
|
|
"""Mixin for models that belong to an organization."""
|
|
|
|
__abstract__ = True
|
|
|
|
organization_id: Mapped[str] = mapped_column(String, ForeignKey("organizations.id"))
|
|
|
|
|
|
class UserMixin(Base):
|
|
"""Mixin for models that belong to a user."""
|
|
|
|
__abstract__ = True
|
|
|
|
user_id: Mapped[str] = mapped_column(String, ForeignKey("users.id"))
|
|
|
|
|
|
class AgentMixin(Base):
|
|
"""Mixin for models that belong to an agent."""
|
|
|
|
__abstract__ = True
|
|
|
|
agent_id: Mapped[str] = mapped_column(String, ForeignKey("agents.id", ondelete="CASCADE"))
|
|
|
|
|
|
class FileMixin(Base):
|
|
"""Mixin for models that belong to a file."""
|
|
|
|
__abstract__ = True
|
|
|
|
file_id: Mapped[Optional[str]] = mapped_column(String, ForeignKey("files.id", ondelete="CASCADE"))
|
|
|
|
|
|
class SourceMixin(Base):
|
|
"""Mixin for models (e.g. file) that belong to a source."""
|
|
|
|
__abstract__ = True
|
|
|
|
source_id: Mapped[str] = mapped_column(String, ForeignKey("sources.id", ondelete="CASCADE"), nullable=False)
|
|
|
|
|
|
class SandboxConfigMixin(Base):
|
|
"""Mixin for models that belong to a SandboxConfig."""
|
|
|
|
__abstract__ = True
|
|
|
|
sandbox_config_id: Mapped[str] = mapped_column(String, ForeignKey("sandbox_configs.id"))
|