浏览代码

feat: add retry mechanism for zhipuai (#5926)

tags/0.6.13
orangeclk 1年前
父节点
当前提交
f8aaa57f31
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3
    0
      api/core/model_runtime/model_providers/zhipuai/zhipuai_sdk/core/_http_client.py

+ 3
- 0
api/core/model_runtime/model_providers/zhipuai/zhipuai_sdk/core/_http_client.py 查看文件

@@ -7,6 +7,8 @@ from typing import Any, Union, cast
import httpx
import pydantic
from httpx import URL, Timeout
from tenacity import retry
from tenacity.stop import stop_after_attempt

from . import _errors
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):
self.close()

@retry(stop=stop_after_attempt(ZHIPUAI_DEFAULT_MAX_RETRIES))
def request(
self,
*,

正在加载...
取消
保存