agentvalet 0.2.2__tar.gz → 0.3.0__tar.gz

{agentvalet-0.2.2 → agentvalet-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: agentvalet
-Version: 0.2.2
+Version: 0.3.0
 Summary: Call approved SaaS platforms from any Python agent. Your agent never holds the downstream credential.
 Project-URL: Homepage, https://agentvalet.ai
 Project-URL: Documentation, https://docs.agentvalet.ai
@@ -164,6 +164,19 @@ result = av.wait_for_approval(approval_id)
 
 Pass `approval_timeout_s=0` if you never want to block.
 
+### Network timeouts
+
+`call()` waits up to `action_timeout_s` (default 130s) for the action request,
+because it waits on the platform itself and some tools run for a minute or
+more. Every other request uses `timeout_s` (default 15s). Waiting on an owner
+approval is separate and set by `approval_timeout_s`. AgentValet's edge ends any
+single response at 100s, so for a tool that runs longer, use its
+start-and-poll variant if it has one.
+
+```python
+av = AgentValet.from_env(action_timeout_s=60, timeout_s=10)
+```
+
 ## Errors you can branch on
 
 Every exception is typed, so you never string-match an error envelope:

{agentvalet-0.2.2 → agentvalet-0.3.0}/README.md

@@ -132,6 +132,19 @@ result = av.wait_for_approval(approval_id)
 
 Pass `approval_timeout_s=0` if you never want to block.
 
+### Network timeouts
+
+`call()` waits up to `action_timeout_s` (default 130s) for the action request,
+because it waits on the platform itself and some tools run for a minute or
+more. Every other request uses `timeout_s` (default 15s). Waiting on an owner
+approval is separate and set by `approval_timeout_s`. AgentValet's edge ends any
+single response at 100s, so for a tool that runs longer, use its
+start-and-poll variant if it has one.
+
+```python
+av = AgentValet.from_env(action_timeout_s=60, timeout_s=10)
+```
+
 ## Errors you can branch on
 
 Every exception is typed, so you never string-match an error envelope:

{agentvalet-0.2.2 → agentvalet-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "agentvalet"
-version = "0.2.2"
+version = "0.3.0"
 description = "Call approved SaaS platforms from any Python agent. Your agent never holds the downstream credential."
 readme = "README.md"
 requires-python = ">=3.9"

{agentvalet-0.2.2 → agentvalet-0.3.0}/src/agentvalet/__init__.py

@@ -37,7 +37,7 @@ from .errors import (
     UpstreamError,
 )
 
-__version__ = "0.2.2"
+__version__ = "0.3.0"
 
 __all__ = [
     "DEFAULT_PROXY_URL",

{agentvalet-0.2.2 → agentvalet-0.3.0}/src/agentvalet/_constants.py

@@ -21,9 +21,15 @@ APPROVAL_TIMEOUT_S = 50.0
 #: How often we ask the proxy whether the owner has decided.
 APPROVAL_POLL_S = 2.0
 
-#: Per-request network timeout.
+#: Network timeout for every request except the action call.
 REQUEST_TIMEOUT_S = 15.0
 
+#: Network timeout for ``POST /v1/actions`` only. That call waits on the
+#: upstream platform, and the proxy gives a slow MCP tool up to its platform's
+#: ``mcp_call_timeout_ms`` (up to 120s today). Matches the Node SDK's
+#: ``actionTimeoutMs`` default and the mcp-server's ACTION_REQUEST_TIMEOUT_MS.
+ACTION_TIMEOUT_S = 130.0
+
 #: Agent assertions are deliberately short-lived; the proxy rejects long-dated
 #: ones. Seconds.
 JWT_LIFETIME_S = 60

{agentvalet-0.2.2 → agentvalet-0.3.0}/src/agentvalet/aio.py

@@ -16,6 +16,7 @@ from typing import Any, Callable, Union
 import httpx
 
 from ._constants import (
+    ACTION_TIMEOUT_S,
     APPROVAL_POLL_S,
     APPROVAL_TIMEOUT_S,
     DEFAULT_PROXY_URL,
@@ -61,6 +62,7 @@ class AsyncAgentValet:
         session_id: str | None = None,
         *,
         bearer_token: str | None = None,
+        action_timeout_s: float = ACTION_TIMEOUT_S,
     ) -> None:
         self._bearer_token = bearer_token
         if bearer_token is None:
@@ -77,6 +79,7 @@ class AsyncAgentValet:
         self._private_key = private_key or ""
         self.session_id = validate_session_id(session_id)
         self._timeout_s = timeout_s
+        self._action_timeout_s = action_timeout_s
         self._approval_timeout_s = approval_timeout_s
         self._approval_poll_s = approval_poll_s
         self._on_approval_pending = on_approval_pending
@@ -145,7 +148,9 @@ class AsyncAgentValet:
             connection_id=connection_id,
             reason=reason,
         )
-        response = await self._authed_request("POST", "/v1/actions", json=body)
+        response = await self._authed_request(
+            "POST", "/v1/actions", timeout_s=self._action_timeout_s, json=body
+        )
         text = response.text
 
         if response.status_code == 202:
@@ -296,8 +301,11 @@ class AsyncAgentValet:
             headers["Authorization"] = f"Bearer {self.sign_jwt()}"
         return headers
 
-    async def _authed_request(self, method: str, path: str, **kwargs: Any) -> httpx.Response:
+    async def _authed_request(
+        self, method: str, path: str, *, timeout_s: float | None = None, **kwargs: Any
+    ) -> httpx.Response:
         url = f"{self.proxy_url}{path}"
+        timeout = self._timeout_s if timeout_s is None else timeout_s
 
         async def once() -> httpx.Response:
             try:
@@ -305,7 +313,7 @@ class AsyncAgentValet:
                     method,
                     url,
                     headers=self._auth_headers(),
-                    timeout=self._timeout_s,
+                    timeout=timeout,
                     **kwargs,
                 )
             except httpx.HTTPError as err:

{agentvalet-0.2.2 → agentvalet-0.3.0}/src/agentvalet/client.py

@@ -12,6 +12,7 @@ from typing import Any, Callable
 import httpx
 
 from ._constants import (
+    ACTION_TIMEOUT_S,
     APPROVAL_POLL_S,
     APPROVAL_TIMEOUT_S,
     DEFAULT_PROXY_URL,
@@ -63,6 +64,7 @@ class AgentValet:
         session_id: str | None = None,
         *,
         bearer_token: str | None = None,
+        action_timeout_s: float = ACTION_TIMEOUT_S,
     ) -> None:
         self._bearer_token = bearer_token
         if bearer_token is None:
@@ -79,6 +81,7 @@ class AgentValet:
         self._private_key = private_key or ""
         self.session_id = validate_session_id(session_id)
         self._timeout_s = timeout_s
+        self._action_timeout_s = action_timeout_s
         self._approval_timeout_s = approval_timeout_s
         self._approval_poll_s = approval_poll_s
         self._on_approval_pending = on_approval_pending
@@ -181,7 +184,9 @@ class AgentValet:
             connection_id=connection_id,
             reason=reason,
         )
