Skip to content

Commit d8f3ae3

Browse files
committed
Merge 'fix/arkruntime/error_format' into 'feat/support_json_schema'
fix: optional positional fields for ark api timeout error See merge request: !624
2 parents 57cb2ef + 2e285e8 commit d8f3ae3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

volcenginesdkarkruntime/resources/batch_chat/completions.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def create(
204204
while True:
205205
model_breaker.wait()
206206
if datetime.now() > last_time:
207-
raise ArkAPITimeoutError()
207+
raise ArkAPITimeoutError(None, None)
208208
try:
209209
resp = self._post_without_retry(
210210
"/batch/chat/completions",
@@ -241,7 +241,7 @@ def create(
241241
except (ArkAPITimeoutError, ArkAPIConnectionError):
242242
waitTime = _calculate_retry_timeout(retryTimes)
243243
if datetime.now() + timedelta(seconds=waitTime) > last_time:
244-
raise ArkAPITimeoutError()
244+
raise ArkAPITimeoutError(None, None)
245245
time.sleep(waitTime)
246246
retryTimes = retryTimes + 1
247247
continue
@@ -349,7 +349,7 @@ async def create(
349349
while True:
350350
await model_breaker.asyncwait()
351351
if datetime.now() > last_time:
352-
raise ArkAPITimeoutError()
352+
raise ArkAPITimeoutError(None, None)
353353
try:
354354
resp = await self._post_without_retry(
355355
"/batch/chat/completions",
@@ -386,7 +386,7 @@ async def create(
386386
except (ArkAPITimeoutError, ArkAPIConnectionError):
387387
waitTime = _calculate_retry_timeout(retryTimes)
388388
if datetime.now() + timedelta(seconds=waitTime) > last_time:
389-
raise ArkAPITimeoutError()
389+
raise ArkAPITimeoutError(None, None)
390390
await asyncio.sleep(waitTime)
391391
retryTimes = retryTimes + 1
392392
continue

0 commit comments

Comments
 (0)