Skip to content

Commit f24c0e0

Browse files
committed
fix(jig): use Together.get instead of Together._client.get so that registry errors are handled correctly
1 parent 48f11e8 commit f24c0e0

File tree

1 file changed

+2
-4
lines changed
  • src/together/lib/cli/api/beta/jig

1 file changed

+2
-4
lines changed

src/together/lib/cli/api/beta/jig/jig.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -505,11 +505,9 @@ def state(self) -> State:
505505
def registry(self) -> str:
506506
"""Get registry and namespace for current user"""
507507
if not self.state.registry_base_path:
508-
response = self.together._client.get("/image-repositories/base-path", headers=self.together.auth_headers)
509-
if not response.is_success:
510-
raise JigError(f"Failed to get registry path (HTTP {response.status_code})")
508+
response = self.together.get("/image-repositories/base-path", cast_to=dict[str, str])
511509
# strip protocol for docker image format
512-
self.state.registry_base_path = response.json()["base-path"].split("://", 1)[-1]
510+
self.state.registry_base_path = response["base-path"].split("://", 1)[-1]
513511
self.state.save()
514512
return self.state.registry_base_path + "/"
515513

0 commit comments

Comments
 (0)