Skip to content

Commit 2e285e8

Browse files
committed
fix: misc
1 parent 6472e93 commit 2e285e8

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

volcenginesdkarkruntime/_exceptions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def __init__(
123123

124124

125125
class ArkAPITimeoutError(ArkAPIConnectionError):
126-
def __init__(self, request: Optional[httpx.Request] = None, request_id: Optional[str] = None) -> None:
126+
def __init__(self, request: httpx.Request, request_id: str) -> None:
127127
super().__init__(
128128
message="Request timed out.", request=request, request_id=request_id
129129
)

volcenginesdkarkruntime/resources/batch_chat/completions.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def create(
182182
while True:
183183
model_breaker.wait()
184184
if datetime.now() > last_time:
185-
raise ArkAPITimeoutError()
185+
raise ArkAPITimeoutError(None, None)
186186
try:
187187
resp = self._post_without_retry(
188188
"/batch/chat/completions",
@@ -219,7 +219,7 @@ def create(
219219
except (ArkAPITimeoutError, ArkAPIConnectionError):
220220
waitTime = _calculate_retry_timeout(retryTimes)
221221
if datetime.now() + timedelta(seconds=waitTime) > last_time:
222-
raise ArkAPITimeoutError()
222+
raise ArkAPITimeoutError(None, None)
223223
time.sleep(waitTime)
224224
retryTimes = retryTimes + 1
225225
continue
@@ -327,7 +327,7 @@ async def create(
327327
while True:
328328
await model_breaker.asyncwait()
329329
if datetime.now() > last_time:
330-
raise ArkAPITimeoutError()
330+
raise ArkAPITimeoutError(None, None)
331331
try:
332332
resp = await self._post_without_retry(
333333
"/batch/chat/completions",
@@ -364,7 +364,7 @@ async def create(
364364
except (ArkAPITimeoutError, ArkAPIConnectionError):
365365
waitTime = _calculate_retry_timeout(retryTimes)
366366
if datetime.now() + timedelta(seconds=waitTime) > last_time:
367-
raise ArkAPITimeoutError()
367+
raise ArkAPITimeoutError(None, None)
368368
await asyncio.sleep(waitTime)
369369
retryTimes = retryTimes + 1
370370
continue

0 commit comments

Comments
 (0)