Skip to content

Commit b0562a1

Browse files
author
liuhuiqi.7
committed
feat(arkruntime): copy msg
1 parent 64160af commit b0562a1

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

volcenginesdkarkruntime/resources/batch_chat/completions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from ..._exceptions import ArkAPITimeoutError, ArkAPIConnectionError, ArkAPIStatusError
1414
from ..._types import Body, Query, Headers
15-
from ..._utils._utils import with_sts_token, async_with_sts_token
15+
from ..._utils._utils import with_sts_token, async_with_sts_token, deepcopy_minimal
1616
from ..._utils._key_agreement import aes_gcm_decrypt_base64_string
1717
from ..._base_client import make_request_options
1818
from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -202,6 +202,7 @@ def create(
202202
and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
203203
):
204204
is_encrypt = True
205+
messages = deepcopy_minimal(messages)
205206
e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
206207
retryTimes = 0
207208
last_time = self._get_request_last_time(timeout)
@@ -353,6 +354,7 @@ async def create(
353354
and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
354355
):
355356
is_encrypt = True
357+
messages = deepcopy_minimal(messages)
356358
e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
357359

358360
retryTimes = 0

volcenginesdkarkruntime/resources/chat/completions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
from typing_extensions import Literal
1717

1818
from ..._types import Body, Query, Headers
19-
from ..._utils._utils import with_sts_token, async_with_sts_token
19+
from ..._utils._utils import deepcopy_minimal, with_sts_token, async_with_sts_token
2020
from ..._utils._key_agreement import aes_gcm_decrypt_base64_string
2121
from ..._base_client import make_request_options
2222
from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -179,6 +179,7 @@ def create(
179179
and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
180180
):
181181
is_encrypt = True
182+
messages = deepcopy_minimal(messages)
182183
e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
183184

184185
resp = self._post(
@@ -327,6 +328,7 @@ async def create(
327328
and extra_headers.get(ARK_E2E_ENCRYPTION_HEADER, None) == "true"
328329
):
329330
is_encrypt = True
331+
messages = deepcopy_minimal(messages)
330332
e2e_key, e2e_nonce = self._encrypt(model, messages, extra_headers)
331333

332334
resp = await self._post(

0 commit comments

Comments
 (0)