From 718ba9cb4fb6bf30a5f8f94a4a7bf2ea09ff046a Mon Sep 17 00:00:00 2001 From: "Richard Kuo (Danswer)" Date: Wed, 12 Mar 2025 16:24:51 -0700 Subject: [PATCH] unique constraint here doesn't work --- .../df46c75b714e_add_default_vision_provider_to_llm_.py | 9 --------- backend/onyx/db/models.py | 4 +--- 2 files changed, 1 insertion(+), 12 deletions(-) diff --git a/backend/alembic/versions/df46c75b714e_add_default_vision_provider_to_llm_.py b/backend/alembic/versions/df46c75b714e_add_default_vision_provider_to_llm_.py index 15f0760ed..026982d08 100644 --- a/backend/alembic/versions/df46c75b714e_add_default_vision_provider_to_llm_.py +++ b/backend/alembic/versions/df46c75b714e_add_default_vision_provider_to_llm_.py @@ -29,17 +29,8 @@ def upgrade() -> None: op.add_column( "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: - 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", "is_default_vision_provider") diff --git a/backend/onyx/db/models.py b/backend/onyx/db/models.py index 32aa3f189..c0865e523 100644 --- a/backend/onyx/db/models.py +++ b/backend/onyx/db/models.py @@ -1489,9 +1489,7 @@ class LLMProvider(Base): # should only be set for a single provider is_default_provider: Mapped[bool | None] = mapped_column(Boolean, unique=True) - is_default_vision_provider: Mapped[bool | None] = mapped_column( - Boolean, unique=True - ) + is_default_vision_provider: Mapped[bool | None] = mapped_column(Boolean) default_vision_model: Mapped[str | None] = mapped_column(String, nullable=True) # EE only is_public: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)