reasoning-deployment-service 0.5.5__tar.gz → 0.5.7__tar.gz

{reasoning_deployment_service-0.5.5 → reasoning_deployment_service-0.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reasoning-deployment-service
-Version: 0.5.5
+Version: 0.5.7
 Summary: Deployment helper for Vertex AI Reasoning Engines & Agent Spaces
 Author-email: Sergio Estrada <sergio.estrada@accenture.com>
 License: Apache-2.0

{reasoning_deployment_service-0.5.5 → reasoning_deployment_service-0.5.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "reasoning-deployment-service"
-version = "0.5.5"
+version = "0.5.7"
 description = "Deployment helper for Vertex AI Reasoning Engines & Agent Spaces"
 readme = "README.md"
 license = { text = "Apache-2.0" }

{reasoning_deployment_service-0.5.5 → reasoning_deployment_service-0.5.7}/reasoning_deployment_service/reasoning_deployment_service.py

@@ -815,6 +815,30 @@ class ReasoningEngineDeploymentService:
     def one_githhub_deployment_to_go_with_skip(self):
         return self.one_github_deployment_to_go(skip_engine=True)
 
+    def update_authorization_scopes(self, auth_id: str, scopes: list, oauth_client_id: str) -> dict:
+        """Patch the scopes for a specific authorization by ID, updating the authorizationUri as well."""
+        scopes_str = " ".join(scopes)
+        authorization_uri = (
+            "https://accounts.google.com/o/oauth2/auth"
+            "?response_type=code"
+            f"&client_id={oauth_client_id}"
+            f"&scope={scopes_str}"
+            "&access_type=offline&prompt=consent"
+        )
+        payload = {
+            "serverSideOauth2": {
+                "authorizationUri": authorization_uri
+            }
+        }
+        url = f"{DISCOVERY_ENGINE_URL}/projects/{self._project_id}/locations/global/authorizations/{auth_id}?update_mask=server_side_oauth2.authorization_uri"
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self._project_number,
+        }
+        r = self._http.patch(url, headers=headers, json=payload, timeout=60)
+        r.raise_for_status()
+        return r.json()
 
     def one_github_deployment_to_go(self, skip_engine=False):
         """
@@ -918,29 +942,7 @@ class ReasoningEngineDeploymentService:
 
                 if existing_scopes != want_scopes:
                     self.info("[AUTH] Scopes changed. Patching authorization...")
-                    new_auth_uri = self._build_authorization_uri(self._oauth_client_id, list(want_scopes))
-                    patch_payload = {
-                        "serverSideOauth2": {
-                            "clientId": self._oauth_client_id,
-                            "clientSecret": "***" if self._oauth_client_secret else None,
-                            "authorizationUri": new_auth_uri,
-                            "tokenUri": "https://oauth2.googleapis.com/token",
-                        }
-                    }
-                    if patch_payload["serverSideOauth2"]["clientSecret"] is None:
-                        patch_payload["serverSideOauth2"].pop("clientSecret", None)
-
-                    hdrs2 = self._get_headers().copy()
-                    if "Authorization" in hdrs2:
-                        hdrs2["Authorization"] = "Bearer ***"
-                    self.info(f"[AUTH] PATCH {auth_url} headers={json.dumps(hdrs2)} payload={json.dumps(patch_payload, indent=2)}")
-                    pr = self._http.patch(auth_url, headers=self._get_headers(), json=patch_payload, timeout=60)
-                    self.info(f"[AUTH] PATCH status={pr.status_code} ct={pr.headers.get('content-type','')}")
-                    try:
-                        self.info(f"[AUTH] PATCH body={json.dumps(pr.json(), indent=2)[:4000]}")
-                    except Exception:
-                        self.info(f"[AUTH] PATCH text={(pr.text or '')[:1000]}")
-                    pr.raise_for_status()
+                    self.update_authorization_scopes(self._authorization_id, list(want_scopes), self._oauth_client_id)
                     self.info("[AUTH] Authorization updated.")
                 else:
                     self.info("[AUTH] Scopes unchanged; no update needed.")

{reasoning_deployment_service-0.5.5 → reasoning_deployment_service-0.5.7}/reasoning_deployment_service.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reasoning-deployment-service
-Version: 0.5.5
+Version: 0.5.7
 Summary: Deployment helper for Vertex AI Reasoning Engines & Agent Spaces
 Author-email: Sergio Estrada <sergio.estrada@accenture.com>
 License: Apache-2.0