cast timeout env var to int

This commit is contained in:
saltacc
2024-03-11 14:53:23 -07:00
committed by GitHub
parent 1eb8db6dd0
commit 4ebc2e3d42
@@ -5,7 +5,7 @@ import json
import time
SERVER_IP = os.environ.get("SERVER_IP", "127.0.0.1:8188")
TIMEOUT_SECONDS = os.environ.get("TIMEOUT_SECONDS", 1000)
TIMEOUT_SECONDS = int(os.environ.get("TIMEOUT_SECONDS", 1000))
PROMPT_ENDPOINT = f"http://{SERVER_IP}/prompt"
def parse_args():