reasoning-deployment-service 0.5.5__py3-none-any.whl → 0.5.6__py3-none-any.whl

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_id,
+        }
+        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.dist-info → reasoning_deployment_service-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reasoning-deployment-service
-Version: 0.5.5
+Version: 0.5.6
 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.dist-info → reasoning_deployment_service-0.5.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 reasoning_deployment_service/__init__.py,sha256=xDuKt9gGviQiTV6vXBdkBvygnlAOIrwnUjVaMGZy0L4,670
-reasoning_deployment_service/reasoning_deployment_service.py,sha256=PtjZvn5qcfgJFoRGPGnPHNAmqKUi3NNY2BGlSwetQ64,43371
+reasoning_deployment_service/reasoning_deployment_service.py,sha256=KGng1H9pXMsBSWsnz3JvIVnQLHADD2YbwgWLeLzldhM,43104
 reasoning_deployment_service/runner.py,sha256=qWN0t66lQ1G4ht48gIHSF2JvedcheHRu8PmUz5TaKTI,5619
 reasoning_deployment_service/cli_editor/__init__.py,sha256=bN8NPkw8riB92pj2lAwJZuEMOQIO_RRuge0ehnJTW1I,118
 reasoning_deployment_service/cli_editor/api_client.py,sha256=Kzx5iYp0MmowggrSmPLE7I2kt1-8xvdGBAgde9a1gCY,33681
@@ -22,8 +22,8 @@ reasoning_deployment_service/gui_editor/src/ui/authorization_view.py,sha256=BoNc
 reasoning_deployment_service/gui_editor/src/ui/reasoning_engine_view.py,sha256=tCvSPEf4dW0NRdAqfs3yT5Pa873gYeLzCMMIt2r2T4o,14644
 reasoning_deployment_service/gui_editor/src/ui/reasoning_engines_view.py,sha256=IRjFlBbY98usAZa0roOonjvWQOsF6NBW4bBg_k8KnKI,7860
 reasoning_deployment_service/gui_editor/src/ui/ui_components.py,sha256=HdQHy-oSZ3GobQ3FNdH7y_w3ANbFiuf2rMoflAmff0A,55366
-reasoning_deployment_service-0.5.5.dist-info/METADATA,sha256=t0A29OMC-bBjmlvQfPftx4tooc2ak1NDP4ZD9rJJC7A,5302
-reasoning_deployment_service-0.5.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-reasoning_deployment_service-0.5.5.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
-reasoning_deployment_service-0.5.5.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
-reasoning_deployment_service-0.5.5.dist-info/RECORD,,
+reasoning_deployment_service-0.5.6.dist-info/METADATA,sha256=wAp6aT6gwNUHL-JPZ4xPz3RA9wzFBRbOtqRK6qQENOM,5302
+reasoning_deployment_service-0.5.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+reasoning_deployment_service-0.5.6.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
+reasoning_deployment_service-0.5.6.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
+reasoning_deployment_service-0.5.6.dist-info/RECORD,,