jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add retry mechanism for zhipuai (#5926)

Browse source
This commit is contained in:
orangeclk 2024-07-05 10:49:18 +08:00 • committed by GitHub
commit f8aaa57f31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/core/model_runtime/model_providers/zhipuai/zhipuai_sdk/core/_http_client.py
View file

@ -7,6 +7,8 @@ from typing import Any, Union, cast
import httpx import httpx
import pydantic import pydantic
from httpx import URL, Timeout from httpx import URL, Timeout
from tenacity import retry
from tenacity.stop import stop_after_attempt
from . import _errors from . import _errors
from ._base_type import NOT_GIVEN, Body, Data, Headers, NotGiven, Query, RequestFiles, ResponseT from ._base_type import NOT_GIVEN, Body, Data, Headers, NotGiven, Query, RequestFiles, ResponseT
@ -221,6 +223,7 @@ class HttpClient:
def __exit__(self, exc_type, exc_val, exc_tb): def __exit__(self, exc_type, exc_val, exc_tb):
self.close() self.close()
@retry(stop=stop_after_attempt(ZHIPUAI_DEFAULT_MAX_RETRIES))
def request( def request(
self, self,
*, *,