fix endpoint_util not using the correct instance's endpoint

2025-08-11 11:49:51 -07:00
parent 8804e17201
commit 52ac4c0c1a
1 changed files with 6 additions and 1 deletions
@@ -30,7 +30,12 @@ class Endpoint:
        Returns:
            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}"}

        try: