diff --git a/api/tests/unit_tests/oss/tencent_cos/test_tencent_cos.py b/api/tests/unit_tests/oss/tencent_cos/test_tencent_cos.py
index 694ded0c89..a0fed1aa14 100644
--- a/api/tests/unit_tests/oss/tencent_cos/test_tencent_cos.py
+++ b/api/tests/unit_tests/oss/tencent_cos/test_tencent_cos.py
@@ -36,9 +36,13 @@ class TestTencentCosConfiguration:
         mock_config_instance = MagicMock()
         mock_client = MagicMock()
 
-        with patch("extensions.storage.tencent_cos_storage.dify_config", mock_dify_config), patch(
-            "extensions.storage.tencent_cos_storage.CosConfig", return_value=mock_config_instance
-        ) as mock_cos_config, patch("extensions.storage.tencent_cos_storage.CosS3Client", return_value=mock_client):
+        with (
+            patch("extensions.storage.tencent_cos_storage.dify_config", mock_dify_config),
+            patch(
+                "extensions.storage.tencent_cos_storage.CosConfig", return_value=mock_config_instance
+            ) as mock_cos_config,
+            patch("extensions.storage.tencent_cos_storage.CosS3Client", return_value=mock_client),
+        ):
             TencentCosStorage()
 
             # Verify CosConfig was called with Domain parameter (not Region)
@@ -65,9 +69,13 @@ class TestTencentCosConfiguration:
         mock_config_instance = MagicMock()
         mock_client = MagicMock()
 
-        with patch("extensions.storage.tencent_cos_storage.dify_config", mock_dify_config), patch(
-            "extensions.storage.tencent_cos_storage.CosConfig", return_value=mock_config_instance
-        ) as mock_cos_config, patch("extensions.storage.tencent_cos_storage.CosS3Client", return_value=mock_client):
+        with (
+            patch("extensions.storage.tencent_cos_storage.dify_config", mock_dify_config),
+            patch(
+                "extensions.storage.tencent_cos_storage.CosConfig", return_value=mock_config_instance
+            ) as mock_cos_config,
+            patch("extensions.storage.tencent_cos_storage.CosS3Client", return_value=mock_client),
+        ):
             TencentCosStorage()
 
             # Verify CosConfig was called with Region parameter (not Domain)
diff --git a/docker/docker-compose.yaml b/docker/docker-compose.yaml
index 964b9fe724..bbfcc973f8 100644
--- a/docker/docker-compose.yaml
+++ b/docker/docker-compose.yaml
@@ -140,6 +140,7 @@ x-shared-env: &shared-api-worker-env
   TENCENT_COS_SECRET_ID: ${TENCENT_COS_SECRET_ID:-your-secret-id}
   TENCENT_COS_REGION: ${TENCENT_COS_REGION:-your-region}
   TENCENT_COS_SCHEME: ${TENCENT_COS_SCHEME:-your-scheme}
+  TENCENT_COS_CUSTOM_DOMAIN: ${TENCENT_COS_CUSTOM_DOMAIN:-your-custom-domain}
   OCI_ENDPOINT: ${OCI_ENDPOINT:-https://your-object-storage-namespace.compat.objectstorage.us-ashburn-1.oraclecloud.com}
   OCI_BUCKET_NAME: ${OCI_BUCKET_NAME:-your-bucket-name}
   OCI_ACCESS_KEY: ${OCI_ACCESS_KEY:-your-access-key}