瀏覽代碼

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

Loading…
取消
儲存