Merge branch 'feat/ark_encryption_error' into 'integration_2025-07-10_995600327426'

feat: [development task] ark runtime (1438955)

See merge request iaasng/volcengine-python-sdk!708

Author: BitsAdmin

volcenginesdkarkruntime/_client.py

@@ -492,7 +492,14 @@ def _init_local_cert_cache(self):
         self._cert_expiration_seconds = 14 * 24 * 60 * 60  # 14 days
 
         if not os.path.exists(self._cert_storage_path):
-            os.makedirs(self._cert_storage_path)
+            try:
+                os.makedirs(self._cert_storage_path)
+            except FileExistsError:
+                pass
+            except Exception as e:
+                raise ArkAPIError(
+                    "failed to create certificate directory %s: %s\n" % (self._cert_storage_path, e)
+                )
 
     def get(self, ep: str) -> key_agreement_client:
         if ep not in self._certificate_manager:

volcenginesdkarkruntime/resources/batch_chat/completions.py

@@ -12,7 +12,7 @@
 
 from ..._exceptions import ArkAPITimeoutError, ArkAPIConnectionError, ArkAPIStatusError
 from ..._types import Body, Query, Headers
-from ..._utils._utils import with_sts_token, async_with_sts_token
+from ..._utils._utils import with_sts_token, async_with_sts_token, deepcopy_minimal
 from ..._utils._key_agreement import aes_gcm_decrypt_base64_string
 from ..._base_client import make_request_options
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -202,6 +202,7 @@ def create(
             and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
         ):
             is_encrypt = True
+            messages = deepcopy_minimal(messages)
             e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
         retryTimes = 0
         last_time = self._get_request_last_time(timeout)
@@ -353,6 +354,7 @@ async def create(
             and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
         ):
             is_encrypt = True
+            messages = deepcopy_minimal(messages)
             e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
 
         retryTimes = 0

volcenginesdkarkruntime/resources/chat/completions.py

@@ -16,7 +16,7 @@
 from typing_extensions import Literal
 
 from ..._types import Body, Query, Headers
-from ..._utils._utils import with_sts_token, async_with_sts_token
+from ..._utils._utils import deepcopy_minimal, with_sts_token, async_with_sts_token
 from ..._utils._key_agreement import aes_gcm_decrypt_base64_string
 from ..._base_client import make_request_options
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -179,6 +179,7 @@ def create(
             and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
         ):
             is_encrypt = True
+            messages = deepcopy_minimal(messages)
             e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
 
         resp = self._post(
@@ -327,6 +328,7 @@ async def create(
             and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
         ):
             is_encrypt = True
+            messages = deepcopy_minimal(messages)
             e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
 
         resp = await self._post(