Procházet zdrojové kódy

Feat/api tool custom timeout (#3420)

tags/0.6.3
Yeuoly před 1 rokem
rodič
revize
ae1f3780f8
Žádný účet není propojen s e-mailovou adresou tvůrce revize
2 změnil soubory, kde provedl 9 přidání a 1 odebrání
  1. 4
    0
      api/.env.example
  2. 5
    1
      api/core/tools/tool/api_tool.py

+ 4
- 0
api/.env.example Zobrazit soubor

CODE_MAX_STRING_ARRAY_LENGTH=30 CODE_MAX_STRING_ARRAY_LENGTH=30
CODE_MAX_OBJECT_ARRAY_LENGTH=30 CODE_MAX_OBJECT_ARRAY_LENGTH=30
CODE_MAX_NUMBER_ARRAY_LENGTH=1000 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 Zobrazit soubor

import json import json
from json import dumps from json import dumps
from os import getenv
from typing import Any, Union from typing import Any, Union
from urllib.parse import urlencode from urllib.parse import urlencode


from core.tools.errors import ToolInvokeError, ToolParameterValidationError, ToolProviderCredentialValidationError from core.tools.errors import ToolInvokeError, ToolParameterValidationError, ToolProviderCredentialValidationError
from core.tools.tool.tool import Tool 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): class ApiTool(Tool):
api_bundle: ApiBasedToolBundle api_bundle: ApiBasedToolBundle

Načítá se…
Zrušit
Uložit