138
138
from onyx .server .models import StatusResponse
139
139
from onyx .utils .logger import setup_logger
140
140
from onyx .utils .telemetry import create_milestone_and_report
141
+ from onyx .utils .threadpool_concurrency import CallableProtocol
141
142
from onyx .utils .threadpool_concurrency import run_functions_tuples_in_parallel
142
143
from onyx .utils .variable_functionality import fetch_ee_implementation_or_noop
143
144
from shared_configs .contextvars import get_current_tenant_id
@@ -889,16 +890,6 @@ def get_connector_indexing_status_with_pagination(
889
890
) -> list [ConnectorIndexingStatusLiteResponse ]:
890
891
tenant_id = get_current_tenant_id ()
891
892
892
- if MOCK_CONNECTOR_FILE_PATH :
893
- import json
894
-
895
- with open (MOCK_CONNECTOR_FILE_PATH , "r" ) as f :
896
- raw_data = json .load (f )
897
- connector_indexing_statuses = [
898
- ConnectorIndexingStatus (** status ) for status in raw_data
899
- ]
900
- return connector_indexing_statuses
901
-
902
893
# NOTE: If the connector is deleting behind the scenes,
903
894
# accessing cc_pairs can be inconsistent and members like
904
895
# connector or credential may be None.
@@ -910,7 +901,7 @@ def get_connector_indexing_status_with_pagination(
910
901
# sqlalchemy-method-connection-for-bind-is-already-in-progress
911
902
# for why we can't pass in the current db_session to these functions
912
903
913
- parallel_functions = [
904
+ parallel_functions : list [ tuple [ CallableProtocol , tuple [ Any , ...]]] = [
914
905
# Get editable connector/credential pairs
915
906
(
916
907
get_connector_credential_pairs_for_user_parallel ,
0 commit comments