-        response = self._authed_request("POST", "/v1/actions", json=body)
+        response = self._authed_request(
+            "POST", "/v1/actions", timeout_s=self._action_timeout_s, json=body
+        )
         text = response.text
 
         if response.status_code == 202:
@@ -379,8 +384,11 @@ class AgentValet:
             headers["Authorization"] = f"Bearer {self.sign_jwt()}"
         return headers
 
-    def _authed_request(self, method: str, path: str, **kwargs: Any) -> httpx.Response:
+    def _authed_request(
+        self, method: str, path: str, *, timeout_s: float | None = None, **kwargs: Any
+    ) -> httpx.Response:
         url = f"{self.proxy_url}{path}"
+        timeout = self._timeout_s if timeout_s is None else timeout_s
 
         def once() -> httpx.Response:
             try:
@@ -388,7 +396,7 @@ class AgentValet:
                     method,
                     url,
                     headers=self._auth_headers(),
-                    timeout=self._timeout_s,
+                    timeout=timeout,
                     **kwargs,
                 )
             except httpx.HTTPError as err:

agentvalet-0.3.0/tests/test_action_timeout.py

@@ -0,0 +1,71 @@
+"""``call()``'s ``POST /v1/actions`` waits up to ``action_timeout_s`` (130s);
+every other request keeps ``timeout_s`` (15s).
+
+The proxy gives a slow MCP tool up to its platform's ``mcp_call_timeout_ms``
+(up to 120s today), so a 15s action timeout gave up on calls the proxy would
+have completed. Each test runs against both the sync and async client.
+The timeout asserted is the one httpx actually attached to the request.
+"""
+
+import httpx
+import pytest
+import respx
+from conftest import PROXY, invoke
+
+from agentvalet import AgentValet, NetworkError
+from agentvalet import _constants as c
+
+
+def read_timeout(request: httpx.Request) -> float:
+    return request.extensions["timeout"]["read"]
+
+
+def test_constants():
+    assert c.ACTION_TIMEOUT_S == 130.0
+    assert c.REQUEST_TIMEOUT_S == 15.0
+
+
+@respx.mock
+async def test_call_uses_the_action_timeout(client):
+    route = respx.post(f"{PROXY}/v1/actions").mock(return_value=httpx.Response(200, json={"ok": True}))
+    await invoke(client, "call", "mcp-treg", "check", "mcp-treg:check", method="POST")
+    assert read_timeout(route.calls.last.request) == 130.0
+
+
+@respx.mock
+async def test_other_requests_keep_the_request_timeout(client):
+    route = respx.get(f"{PROXY}/v1/agent/permissions").mock(return_value=httpx.Response(200, json={"platforms": []}))
+    await invoke(client, "list_platforms")
+    assert read_timeout(route.calls.last.request) == 15.0
+
+
+@respx.mock
+async def test_the_two_timeouts_are_independent(make_client):
+    client = make_client(action_timeout_s=40.0, timeout_s=5.0)
+    actions = respx.post(f"{PROXY}/v1/actions").mock(return_value=httpx.Response(200, json={}))
+    perms = respx.get(f"{PROXY}/v1/agent/permissions").mock(return_value=httpx.Response(200, json={}))
+    await invoke(client, "call", "p", "/x", "s")
+    await invoke(client, "list_platforms")
+    assert read_timeout(actions.calls.last.request) == 40.0
+    assert read_timeout(perms.calls.last.request) == 5.0
+
+
+@respx.mock
+async def test_the_401_retry_keeps_the_action_timeout(client):
+    route = respx.post(f"{PROXY}/v1/actions").mock(
+        side_effect=[httpx.Response(401, json={"error": "skew"}), httpx.Response(200, json={})]
+    )
+    await invoke(client, "call", "p", "/x", "s")
+    assert [read_timeout(call.request) for call in route.calls] == [130.0, 130.0]
+
+
+@respx.mock
+async def test_an_action_timeout_is_a_network_error(client):
+    respx.post(f"{PROXY}/v1/actions").mock(side_effect=httpx.ReadTimeout("timed out"))
+    with pytest.raises(NetworkError):
+        await invoke(client, "call", "p", "/x", "s")
+
+
+def test_from_bearer_accepts_an_action_timeout():
+    child = AgentValet.from_bearer("h.e30.s", action_timeout_s=90.0)
+    assert child._action_timeout_s == 90.0