tests : add DEFAULT_HTTP_TIMEOUT

This commit is contained in:
Georgi Gerganov 2025-01-19 19:28:26 +02:00
parent 9d65a3d65b
commit 08a756aea0
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

View file

@ -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":