浏览代码

Feat/api tool custom timeout (#3420)

tags/0.6.3
Yeuoly 1年前
父节点
当前提交
ae1f3780f8
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 9 次插入1 次删除
  1. 4
    0
      api/.env.example
  2. 5
    1
      api/core/tools/tool/api_tool.py

+ 4
- 0
api/.env.example 查看文件

@@ -149,3 +149,7 @@ TEMPLATE_TRANSFORM_MAX_LENGTH=80000
CODE_MAX_STRING_ARRAY_LENGTH=30
CODE_MAX_OBJECT_ARRAY_LENGTH=30
CODE_MAX_NUMBER_ARRAY_LENGTH=1000

# API Tool configuration
API_TOOL_DEFAULT_CONNECT_TIMEOUT=10
API_TOOL_DEFAULT_READ_TIMEOUT=60

+ 5
- 1
api/core/tools/tool/api_tool.py 查看文件

@@ -1,5 +1,6 @@
import json
from json import dumps
from os import getenv
from typing import Any, Union
from urllib.parse import urlencode

@@ -13,7 +14,10 @@ from core.tools.entities.user_entities import UserToolProvider
from core.tools.errors import ToolInvokeError, ToolParameterValidationError, ToolProviderCredentialValidationError
from core.tools.tool.tool import Tool

API_TOOL_DEFAULT_TIMEOUT = (10, 60)
API_TOOL_DEFAULT_TIMEOUT = (
int(getenv('API_TOOL_DEFAULT_CONNECT_TIMEOUT', '10')),
int(getenv('API_TOOL_DEFAULT_READ_TIMEOUT', '60'))
)

class ApiTool(Tool):
api_bundle: ApiBasedToolBundle

正在加载...
取消
保存