enh: AIOHTTP_CLIENT_TIMEOUT None support

8e62c361 · Timothy J. Baek · f76a5695 · 8e62c361
Commit 8e62c361 authored Jun 16, 2024 by Timothy J. Baek
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

backend/config.py backend/config.py +8 -1

No files found.
--- a/backend/config.py
+++ b/backend/config.py
@@ -425,7 +425,14 @@ OLLAMA_API_BASE_URL = os.environ.get(
 )

 OLLAMA_BASE_URL = os.environ.get("OLLAMA_BASE_URL", "")
-AIOHTTP_CLIENT_TIMEOUT = int(os.environ.get("AIOHTTP_CLIENT_TIMEOUT", "300"))
+AIOHTTP_CLIENT_TIMEOUT = os.environ.get("AIOHTTP_CLIENT_TIMEOUT", "300")
+
+if AIOHTTP_CLIENT_TIMEOUT == "":
+    AIOHTTP_CLIENT_TIMEOUT = None
+else:
+    AIOHTTP_CLIENT_TIMEOUT = int(AIOHTTP_CLIENT_TIMEOUT)
+
+
 K8S_FLAG = os.environ.get("K8S_FLAG", "")
 USE_OLLAMA_DOCKER = os.environ.get("USE_OLLAMA_DOCKER", "false")