Fix chains invocation URL domain.

basetenlabs · Aug 8, 2024 · 974cb3c · 974cb3c
1 parent b2e125e
commit 974cb3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/truss/remote/baseten/remote.py b/truss/remote/baseten/remote.py
@@ -100,7 +100,7 @@ def get_chainlets(
                     chainlet["id"],
                 ),
                 oracle_predict_url=URLConfig.invocation_url(
-                    self.remote_url,
+                    self._api.rest_api_url,
                     URLConfig.MODEL,
                     chainlet["oracle"]["id"],
                     chainlet["oracle_version"]["id"],

diff --git a/truss/remote/baseten/service.py b/truss/remote/baseten/service.py
@@ -170,7 +170,7 @@ def logs_url(self) -> str:
     @property
     def predict_url(self) -> str:
         return URLConfig.invocation_url(
-            self._api.app_url,
+            self._api.rest_api_url,
             URLConfig.MODEL,
             self.model_id,
             self._model_version_id,