Testing

.github/workflows/docker-build-backend-container-on-merge-group.yml

@@ -1,33 +0,0 @@
-name: Build Backend Image on Merge Group
-
-on:
-  merge_group:
-    types: [checks_requested]
-
-env:
-  REGISTRY_IMAGE: danswer/danswer-backend
-
-jobs:
-  build:
-    # TODO: make this a matrix build like the web containers
-    runs-on: 
-      group: amd64-image-builders
-      
-    steps:
-    - name: Checkout code
-      uses: actions/checkout@v2
-
-    - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
-
-    - name: Backend Image Docker Build
-      uses: docker/build-push-action@v5
-      with:
-        context: ./backend
-        file: ./backend/Dockerfile
-        platforms: linux/amd64,linux/arm64
-        push: false
-        tags: |
-          ${{ env.REGISTRY_IMAGE }}:latest
-        build-args: |
-          DANSWER_VERSION=v0.0.1

.github/workflows/docker-build-push-backend-container-on-tag.yml

@@ -5,45 +5,31 @@ on:
     tags:
       - '*'
 
-env:
-  REGISTRY_IMAGE: danswer/danswer-backend
-
 jobs:
   build-and-push:
-    # TODO: make this a matrix build like the web containers
-    runs-on: 
-      group: amd64-image-builders
+    runs-on: ubuntu-latest
 
     steps:
     - name: Checkout code
       uses: actions/checkout@v2
 
     - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
+      uses: docker/setup-buildx-action@v1
 
     - name: Login to Docker Hub
-      uses: docker/login-action@v3
+      uses: docker/login-action@v1
       with:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_TOKEN }}
 
     - name: Backend Image Docker Build and Push
-      uses: docker/build-push-action@v5
+      uses: docker/build-push-action@v2
       with:
         context: ./backend
         file: ./backend/Dockerfile
         platforms: linux/amd64,linux/arm64
         push: true
         tags: |
-          ${{ env.REGISTRY_IMAGE }}:${{ github.ref_name }}
-          ${{ env.REGISTRY_IMAGE }}:latest
+          danswer/danswer-backend:${{ github.ref_name }}
         build-args: |
           DANSWER_VERSION=${{ github.ref_name }}
-
-    - name: Run Trivy vulnerability scanner
-      uses: aquasecurity/trivy-action@master
-      with:
-        # To run locally: trivy image --severity HIGH,CRITICAL danswer/danswer-backend
-        image-ref: docker.io/${{ env.REGISTRY_IMAGE }}:${{ github.ref_name }}
-        severity: 'CRITICAL,HIGH'
-        trivyignores: ./backend/.trivyignore

.github/workflows/docker-build-push-model-server-container-on-tag.yml

@@ -14,16 +14,16 @@ jobs:
       uses: actions/checkout@v2
 
     - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
+      uses: docker/setup-buildx-action@v1
 
     - name: Login to Docker Hub
-      uses: docker/login-action@v3
+      uses: docker/login-action@v1
       with:
         username: ${{ secrets.DOCKER_USERNAME }}
         password: ${{ secrets.DOCKER_TOKEN }}
 
     - name: Model Server Image Docker Build and Push
-      uses: docker/build-push-action@v5
+      uses: docker/build-push-action@v2
       with:
         context: ./backend
         file: ./backend/Dockerfile.model_server
@@ -31,12 +31,5 @@ jobs:
         push: true
         tags: |
           danswer/danswer-model-server:${{ github.ref_name }}
-          danswer/danswer-model-server:latest
         build-args: |
           DANSWER_VERSION=${{ github.ref_name }}
-
-    - name: Run Trivy vulnerability scanner
-      uses: aquasecurity/trivy-action@master
-      with:
-        image-ref: docker.io/danswer/danswer-model-server:${{ github.ref_name }}
-        severity: 'CRITICAL,HIGH'

.github/workflows/docker-build-push-web-container-on-tag.yml

@@ -5,115 +5,31 @@ on:
     tags:
       - '*'
 
-env:
-  REGISTRY_IMAGE: danswer/danswer-web-server
-
 jobs:
