mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-06-23 14:30:57 +02:00
299cb5035c
* add litellm proxy * formatting * move `api_url` to cloud provider + nits * remove log * typing * quick tuyping fix * update LiteLLM selection logic * remove logs + validate functionality * rename proxy var * update path casing * remove pricing for custom models * functional values
27 lines
530 B
Python
27 lines
530 B
Python
"""Add base_url to CloudEmbeddingProvider
|
|
|
|
Revision ID: bceb1e139447
|
|
Revises: 1f60f60c3401
|
|
Create Date: 2024-08-28 17:00:52.554580
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "bceb1e139447"
|
|
down_revision = "1f60f60c3401"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
"embedding_provider", sa.Column("api_url", sa.String(), nullable=True)
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_column("embedding_provider", "api_url")
|