4ec6649caf
* feat: byok provider models in db also * make tests and sync api * fix inconsistent state with recreating provider of same name * fix sync on byok creation * update revision * move stripe code for testing purposes * revert * add refresh byok models endpoint * just stage publish api * add tests * reorder revision * add test for name clashes
53 lines
2.5 KiB
Python
53 lines
2.5 KiB
Python
from datetime import datetime
|
|
from typing import TYPE_CHECKING, Optional
|
|
|
|
from sqlalchemy import DateTime, ForeignKey, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from letta.orm.mixins import OrganizationMixin
|
|
from letta.orm.sqlalchemy_base import SqlalchemyBase
|
|
from letta.schemas.providers import Provider as PydanticProvider
|
|
|
|
if TYPE_CHECKING:
|
|
from letta.orm.organization import Organization
|
|
from letta.orm.provider_model import ProviderModel
|
|
|
|
|
|
class Provider(SqlalchemyBase, OrganizationMixin):
|
|
"""Provider ORM class"""
|
|
|
|
__tablename__ = "providers"
|
|
__pydantic_model__ = PydanticProvider
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"name",
|
|
"organization_id",
|
|
name="unique_name_organization_id",
|
|
),
|
|
)
|
|
|
|
# Override organization_id to make it nullable for base providers
|
|
organization_id: Mapped[Optional[str]] = mapped_column(String, ForeignKey("organizations.id"), nullable=True)
|
|
|
|
name: Mapped[str] = mapped_column(nullable=False, doc="The name of the provider")
|
|
provider_type: Mapped[str] = mapped_column(nullable=True, doc="The type of the provider")
|
|
provider_category: Mapped[str] = mapped_column(nullable=True, doc="The category of the provider (base or byok)")
|
|
api_key: Mapped[str] = mapped_column(nullable=True, doc="API key or secret key used for requests to the provider.")
|
|
base_url: Mapped[str] = mapped_column(nullable=True, doc="Base URL for the provider.")
|
|
access_key: Mapped[str] = mapped_column(nullable=True, doc="Access key used for requests to the provider.")
|
|
region: Mapped[str] = mapped_column(nullable=True, doc="Region used for requests to the provider.")
|
|
api_version: Mapped[str] = mapped_column(nullable=True, doc="API version used for requests to the provider.")
|
|
|
|
# encrypted columns
|
|
api_key_enc: Mapped[Optional[str]] = mapped_column(Text, nullable=True, doc="Encrypted API key or secret key for the provider.")
|
|
access_key_enc: Mapped[Optional[str]] = mapped_column(Text, nullable=True, doc="Encrypted access key for the provider.")
|
|
|
|
# sync tracking
|
|
last_synced: Mapped[Optional[datetime]] = mapped_column(
|
|
DateTime(timezone=True), nullable=True, doc="Last time models were synced for this provider."
|
|
)
|
|
|
|
# relationships
|
|
organization: Mapped["Organization"] = relationship("Organization", back_populates="providers")
|
|
models: Mapped[list["ProviderModel"]] = relationship("ProviderModel", back_populates="provider", cascade="all, delete-orphan")
|