-  build:
-    runs-on: 
-      group: ${{ matrix.platform == 'linux/amd64' && 'amd64-image-builders' || 'arm64-image-builders' }}
-    strategy:
-      fail-fast: false
-      matrix:
-        platform:
-          - linux/amd64
-          - linux/arm64
-
-    steps:
-      - name: Prepare
-        run: |
-          platform=${{ matrix.platform }}
-          echo "PLATFORM_PAIR=${platform//\//-}" >> $GITHUB_ENV          
-      
-      - name: Checkout
-        uses: actions/checkout@v4
-      
-      - name: Docker meta
-        id: meta
-        uses: docker/metadata-action@v5
-        with:
-          images: ${{ env.REGISTRY_IMAGE }}
-          tags: |
-            type=raw,value=${{ env.REGISTRY_IMAGE }}:${{ github.ref_name }}
-            type=raw,value=${{ env.REGISTRY_IMAGE }}:latest
-      
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-      
-      - name: Login to Docker Hub
-        uses: docker/login-action@v3
-        with:
-          username: ${{ secrets.DOCKER_USERNAME }}
-          password: ${{ secrets.DOCKER_TOKEN }}
-    
-      - name: Build and push by digest
-        id: build
-        uses: docker/build-push-action@v5
-        with:
-          context: ./web
-          file: ./web/Dockerfile
-          platforms: ${{ matrix.platform }}
-          push: true
-          build-args: |
-            DANSWER_VERSION=${{ github.ref_name }}
-          # needed due to weird interactions with the builds for different platforms  
-          no-cache: true
-          labels: ${{ steps.meta.outputs.labels }}
-          outputs: type=image,name=${{ env.REGISTRY_IMAGE }},push-by-digest=true,name-canonical=true,push=true
-      
-      - name: Export digest
-        run: |
-          mkdir -p /tmp/digests
-          digest="${{ steps.build.outputs.digest }}"
-          touch "/tmp/digests/${digest#sha256:}"          
-      
-      - name: Upload digest
-        uses: actions/upload-artifact@v4
-        with:
-          name: digests-${{ env.PLATFORM_PAIR }}
-          path: /tmp/digests/*
-          if-no-files-found: error
-          retention-days: 1
-
-  merge:
+  build-and-push:
     runs-on: ubuntu-latest
-    needs:
-      - build
-    steps:
-      - name: Download digests
-        uses: actions/download-artifact@v4
-        with:
-          path: /tmp/digests
-          pattern: digests-*
-          merge-multiple: true
-      
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-      
-      - name: Docker meta
-        id: meta
-        uses: docker/metadata-action@v5
-        with:
-          images: ${{ env.REGISTRY_IMAGE }}
-      
-      - name: Login to Docker Hub
-        uses: docker/login-action@v3
-        with:
-          username: ${{ secrets.DOCKER_USERNAME }}
-          password: ${{ secrets.DOCKER_TOKEN }}
-      
-      - name: Create manifest list and push
-        working-directory: /tmp/digests
-        run: |
-          docker buildx imagetools create $(jq -cr '.tags | map("-t " + .) | join(" ")' <<< "$DOCKER_METADATA_OUTPUT_JSON") \
-            $(printf '${{ env.REGISTRY_IMAGE }}@sha256:%s ' *)          
-      
-      - name: Inspect image
-        run: |
-          docker buildx imagetools inspect ${{ env.REGISTRY_IMAGE }}:${{ steps.meta.outputs.version }}
 
-      - name: Run Trivy vulnerability scanner
-        uses: aquasecurity/trivy-action@master
-        with:
-          image-ref: docker.io/${{ env.REGISTRY_IMAGE }}:${{ github.ref_name }}
-          severity: 'CRITICAL,HIGH'
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v2
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v1
+
+    - name: Login to Docker Hub
+      uses: docker/login-action@v1
+      with:
+        username: ${{ secrets.DOCKER_USERNAME }}
+        password: ${{ secrets.DOCKER_TOKEN }}
+
+    - name: Web Image Docker Build and Push
+      uses: docker/build-push-action@v2
+      with:
+        context: ./web
+        file: ./web/Dockerfile
+        platforms: linux/amd64,linux/arm64
+        push: true
+        tags: |
+          danswer/danswer-web-server:${{ github.ref_name }}
+        build-args: |
+          DANSWER_VERSION=${{ github.ref_name }}

.github/workflows/docker-build-web-container-on-merge-group.yml

@@ -1,53 +0,0 @@
-name: Build Web Image on Merge Group
-
-on:
-  merge_group:
-    types: [checks_requested]
-
-env:
-  REGISTRY_IMAGE: danswer/danswer-web-server
-
-jobs:
-  build:
-    runs-on: 
-      group: ${{ matrix.platform == 'linux/amd64' && 'amd64-image-builders' || 'arm64-image-builders' }}
-    strategy:
-      fail-fast: false
-      matrix:
-        platform:
-          - linux/amd64
-          - linux/arm64
-
-    steps:
-      - name: Prepare
-        run: |
-          platform=${{ matrix.platform }}
-          echo "PLATFORM_PAIR=${platform//\//-}" >> $GITHUB_ENV          
-      
-      - name: Checkout
-        uses: actions/checkout@v4
-      
-      - name: Docker meta
-        id: meta
-        uses: docker/metadata-action@v5
-        with:
-          images: ${{ env.REGISTRY_IMAGE }}
-          tags: |
-            type=raw,value=${{ env.REGISTRY_IMAGE }}:latest
-      
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-    
-      - name: Build by digest
-        id: build
-        uses: docker/build-push-action@v5
-        with:
-          context: ./web
-          file: ./web/Dockerfile
-          platforms: ${{ matrix.platform }}
-          push: false
-          build-args: |
-            DANSWER_VERSION=v0.0.1
-          # needed due to weird interactions with the builds for different platforms  
-          no-cache: true
-          labels: ${{ steps.meta.outputs.labels }}

.gitignore

@@ -5,5 +5,3 @@
 .idea
 /deployment/data/nginx/app.conf
 .vscode/launch.json
-*.sw?
-/backend/tests/regression/answer_quality/search_test_config.yaml

.vscode/env_template.txt

@@ -1,52 +0,0 @@
-# Copy this file to .env at the base of the repo and fill in the <REPLACE THIS> values
-# This will help with development iteration speed and reduce repeat tasks for dev
-# Also check out danswer/backend/scripts/restart_containers.sh for a script to restart the containers which Danswer relies on outside of VSCode/Cursor processes
-
-# For local dev, often user Authentication is not needed
-AUTH_TYPE=disabled
-
-
-# Always keep these on for Dev
-# Logs all model prompts to stdout
-LOG_DANSWER_MODEL_INTERACTIONS=True
-# More verbose logging
-LOG_LEVEL=debug
-
-
-# This passes top N results to LLM an additional time for reranking prior to answer generation
-# This step is quite heavy on token usage so we disable it for dev generally
-DISABLE_LLM_CHUNK_FILTER=True
-
-
-# Useful if you want to toggle auth on/off (google_oauth/OIDC specifically)
-OAUTH_CLIENT_ID=<REPLACE THIS>
-OAUTH_CLIENT_SECRET=<REPLACE THIS>
-# Generally not useful for dev, we don't generally want to set up an SMTP server for dev
-REQUIRE_EMAIL_VERIFICATION=False
-
-
-# Set these so if you wipe the DB, you don't end up having to go through the UI every time
-GEN_AI_API_KEY=<REPLACE THIS>
-# If answer quality isn't important for dev, use 3.5 turbo due to it being cheaper
-GEN_AI_MODEL_VERSION=gpt-3.5-turbo
-FAST_GEN_AI_MODEL_VERSION=gpt-3.5-turbo
-
-# For Danswer Slack Bot, overrides the UI values so no need to set this up via UI every time
-# Only needed if using DanswerBot
-#DANSWER_BOT_SLACK_APP_TOKEN=<REPLACE THIS>
-#DANSWER_BOT_SLACK_BOT_TOKEN=<REPLACE THIS>
-
-
-# Python stuff
-PYTHONPATH=./backend
-PYTHONUNBUFFERED=1
-
-
-# Internet Search 
-BING_API_KEY=<REPLACE THIS>
-
-
-# Enable the full set of Danswer Enterprise Edition features
-# NOTE: DO NOT ENABLE THIS UNLESS YOU HAVE A PAID ENTERPRISE LICENSE (or if you are using this for local testing/development)
-ENABLE_PAID_ENTERPRISE_EDITION_FEATURES=False
-

.vscode/launch.template.jsonc

@@ -11,47 +11,17 @@
     // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
     "version": "0.2.0",
     "configurations": [
-        {
-            "name": "Web Server",
-            "type": "node",
-            "request": "launch",
-            "cwd": "${workspaceRoot}/web",
-            "runtimeExecutable": "npm",
-            "envFile": "${workspaceFolder}/.env",
-            "runtimeArgs": [
-                "run", "dev"
-            ],
-            "console": "integratedTerminal"
-        },
-        {
-            "name": "Model Server",
-            "type": "python",
-            "request": "launch",
-            "module": "uvicorn",
-            "cwd": "${workspaceFolder}/backend",
-            "envFile": "${workspaceFolder}/.env",
-            "env": {
-                "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1"
-            },
-            "args": [
-                "model_server.main:app",
-                "--reload",
-                "--port",
-                "9000"
-            ]
-        },
         {
             "name": "API Server",
             "type": "python",
             "request": "launch",
             "module": "uvicorn",
             "cwd": "${workspaceFolder}/backend",
-            "envFile": "${workspaceFolder}/.env",
             "env": {
-                "LOG_DANSWER_MODEL_INTERACTIONS": "True",
                 "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1"
+                "DISABLE_AUTH": "True",
+                "TYPESENSE_API_KEY": "typesense_api_key",
+                "DYNAMIC_CONFIG_DIR_PATH": "./dynamic_config_storage"
             },
             "args": [
                 "danswer.main:app",
@@ -61,69 +31,52 @@
             ]
         },
         {
-            "name": "Indexing",
+            "name": "Indexer",
             "type": "python",
             "request": "launch",
             "program": "danswer/background/update.py",
             "cwd": "${workspaceFolder}/backend",
-            "envFile": "${workspaceFolder}/.env",
             "env": {
-                "ENABLE_MINI_CHUNK": "false",
                 "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1",
-                "PYTHONPATH": "."
+                "PYTHONPATH": ".",
+                "TYPESENSE_API_KEY": "typesense_api_key",
+                "DYNAMIC_CONFIG_DIR_PATH": "./dynamic_config_storage"
             }
         },
-        // Celery and all async jobs, usually would include indexing as well but this is handled separately above for dev
         {
-            "name": "Background Jobs",
+            "name": "Temp File Deletion",
             "type": "python",
             "request": "launch",
-            "program": "scripts/dev_run_background_jobs.py",
+            "program": "danswer/background/file_deletion.py",
             "cwd": "${workspaceFolder}/backend",
-            "envFile": "${workspaceFolder}/.env",
             "env": {
-                "LOG_DANSWER_MODEL_INTERACTIONS": "True",
                 "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1",
-                "PYTHONPATH": "."
-            },
-            "args": [
-                "--no-indexing"
-            ]
+                "PYTHONPATH": "${workspaceFolder}/backend"
+            }
         },
         // For the listner to access the Slack API,
         // DANSWER_BOT_SLACK_APP_TOKEN & DANSWER_BOT_SLACK_BOT_TOKEN need to be set in .env file located in the root of the project
         {
-            "name": "Slack Bot",
+            "name": "Slack Bot Listener",
             "type": "python",
             "request": "launch",
-            "program": "danswer/danswerbot/slack/listener.py",
+            "program": "danswer/listeners/slack_listener.py",
             "cwd": "${workspaceFolder}/backend",
             "envFile": "${workspaceFolder}/.env",
             "env": {
-                "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1",
-                "PYTHONPATH": "."
+                "LOG_LEVEL": "DEBUG"
             }
         },
         {
-            "name": "Pytest",
-            "type": "python",
+            "name": "Web Server",
+            "type": "node",
             "request": "launch",
-            "module": "pytest",
-            "cwd": "${workspaceFolder}/backend",
-            "envFile": "${workspaceFolder}/.env",
-            "env": {
-                "LOG_LEVEL": "DEBUG",
-                "PYTHONUNBUFFERED": "1",
-                "PYTHONPATH": "."
-            },
-            "args": [
-                "-v"
-                // Specify a sepcific module/test to run or provide nothing to run all tests
-                //"tests/unit/danswer/llm/answering/test_prune_and_merge.py"
-            ]
+            "cwd": "${workspaceRoot}/web",
+            "runtimeExecutable": "npm",
+            "runtimeArgs": [
+                "run", "dev"
+            ],
+            "console": "integratedTerminal"
         }
     ]
-}
+}

CONTRIBUTING.md

@@ -72,10 +72,6 @@ For convenience here's a command for it:
 python -m venv .venv
 source .venv/bin/activate
 ```
-
---> Note that this virtual environment MUST NOT be set up WITHIN the danswer
-directory
-
 _For Windows, activate the virtual environment using Command Prompt:_
 ```bash
 .venv\Scripts\activate

LICENSE

@@ -1,10 +1,6 @@
-Copyright (c) 2023-present DanswerAI, Inc.
+MIT License
 
-Portions of this software are licensed as follows:
-
-* All content that resides under "ee" directories of this repository, if that directory exists, is licensed under the license defined in "backend/ee/LICENSE". Specifically all content under "backend/ee" and "web/src/app/ee" is licensed under the license defined in "backend/ee/LICENSE".
-* All third party components incorporated into the Danswer Software are licensed under the original license provided by the owner of the applicable component.
-* Content outside of the above mentioned directories or restrictions above is available under the "MIT Expat" license as defined below.
+Copyright (c) 2023 Yuhong Sun, Chris Weaver
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -11,7 +11,7 @@
 <a href="https://docs.danswer.dev/" target="_blank">
     <img src="https://img.shields.io/badge/docs-view-blue" alt="Documentation">
 </a>
-<a href="https://join.slack.com/t/danswer/shared_invite/zt-2lcmqw703-071hBuZBfNEOGUsLa5PXvQ" target="_blank">
+<a href="https://join.slack.com/t/danswer/shared_invite/zt-2afut44lv-Rw3kSWu6_OmdAXRpCv80DQ" target="_blank">
     <img src="https://img.shields.io/badge/slack-join-blue.svg?logo=slack" alt="Slack">
 </a>
 <a href="https://discord.gg/TDJ59cGV2X" target="_blank">
@@ -105,25 +105,5 @@ Efficiently pulls the latest changes from:
   * Websites
   * And more ...
 
-## 📚 Editions
-
-There are two editions of Danswer:
-
-  * Danswer Community Edition (CE) is available freely under the MIT Expat license. This version has ALL the core features discussed above. This is the version of Danswer you will get if you follow the Deployment guide above.
-  * Danswer Enterprise Edition (EE) includes extra features that are primarily useful for larger organizations. Specifically, this includes:
-    * Single Sign-On (SSO), with support for both SAML and OIDC
-    * Role-based access control
-    * Document permission inheritance from connected sources
-    * Usage analytics and query history accessible to admins
-    * Whitelabeling
-    * API key authentication
-    * Encryption of secrets
-    * Any many more! Checkout [our website](https://www.danswer.ai/) for the latest.
-
-To try the Danswer Enterprise Edition: 
-
-  1. Checkout our [Cloud product](https://app.danswer.ai/signup).
-  2. For self-hosting, contact us at [founders@danswer.ai](mailto:founders@danswer.ai) or book a call with us on our [Cal](https://cal.com/team/danswer/founders).
-
 ## 💡 Contributing
 Looking to contribute? Please check out the [Contribution Guide](CONTRIBUTING.md) for more details.

backend/.gitignore

@@ -5,7 +5,7 @@ site_crawls/
 .ipynb_checkpoints/
 api_keys.py
 *ipynb
-.env*
+.env
 vespa-app.zip
 dynamic_config_storage/
-celerybeat-schedule*
+celerybeat-schedule

backend/Dockerfile

@@ -1,17 +1,15 @@
 FROM python:3.11.7-slim-bookworm
 
 LABEL com.danswer.maintainer="founders@danswer.ai"
-LABEL com.danswer.description="This image is the web/frontend container of Danswer which \
-contains code for both the Community and Enterprise editions of Danswer. If you do not \
-have a contract or agreement with DanswerAI, you are not permitted to use the Enterprise \
-Edition features outside of personal development or testing purposes. Please reach out to \
-founders@danswer.ai for more information. Please visit https://github.com/danswer-ai/danswer"
+LABEL com.danswer.description="This image is for the backend of Danswer. It is MIT Licensed and \
+free for all to use. You can find it at https://hub.docker.com/r/danswer/danswer-backend. For \
+more details, visit https://github.com/danswer-ai/danswer."
 
 # Default DANSWER_VERSION, typically overriden during builds by GitHub Actions.
 ARG DANSWER_VERSION=0.3-dev
 ENV DANSWER_VERSION=${DANSWER_VERSION}
-
 RUN echo "DANSWER_VERSION: ${DANSWER_VERSION}"
+
 # Install system dependencies
 # cmake needed for psycopg (postgres)
 # libpq-dev needed for psycopg (postgres)
@@ -19,32 +17,18 @@ RUN echo "DANSWER_VERSION: ${DANSWER_VERSION}"
 # zip for Vespa step futher down
 # ca-certificates for HTTPS
 RUN apt-get update && \
-    apt-get install -y \
-        cmake \
-        curl \
-        zip \
-        ca-certificates \
-        libgnutls30=3.7.9-2+deb12u3 \
-        libblkid1=2.38.1-5+deb12u1 \
-        libmount1=2.38.1-5+deb12u1 \
-        libsmartcols1=2.38.1-5+deb12u1 \
-        libuuid1=2.38.1-5+deb12u1 \
-        libxmlsec1-dev \
-        pkg-config \
-        gcc && \
+    apt-get install -y cmake curl zip ca-certificates libgnutls30=3.7.9-2+deb12u2 \
+    libblkid1=2.38.1-5+deb12u1 libmount1=2.38.1-5+deb12u1 libsmartcols1=2.38.1-5+deb12u1 \
+    libuuid1=2.38.1-5+deb12u1 && \
     rm -rf /var/lib/apt/lists/* && \
     apt-get clean
 
 # Install Python dependencies
 # Remove py which is pulled in by retry, py is not needed and is a CVE
 COPY ./requirements/default.txt /tmp/requirements.txt
-COPY ./requirements/ee.txt /tmp/ee-requirements.txt
-RUN pip install --no-cache-dir --upgrade \
-        -r /tmp/requirements.txt \
-        -r /tmp/ee-requirements.txt && \
+RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt && \
     pip uninstall -y py && \
-    playwright install chromium && \
-    playwright install-deps chromium && \
+    playwright install chromium && playwright install-deps chromium && \
     ln -s /usr/local/bin/supervisord /usr/bin/supervisord
 
 # Cleanup for CVEs and size reduction
@@ -52,20 +36,11 @@ RUN pip install --no-cache-dir --upgrade \
 # xserver-common and xvfb included by playwright installation but not needed after
 # perl-base is part of the base Python Debian image but not needed for Danswer functionality
 # perl-base could only be removed with --allow-remove-essential
-RUN apt-get update && \
-    apt-get remove -y --allow-remove-essential \
-        perl-base \
-        xserver-common \
-        xvfb \
-        cmake \
-        libldap-2.5-0 \
-        libxmlsec1-dev \
-        pkg-config \
-        gcc && \
-    apt-get install -y libxmlsec1-openssl && \
+RUN apt-get remove -y --allow-remove-essential perl-base xserver-common xvfb cmake \
+    libldap-2.5-0 libldap-2.5-0 && \
     apt-get autoremove -y && \
     rm -rf /var/lib/apt/lists/* && \
-    rm -f /usr/local/lib/python3.11/site-packages/tornado/test/test.key
+    rm /usr/local/lib/python3.11/site-packages/tornado/test/test.key
 
 # Pre-downloading models for setups with limited egress
 RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('intfloat/e5-base-v2')"
@@ -78,24 +53,12 @@ nltk.download('punkt', quiet=True);"
 
 # Set up application files
 WORKDIR /app
-
-# Enterprise Version Files
-COPY ./ee /app/ee
-COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
-
-# Set up application files
 COPY ./danswer /app/danswer
 COPY ./shared_configs /app/shared_configs
 COPY ./alembic /app/alembic
 COPY ./alembic.ini /app/alembic.ini
 COPY supervisord.conf /usr/etc/supervisord.conf
 
-# Escape hatch
-COPY ./scripts/force_delete_connector_by_id.py /app/scripts/force_delete_connector_by_id.py
-
-# Put logo in assets
-COPY ./assets /app/assets
-
 ENV PYTHONPATH /app
 
 # Default command which does nothing

backend/alembic/versions/0568ccf46a6b_add_thread_specific_model_selection.py

@@ -1,31 +0,0 @@
-"""Add thread specific model selection
-
-Revision ID: 0568ccf46a6b
-Revises: e209dc5a8156
-Create Date: 2024-06-19 14:25:36.376046
-
-"""
-
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "0568ccf46a6b"
-down_revision = "e209dc5a8156"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column(
-        "chat_session",
-        sa.Column("current_alternate_model", sa.String(), nullable=True),
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_column("chat_session", "current_alternate_model")
-    # ### end Alembic commands ###

backend/alembic/versions/05c07bf07c00_add_search_doc_relevance_details.py

@@ -1,32 +0,0 @@
-"""add search doc relevance details
-
-Revision ID: 05c07bf07c00
-Revises: b896bbd0d5a7
-Create Date: 2024-07-10 17:48:15.886653
-
-"""
-
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "05c07bf07c00"
-down_revision = "b896bbd0d5a7"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "search_doc",
-        sa.Column("is_relevant", sa.Boolean(), nullable=True),
-    )
-    op.add_column(
-        "search_doc",
-        sa.Column("relevance_explanation", sa.String(), nullable=True),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("search_doc", "relevance_explanation")
-    op.drop_column("search_doc", "is_relevant")

backend/alembic/versions/0a2b51deb0b8_add_starter_prompts.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "0a2b51deb0b8"
 down_revision = "5f4b8568a221"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/0a98909f2757_enable_encrypted_fields.py

@@ -1,113 +0,0 @@
-"""Enable Encrypted Fields
-
-Revision ID: 0a98909f2757
-Revises: 570282d33c49
-Create Date: 2024-05-05 19:30:34.317972
-
-"""
-from alembic import op
-import sqlalchemy as sa
-from sqlalchemy.sql import table
-from sqlalchemy.dialects import postgresql
-import json
-
-from danswer.utils.encryption import encrypt_string_to_bytes
-
-# revision identifiers, used by Alembic.
-revision = "0a98909f2757"
-down_revision = "570282d33c49"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    connection = op.get_bind()
-
-    op.alter_column("key_value_store", "value", nullable=True)
-    op.add_column(
-        "key_value_store",
-        sa.Column(
-            "encrypted_value",
-            sa.LargeBinary,
-            nullable=True,
-        ),
-    )
-
-    # Need a temporary column to translate the JSONB to binary
-    op.add_column("credential", sa.Column("temp_column", sa.LargeBinary()))
-
-    creds_table = table(
-        "credential",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column(
-            "credential_json",
-            postgresql.JSONB(astext_type=sa.Text()),
-            nullable=False,
-        ),
-        sa.Column(
-            "temp_column",
-            sa.LargeBinary(),
-            nullable=False,
-        ),
-    )
-
-    results = connection.execute(sa.select(creds_table))
-
-    # This uses the MIT encrypt which does not actually encrypt the credentials
-    # In other words, this upgrade does not apply the encryption. Porting existing sensitive data
-    # and key rotation currently is not supported and will come out in the future
-    for row_id, creds, _ in results:
-        creds_binary = encrypt_string_to_bytes(json.dumps(creds))
-        connection.execute(
-            creds_table.update()
-            .where(creds_table.c.id == row_id)
-            .values(temp_column=creds_binary)
-        )
-
-    op.drop_column("credential", "credential_json")
-    op.alter_column("credential", "temp_column", new_column_name="credential_json")
-
-    op.add_column("llm_provider", sa.Column("temp_column", sa.LargeBinary()))
-
-    llm_table = table(
-        "llm_provider",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column(
-            "api_key",
-            sa.String(),
-            nullable=False,
-        ),
-        sa.Column(
-            "temp_column",
-            sa.LargeBinary(),
-            nullable=False,
-        ),
-    )
-    results = connection.execute(sa.select(llm_table))
-
-    for row_id, api_key, _ in results:
-        llm_key = encrypt_string_to_bytes(api_key)
-        connection.execute(
-            llm_table.update()
-            .where(llm_table.c.id == row_id)
-            .values(temp_column=llm_key)
-        )
-
-    op.drop_column("llm_provider", "api_key")
-    op.alter_column("llm_provider", "temp_column", new_column_name="api_key")
-
-
-def downgrade() -> None:
-    # Some information loss but this is ok. Should not allow decryption via downgrade.
-    op.drop_column("credential", "credential_json")
-    op.drop_column("llm_provider", "api_key")
-
-    op.add_column("llm_provider", sa.Column("api_key", sa.String()))
-    op.add_column(
-        "credential",
-        sa.Column("credential_json", postgresql.JSONB(astext_type=sa.Text())),
-    )
-
-    op.execute("DELETE FROM key_value_store WHERE value IS NULL")
-    op.alter_column("key_value_store", "value", nullable=False)
-    op.drop_column("key_value_store", "encrypted_value")

backend/alembic/versions/15326fcec57e_introduce_danswer_apis.py

@@ -13,8 +13,8 @@ from danswer.configs.constants import DocumentSource
 # revision identifiers, used by Alembic.
 revision = "15326fcec57e"
 down_revision = "77d07dffae64"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/173cae5bba26_port_config_store.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "173cae5bba26"
 down_revision = "e50154680a5c"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/23957775e5f5_remove_feedback_foreignkey_constraint.py

@@ -1,86 +0,0 @@
-"""remove-feedback-foreignkey-constraint
-
-Revision ID: 23957775e5f5
-Revises: bc9771dccadf
-Create Date: 2024-06-27 16:04:51.480437
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "23957775e5f5"
-down_revision = "bc9771dccadf"
-branch_labels = None  # type: ignore
-depends_on = None  # type: ignore
-
-
-def upgrade() -> None:
-    op.drop_constraint(
-        "chat_feedback__chat_message_fk", "chat_feedback", type_="foreignkey"
-    )
-    op.create_foreign_key(
-        "chat_feedback__chat_message_fk",
-        "chat_feedback",
-        "chat_message",
-        ["chat_message_id"],
-        ["id"],
-        ondelete="SET NULL",
-    )
-    op.alter_column(
-        "chat_feedback", "chat_message_id", existing_type=sa.Integer(), nullable=True
-    )
-    op.drop_constraint(
-        "document_retrieval_feedback__chat_message_fk",
-        "document_retrieval_feedback",
-        type_="foreignkey",
-    )
-    op.create_foreign_key(
-        "document_retrieval_feedback__chat_message_fk",
-        "document_retrieval_feedback",
-        "chat_message",
-        ["chat_message_id"],
-        ["id"],
-        ondelete="SET NULL",
-    )
-    op.alter_column(
-        "document_retrieval_feedback",
-        "chat_message_id",
-        existing_type=sa.Integer(),
-        nullable=True,
-    )
-
-
-def downgrade() -> None:
-    op.alter_column(
-        "chat_feedback", "chat_message_id", existing_type=sa.Integer(), nullable=False
-    )
-    op.drop_constraint(
-        "chat_feedback__chat_message_fk", "chat_feedback", type_="foreignkey"
-    )
-    op.create_foreign_key(
-        "chat_feedback__chat_message_fk",
-        "chat_feedback",
-        "chat_message",
-        ["chat_message_id"],
-        ["id"],
-    )
-
-    op.alter_column(
-        "document_retrieval_feedback",
-        "chat_message_id",
-        existing_type=sa.Integer(),
-        nullable=False,
-    )
-    op.drop_constraint(
-        "document_retrieval_feedback__chat_message_fk",
-        "document_retrieval_feedback",
-        type_="foreignkey",
-    )
-    op.create_foreign_key(
-        "document_retrieval_feedback__chat_message_fk",
-        "document_retrieval",
-        "chat_message",
-        ["chat_message_id"],
-        ["id"],
-    )

backend/alembic/versions/2666d766cb9b_google_oauth2.py

@@ -13,8 +13,8 @@ from alembic import op
 # revision identifiers, used by Alembic.
 revision = "2666d766cb9b"
 down_revision = "6d387b3196c2"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/27c6ecc08586_permission_framework.py

@@ -13,8 +13,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "27c6ecc08586"
 down_revision = "2666d766cb9b"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/30c1d5744104_persona_datetime_aware.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "30c1d5744104"
 down_revision = "7f99be1cb9f5"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/3879338f8ba1_add_tool_table.py

@@ -1,45 +0,0 @@
-"""Add tool table
-
-Revision ID: 3879338f8ba1
-Revises: f1c6478c3fd8
-Create Date: 2024-05-11 16:11:23.718084
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "3879338f8ba1"
-down_revision = "f1c6478c3fd8"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "tool",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("name", sa.String(), nullable=False),
-        sa.Column("description", sa.Text(), nullable=True),
-        sa.Column("in_code_tool_id", sa.String(), nullable=True),
-        sa.PrimaryKeyConstraint("id"),
-    )
-    op.create_table(
-        "persona__tool",
-        sa.Column("persona_id", sa.Integer(), nullable=False),
-        sa.Column("tool_id", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["persona_id"],
-            ["persona.id"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["tool_id"],
-            ["tool.id"],
-        ),
-        sa.PrimaryKeyConstraint("persona_id", "tool_id"),
-    )
-
-
-def downgrade() -> None:
-    op.drop_table("persona__tool")
-    op.drop_table("tool")

backend/alembic/versions/38eda64af7fe_add_chat_session_sharing.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "38eda64af7fe"
 down_revision = "776b3bbe9092"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/3a7802814195_add_alternate_assistant_to_chat_message.py

@@ -1,38 +0,0 @@
-"""add alternate assistant to chat message
-
-Revision ID: 3a7802814195
-Revises: 23957775e5f5
-Create Date: 2024-06-05 11:18:49.966333
-
-"""
-
-from alembic import op
-import sqlalchemy as sa
-
-
-# revision identifiers, used by Alembic.
-revision = "3a7802814195"
-down_revision = "23957775e5f5"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column(
-        "chat_message", sa.Column("alternate_assistant_id", sa.Integer(), nullable=True)
-    )
-    op.create_foreign_key(
-        "fk_chat_message_persona",
-        "chat_message",
-        "persona",
-        ["alternate_assistant_id"],
-        ["id"],
-    )
-    # ### end Alembic commands ###
-
-
-def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_constraint("fk_chat_message_persona", "chat_message", type_="foreignkey")
-    op.drop_column("chat_message", "alternate_assistant_id")

backend/alembic/versions/3b25685ff73c_move_is_public_to_cc_pair.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "3b25685ff73c"
 down_revision = "e0a68a81d434"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/3c5e35aa9af0_polling_document_count.py

@@ -12,8 +12,8 @@ from alembic import op
 # revision identifiers, used by Alembic.
 revision = "3c5e35aa9af0"
 down_revision = "27c6ecc08586"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/401c1ac29467_add_tables_for_ui_based_llm_.py

@@ -1,49 +0,0 @@
-"""Add tables for UI-based LLM configuration
-
-Revision ID: 401c1ac29467
-Revises: 703313b75876
-Create Date: 2024-04-13 18:07:29.153817
-
-"""
-from alembic import op
-import sqlalchemy as sa
-from sqlalchemy.dialects import postgresql
-
-# revision identifiers, used by Alembic.
-revision = "401c1ac29467"
-down_revision = "703313b75876"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "llm_provider",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("name", sa.String(), nullable=False),
-        sa.Column("api_key", sa.String(), nullable=True),
-        sa.Column("api_base", sa.String(), nullable=True),
-        sa.Column("api_version", sa.String(), nullable=True),
-        sa.Column(
-            "custom_config",
-            postgresql.JSONB(astext_type=sa.Text()),
-            nullable=True,
-        ),
-        sa.Column("default_model_name", sa.String(), nullable=False),
-        sa.Column("fast_default_model_name", sa.String(), nullable=True),
-        sa.Column("is_default_provider", sa.Boolean(), unique=True, nullable=True),
-        sa.Column("model_names", postgresql.ARRAY(sa.String()), nullable=True),
-        sa.PrimaryKeyConstraint("id"),
-        sa.UniqueConstraint("name"),
-    )
-
-    op.add_column(
-        "persona",
-        sa.Column("llm_model_provider_override", sa.String(), nullable=True),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("persona", "llm_model_provider_override")
-
-    op.drop_table("llm_provider")

backend/alembic/versions/44f856ae2a4a_add_cloud_embedding_model.py

@@ -1,65 +0,0 @@
-"""add cloud embedding model and update embedding_model
-
-Revision ID: 44f856ae2a4a
-Revises: d716b0791ddd
-Create Date: 2024-06-28 20:01:05.927647
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "44f856ae2a4a"
-down_revision = "d716b0791ddd"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    # Create embedding_provider table
-    op.create_table(
-        "embedding_provider",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("name", sa.String(), nullable=False),
-        sa.Column("api_key", sa.LargeBinary(), nullable=True),
-        sa.Column("default_model_id", sa.Integer(), nullable=True),
-        sa.PrimaryKeyConstraint("id"),
-        sa.UniqueConstraint("name"),
-    )
-
-    # Add cloud_provider_id to embedding_model table
-    op.add_column(
-        "embedding_model", sa.Column("cloud_provider_id", sa.Integer(), nullable=True)
-    )
-
-    # Add foreign key constraints
-    op.create_foreign_key(
-        "fk_embedding_model_cloud_provider",
-        "embedding_model",
-        "embedding_provider",
-        ["cloud_provider_id"],
-        ["id"],
-    )
-    op.create_foreign_key(
-        "fk_embedding_provider_default_model",
-        "embedding_provider",
-        "embedding_model",
-        ["default_model_id"],
-        ["id"],
-    )
-
-
-def downgrade() -> None:
-    # Remove foreign key constraints
-    op.drop_constraint(
-        "fk_embedding_model_cloud_provider", "embedding_model", type_="foreignkey"
-    )
-    op.drop_constraint(
-        "fk_embedding_provider_default_model", "embedding_provider", type_="foreignkey"
-    )
-
-    # Remove cloud_provider_id column
-    op.drop_column("embedding_model", "cloud_provider_id")
-
-    # Drop embedding_provider table
-    op.drop_table("embedding_provider")

backend/alembic/versions/4505fd7302e1_added_is_internet_to_dbdoc.py

@@ -1,23 +0,0 @@
-"""added is_internet to DBDoc
-
-Revision ID: 4505fd7302e1
-Revises: c18cdf4b497e
-Create Date: 2024-06-18 20:46:09.095034
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "4505fd7302e1"
-down_revision = "c18cdf4b497e"
-
-
-def upgrade() -> None:
-    op.add_column("search_doc", sa.Column("is_internet", sa.Boolean(), nullable=True))
-    op.add_column("tool", sa.Column("display_name", sa.String(), nullable=True))
-
-
-def downgrade() -> None:
-    op.drop_column("tool", "display_name")
-    op.drop_column("search_doc", "is_internet")

backend/alembic/versions/465f78d9b7f9_larger_access_tokens_for_oauth.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "465f78d9b7f9"
 down_revision = "3c5e35aa9af0"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/46625e4745d4_remove_native_enum.py

@@ -11,8 +11,8 @@ from sqlalchemy import String
 # revision identifiers, used by Alembic.
 revision = "46625e4745d4"
 down_revision = "9d97fecfab7f"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/4738e4b3bae1_pg_file_store.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "4738e4b3bae1"
 down_revision = "e91df4e935ef"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/47433d30de82_create_indexattempt_table.py

@@ -11,9 +11,9 @@ from sqlalchemy.dialects import postgresql
 
 # revision identifiers, used by Alembic.
 revision = "47433d30de82"
-down_revision: None = None
-branch_labels: None = None
-depends_on: None = None
+down_revision = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/475fcefe8826_add_name_to_api_key.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "475fcefe8826"
 down_revision = "ecab2b3f1a3b"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/48d14957fe80_add_support_for_custom_tools.py

@@ -1,61 +0,0 @@
-"""Add support for custom tools
-
-Revision ID: 48d14957fe80
-Revises: b85f02ec1308
-Create Date: 2024-06-09 14:58:19.946509
-
-"""
-from alembic import op
-import fastapi_users_db_sqlalchemy
-import sqlalchemy as sa
-from sqlalchemy.dialects import postgresql
-
-# revision identifiers, used by Alembic.
-revision = "48d14957fe80"
-down_revision = "b85f02ec1308"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "tool",
-        sa.Column(
-            "openapi_schema",
-            postgresql.JSONB(astext_type=sa.Text()),
-            nullable=True,
-        ),
-    )
-    op.add_column(
-        "tool",
-        sa.Column(
-            "user_id",
-            fastapi_users_db_sqlalchemy.generics.GUID(),
-            nullable=True,
-        ),
-    )
-    op.create_foreign_key("tool_user_fk", "tool", "user", ["user_id"], ["id"])
-
-    op.create_table(
-        "tool_call",
-        sa.Column("id", sa.Integer(), primary_key=True),
-        sa.Column("tool_id", sa.Integer(), nullable=False),
-        sa.Column("tool_name", sa.String(), nullable=False),
-        sa.Column(
-            "tool_arguments", postgresql.JSONB(astext_type=sa.Text()), nullable=False
-        ),
-        sa.Column(
-            "tool_result", postgresql.JSONB(astext_type=sa.Text()), nullable=False
-        ),
-        sa.Column(
-            "message_id", sa.Integer(), sa.ForeignKey("chat_message.id"), nullable=False
-        ),
-    )
-
-
-def downgrade() -> None:
-    op.drop_table("tool_call")
-
-    op.drop_constraint("tool_user_fk", "tool", type_="foreignkey")
-    op.drop_column("tool", "user_id")
-    op.drop_column("tool", "openapi_schema")

backend/alembic/versions/50b683a8295c_add_additional_retrieval_controls_to_.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "50b683a8295c"
 down_revision = "7da0ae5ad583"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/570282d33c49_track_danswerbot_explicitly.py

@@ -1,27 +0,0 @@
-"""Track Danswerbot Explicitly
-
-Revision ID: 570282d33c49
-Revises: 7547d982db8f
-Create Date: 2024-05-04 17:49:28.568109
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "570282d33c49"
-down_revision = "7547d982db8f"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "chat_session", sa.Column("danswerbot_flow", sa.Boolean(), nullable=True)
-    )
-    op.execute("UPDATE chat_session SET danswerbot_flow = one_shot")
-    op.alter_column("chat_session", "danswerbot_flow", nullable=False)
-
-
-def downgrade() -> None:
-    op.drop_column("chat_session", "danswerbot_flow")

backend/alembic/versions/57b53544726e_add_document_set_tables.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "57b53544726e"
 down_revision = "800f48024ae9"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/5809c0787398_add_chat_sessions.py

@@ -13,8 +13,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "5809c0787398"
 down_revision = "d929f0c1c6af"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/5e84129c8be3_add_docs_indexed_column_to_index_.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "5e84129c8be3"
 down_revision = "e6a4bbc13fe4"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/5f4b8568a221_add_removed_documents_to_index_attempt.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "5f4b8568a221"
 down_revision = "8987770549c0"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/643a84a42a33_add_user_configured_names_to_llmprovider.py

@@ -1,45 +0,0 @@
-"""Add user-configured names to LLMProvider
-
-Revision ID: 643a84a42a33
-Revises: 0a98909f2757
-Create Date: 2024-05-07 14:54:55.493100
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "643a84a42a33"
-down_revision = "0a98909f2757"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column("llm_provider", sa.Column("provider", sa.String(), nullable=True))
-    # move "name" -> "provider" to match the new schema
-    op.execute("UPDATE llm_provider SET provider = name")
-    # pretty up display name
-    op.execute("UPDATE llm_provider SET name = 'OpenAI' WHERE name = 'openai'")
-    op.execute("UPDATE llm_provider SET name = 'Anthropic' WHERE name = 'anthropic'")
-    op.execute("UPDATE llm_provider SET name = 'Azure OpenAI' WHERE name = 'azure'")
-    op.execute("UPDATE llm_provider SET name = 'AWS Bedrock' WHERE name = 'bedrock'")
-
-    # update personas to use the new provider names
-    op.execute(
-        "UPDATE persona SET llm_model_provider_override = 'OpenAI' WHERE llm_model_provider_override = 'openai'"
-    )
-    op.execute(
-        "UPDATE persona SET llm_model_provider_override = 'Anthropic' WHERE llm_model_provider_override = 'anthropic'"
-    )
-    op.execute(
-        "UPDATE persona SET llm_model_provider_override = 'Azure OpenAI' WHERE llm_model_provider_override = 'azure'"
-    )
-    op.execute(
-        "UPDATE persona SET llm_model_provider_override = 'AWS Bedrock' WHERE llm_model_provider_override = 'bedrock'"
-    )
-
-
-def downgrade() -> None:
-    op.execute("UPDATE llm_provider SET name = provider")
-    op.drop_column("llm_provider", "provider")

backend/alembic/versions/6d387b3196c2_basic_auth.py

@@ -13,8 +13,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "6d387b3196c2"
 down_revision = "47433d30de82"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/703313b75876_add_tokenratelimit_tables.py

@@ -1,83 +0,0 @@
-"""Add TokenRateLimit Tables
-
-Revision ID: 703313b75876
-Revises: fad14119fb92
-Create Date: 2024-04-15 01:36:02.952809
-
-"""
-import json
-from typing import cast
-from alembic import op
-import sqlalchemy as sa
-from danswer.dynamic_configs.factory import get_dynamic_config_store
-
-# revision identifiers, used by Alembic.
-revision = "703313b75876"
-down_revision = "fad14119fb92"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "token_rate_limit",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("enabled", sa.Boolean(), nullable=False),
-        sa.Column("token_budget", sa.Integer(), nullable=False),
-        sa.Column("period_hours", sa.Integer(), nullable=False),
-        sa.Column(
-            "scope",
-            sa.String(length=10),
-            nullable=False,
-        ),
-        sa.Column(
-            "created_at",
-            sa.DateTime(timezone=True),
-            server_default=sa.text("now()"),
-            nullable=False,
-        ),
-        sa.PrimaryKeyConstraint("id"),
-    )
-    op.create_table(
-        "token_rate_limit__user_group",
-        sa.Column("rate_limit_id", sa.Integer(), nullable=False),
-        sa.Column("user_group_id", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["rate_limit_id"],
-            ["token_rate_limit.id"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["user_group_id"],
-            ["user_group.id"],
-        ),
-        sa.PrimaryKeyConstraint("rate_limit_id", "user_group_id"),
-    )
-
-    try:
-        settings_json = cast(
-            str, get_dynamic_config_store().load("token_budget_settings")
-        )
-        settings = json.loads(settings_json)
-
-        is_enabled = settings.get("enable_token_budget", False)
-        token_budget = settings.get("token_budget", -1)
-        period_hours = settings.get("period_hours", -1)
-
-        if is_enabled and token_budget > 0 and period_hours > 0:
-            op.execute(
-                f"INSERT INTO token_rate_limit \
-                    (enabled, token_budget, period_hours, scope) VALUES \
-                        ({is_enabled}, {token_budget}, {period_hours}, 'GLOBAL')"
-            )
-
-        # Delete the dynamic config
-        get_dynamic_config_store().delete("token_budget_settings")
-
-    except Exception:
-        # Ignore if the dynamic config is not found
-        pass
-
-
-def downgrade() -> None:
-    op.drop_table("token_rate_limit__user_group")
-    op.drop_table("token_rate_limit")

backend/alembic/versions/70f00c45c0f2_more_descriptive_filestore.py

@@ -1,68 +0,0 @@
-"""More Descriptive Filestore
-
-Revision ID: 70f00c45c0f2
-Revises: 3879338f8ba1
-Create Date: 2024-05-17 17:51:41.926893
-
-"""
-from alembic import op
-import sqlalchemy as sa
-from sqlalchemy.dialects import postgresql
-
-# revision identifiers, used by Alembic.
-revision = "70f00c45c0f2"
-down_revision = "3879338f8ba1"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column("file_store", sa.Column("display_name", sa.String(), nullable=True))
-    op.add_column(
-        "file_store",
-        sa.Column(
-            "file_origin",
-            sa.String(),
-            nullable=False,
-            server_default="connector",  # Default to connector
-        ),
-    )
-    op.add_column(
-        "file_store",
-        sa.Column(
-            "file_type", sa.String(), nullable=False, server_default="text/plain"
-        ),
-    )
-    op.add_column(
-        "file_store",
-        sa.Column(
-            "file_metadata",
-            postgresql.JSONB(astext_type=sa.Text()),
-            nullable=True,
-        ),
-    )
-
-    op.execute(
-        """
-        UPDATE file_store
-        SET file_origin = CASE
-            WHEN file_name LIKE 'chat__%' THEN 'chat_upload'
-            ELSE 'connector'
-        END,
-        file_name = CASE
-            WHEN file_name LIKE 'chat__%' THEN SUBSTR(file_name, 7)
-            ELSE file_name
-        END,
-        file_type = CASE
-            WHEN file_name LIKE 'chat__%' THEN 'image/png'
-            ELSE 'text/plain'
-        END
-    """
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("file_store", "file_metadata")
-    op.drop_column("file_store", "file_type")
-    op.drop_column("file_store", "file_origin")
-    op.drop_column("file_store", "display_name")

backend/alembic/versions/72bdc9929a46_permission_auto_sync_framework.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "72bdc9929a46"
 down_revision = "475fcefe8826"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7547d982db8f_chat_folders.py

@@ -1,51 +0,0 @@
-"""Chat Folders
-
-Revision ID: 7547d982db8f
-Revises: ef7da92f7213
-Create Date: 2024-05-02 15:18:56.573347
-
-"""
-from alembic import op
-import sqlalchemy as sa
-import fastapi_users_db_sqlalchemy
-
-# revision identifiers, used by Alembic.
-revision = "7547d982db8f"
-down_revision = "ef7da92f7213"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "chat_folder",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column(
-            "user_id",
-            fastapi_users_db_sqlalchemy.generics.GUID(),
-            nullable=True,
-        ),
-        sa.Column("name", sa.String(), nullable=True),
-        sa.Column("display_priority", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["user_id"],
-            ["user.id"],
-        ),
-        sa.PrimaryKeyConstraint("id"),
-    )
-    op.add_column("chat_session", sa.Column("folder_id", sa.Integer(), nullable=True))
-    op.create_foreign_key(
-        "chat_session_chat_folder_fk",
-        "chat_session",
-        "chat_folder",
-        ["folder_id"],
-        ["id"],
-    )
-
-
-def downgrade() -> None:
-    op.drop_constraint(
-        "chat_session_chat_folder_fk", "chat_session", type_="foreignkey"
-    )
-    op.drop_column("chat_session", "folder_id")
-    op.drop_table("chat_folder")

backend/alembic/versions/767f1c2a00eb_count_chat_tokens.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "767f1c2a00eb"
 down_revision = "dba7f71618f5"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/76b60d407dfb_cc_pair_name_not_unique.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "76b60d407dfb"
 down_revision = "b156fa702355"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/776b3bbe9092_remove_remaining_enums.py

@@ -15,8 +15,8 @@ from danswer.search.models import SearchType
 # revision identifiers, used by Alembic.
 revision = "776b3bbe9092"
 down_revision = "4738e4b3bae1"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/77d07dffae64_forcibly_remove_more_enum_types_from_.py

@@ -12,8 +12,8 @@ from sqlalchemy import String
 # revision identifiers, used by Alembic.
 revision = "77d07dffae64"
 down_revision = "d61e513bef0a"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/78dbe7e38469_task_tracking.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "78dbe7e38469"
 down_revision = "7ccea01261f6"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/79acd316403a_add_api_key_table.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "79acd316403a"
 down_revision = "904e5138fffb"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7aea705850d5_added_slack_auto_filter.py

@@ -1,35 +0,0 @@
-"""added slack_auto_filter
-
-Revision ID: 7aea705850d5
-Revises: 4505fd7302e1
-Create Date: 2024-07-10 11:01:23.581015
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-revision = "7aea705850d5"
-down_revision = "4505fd7302e1"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "slack_bot_config",
-        sa.Column("enable_auto_filters", sa.Boolean(), nullable=True),
-    )
-    op.execute(
-        "UPDATE slack_bot_config SET enable_auto_filters = FALSE WHERE enable_auto_filters IS NULL"
-    )
-    op.alter_column(
-        "slack_bot_config",
-        "enable_auto_filters",
-        existing_type=sa.Boolean(),
-        nullable=False,
-        server_default=sa.false(),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("slack_bot_config", "enable_auto_filters")

backend/alembic/versions/7ccea01261f6_store_chat_retrieval_docs.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "7ccea01261f6"
 down_revision = "a570b80a5f20"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7da0ae5ad583_add_description_to_persona.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "7da0ae5ad583"
 down_revision = "e86866a9c78a"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7da543f5672f_add_slackbotconfig_table.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "7da543f5672f"
 down_revision = "febe9eaa0644"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7f726bad5367_slack_followup.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "7f726bad5367"
 down_revision = "79acd316403a"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/7f99be1cb9f5_add_index_for_getting_documents_just_by_.py

@@ -11,8 +11,8 @@ from alembic import op
 # revision identifiers, used by Alembic.
 revision = "7f99be1cb9f5"
 down_revision = "78dbe7e38469"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/800f48024ae9_add_id_to_connectorcredentialpair.py

@@ -12,8 +12,8 @@ from sqlalchemy.schema import Sequence, CreateSequence
 # revision identifiers, used by Alembic.
 revision = "800f48024ae9"
 down_revision = "767f1c2a00eb"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/80696cf850ae_add_chat_session_to_query_event.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "80696cf850ae"
 down_revision = "15326fcec57e"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/891cd83c87a8_add_is_visible_to_persona.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "891cd83c87a8"
 down_revision = "76b60d407dfb"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/8987770549c0_add_full_exception_stack_trace.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "8987770549c0"
 down_revision = "ec3ec2eabf7b"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/8aabb57f3b49_restructure_document_indices.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "8aabb57f3b49"
 down_revision = "5e84129c8be3"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/8e26726b7683_chat_context_addition.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "8e26726b7683"
 down_revision = "5809c0787398"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/904451035c9b_store_tool_details.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "904451035c9b"
 down_revision = "3b25685ff73c"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/904e5138fffb_tags.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "904e5138fffb"
 down_revision = "891cd83c87a8"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/91fd3b470d1a_remove_documentsource_from_tag.py

@@ -12,8 +12,8 @@ from danswer.configs.constants import DocumentSource
 # revision identifiers, used by Alembic.
 revision = "91fd3b470d1a"
 down_revision = "173cae5bba26"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/9d97fecfab7f_added_retrieved_docs_to_query_event.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "9d97fecfab7f"
 down_revision = "ffc707a226b4"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/a3bfd0d64902_add_chosen_assistants_to_user_table.py

@@ -1,27 +0,0 @@
-"""Add chosen_assistants to User table
-
-Revision ID: a3bfd0d64902
-Revises: ec85f2b3c544
-Create Date: 2024-05-26 17:22:24.834741
-
-"""
-from alembic import op
-import sqlalchemy as sa
-from sqlalchemy.dialects import postgresql
-
-# revision identifiers, used by Alembic.
-revision = "a3bfd0d64902"
-down_revision = "ec85f2b3c544"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "user",
-        sa.Column("chosen_assistants", postgresql.ARRAY(sa.Integer()), nullable=True),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("user", "chosen_assistants")

backend/alembic/versions/a570b80a5f20_usergroup_tables.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "a570b80a5f20"
 down_revision = "904451035c9b"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/ae62505e3acc_add_saml_accounts.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "ae62505e3acc"
 down_revision = "7da543f5672f"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/b082fec533f0_make_last_attempt_status_nullable.py

@@ -11,8 +11,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "b082fec533f0"
 down_revision = "df0c7ad8a076"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/b156fa702355_chat_reworked.py

@@ -15,8 +15,8 @@ from danswer.configs.constants import DocumentSource
 # revision identifiers, used by Alembic.
 revision = "b156fa702355"
 down_revision = "baf71f781b9e"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 searchtype_enum = ENUM(

backend/alembic/versions/b85f02ec1308_fix_file_type_migration.py

@@ -1,28 +0,0 @@
-"""fix-file-type-migration
-
-Revision ID: b85f02ec1308
-Revises: a3bfd0d64902
-Create Date: 2024-05-31 18:09:26.658164
-
-"""
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision = "b85f02ec1308"
-down_revision = "a3bfd0d64902"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.execute(
-        """
-        UPDATE file_store
-        SET file_origin = UPPER(file_origin)
-    """
-    )
-
-
-def downgrade() -> None:
-    # Let's not break anything on purpose :)
-    pass

backend/alembic/versions/b896bbd0d5a7_backfill_is_internet_data_to_false.py

@@ -1,23 +0,0 @@
-"""backfill is_internet data to False
-
-Revision ID: b896bbd0d5a7
-Revises: 44f856ae2a4a
-Create Date: 2024-07-16 15:21:05.718571
-
-"""
-from alembic import op
-
-
-# revision identifiers, used by Alembic.
-revision = "b896bbd0d5a7"
-down_revision = "44f856ae2a4a"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.execute("UPDATE search_doc SET is_internet = FALSE WHERE is_internet IS NULL")
-
-
-def downgrade() -> None:
-    pass

backend/alembic/versions/baf71f781b9e_add_llm_model_version_override_to_.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "baf71f781b9e"
 down_revision = "50b683a8295c"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/bc9771dccadf_create_usage_reports_table.py

@@ -1,51 +0,0 @@
-"""create usage reports table
-
-Revision ID: bc9771dccadf
-Revises: 0568ccf46a6b
-Create Date: 2024-06-18 10:04:26.800282
-
-"""
-from alembic import op
-import sqlalchemy as sa
-import fastapi_users_db_sqlalchemy
-
-# revision identifiers, used by Alembic.
-revision = "bc9771dccadf"
-down_revision = "0568ccf46a6b"
-
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "usage_reports",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("report_name", sa.String(), nullable=False),
-        sa.Column(
-            "requestor_user_id",
-            fastapi_users_db_sqlalchemy.generics.GUID(),
-            nullable=True,
-        ),
-        sa.Column(
-            "time_created",
-            sa.DateTime(timezone=True),
-            server_default=sa.text("now()"),
-            nullable=False,
-        ),
-        sa.Column("period_from", sa.DateTime(timezone=True), nullable=True),
-        sa.Column("period_to", sa.DateTime(timezone=True), nullable=True),
-        sa.ForeignKeyConstraint(
-            ["report_name"],
-            ["file_store.file_name"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["requestor_user_id"],
-            ["user.id"],
-        ),
-        sa.PrimaryKeyConstraint("id"),
-    )
-
-
-def downgrade() -> None:
-    op.drop_table("usage_reports")

backend/alembic/versions/c18cdf4b497e_add_standard_answer_tables.py

@@ -1,75 +0,0 @@
-"""Add standard_answer tables
-
-Revision ID: c18cdf4b497e
-Revises: 3a7802814195
-Create Date: 2024-06-06 15:15:02.000648
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "c18cdf4b497e"
-down_revision = "3a7802814195"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.create_table(
-        "standard_answer",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("keyword", sa.String(), nullable=False),
-        sa.Column("answer", sa.String(), nullable=False),
-        sa.Column("active", sa.Boolean(), nullable=False),
-        sa.PrimaryKeyConstraint("id"),
-        sa.UniqueConstraint("keyword"),
-    )
-    op.create_table(
-        "standard_answer_category",
-        sa.Column("id", sa.Integer(), nullable=False),
-        sa.Column("name", sa.String(), nullable=False),
-        sa.PrimaryKeyConstraint("id"),
-        sa.UniqueConstraint("name"),
-    )
-    op.create_table(
-        "standard_answer__standard_answer_category",
-        sa.Column("standard_answer_id", sa.Integer(), nullable=False),
-        sa.Column("standard_answer_category_id", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["standard_answer_category_id"],
-            ["standard_answer_category.id"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["standard_answer_id"],
-            ["standard_answer.id"],
-        ),
-        sa.PrimaryKeyConstraint("standard_answer_id", "standard_answer_category_id"),
-    )
-    op.create_table(
-        "slack_bot_config__standard_answer_category",
-        sa.Column("slack_bot_config_id", sa.Integer(), nullable=False),
-        sa.Column("standard_answer_category_id", sa.Integer(), nullable=False),
-        sa.ForeignKeyConstraint(
-            ["slack_bot_config_id"],
-            ["slack_bot_config.id"],
-        ),
-        sa.ForeignKeyConstraint(
-            ["standard_answer_category_id"],
-            ["standard_answer_category.id"],
-        ),
-        sa.PrimaryKeyConstraint("slack_bot_config_id", "standard_answer_category_id"),
-    )
-
-    op.add_column(
-        "chat_session", sa.Column("slack_thread_id", sa.String(), nullable=True)
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("chat_session", "slack_thread_id")
-
-    op.drop_table("slack_bot_config__standard_answer_category")
-    op.drop_table("standard_answer__standard_answer_category")
-    op.drop_table("standard_answer_category")
-    op.drop_table("standard_answer")

backend/alembic/versions/d5645c915d0e_remove_deletion_attempt_table.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "d5645c915d0e"
 down_revision = "8e26726b7683"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/d61e513bef0a_add_total_docs_for_index_attempt.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "d61e513bef0a"
 down_revision = "46625e4745d4"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/d7111c1238cd_remove_document_ids.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "d7111c1238cd"
 down_revision = "465f78d9b7f9"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/d716b0791ddd_combined_slack_id_fields.py

@@ -1,45 +0,0 @@
-"""combined slack id fields
-
-Revision ID: d716b0791ddd
-Revises: 7aea705850d5
-Create Date: 2024-07-10 17:57:45.630550
-
-"""
-from alembic import op
-
-# revision identifiers, used by Alembic.
-revision = "d716b0791ddd"
-down_revision = "7aea705850d5"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.execute(
-        """
-    UPDATE slack_bot_config
-    SET channel_config = jsonb_set(
-        channel_config,
-        '{respond_member_group_list}',
-        coalesce(channel_config->'respond_team_member_list', '[]'::jsonb) ||
-        coalesce(channel_config->'respond_slack_group_list', '[]'::jsonb)
-    ) - 'respond_team_member_list' - 'respond_slack_group_list'
-    """
-    )
-
-
-def downgrade() -> None:
-    op.execute(
-        """
-    UPDATE slack_bot_config
-    SET channel_config = jsonb_set(
-        jsonb_set(
-            channel_config - 'respond_member_group_list',
-            '{respond_team_member_list}',
-            '[]'::jsonb
-        ),
-        '{respond_slack_group_list}',
-        '[]'::jsonb
-    )
-    """
-    )

backend/alembic/versions/d929f0c1c6af_feedback_feature.py

@@ -13,8 +13,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "d929f0c1c6af"
 down_revision = "8aabb57f3b49"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/dba7f71618f5_danswer_custom_tool_flow.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "dba7f71618f5"
 down_revision = "d5645c915d0e"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/dbaa756c2ccf_embedding_models.py

@@ -19,8 +19,8 @@ from danswer.db.models import IndexModelStatus
 # revision identifiers, used by Alembic.
 revision = "dbaa756c2ccf"
 down_revision = "7f726bad5367"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/df0c7ad8a076_added_deletion_attempt_table.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "df0c7ad8a076"
 down_revision = "d7111c1238cd"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/e0a68a81d434_add_chat_feedback.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "e0a68a81d434"
 down_revision = "ae62505e3acc"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/e209dc5a8156_added_prune_frequency.py

@@ -1,22 +0,0 @@
-"""added-prune-frequency
-
-Revision ID: e209dc5a8156
-Revises: 48d14957fe80
-Create Date: 2024-06-16 16:02:35.273231
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-revision = "e209dc5a8156"
-down_revision = "48d14957fe80"
-branch_labels = None  # type: ignore
-depends_on = None  # type: ignore
-
-
-def upgrade() -> None:
-    op.add_column("connector", sa.Column("prune_freq", sa.Integer(), nullable=True))
-
-
-def downgrade() -> None:
-    op.drop_column("connector", "prune_freq")

backend/alembic/versions/e50154680a5c_no_source_enum.py

@@ -13,8 +13,8 @@ from danswer.configs.constants import DocumentSource
 # revision identifiers, used by Alembic.
 revision = "e50154680a5c"
 down_revision = "fcd135795f21"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/e6a4bbc13fe4_add_index_for_retrieving_latest_index_.py

@@ -11,8 +11,8 @@ from alembic import op
 # revision identifiers, used by Alembic.
 revision = "e6a4bbc13fe4"
 down_revision = "b082fec533f0"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/e86866a9c78a_add_persona_to_chat_session.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "e86866a9c78a"
 down_revision = "80696cf850ae"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/e91df4e935ef_private_personas_documentsets.py

@@ -12,8 +12,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "e91df4e935ef"
 down_revision = "91fd3b470d1a"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/ec3ec2eabf7b_index_from_beginning.py

@@ -11,8 +11,8 @@ import sqlalchemy as sa
 # revision identifiers, used by Alembic.
 revision = "ec3ec2eabf7b"
 down_revision = "dbaa756c2ccf"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/ec85f2b3c544_remove_last_attempt_status_from_cc_pair.py

@@ -1,31 +0,0 @@
-"""Remove Last Attempt Status from CC Pair
-
-Revision ID: ec85f2b3c544
-Revises: 3879338f8ba1
-Create Date: 2024-05-23 21:39:46.126010
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-# revision identifiers, used by Alembic.
-revision = "ec85f2b3c544"
-down_revision = "70f00c45c0f2"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.drop_column("connector_credential_pair", "last_attempt_status")
-
-
-def downgrade() -> None:
-    op.add_column(
-        "connector_credential_pair",
-        sa.Column(
-            "last_attempt_status",
-            sa.VARCHAR(),
-            autoincrement=False,
-            nullable=True,
-        ),
-    )

backend/alembic/versions/ecab2b3f1a3b_add_overrides_to_the_chat_session.py

@@ -12,8 +12,8 @@ from sqlalchemy.dialects import postgresql
 # revision identifiers, used by Alembic.
 revision = "ecab2b3f1a3b"
 down_revision = "38eda64af7fe"
-branch_labels: None = None
-depends_on: None = None
+branch_labels = None
+depends_on = None
 
 
 def upgrade() -> None:

backend/alembic/versions/ef7da92f7213_add_files_to_chatmessage.py

@@ -1,27 +0,0 @@
-"""Add files to ChatMessage
-
-Revision ID: ef7da92f7213
-Revises: 401c1ac29467
-Create Date: 2024-04-28 16:59:33.199153
-
-"""
-from alembic import op
-import sqlalchemy as sa
-from sqlalchemy.dialects import postgresql
-
-# revision identifiers, used by Alembic.
-revision = "ef7da92f7213"
-down_revision = "401c1ac29467"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "chat_message",
-        sa.Column("files", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("chat_message", "files")

backend/alembic/versions/f1c6478c3fd8_add_pre_defined_feedback.py

@@ -1,25 +0,0 @@
-"""Add pre-defined feedback
-
-Revision ID: f1c6478c3fd8
-Revises: 643a84a42a33
-Create Date: 2024-05-09 18:11:49.210667
-
-"""
-from alembic import op
-import sqlalchemy as sa
-
-revision = "f1c6478c3fd8"
-down_revision = "643a84a42a33"
-branch_labels: None = None
-depends_on: None = None
-
-
-def upgrade() -> None:
-    op.add_column(
-        "chat_feedback",
-        sa.Column("predefined_feedback", sa.String(), nullable=True),
-    )
-
-
-def downgrade() -> None:
-    op.drop_column("chat_feedback", "predefined_feedback")