From 52ac4c0c1ae84de6923940c828b3947678d09c3e Mon Sep 17 00:00:00 2001
From: Nader Arbabian <nader@vast.ai>
Date: Mon, 11 Aug 2025 11:49:51 -0700
Subject: [PATCH] fix endpoint_util not using the correct instance's endpoint

---
 utils/endpoint_util.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/utils/endpoint_util.py b/utils/endpoint_util.py
index 48c4cdb..30cf074 100644
--- a/utils/endpoint_util.py
+++ b/utils/endpoint_util.py
@@ -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: