chore: increase default requests timeout

prefeitura-rio · Apr 15, 2024 · b42746a · b42746a
1 parent e62b3d7
commit b42746a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/app/config/base.py b/app/config/base.py
@@ -41,3 +41,6 @@
 
 # Host
 HOST = socket.gethostname()
+
+# Requests timeout
+REQUESTS_DEFAULT_TIMEOUT = int(getenv_or_action("REQUESTS_DEFAULT_TIMEOUT", default="30"))
diff --git a/app/routers/proxy.py b/app/routers/proxy.py
@@ -27,7 +27,7 @@ async def options():
     """
     Proxies requests to the Prefect API.
     """
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=config.REQUESTS_DEFAULT_TIMEOUT) as client:
         response = await client.options(
             config.PREFECT_API_URL,
         )

diff --git a/app/utils.py b/app/utils.py
@@ -475,7 +475,7 @@ async def graphql_request(operations: dict) -> httpx.Response:
     Returns:
         bytes: The response from Prefect backend.
     """
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=config.REQUESTS_DEFAULT_TIMEOUT) as client:
         response = await client.post(
             config.PREFECT_API_URL,
             json=operations,