reasoning-deployment-service 0.3.5__tar.gz → 0.3.6__tar.gz

reasoning_deployment_service-0.3.6/PKG-INFO

@@ -0,0 +1,211 @@
+Metadata-Version: 2.4
+Name: reasoning-deployment-service
+Version: 0.3.6
+Summary: Deployment helper for Vertex AI Reasoning Engines & Agent Spaces
+Author-email: Sergio Estrada <sergio.estrada@accenture.com>
+License: Apache-2.0
+Project-URL: Homepage, https://github.com/AxG-AI-Exchange-GenAI-Initiative/AIXAgentDeploymentService
+Project-URL: Repository, https://github.com/AxG-AI-Exchange-GenAI-Initiative/AIXAgentDeploymentService
+Project-URL: Issues, https://github.com/AxG-AI-Exchange-GenAI-Initiative/AIXAgentDeploymentService/issues
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.28
+Requires-Dist: python-dotenv>=1.0
+Requires-Dist: google-auth>=2.20
+Requires-Dist: google-cloud-storage>=2.16
+Requires-Dist: google-cloud-aiplatform[adk,agent-engines]>=1.88.0
+Requires-Dist: protobuf<7.0.0,>=4.21
+Requires-Dist: keyring>=24.0
+Requires-Dist: keyrings.google-artifactregistry-auth>=1.1
+Requires-Dist: PyYAML>=6.0
+Requires-Dist: click>=8.0
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+
+# Reasoning Deployment Service
+
+Helper package for deploying Vertex AI Reasoning Engines and Agent Spaces
+
+---
+
+## 📦 Installation
+
+Install the package directly from PyPI (version 0.3.5):
+
+```bash
+pip install reasoning-deployment-service==0.3.5
+```
+
+---
+
+## 🚀 First Run
+
+Run the CLI tool with no arguments:
+
+```bash
+reasoning-deploy
+```
+
+### What happens
+
+- You will see a list of available modes:
+
+```
+Choose an operation:
+1) Create/Update
+2) Auth only
+3) CLI
+4) GUI
+q) Quit
+```
+
+- If required files are missing, usage will be limited until you bootstrap the project
+- The tool will guide you to create the starter files:
+  - `your_agent_import.py` (shim pointing to your agent folder)
+  - `.env.agent` (environment and project settings)
+  - `aix_agent.yaml` (agent engine and space definition)
+
+Once these are in place, all modes will be fully available
+
+⚠️ Important: You must be running inside the same virtual environment where you installed reasoning-deployment-service
+
+---
+
+## ⚙️ Required Project Structure
+
+Your project must have the following at minimum:
+
+```
+my_project/
+├── requirements.txt       # pinned dependencies
+├── .env.agent             # environment and project config
+├── aix_agent.yaml         # engine and agent space definition
+├── my_agent/              # your agent folder
+│   └── agent.py           # defines root_agent
+```
+
+---
+
+### ✅ requirements.txt
+
+This file must be in your project root and contain all dependencies
+
+Example:
+
+```text
+google-adk==1.0.0
+google-cloud-aiplatform[adk,agent_engines]>=1.88.0
+google-generativeai>=0.7.0
+pydantic>=2.0.0
+python-dotenv>=1.0.0
+python-docx>=1.1.0
+google-api-python-client>=2.100.0
+google-auth>=2.25.0
+google-auth-httplib2>=0.2.0
+google-auth-oauthlib>=1.2.0
+absl-py>=2.0.0
+PyPDF2>=3.0.0
+deprecated>=1.2.14
+reasoning-deployment-service==0.3.5
+```
+
+Install with:
+
+```bash
+pip install -r requirements.txt
+```
+
+---
+
+### ✅ my_agent/agent.py
+
+Inside your agent folder (for example `my_agent`) you must define a variable called `root_agent`
+
+Example:
+
+```python
+from google.adk.agents import LlmAgent
+
+root_agent = LlmAgent(
+    name="my_agent",
+    model="gemini-2.5-flash",
+    description="My reasoning agent",
+    instruction="You do not do a whole lot yet"
+)
+```
+
+This `root_agent` instance is what gets deployed  
+It can be an `LlmAgent`, `Agent`, or any derived class
+
+---
+
+### ✅ .env.agent
+
+Environment specific config for DEV and PROD
+
+Example:
+
+```env
+DEV_PROJECT_ID=your-dev-project-id
+DEV_PROJECT_NUMBER=123456789012
+DEV_PROJECT_LOCATION=us-central1
+DEV_STAGING_BUCKET=gs://your-dev-staging-bucket
+DEV_AGENT_SPACE_ENGINE=your-dev-agentspace-engine
+DEV_OAUTH_CLIENT_ID=your-dev-oauth-client-id.apps.googleusercontent.com
+DEV_OAUTH_CLIENT_SECRET=your-dev-oauth-client-secret
+
+PROD_PROJECT_ID=
+PROD_PROJECT_NUMBER=
+PROD_PROJECT_LOCATION=
+PROD_STAGING_BUCKET=
+PROD_AGENT_SPACE_ENGINE=
+PROD_OAUTH_CLIENT_ID=
+PROD_OAUTH_CLIENT_SECRET=
+
+REASONING_DEPLOYMENT_VERSION=0.3.5
+
+DEVELOPER=your.name
+```
+
+---
+
+### ✅ aix_agent.yaml
+
+Example starter file:
+
+```yaml
+defaults:
+  scopes:
+    - https://www.googleapis.com/auth/cloud-platform
+    - https://www.googleapis.com/auth/userinfo.email
+  metadata:
+    reasoning_engine_name: reasoning-engine-dev
+    reasoning_engine_description: My Engine
+    agent_space_name: My Agent Space
+    agent_space_description: Example Agent Space
+    agent_space_tool_description: Example Tool
+  agent_folder: my_agent
+  auth:
+    oauth_authorization_id: test_auth
+  environment_variables:
+    - DEVELOPER
+```
+
+---
+
+## 🛠 Potential Issues
+
+- If you cannot run `reasoning-deploy`, try deleting your venv and making a new one
+- If you see a message saying you can only install on user site-packages, you will most likely need a fresh venv and new requirements install
+- Always ensure you are running `reasoning-deploy` inside the venv where it was installed
+
+---
+
+## ⚠️ Experimental Notice
+
+This tool is still experimental  
+If it takes longer than 20 to 30 minutes to set up, please go back to what already works for you and send logs and feedback to the development team

reasoning_deployment_service-0.3.6/README.md

@@ -0,0 +1,183 @@
+# Reasoning Deployment Service
+
+Helper package for deploying Vertex AI Reasoning Engines and Agent Spaces
+
+---
+
+## 📦 Installation
+
+Install the package directly from PyPI (version 0.3.5):
+
+```bash
+pip install reasoning-deployment-service==0.3.5
+```
+
+---
+
+## 🚀 First Run
+
+Run the CLI tool with no arguments:
+
+```bash
+reasoning-deploy
+```
+
+### What happens
+
+- You will see a list of available modes:
+
+```
+Choose an operation:
+1) Create/Update
+2) Auth only
+3) CLI
+4) GUI
+q) Quit
+```
+
+- If required files are missing, usage will be limited until you bootstrap the project
+- The tool will guide you to create the starter files:
+  - `your_agent_import.py` (shim pointing to your agent folder)
+  - `.env.agent` (environment and project settings)
+  - `aix_agent.yaml` (agent engine and space definition)
+
+Once these are in place, all modes will be fully available
+
+⚠️ Important: You must be running inside the same virtual environment where you installed reasoning-deployment-service
+
+---
+
+## ⚙️ Required Project Structure
+
+Your project must have the following at minimum:
+
+```
+my_project/
+├── requirements.txt       # pinned dependencies
+├── .env.agent             # environment and project config
+├── aix_agent.yaml         # engine and agent space definition
+├── my_agent/              # your agent folder
+│   └── agent.py           # defines root_agent
+```
+
+---
+
+### ✅ requirements.txt
+
+This file must be in your project root and contain all dependencies
+
+Example:
+
+```text
+google-adk==1.0.0
+google-cloud-aiplatform[adk,agent_engines]>=1.88.0
+google-generativeai>=0.7.0
+pydantic>=2.0.0
+python-dotenv>=1.0.0
+python-docx>=1.1.0
+google-api-python-client>=2.100.0
+google-auth>=2.25.0
+google-auth-httplib2>=0.2.0
+google-auth-oauthlib>=1.2.0
+absl-py>=2.0.0
+PyPDF2>=3.0.0
+deprecated>=1.2.14
+reasoning-deployment-service==0.3.5
+```
+
+Install with:
+
+```bash
+pip install -r requirements.txt
+```
+
+---
+
+### ✅ my_agent/agent.py
+
+Inside your agent folder (for example `my_agent`) you must define a variable called `root_agent`
+
+Example:
+
+```python
+from google.adk.agents import LlmAgent
+
+root_agent = LlmAgent(
+    name="my_agent",
+    model="gemini-2.5-flash",
+    description="My reasoning agent",
+    instruction="You do not do a whole lot yet"
+)
+```
+
+This `root_agent` instance is what gets deployed  
+It can be an `LlmAgent`, `Agent`, or any derived class
+
+---
+
+### ✅ .env.agent
+
+Environment specific config for DEV and PROD
+
+Example:
+
+```env
+DEV_PROJECT_ID=your-dev-project-id
+DEV_PROJECT_NUMBER=123456789012
+DEV_PROJECT_LOCATION=us-central1
+DEV_STAGING_BUCKET=gs://your-dev-staging-bucket
+DEV_AGENT_SPACE_ENGINE=your-dev-agentspace-engine
+DEV_OAUTH_CLIENT_ID=your-dev-oauth-client-id.apps.googleusercontent.com
+DEV_OAUTH_CLIENT_SECRET=your-dev-oauth-client-secret
+
+PROD_PROJECT_ID=
+PROD_PROJECT_NUMBER=
+PROD_PROJECT_LOCATION=
+PROD_STAGING_BUCKET=
+PROD_AGENT_SPACE_ENGINE=
+PROD_OAUTH_CLIENT_ID=
+PROD_OAUTH_CLIENT_SECRET=
+
+REASONING_DEPLOYMENT_VERSION=0.3.5
+
+DEVELOPER=your.name
+```
+
+---
+
+### ✅ aix_agent.yaml
+
+Example starter file:
+
+```yaml
+defaults:
+  scopes:
+    - https://www.googleapis.com/auth/cloud-platform
+    - https://www.googleapis.com/auth/userinfo.email
+  metadata:
+    reasoning_engine_name: reasoning-engine-dev
+    reasoning_engine_description: My Engine
+    agent_space_name: My Agent Space
+    agent_space_description: Example Agent Space
+    agent_space_tool_description: Example Tool
+  agent_folder: my_agent
+  auth:
+    oauth_authorization_id: test_auth
+  environment_variables:
+    - DEVELOPER
+```
+
+---
+
+## 🛠 Potential Issues
+
+- If you cannot run `reasoning-deploy`, try deleting your venv and making a new one
+- If you see a message saying you can only install on user site-packages, you will most likely need a fresh venv and new requirements install
+- Always ensure you are running `reasoning-deploy` inside the venv where it was installed
+
+---
+
+## ⚠️ Experimental Notice
+
+This tool is still experimental  
+If it takes longer than 20 to 30 minutes to set up, please go back to what already works for you and send logs and feedback to the development team

