reasoning-deployment-service 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

reasoning_deployment_service/reasoning_deployment_service.py

@@ -766,6 +766,76 @@ class ReasoningEngineDeploymentService:
             )
         
         return matches[0]
+    
+    def patch_agent_space_metadata_and_auth(
+        self,
+        agent_id: str,
+        new_display_name: Optional[str] = None,
+        new_description: Optional[str] = None,
+        new_reasoning_engine: Optional[str] = None,
+        new_authorizations: Optional[list[str]] = None,
+    ) -> dict:
+        """
+        Patch only metadata (displayName, description) and linkage fields
+        (reasoningEngine, authorizations) for an existing Agent Space agent.
+        """
+        url = (
+            f"{DISCOVERY_ENGINE_URL}/projects/{self._project_id}/locations/global/"
+            f"collections/default_collection/engines/{self._agent_space_engine}/"
+            f"assistants/default_assistant/agents/{agent_id}"
+        )
+        headers = self._get_headers()
+
+        # GET current agent
+        resp = self._http.get(url, headers=headers, timeout=60)
+        resp.raise_for_status()
+        agent_data = resp.json()
+
+        # Prepare PATCH payload
+        patch_payload = {}
+
+        if new_display_name:
+            patch_payload["displayName"] = new_display_name
+        if new_description:
+            patch_payload["description"] = new_description
+
+        adk_def = agent_data.get("adkAgentDefinition", {})
+        patch_adk_def = {}
+
+        if new_reasoning_engine:
+            patch_adk_def.setdefault("provisionedReasoningEngine", {})[
+                "reasoningEngine"
+            ] = new_reasoning_engine
+
+        if new_authorizations is not None:
+            patch_adk_def["authorizations"] = new_authorizations
+
+        if patch_adk_def:
+            patch_payload["adkAgentDefinition"] = patch_adk_def
+
+        # Build updateMask
+        update_mask = []
+        if "displayName" in patch_payload:
+            update_mask.append("display_name")
+        if "description" in patch_payload:
+            update_mask.append("description")
+        if "adkAgentDefinition" in patch_payload:
+            if "provisionedReasoningEngine" in patch_payload["adkAgentDefinition"]:
+                update_mask.append("adk_agent_definition.provisioned_reasoning_engine")
+            if "authorizations" in patch_payload["adkAgentDefinition"]:
+                update_mask.append("adk_agent_definition.authorizations")
+
+        if not update_mask:
+            self.info("[AGENT PATCH] Nothing to update.")
+            return agent_data
+
+        patch_url = f"{url}?updateMask={','.join(update_mask)}"
+        self.info(f"[AGENT PATCH] PATCH {patch_url} with payload={json.dumps(patch_payload, indent=2)}")
+
+        patch_resp = self._http.patch(patch_url, headers=headers, json=patch_payload, timeout=60)
+        patch_resp.raise_for_status()
+
+        return patch_resp.json()
 
     def one_github_deployment_to_go(self):
         """
@@ -877,7 +947,6 @@ class ReasoningEngineDeploymentService:
                             "tokenUri": "https://oauth2.googleapis.com/token",
                         }
                     }
-                    # remove None to keep payload clean
                     if patch_payload["serverSideOauth2"]["clientSecret"] is None:
                         patch_payload["serverSideOauth2"].pop("clientSecret", None)
 
@@ -899,59 +968,38 @@ class ReasoningEngineDeploymentService:
             self.info("[AUTH] No authorization_id configured; skipping authorization step.")
 
         # -----------------------------
-        # 3) Agent Space Agent (create or update by displayName under engine)
+        # 3) Agent Space Agent (create or update)
         # -----------------------------
         self.info(f"[AGENT] Resolving by display_name={self._agent_space_name}")
         existing_agent = self.find_agent_space_agents_by_display(self._agent_space_name)
         self.info(f"[AGENT] find_agent_space_agents_by_display -> {json.dumps(existing_agent, indent=2)}")
 
-        headers, payload = self._get_agent_space_payload(engine_rn)
-        headers_log = headers.copy()
-        if "Authorization" in headers_log:
-            headers_log["Authorization"] = "Bearer ***"
-        self.info(f"[AGENT] request headers={json.dumps(headers_log)}")
-        self.info(f"[AGENT] payload={json.dumps(payload, indent=2)}")
-
         if not existing_agent:
-            self.info(f"[AGENT] '{self._agent_space_name}' not found. Creating...")
+            # Fresh create
+            headers, payload = self._get_agent_space_payload(engine_rn)
             create_url = self._get_agent_space_agent_url_new()
             self.info(f"[AGENT] POST {create_url}")
             cr = self._http.post(create_url, headers=headers, json=payload, timeout=90)
-            self.info(f"[AGENT] POST status={cr.status_code} ct={cr.headers.get('content-type','')}")
-            try:
-                self.info(f"[AGENT] POST body={json.dumps(cr.json(), indent=2)[:4000]}")
-            except Exception:
-                self.info(f"[AGENT] POST text={(cr.text or '')[:1000]}")
-            if cr.status_code >= 400:
-                self.error(f"[AGENT] Creation failed [{cr.status_code}].")
-                cr.raise_for_status()
-
-            try:
-                j = cr.json() or {}
-            except Exception:
-                j = {}
-            agent_name = j.get("name")
-            self.info(f"[AGENT] create response name={agent_name}")
+            self.info(f"[AGENT] POST status={cr.status_code}")
+            cr.raise_for_status()
+            agent_name = (cr.json() or {}).get("name")
             if agent_name:
                 self._write_engine_deployment({"agent_space_id": agent_name})
                 self.info(f"[AGENT] Created: {agent_name}")
             else:
                 self.warning("[AGENT] Created but response missing name. Verify in console.")
         else:
-            self.info(f"[AGENT] '{self._agent_space_name}' exists. Updating...")
-            patch_url = f"{DISCOVERY_ENGINE_URL}/{existing_agent['full_name']}"
-            self.info(f"[AGENT] PATCH {patch_url}")
-            ur = self._http.patch(patch_url, headers=headers, json=payload, timeout=90)
-            self.info("payload")
-            self.info(payload)
-            self.info(f"[AGENT] PATCH status={ur.status_code} ct={ur.headers.get('content-type','')}")
-            try:
-                self.info(f"[AGENT] PATCH body={json.dumps(ur.json(), indent=2)[:4000]}")
-            except Exception:
-                self.info(f"[AGENT] PATCH text={(ur.text or '')[:1000]}")
-            if ur.status_code >= 400:
-                self.error(f"[AGENT] Update failed [{ur.status_code}].")
-                ur.raise_for_status()
-            self.info("[AGENT] Agent Space agent updated.")
+            # Safe patch using our new helper
+            self.info(f"[AGENT] '{self._agent_space_name}' exists. Patching metadata and auth only...")
+            patched = self.patch_agent_space_metadata_and_auth(
+                agent_id=existing_agent["id"],
+                new_display_name=self._agent_space_name,
+                new_description=self._agent_space_description,
+                new_reasoning_engine=engine_rn,
+                new_authorizations=[
+                    f"projects/{self._project_id}/locations/global/authorizations/{self._authorization_id}"
+                ] if self._authorization_id else None,
+            )
+            self.info(f"[AGENT] PATCH result -> {json.dumps(patched, indent=2)[:2000]}")
 
         self.info("GitHub deployment completed successfully.")

{reasoning_deployment_service-0.4.3.dist-info → reasoning_deployment_service-0.4.4.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 reasoning_deployment_service/__init__.py,sha256=xDuKt9gGviQiTV6vXBdkBvygnlAOIrwnUjVaMGZy0L4,670
-reasoning_deployment_service/reasoning_deployment_service.py,sha256=Bug9Ea_Vk4egMyTUt9C8Jfx0BQv57gSjOJBZx4tQQl8,42304
+reasoning_deployment_service/reasoning_deployment_service.py,sha256=IEYDkhjZa0xE2uxTqsF5n5DHPhagHpnbeicHycrdkjk,43897
 reasoning_deployment_service/runner.py,sha256=Y_SPGp26eoXRr8Ql6ugRF7jNYq3OwaytFIivuym6ICA,5428
 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.4.3.dist-info/METADATA,sha256=rijIu4VInnB_noffloRO73jVUbrHW0wwDTGLAiKXzKg,5302
-reasoning_deployment_service-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-reasoning_deployment_service-0.4.3.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
-reasoning_deployment_service-0.4.3.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
-reasoning_deployment_service-0.4.3.dist-info/RECORD,,
+reasoning_deployment_service-0.4.4.dist-info/METADATA,sha256=kB2YwVjqXGrfOw2J2u8G50qeV89W0coPI8r1K5Ucg6s,5302
+reasoning_deployment_service-0.4.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+reasoning_deployment_service-0.4.4.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
+reasoning_deployment_service-0.4.4.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
+reasoning_deployment_service-0.4.4.dist-info/RECORD,,