Align workspace_access client workspace_id logic

PrefectHQ · Dec 13, 2024 · 06df965 · 06df965
1 parent 35513b9
commit 06df965
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/client/workspace_access.go b/internal/client/workspace_access.go
@@ -26,11 +26,13 @@ func (c *Client) WorkspaceAccess(accountID uuid.UUID, workspaceID uuid.UUID) (ap
 	if accountID == uuid.Nil {
 		accountID = c.defaultAccountID
 	}
+
 	if workspaceID == uuid.Nil {
 		workspaceID = c.defaultWorkspaceID
 	}
-	if accountID == uuid.Nil || workspaceID == uuid.Nil {
-		return nil, fmt.Errorf("both accountID and workspaceID must be defined")
+
+	if err := validateCloudEndpoint(c.endpoint, accountID, workspaceID); err != nil {
+		return nil, err
 	}
 
 	return &WorkspaceAccessClient{