{reasoning_deployment_service-0.3.5 → reasoning_deployment_service-0.3.6}/pyproject.toml

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

{reasoning_deployment_service-0.3.5 → reasoning_deployment_service-0.3.6}/reasoning_deployment_service/cli_editor/api_client.py

@@ -30,6 +30,27 @@ EXCLUDES = [
 
 
 class ApiClient:
+    def get_authorization_info(self, auth_id: str) -> dict:
+        """Get details for a specific authorization by ID."""
+        if not self.is_live:
+            # Return mock info for testing
+            return {
+                "id": auth_id,
+                "scopes": [
+                    "https://www.googleapis.com/auth/cloud-platform",
+                    "https://www.googleapis.com/auth/userinfo.email"
+                ],
+                "status": "mock"
+            }
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self.project_id,
+        }
+        url = f"https://discoveryengine.googleapis.com/v1alpha/projects/{self.project_id}/locations/global/authorizations/{auth_id}"
+        r = self._http.get(url, headers=headers, timeout=60)
+        r.raise_for_status()
+        return r.json()
     """
     Single responsibility: hold configuration & credentials and expose API calls.
     This class has both 'live' and 'mock' modes; the public surface is identical.
@@ -559,6 +580,63 @@ class ApiClient:
         else:
             return ("failed", f"{r.status_code} {r.text}")
 
+    def get_authorization_info(self, auth_id: str) -> Dict[str, Any]:
+        """Get details for a specific authorization by ID."""
+        if not self.is_live:
+            # Return mock info for testing
+            return {
+                "id": auth_id,
+                "scopes": [
+                    "https://www.googleapis.com/auth/cloud-platform",
+                    "https://www.googleapis.com/auth/userinfo.email"
+                ],
+                "status": "mock"
+            }
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self.project_id,
+        }
+        url = f"{BASE_URL}/projects/{self.project_id}/locations/global/authorizations/{auth_id}"
+        r = self._http.get(url, headers=headers, timeout=60)
+        r.raise_for_status()
+        return r.json()
+
+    def update_authorization_scopes(self, auth_id: str, scopes: list) -> Dict[str, Any]:
+        """Patch the scopes for a specific authorization by ID, updating the authorizationUri as well."""
+        if not self.is_live:
+            return {
+                "id": auth_id,
+                "scopes": scopes,
+                "status": "mock-patched"
+            }
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self.project_number,
+        }
+        url = f"{BASE_URL}/projects/{self.project_number}/locations/global/authorizations/{auth_id}?update_mask=server_side_oauth2.authorization_uri"
+        scopes_str = "%20".join(scopes)
+        authorization_uri = (
+            "https://accounts.google.com/o/oauth2/auth"
+            "?response_type=code"
+            f"&client_id={self.oauth_client_id or 'your-client-id'}"
+            f"&scope={scopes_str}"
+            "&access_type=offline&prompt=consent"
+        )
+
+        payload = {
+            "serverSideOauth2": {
+                "authorizationUri": authorization_uri
+            }
+        }
+        r = self._http.patch(url, headers=headers, json=payload, timeout=60)
+        print(r.status_code)
+        print(r.text)
+        print(r.json())
+        r.raise_for_status()
+        return r.json()
+
     def _ensure_authorization(self, auth_name: str) -> Tuple[bool, str]:
         if not self.is_live:
             time.sleep(0.02)
@@ -663,4 +741,60 @@ class ApiClient:
         }
         self._profile["agent_space_agent_id"] = full
         self._save_profile()
-        return ("created", "Deployed", item)
+        return ("created", "Deployed", item)
+
+    def remove_authorization_from_agent(self, agent_id: str) -> Dict[str, Any]:
+        """Remove all authorizations from an agent by PATCHing with an empty authorizations list."""
+        if not self.is_live:
+            return {
+                "id": agent_id,
+                "authorizations": [],
+                "status": "mock-removed"
+            }
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self.project_id,
+        }
+        url = f"{BASE_URL}/projects/{self.project_id}/locations/global/assistants/default_assistant/agents/{agent_id}"
+        payload = {"authorizations": []}
+        r = self._http.patch(url, headers=headers, json=payload, timeout=60)
+        r.raise_for_status()
+        return r.json()
+
+    def drop_agent_authorizations(self, agent_id: str) -> Dict[str, Any]:
+        """Drop all authorizations for an agent space agent by PATCHing with all attributes except authorizations (fully omitted)."""
+        if not self.is_live:
+            return {
+                "id": agent_id,
+                "authorizations": [],
+                "status": "mock-dropped"
+            }
+        headers = {
+            "Authorization": f"Bearer {self._access_token()}",
+            "Content-Type": "application/json",
+            "X-Goog-User-Project": self.project_id,
+        }
+        url = (f"{BASE_URL}/projects/{self.project_id}/locations/global/collections/default_collection/"
+               f"engines/{self.engine_name}/assistants/default_assistant/agents/{agent_id}")
+        # GET the current agent definition
+        get_resp = self._http.get(url, headers=headers, timeout=60)
+        get_resp.raise_for_status()
+        agent_data = get_resp.json()
+        # Build PATCH payload: copy all attributes except authorizations
+        patch_payload = {}
+        for key in ["displayName", "description"]:
+            if key in agent_data:
+                patch_payload[key] = agent_data[key]
+        adk_def = agent_data.get("adkAgentDefinition", {})
+        patch_adk_def = {}
+        # Copy all adkAgentDefinition fields except 'authorizations'
+        for k, v in adk_def.items():
+            if k != "authorizations":
+                patch_adk_def[k] = v
+        if patch_adk_def:
+            patch_payload["adk_agent_definition"] = patch_adk_def
+        # PATCH with the new payload
+        patch_resp = self._http.patch(url, headers=headers, json=patch_payload, timeout=60)
+        patch_resp.raise_for_status()
+        return patch_resp.json()

{reasoning_deployment_service-0.3.5 → reasoning_deployment_service-0.3.6}/reasoning_deployment_service/cli_editor/cli_runner.py

@@ -43,6 +43,14 @@ class CLIRunner:
         subparsers.add_parser("list-authorizations", help="List all authorizations in the project")
         delete_auth_parser = subparsers.add_parser("delete-authorization", help="Delete an authorization by list position")
         delete_auth_parser.add_argument("position", help="Position number from the list (e.g., 1, 2, 3...)")
+        # New commands
+        get_auth_info_parser = subparsers.add_parser("get-authorization-info", help="Get details for a specific authorization")
+        get_auth_info_parser.add_argument("auth_id", help="Authorization ID")
+        update_auth_scopes_parser = subparsers.add_parser("update-authorization-scopes", help="Update scopes for a specific authorization")
+        update_auth_scopes_parser.add_argument("auth_id", help="Authorization ID")
+        update_auth_scopes_parser.add_argument("scopes", help="Comma-separated list of scopes")
+        drop_agent_auth_parser = subparsers.add_parser("drop-agent-authorizations", help="Drop all authorizations for an agent space agent")
+        drop_agent_auth_parser.add_argument("agent_id", help="Agent ID")
 
     def run(self):
         # Load environment variables from .env.agent file (same as GUI editor)
@@ -113,12 +121,20 @@ class CLIRunner:
                     self._list_authorizations()
                 elif args.command == "delete-authorization":
                     self._delete_authorization(args.position)
+                elif args.command == "get-authorization-info":
+                    self._get_authorization_info(args.auth_id)
+                elif args.command == "update-authorization-scopes":
+                    scopes = [s.strip() for s in args.scopes.split(",") if s.strip()]
+                    self._update_authorization_scopes(args.auth_id, scopes)
+                elif args.command == "drop-agent-authorizations":
+                    self._drop_agent_authorizations(args.agent_id)
                 else:
                     self.parser.print_help()
             except SystemExit:
                 # Catch argparse's SystemExit to prevent CLI from closing
                 print("Invalid command. Please try again.")
             except Exception as e:
+                raise
                 print(f"An error occurred: {e}")
         print("\nAvailable commands:")
         print("  list-engines            List all reasoning engines in the project")
@@ -339,5 +355,35 @@ class CLIRunner:
         except Exception as e:
             print(f"❌ Error deleting authorization: {e}")
 
+    def _get_authorization_info(self, auth_id):
+        """Get details for a specific authorization."""
+        print(f"🔍 Getting info for authorization: {auth_id}")
+        try:
+            info = self.api_client.get_authorization_info(auth_id)
+            from pprint import pprint
+            pprint(info)
+        except Exception as e:
+            print(f"❌ Error getting authorization info: {e}")
+
+    def _update_authorization_scopes(self, auth_id, scopes):
+        """Update scopes for a specific authorization."""
+        print(f"🔧 Updating scopes for authorization: {auth_id}")
+        print(f"New scopcdscsdes: {scopes}")
+
+        result = self.api_client.update_authorization_scopes(auth_id, scopes)
+        from pprint import pprint
+        pprint(result)
+
+    def _drop_agent_authorizations(self, agent_id):
+        """Drop all authorizations for an agent space agent."""
+        print(f"🗑️ Dropping all authorizations for agent: {agent_id}")
+        try:
+            result = self.api_client.drop_agent_authorizations(agent_id)
+            from pprint import pprint
+            pprint(result)
+        except Exception as e:
+            raise
+            print(f"❌ Error dropping agent authorizations: {e}")
+
 if __name__ == "__main__":
     CLIRunner().run()

reasoning_deployment_service-0.3.6/reasoning_deployment_service/gui_editor/agent_checkbox_list.py

@@ -0,0 +1,48 @@
+"""
+Scrollable agent list with checkboxes for mass selection and actions.
+"""
+import tkinter as tk
+from tkinter import ttk, messagebox
+
+class AgentCheckboxList(tk.Frame):
+    def __init__(self, master, agents, on_mass_action):
+        super().__init__(master)
+        self.agents = agents
+        self.on_mass_action = on_mass_action
+        self.vars = {}
+        self._setup_ui()
+
+    def _setup_ui(self):
+        canvas = tk.Canvas(self)
+        scrollbar = ttk.Scrollbar(self, orient="vertical", command=canvas.yview)
+        self.scrollable_frame = tk.Frame(canvas)
+
+        self.scrollable_frame.bind(
+            "<Configure>",
+            lambda e: canvas.configure(scrollregion=canvas.bbox("all"))
+        )
+
+        canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw")
+        canvas.configure(yscrollcommand=scrollbar.set, height=300)
+
+        canvas.pack(side="left", fill="both", expand=True)
+        scrollbar.pack(side="right", fill="y")
+
+        for agent in self.agents:
+            var = tk.BooleanVar()
+            self.vars[agent["id"]] = var
+            cb = tk.Checkbutton(self.scrollable_frame, text=f"{agent.get('display_name', agent['id'])}", variable=var)
+            cb.pack(anchor="w", padx=4, pady=2)
+
+        self.action_btn = ttk.Button(self, text="Drop Authorizations for Checked Agents", command=self._do_mass_action)
+        self.action_btn.pack(fill="x", pady=8)
+
+    def get_checked_agent_ids(self):
+        return [aid for aid, var in self.vars.items() if var.get()]
+
+    def _do_mass_action(self):
+        checked = self.get_checked_agent_ids()
+        if not checked:
+            messagebox.showinfo("No agents selected", "Please check agents to drop authorizations.")
+            return
+        self.on_mass_action(checked)