unique constraint here doesn't work (#4271)

Co-authored-by: Richard Kuo (Danswer) <rkuo@onyx.app>
This commit is contained in:
rkuo-danswer 2025-03-12 16:25:27 -07:00 committed by GitHub
parent f87e559cc4
commit a2ac9f02fb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 12 deletions

View File

@ -29,17 +29,8 @@ def upgrade() -> None:
op.add_column( op.add_column(
"llm_provider", sa.Column("default_vision_model", sa.String(), nullable=True) "llm_provider", sa.Column("default_vision_model", sa.String(), nullable=True)
) )
# Add unique constraint for is_default_vision_provider
op.create_unique_constraint(
"uq_llm_provider_is_default_vision_provider",
"llm_provider",
["is_default_vision_provider"],
)
def downgrade() -> None: def downgrade() -> None:
op.drop_constraint(
"uq_llm_provider_is_default_vision_provider", "llm_provider", type_="unique"
)
op.drop_column("llm_provider", "default_vision_model") op.drop_column("llm_provider", "default_vision_model")
op.drop_column("llm_provider", "is_default_vision_provider") op.drop_column("llm_provider", "is_default_vision_provider")

View File

@ -1489,9 +1489,7 @@ class LLMProvider(Base):
# should only be set for a single provider # should only be set for a single provider
is_default_provider: Mapped[bool | None] = mapped_column(Boolean, unique=True) is_default_provider: Mapped[bool | None] = mapped_column(Boolean, unique=True)
is_default_vision_provider: Mapped[bool | None] = mapped_column( is_default_vision_provider: Mapped[bool | None] = mapped_column(Boolean)
Boolean, unique=True
)
default_vision_model: Mapped[str | None] = mapped_column(String, nullable=True) default_vision_model: Mapped[str | None] = mapped_column(String, nullable=True)
# EE only # EE only
is_public: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True) is_public: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)