From 08a756aea092d01772c30b6ffb582888061ca780 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Sun, 19 Jan 2025 19:28:26 +0200
Subject: [PATCH] tests : add DEFAULT_HTTP_TIMEOUT

---
 examples/server/tests/utils.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/server/tests/utils.py b/examples/server/tests/utils.py
index b52119532..9d1a7a5b0 100644
--- a/examples/server/tests/utils.py
+++ b/examples/server/tests/utils.py
@@ -26,6 +26,9 @@ from re import RegexFlag
 import wget
 
 
+DEFAULT_HTTP_TIMEOUT = 10 if "LLAMA_SANITIZE" not in os.environ else 30
+
+
 class ServerResponse:
     headers: dict
     status_code: int
@@ -88,10 +91,7 @@ class ServerProcess:
         if "PORT" in os.environ:
             self.server_port = int(os.environ["PORT"])
 
-    def start(self, timeout_seconds: int = 10) -> None:
-        if "LLAMA_SANITIZE" in os.environ:
-            timeout_seconds = 30
-
+    def start(self, timeout_seconds: int | None = DEFAULT_HTTP_TIMEOUT) -> None:
         if "LLAMA_SERVER_BIN_PATH" in os.environ:
             server_path = os.environ["LLAMA_SERVER_BIN_PATH"]
         elif os.name == "nt":