From 2e9f937ce63883e490e593a74c2b2c3504ac28e4 Mon Sep 17 00:00:00 2001
From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com>
Date: Wed, 15 Apr 2026 03:21:46 +0000
Subject: [PATCH] [autofix.ci] apply automated fixes

---
 .../vdb/tidb_on_qdrant/tidb_service.py        | 19 +++++++++++++------
 .../tidb_on_qdrant_vector.py                  |  6 ++++--
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py b/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
index 8b425164e4..40f6794af2 100644
--- a/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
+++ b/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
@@ -109,7 +109,9 @@ class TidbService:
                     qdrant_endpoint = TidbService.fetch_qdrant_endpoint(api_url, public_key, private_key, cluster_id)
                     logger.info(
                         "Cluster %s is ACTIVE, user_prefix=%s, qdrant_endpoint=%s",
-                        cluster_id, user_prefix, qdrant_endpoint,
+                        cluster_id,
+                        user_prefix,
+                        qdrant_endpoint,
                     )
                     return {
                         "cluster_id": cluster_id,
@@ -118,7 +120,13 @@ class TidbService:
                         "password": password,
                         "qdrant_endpoint": qdrant_endpoint,
                     }
-                logger.info("Cluster %s state=%s, retry %d/%d", cluster_id, cluster_response["state"], retry_count + 1, max_retries)
+                logger.info(
+                    "Cluster %s state=%s, retry %d/%d",
+                    cluster_id,
+                    cluster_response["state"],
+                    retry_count + 1,
+                    max_retries,
+                )
                 time.sleep(30)
                 retry_count += 1
             logger.error("Cluster %s did not become ACTIVE after %d retries", cluster_id, max_retries)
@@ -295,12 +303,11 @@ class TidbService:
                 if not cached_password:
                     logger.warning("No cached password for cluster %s, skipping", item["displayName"])
                     continue
-                qdrant_endpoint = TidbService.fetch_qdrant_endpoint(
-                    api_url, public_key, private_key, item["clusterId"]
-                )
+                qdrant_endpoint = TidbService.fetch_qdrant_endpoint(api_url, public_key, private_key, item["clusterId"])
                 logger.info(
                     "Batch cluster %s: qdrant_endpoint=%s",
-                    item["clusterId"], qdrant_endpoint,
+                    item["clusterId"],
+                    qdrant_endpoint,
                 )
                 cluster_info = {
                     "cluster_id": item["clusterId"],
diff --git a/api/providers/vdb/vdb-tidb-on-qdrant/src/dify_vdb_tidb_on_qdrant/tidb_on_qdrant_vector.py b/api/providers/vdb/vdb-tidb-on-qdrant/src/dify_vdb_tidb_on_qdrant/tidb_on_qdrant_vector.py
index a7b14b84e7..abca55f540 100644
--- a/api/providers/vdb/vdb-tidb-on-qdrant/src/dify_vdb_tidb_on_qdrant/tidb_on_qdrant_vector.py
+++ b/api/providers/vdb/vdb-tidb-on-qdrant/src/dify_vdb_tidb_on_qdrant/tidb_on_qdrant_vector.py
@@ -445,7 +445,8 @@ class TidbOnQdrantVectorFactory(AbstractVectorFactory):
                     if idle_tidb_auth_binding:
                         logger.info(
                             "Assigning idle cluster %s to tenant %s",
-                            idle_tidb_auth_binding.cluster_id, dataset.tenant_id,
+                            idle_tidb_auth_binding.cluster_id,
+                            dataset.tenant_id,
                         )
                         idle_tidb_auth_binding.active = True
                         idle_tidb_auth_binding.tenant_id = dataset.tenant_id
@@ -464,7 +465,8 @@ class TidbOnQdrantVectorFactory(AbstractVectorFactory):
                         )
                         logger.info(
                             "New cluster created: cluster_id=%s, qdrant_endpoint=%s",
-                            new_cluster["cluster_id"], new_cluster.get("qdrant_endpoint"),
+                            new_cluster["cluster_id"],
+                            new_cluster.get("qdrant_endpoint"),
                         )
                         new_tidb_auth_binding = TidbAuthBinding(
                             cluster_id=new_cluster["cluster_id"],