fix endpoint_util not using the correct instance's endpoint

This commit is contained in:
Nader Arbabian
2025-08-11 11:49:51 -07:00
parent 8804e17201
commit 52ac4c0c1a
+6 -1
View File
@@ -30,7 +30,12 @@ class Endpoint:
Returns: Returns:
Endpoint API key if successful, None otherwise Endpoint API key if successful, None otherwise
""" """
vast_console_url = "https://console.vast.ai/api/v0/endptjobs/" endpoints = {
"alpha": "alpha",
"candidate": "candidate",
"prod": "console",
}
vast_console_url = f"https://{endpoints[instance]}.vast.ai/api/v0/endptjobs/"
headers = {"Authorization": f"Bearer {account_api_key}"} headers = {"Authorization": f"Bearer {account_api_key}"}
try: try: