aixtools 0.2.20__tar.gz → 0.2.21__tar.gz

{aixtools-0.2.20 → aixtools-0.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.2.20
+Version: 0.2.21
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown

{aixtools-0.2.20 → aixtools-0.2.21}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.20'
-__version_tuple__ = version_tuple = (0, 2, 20)
+__version__ = version = '0.2.21'
+__version_tuple__ = version_tuple = (0, 2, 21)
 
-__commit_id__ = commit_id = 'ge7f4f14bb'
+__commit_id__ = commit_id = 'gbbd8ec97d'

{aixtools-0.2.20 → aixtools-0.2.21}/aixtools/mcp/client.py

@@ -23,6 +23,7 @@ from pydantic_ai.toolsets.abstract import ToolsetTool
 from aixtools.context import SessionIdTuple
 from aixtools.logging.logging_config import get_logger
 from aixtools.server.utils import create_session_headers
+from aixtools.utils.config import MCP_TOOLS_MAX_RETRIES
 
 MCP_TOOL_CACHE_TTL = 300  # 5 minutes
 DEFAULT_MCP_CONNECTION_TIMEOUT = 30
@@ -87,6 +88,7 @@ def get_mcp_servers(
     mcp_configs: list[MCPConfig],
     session_id_tuple: SessionIdTuple,
     auth_token: str = None,
+    max_retries: int = MCP_TOOLS_MAX_RETRIES,
     *,
     timeout: float = DEFAULT_MCP_CONNECTION_TIMEOUT,
 ):
@@ -115,7 +117,7 @@ def get_mcp_servers(
     servers = []
     for config in mcp_configs:
         server = CachedMCPServerStreamableHTTP(
-            url=config.url, headers=headers, timeout=timeout, read_timeout=config.read_timeout
+            url=config.url, headers=headers, timeout=timeout, read_timeout=config.read_timeout, max_retries=max_retries
         )
         logger.info("Using MCP server at %s", config.url)
         servers.append(server)

{aixtools-0.2.20 → aixtools-0.2.21}/aixtools/utils/config.py

@@ -144,3 +144,5 @@ SKIP_MCP_AUTHORIZATION = str2bool(get_variable_env("SKIP_MCP_AUTHORIZATION", Tru
 APP_DEFAULT_SCOPE = get_variable_env("APP_DEFAULT_SCOPE", allow_empty=True)
 
 AUTH_TEST_TOKEN = get_variable_env("AUTH_TEST_TOKEN", allow_empty=True)
+
+MCP_TOOLS_MAX_RETRIES = int(get_variable_env("MCP_TOOLS_MAX_RETRIES", default=10))