From 568775a7f48ba66348616a1459f079a2bb4bd806 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Sun, 16 Mar 2025 18:11:25 -0700
Subject: [PATCH] nocdhelp script (#1948)

---
 skyvern/agent/client.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/skyvern/agent/client.py b/skyvern/agent/client.py
index e6c42127..8e7d8bc1 100644
--- a/skyvern/agent/client.py
+++ b/skyvern/agent/client.py
@@ -1,4 +1,5 @@
 from enum import StrEnum
+from typing import Any
 
 import httpx
 
@@ -51,7 +52,7 @@ class SkyvernClient:
         totp_identifier: str | None = None,
         totp_url: str | None = None,
     ) -> RunWorkflowResponse:
-        data = {
+        data: dict[str, Any] = {
             "webhook_callback_url": webhook_url,
             "proxy_location": proxy_location,
             "totp_identifier": totp_identifier,
@@ -86,6 +87,7 @@ class SkyvernClient:
             response = await client.get(
                 f"{self.base_url}/api/v1/workflows/runs/{workflow_run_id}",
                 headers={"x-api-key": self.api_key},
+                timeout=60,
             )
             if response.status_code != 200:
                 raise SkyvernClientException(