Added caller for REPORT_ADDR to backend.py

Merge pull request #51 from vast-ai/delete-reqs-hotfix
Redis subscriber queue patch
2025-10-29 18:02:17 -07:00 · 2025-10-28 16:07:28 -07:00 · 2025-10-28 16:03:50 -07:00 · 2025-10-28 16:01:32 -07:00
2 changed files with 20 additions and 17 deletions
@@ -66,6 +66,9 @@ class Backend:
    unsecured: bool = dataclasses.field(
        default_factory=lambda: bool(strtobool(os.environ.get("UNSECURED", "false"))),
    )
+    report_addr: str = dataclasses.field(
+        default_factory=lambda: os.environ.get("REPORT_ADDR", "https://run.vast.ai")
+    )

    def __post_init__(self):
        self.metrics = Metrics()
@@ -104,23 +107,19 @@ class Backend:

    #######################################Private#######################################
    def _fetch_pubkey(self):
-        command = ["curl", "-X", "GET", "https://run.vast.ai/pubkey/"]
-        result = subprocess.check_output(command, universal_newlines=True)
-        log.debug("public key:")
-        log.debug(result)
-        key = None
-        for _ in range(5):
-            try:
-                key = RSA.import_key(result)
-                break
-            except ValueError as e:
-                log.debug(f"Error downloading key: {e}")
-                time.sleep(15)
-        if key is None:
-            self._total_pubkey_fetch_errors += 1
-            if self._total_pubkey_fetch_errors >= MAX_PUBKEY_FETCH_ATTEMPTS:
-                self.backend_errored("Failed to get autoscaler pubkey")
-        return key
+        report_addr = self.report_addr.rstrip("/")
+        command = ["curl", "-X", "GET", f"{report_addr}/pubkey/"]
+        try:
+            result = subprocess.check_output(command, universal_newlines=True)
+            log.debug("public key:")
+            log.debug(result)
+            key = RSA.import_key(result)
+            if key is not None:
+                return key
+        except (ValueError , subprocess.CalledProcessError) as e:
+            log.debug(f"Error downloading key: {e}")
+        self.backend_errored("Failed to get autoscaler pubkey")
+       

    async def __handle_request(
        self,
@@ -180,6 +180,10 @@ class Metrics:
            return  # nothing to do

        for report_addr in self.report_addr:
+            # TODO: Add a Redis subscriber queue for delete_requests
+            if report_addr == "https://cloud.vast.ai/api/v0":
+                # Patch: ignore the Redis API report_addr
+                continue
            sent_success = True
            sent_failed  = True
Author	SHA1	Message	Date
Abiola Akinnubi	7437028cb2	Added caller for REPORT_ADDR to backend.py	2025-10-29 18:02:17 -07:00
LucasArmandVast	9f5a432513	Merge pull request #51 from vast-ai/delete-reqs-hotfix Redis subscriber queue patch	2025-10-28 16:07:28 -07:00
Lucas Armand	e09f1fa953	patch for redis queue	2025-10-28 16:03:50 -07:00
edgaratvast	ba6f1c2e4b	Fix signature (#50 ) * change order of fields in auth_data to match autoscaler for signature verification * also ignore __request_id * Revert "change order of fields in auth_data to match autoscaler for signature verification" so that it's alphabetical again This reverts commit `b8223879c9`. * enforce alphabetical json dumping of message for signature verification --------- Co-authored-by: Edgar Lin <edgarlin2000@gmail.com>	2025-10-28 16:01:32 -07:00