black format again

Author: Richard Kuo (Onyx)

Diff for: backend/alembic_tenants/versions/34e3630c7f32_lowercase_multi_tenant_user_auth.py

@@ -5,6 +5,7 @@
 Create Date: 2025-02-26 15:03:01.211894
 
 """
+
 from alembic import op
 
 

Diff for: backend/alembic_tenants/versions/3b45e0018bf1_add_new_available_tenant_table.py

@@ -5,6 +5,7 @@
 Create Date: 2025-03-06 09:55:18.229910
 
 """
+
 import sqlalchemy as sa
 
 from alembic import op

Diff for: backend/alembic_tenants/versions/a4f6ee863c47_mapping_for_anonymous_user_path.py

@@ -5,6 +5,7 @@
 Create Date: 2025-01-04 14:16:58.697451
 
 """
+
 import sqlalchemy as sa
 
 from alembic import op

Diff for: backend/alembic_tenants/versions/ac842f85f932_new_column_user_tenant_mapping.py

@@ -5,6 +5,7 @@
 Create Date: 2025-03-03 13:30:14.802874
 
 """
+
 import sqlalchemy as sa
 
 from alembic import op

Diff for: backend/ee/onyx/external_permissions/confluence/doc_sync.py

@@ -2,6 +2,7 @@
 Rules defined here:
 https://confluence.atlassian.com/conf85/check-who-can-view-a-page-1283360557.html
 """
+
 from collections.abc import Generator
 from typing import Any
 

Diff for: backend/ee/onyx/external_permissions/slack/group_sync.py

@@ -4,6 +4,7 @@
 SO WHEN CHECKING IF A USER CAN ACCESS A DOCUMENT, WE ONLY NEED TO CHECK THEIR EMAIL
 THERE IS NO USERGROUP <-> DOCUMENT PERMISSION MAPPING
 """
+
 from slack_sdk import WebClient
 
 from ee.onyx.db.external_perm import ExternalUserGroup

Diff for: backend/onyx/background/celery/tasks/tenant_provisioning/tasks.py

@@ -1,6 +1,7 @@
 """
 Periodic tasks for tenant pre-provisioning.
 """
+
 import asyncio
 import datetime
 import uuid

Diff for: backend/onyx/background/celery/versioned_apps/beat.py

@@ -1,4 +1,5 @@
 """Factory stub for running celery worker / celery beat."""
+
 from celery import Celery
 
 from onyx.background.celery.apps.beat import celery_app

Diff for: backend/onyx/background/celery/versioned_apps/client.py

@@ -4,6 +4,7 @@
 
 This is an app stub purely for sending tasks as a client.
 """
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import set_is_ee_based_on_env_variable

Diff for: backend/onyx/background/celery/versioned_apps/heavy.py

@@ -1,6 +1,7 @@
 """Factory stub for running celery worker / celery beat.
 This code is different from the primary/beat stubs because there is no EE version to
 fetch. Port over the code in those files if we add an EE version of this worker."""
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import set_is_ee_based_on_env_variable

Diff for: backend/onyx/background/celery/versioned_apps/indexing.py

@@ -1,6 +1,7 @@
 """Factory stub for running celery worker / celery beat.
 This code is different from the primary/beat stubs because there is no EE version to
 fetch. Port over the code in those files if we add an EE version of this worker."""
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import set_is_ee_based_on_env_variable

Diff for: backend/onyx/background/celery/versioned_apps/light.py

@@ -1,6 +1,7 @@
 """Factory stub for running celery worker / celery beat.
 This code is different from the primary/beat stubs because there is no EE version to
 fetch. Port over the code in those files if we add an EE version of this worker."""
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import set_is_ee_based_on_env_variable

Diff for: backend/onyx/background/celery/versioned_apps/monitoring.py

@@ -1,4 +1,5 @@
 """Factory stub for running celery worker / celery beat."""
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import set_is_ee_based_on_env_variable

Diff for: backend/onyx/background/celery/versioned_apps/primary.py

@@ -1,4 +1,5 @@
 """Factory stub for running celery worker / celery beat."""
+
 from celery import Celery
 
 from onyx.utils.variable_functionality import fetch_versioned_implementation

Diff for: backend/onyx/background/indexing/job_client.py

@@ -4,6 +4,7 @@
 
 NOTE: cannot use Celery directly due to
 https://github.com/celery/celery/issues/7007#issuecomment-1740139367"""
+
 import multiprocessing as mp
 import sys
 import traceback

Diff for: backend/onyx/connectors/onyx_jira/utils.py

@@ -1,4 +1,5 @@
 """Module with custom fields processing functions"""
+
 import os
 from typing import Any
 from typing import List

Diff for: backend/onyx/connectors/xenforo/connector.py

@@ -10,6 +10,7 @@
 The `load_from_state` method is used to load documents from the forum. It takes an optional `state` parameter, which
 can be used to specify a state from which to start loading documents.
 """
+
 import re
 from datetime import datetime
 from datetime import timedelta

Diff for: backend/onyx/context/search/enums.py

@@ -1,6 +1,7 @@
 """NOTE: this needs to be separate from models.py because of circular imports.
 Both search/models.py and db/models.py import enums from this file AND
 search/models.py imports from db/models.py."""
+
 from enum import Enum
 
 

Diff for: backend/onyx/file_processing/file_validation.py

@@ -1,6 +1,7 @@
 """
 Centralized file type validation utilities.
 """
+
 # Standard image MIME types supported by most vision LLMs
 IMAGE_MIME_TYPES = [
     "image/png",

Diff for: backend/onyx/llm/override_models.py

@@ -3,6 +3,7 @@
 NOTE: these models are used in many places, so have to be
 kepy in a separate file to avoid circular imports.
 """
+
 from pydantic import BaseModel
 
 

Diff for: backend/onyx/utils/error_handling.py

@@ -1,6 +1,7 @@
 """
 Standardized error handling utilities.
 """
+
 from onyx.configs.app_configs import CONTINUE_ON_CONNECTOR_FAILURE
 from onyx.utils.logger import setup_logger
 

Diff for: backend/scripts/chat_loadtest.py

@@ -8,6 +8,7 @@
 
 For more options, checkout the bottom of the file.
 """
+
 import argparse
 import asyncio
 import logging

Diff for: backend/scripts/debugging/onyx_vespa.py

@@ -27,6 +27,7 @@
   python vespa_debug_tool.py --action list_docs --tenant-id my_tenant --connector-id 1 --n 5
   python vespa_debug_tool.py --action list_connector --tenant-id my_tenant --cc-pair-id 1 --n 5
 """
+
 import argparse
 import json
 from datetime import datetime

Diff for: backend/scripts/query_time_check/seed_dummy_docs.py

@@ -8,6 +8,7 @@
 Run this script to seed the database with dummy documents.
 Then run test_query_times.py to test query times.
 """
+
 import random
 from datetime import datetime
 

Diff for: backend/scripts/query_time_check/test_query_times.py

@@ -1,6 +1,7 @@
 """
 RUN THIS AFTER SEED_DUMMY_DOCS.PY
 """
+
 import random
 import time
 

Diff for: backend/tests/integration/connector_job_tests/slack/slack_api_utils.py

@@ -8,6 +8,7 @@
     - chat:write
     - chat:write.public
 """
+
 from typing import Any
 from uuid import uuid4
 

Diff for: backend/tests/integration/tests/connector/test_connector_deletion.py

@@ -5,6 +5,7 @@
     - updates the document sets and user groups to remove the connector
 - Ensure that deleting a connector that is part of an overlapping document set and/or user group works as expected
 """
+
 import os
 from uuid import uuid4
 

Diff for: backend/tests/integration/tests/permissions/test_cc_pair_permissions.py

@@ -2,6 +2,7 @@
 This file takes the happy path to adding a curator to a user group and then tests
 the permissions of the curator manipulating connector-credential pairs.
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/permissions/test_connector_permissions.py

@@ -2,6 +2,7 @@
 This file takes the happy path to adding a curator to a user group and then tests
 the permissions of the curator manipulating connectors.
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/permissions/test_credential_permissions.py

@@ -2,6 +2,7 @@
 This file takes the happy path to adding a curator to a user group and then tests
 the permissions of the curator manipulating credentials.
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/permissions/test_persona_permissions.py

@@ -4,6 +4,7 @@
 - Curators can edit personas that belong exclusively to groups they curate
 - Admins can edit all personas
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/permissions/test_user_role_permissions.py

@@ -1,6 +1,7 @@
 """
 This file tests the ability of different user types to set the role of other users.
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/permissions/test_whole_curator_flow.py

@@ -1,6 +1,7 @@
 """
 This test tests the happy path for curator permissions
 """
+
 import os
 
 import pytest

Diff for: backend/tests/integration/tests/usergroup/test_user_group_deletion.py

@@ -8,6 +8,7 @@
 - token_rate_limit (Not Implemented)
 - persona
 """
+
 import os
 
 import pytest