vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests : add DEFAULT_HTTP_TIMEOUT

Browse source
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
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
examples/server/tests/utils.py
View file

@ -26,6 +26,9 @@ from re import RegexFlag
import wget import wget
DEFAULT_HTTP_TIMEOUT = 10 if "LLAMA_SANITIZE" not in os.environ else 30
class ServerResponse: class ServerResponse:
headers: dict headers: dict
status_code: int status_code: int
@ -88,10 +91,7 @@ class ServerProcess:
if "PORT" in os.environ: if "PORT" in os.environ:
self.server_port = int(os.environ["PORT"]) self.server_port = int(os.environ["PORT"])
def start(self, timeout_seconds: int = 10) -> None: def start(self, timeout_seconds: int | None = DEFAULT_HTTP_TIMEOUT) -> None:
if "LLAMA_SANITIZE" in os.environ:
timeout_seconds = 30
if "LLAMA_SERVER_BIN_PATH" in os.environ: if "LLAMA_SERVER_BIN_PATH" in os.environ:
server_path = os.environ["LLAMA_SERVER_BIN_PATH"] server_path = os.environ["LLAMA_SERVER_BIN_PATH"]
elif os.name == "nt": elif os.name == "nt":