yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-02 01:49:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add timeout for http tools (#6818)

Browse Source 
Co-authored-by: Victor Dibia <victordibia@microsoft.com>
This commit is contained in:
lo5twind 2025-07-17 04:06:06 +08:00 committed by GitHub
parent b2cef7f47c
commit aa131bb7f7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
python/packages/autogen-ext/src/autogen_ext/tools/http/_http_tool.py
View File

@ -8,6 +8,8 @@ from json_schema_to_pydantic import create_model
from pydantic import BaseModel, Field
from typing_extensions import Self
DEFAULT_TIMEOUT_CONFIG = 5.0
class HttpToolConfig(BaseModel):
name: str
@ -53,6 +55,10 @@ class HttpToolConfig(BaseModel):
"""
The type of response to return from the tool.
"""
timeout: float = DEFAULT_TIMEOUT_CONFIG
"""
The timeout for the tool request in seconds.
"""
class HttpTool(BaseTool[BaseModel, Any], Component[HttpToolConfig]):
@ -73,6 +79,8 @@ class HttpTool(BaseTool[BaseModel, Any], Component[HttpToolConfig]):
Path parameters must also be included in the schema and must be strings.
return_type (Literal["text", "json"], optional): The type of response to return from the tool.
Defaults to "text".
timeout (float, optional): The timeout for HTTP requests in seconds.
Defaults to 5.0.
.. note::
This tool requires the :code:`http-tool` extra for the :code:`autogen-ext` package.
@ -148,6 +156,7 @@ class HttpTool(BaseTool[BaseModel, Any], Component[HttpToolConfig]):
scheme: Literal["http", "https"] = "http",
method: Literal["GET", "POST", "PUT", "DELETE", "PATCH"] = "POST",
return_type: Literal["text", "json"] = "text",
timeout: float = DEFAULT_TIMEOUT_CONFIG,
) -> None:
self.server_params = HttpToolConfig(
name=name,
@ -160,6 +169,7 @@ class HttpTool(BaseTool[BaseModel, Any], Component[HttpToolConfig]):
headers=headers,
json_schema=json_schema,
return_type=return_type,
timeout=timeout,
)
# Use regex to find all path parameters, we will need those later to template the path
@ -211,7 +221,8 @@ class HttpTool(BaseTool[BaseModel, Any], Component[HttpToolConfig]):
port=self.server_params.port,
path=path,
)
async with httpx.AsyncClient() as client:
timeout_config = httpx.Timeout(timeout=self.server_params.timeout)
async with httpx.AsyncClient(timeout=timeout_config) as client:
match self.server_params.method:
case "GET":
response = await client.get(url, headers=self.server_params.headers, params=model_dump)