reasoning-deployment-service 0.4.3__tar.gz → 0.4.6__tar.gz

{reasoning_deployment_service-0.4.3 → reasoning_deployment_service-0.4.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reasoning-deployment-service
-Version: 0.4.3
+Version: 0.4.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.4.3 → reasoning_deployment_service-0.4.6}/pyproject.toml

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

{reasoning_deployment_service-0.4.3 → reasoning_deployment_service-0.4.6}/reasoning_deployment_service/reasoning_deployment_service.py

@@ -766,8 +766,82 @@ 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
 
-    def one_github_deployment_to_go(self):
+        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}"
+        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_githhub_deployment_to_go_with_skip(self):
+        return self.one_github_deployment_to_go(skip_engine=True)
+
+
+    def one_github_deployment_to_go(self, skip_engine=False):
         """
         CI-friendly deploy:
         - Engine: create or update by display_name.
@@ -800,19 +874,20 @@ class ReasoningEngineDeploymentService:
         engine_rn = self.find_engine_by_name(self._reasoning_engine_name)
         self.info(f"[ENGINE] find_engine_by_name -> {engine_rn}")
 
-        if not engine_rn:
-            self.info(f"[ENGINE] '{self._reasoning_engine_name}' not found. Creating...")
-            self.create_reasoning_engine()
-            rec_after_create = self._read_engine_deployment_record()
-            self.info(f"[ENGINE] record after create -> {json.dumps(rec_after_create, indent=2)}")
-            engine_rn = rec_after_create.get("reasoning_engine_id") or self.find_engine_by_name(self._reasoning_engine_name)
-            self.info(f"[ENGINE] post-create resolution -> {engine_rn}")
+        if not skip_engine:
             if not engine_rn:
-                self.error("[ENGINE] Creation did not yield a resource name.")
-                raise RuntimeError("Engine creation failed.")
-        else:
-            self.info(f"[ENGINE] '{self._reasoning_engine_name}' exists. Updating...")
-            self.update_reasoning_engine(engine_rn)
+                self.info(f"[ENGINE] '{self._reasoning_engine_name}' not found. Creating...")
+                self.create_reasoning_engine()
+                rec_after_create = self._read_engine_deployment_record()
+                self.info(f"[ENGINE] record after create -> {json.dumps(rec_after_create, indent=2)}")
+                engine_rn = rec_after_create.get("reasoning_engine_id") or self.find_engine_by_name(self._reasoning_engine_name)
+                self.info(f"[ENGINE] post-create resolution -> {engine_rn}")
+                if not engine_rn:
+                    self.error("[ENGINE] Creation did not yield a resource name.")
+                    raise RuntimeError("Engine creation failed.")
+            else:
+                self.info(f"[ENGINE] '{self._reasoning_engine_name}' exists. Updating...")
+                self.update_reasoning_engine(engine_rn)
 
         self.info(f"[ENGINE] final engine_rn={engine_rn}")
 
@@ -877,7 +952,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 +973,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 → reasoning_deployment_service-0.4.6}/reasoning_deployment_service/runner.py

@@ -13,7 +13,7 @@ class Runner:
         parser = argparse.ArgumentParser(description="Reasoning Engine Deployment Runner")
         parser.add_argument(
             "--mode",
-            choices=["create", "auth", "cli", "gui", "populate_files", "github_deployment"],
+            choices=["create", "auth", "cli", "gui", "populate_files", "github_deployment", "github_deployment_without_engine"],
             help="Operation mode to run",
         )
         parser.add_argument(
@@ -133,12 +133,14 @@ class Runner:
             svc._check_required_files_exist()
         elif mode == "github_deployment":
             svc.one_github_deployment_to_go()
+        elif mode == "github_deployment_without_engine":
+            svc.one_githhub_deployment_to_go_with_skip()
 
     @staticmethod
     def _menu(root_agent):
         print("Choose an operation:\n1) Create/Update\n2) Auth only\n3) CLI\n4) GUI\nq) Quit")
         choice = input("Enter choice: ").strip().lower()
-        mapping = {"1": "create", "2": "auth", "3": "cli", "4": "gui", "5": "github_deployment"}
+        mapping = {"1": "create", "2": "auth", "3": "cli", "4": "gui", "5": "github_deployment", "6": "github_deployment_without_engine"}
         if choice == "q":
             sys.exit(0)
         Runner._dispatch(mapping.get(choice, ""), root_agent)

{reasoning_deployment_service-0.4.3 → reasoning_deployment_service-0.4.6}/reasoning_deployment_service.egg-info/PKG-INFO

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