43 lines
2.0 KiB
Python
43 lines
2.0 KiB
Python
from typing import TYPE_CHECKING, List, Optional
|
|
|
|
from sqlalchemy import JSON
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from letta.orm import FileMetadata
|
|
from letta.orm.custom_columns import EmbeddingConfigColumn
|
|
from letta.orm.mixins import OrganizationMixin
|
|
from letta.orm.sqlalchemy_base import SqlalchemyBase
|
|
from letta.schemas.embedding_config import EmbeddingConfig
|
|
from letta.schemas.source import Source as PydanticSource
|
|
|
|
if TYPE_CHECKING:
|
|
from letta.orm.agent import Agent
|
|
from letta.orm.file import FileMetadata
|
|
from letta.orm.organization import Organization
|
|
from letta.orm.passage import SourcePassage
|
|
|
|
|
|
class Source(SqlalchemyBase, OrganizationMixin):
|
|
"""A source represents an embedded text passage"""
|
|
|
|
__tablename__ = "sources"
|
|
__pydantic_model__ = PydanticSource
|
|
|
|
name: Mapped[str] = mapped_column(doc="the name of the source, must be unique within the org", nullable=False)
|
|
description: Mapped[str] = mapped_column(nullable=True, doc="a human-readable description of the source")
|
|
embedding_config: Mapped[EmbeddingConfig] = mapped_column(EmbeddingConfigColumn, doc="Configuration settings for embedding.")
|
|
metadata_: Mapped[Optional[dict]] = mapped_column(JSON, nullable=True, doc="metadata for the source.")
|
|
|
|
# relationships
|
|
organization: Mapped["Organization"] = relationship("Organization", back_populates="sources")
|
|
files: Mapped[List["FileMetadata"]] = relationship("FileMetadata", back_populates="source", cascade="all, delete-orphan")
|
|
passages: Mapped[List["SourcePassage"]] = relationship("SourcePassage", back_populates="source", cascade="all, delete-orphan")
|
|
agents: Mapped[List["Agent"]] = relationship(
|
|
"Agent",
|
|
secondary="sources_agents",
|
|
back_populates="sources",
|
|
lazy="selectin",
|
|
cascade="all, delete", # Ensures rows in sources_agents are deleted when the source is deleted
|
|
passive_deletes=True, # Allows the database to handle deletion of orphaned rows
|
|
)
|