Skip to content

Commit 0199df4

Browse files
authored
Merge branch 'main' into averikitsch-patch-1
2 parents 438a4d9 + 0510db6 commit 0199df4

18 files changed

Lines changed: 573 additions & 150 deletions

.agents/plugins/marketplace.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
{
2-
"name": "data-cloud-ai-dev-kit-marketplace",
2+
"name": "data-agent-kit-starter-pack-marketplace",
33
"interface": {
4-
"displayName": "data-cloud-ai-dev-kit Marketplace"
4+
"displayName": "Data Agent Kit Starter Pack Marketplace"
55
},
66
"plugins": [
77
{
8-
"name": "data-cloud-ai-dev-kit",
8+
"name": "data-agent-kit-starter-pack",
99
"source": {
1010
"source": "local",
1111
"path": "./"

.claude-plugin/marketplace.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
{
2-
"name": "data-cloud-ai-dev-kit-marketplace",
2+
"name": "data-agent-kit-starter-pack-marketplace",
33
"owner": {
44
"name": "Google LLC"
55
},
66
"plugins": [
77
{
8-
"name": "data-cloud-ai-dev-kit",
8+
"name": "data-agent-kit-starter-pack",
99
"source": "./",
1010
"description": "This plugin provides a specialized suite of skills for data engineers and database practitioners working on Google Cloud. It acts as an expert assistant, allowing you to use natural language prompts in your preferred coding agent to architect complex data pipelines, transform data with dbt, write Spark and BigQuery SQL notebooks, and orchestrate end-to-end workflows across GCP's data ecosystem."
1111
}

.claude-plugin/plugin.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
{
2-
"name": "data-cloud-ai-dev-kit",
2+
"name": "data-agent-kit-starter-pack",
33
"version": "0.1.8",
44
"description": "This plugin provides a specialized suite of skills for data engineers and database practitioners working on Google Cloud. It acts as an expert assistant, allowing you to use natural language prompts in your preferred coding agent to architect complex data pipelines, transform data with dbt, write Spark and BigQuery SQL notebooks, and orchestrate end-to-end workflows across GCP's data ecosystem.",
55
"mcp": "./.mcp.json"

.codex-plugin/plugin.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
{
2-
"name": "data-cloud-ai-dev-kit",
2+
"name": "data-agent-kit-starter-pack",
33
"version": "0.1.8",
44
"description": "This plugin provides a specialized suite of skills for data engineers and database practitioners working on Google Cloud. It acts as an expert assistant, allowing you to use natural language prompts in your preferred coding agent to architect complex data pipelines, transform data with dbt, write Spark and BigQuery SQL notebooks, and orchestrate end-to-end workflows across GCP's data ecosystem.",
55
"skills": "./skills/",
66
"interface": {
7-
"displayName": "Data Cloud AI Dev Kit",
8-
"shortDescription": "Data Cloud AI Dev Kit",
7+
"displayName": "Data Agent Kit Starter Pack",
8+
"shortDescription": "Data Agent Kit Starter Pack",
99
"category": "Productivity"
1010
},
1111
"mcp": "./.mcp.json"

.lycheeignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit/compare/
22
https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit
33
https://raw.githubusercontent.com/gemini-cli-extensions/data-cloud-ai-dev-kit/main/codex-install.sh
4+
https://docs.cloud.google.com/data-cloud-extension/vs-code/install
45
data-cloud-ai-dev-kit@data-cloud-ai-dev-kit-marketplace

.mcp.json

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"command": "npx",
55
"args": [
66
"-y",
7-
"git+https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit.git"
7+
"git+https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack.git"
88
],
99
"env": {},
1010
"env_vars": [
@@ -20,7 +20,7 @@
2020
"bigquery",
2121
"--stdio",
2222
"--user-agent-metadata",
23-
"google-cloud-data-agent-kit-claude-and-codex"
23+
"datacloud.claude-and-codex"
2424
],
2525
"env": {
2626
"BIGQUERY_LOCATION": "",
@@ -36,7 +36,7 @@
3636
"spanner",
3737
"--stdio",
3838
"--user-agent-metadata",
39-
"google-cloud-data-agent-kit-claude-and-codex"
39+
"datacloud.claude-and-codex"
4040
],
4141
"env": {
4242
"SPANNER_DATABASE": "",
@@ -54,7 +54,7 @@
5454
"alloydb-postgres-admin",
5555
"--stdio",
5656
"--user-agent-metadata",
57-
"google-cloud-data-agent-kit-claude-and-codex"
57+
"datacloud.claude-and-codex"
5858
],
5959
"env": {}
6060
},
@@ -67,7 +67,7 @@
6767
"alloydb-postgres",
6868
"--stdio",
6969
"--user-agent-metadata",
70-
"google-cloud-data-agent-kit-claude-and-codex"
70+
"datacloud.claude-and-codex"
7171
],
7272
"env": {
7373
"ALLOYDB_POSTGRES_CLUSTER": "",
@@ -89,7 +89,7 @@
8989
"cloud-sql-postgres-admin",
9090
"--stdio",
9191
"--user-agent-metadata",
92-
"google-cloud-data-agent-kit-claude-and-codex"
92+
"datacloud.claude-and-codex"
9393
],
9494
"env": {}
9595
},
@@ -102,7 +102,7 @@
102102
"cloud-sql-postgres",
103103
"--stdio",
104104
"--user-agent-metadata",
105-
"google-cloud-data-agent-kit-claude-and-codex"
105+
"datacloud.claude-and-codex"
106106
],
107107
"env": {
108108
"CLOUD_SQL_POSTGRES_DATABASE": "",
@@ -123,7 +123,7 @@
123123
"dataplex",
124124
"--stdio",
125125
"--user-agent-metadata",
126-
"google-cloud-data-agent-kit-claude-and-codex"
126+
"datacloud.claude-and-codex"
127127
],
128128
"env": {
129129
"DATAPLEX_PROJECT": ""
@@ -138,7 +138,7 @@
138138
"dataproc",
139139
"--stdio",
140140
"--user-agent-metadata",
141-
"google-cloud-data-agent-kit-claude-and-codex"
141+
"datacloud.claude-and-codex"
142142
],
143143
"env": {
144144
"DATAPROC_PROJECT": "",
@@ -154,7 +154,7 @@
154154
"serverless-spark",
155155
"--stdio",
156156
"--user-agent-metadata",
157-
"google-cloud-data-agent-kit-claude-and-codex"
157+
"datacloud.claude-and-codex"
158158
],
159159
"env": {
160160
"SERVERLESS_SPARK_PROJECT": "",

DEVELOPER.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# DEVELOPER.md
22

33
This document provides instructions for setting up your development environment
4-
and contributing to the Cloud SQL for PostgreSQL Gemini CLI Extension project.
4+
and contributing to the Data Agent Kit Starter Pack project.
55

66
## Prerequisites
77

@@ -22,8 +22,8 @@ The core logic for this extension is handled by a pre-built `toolbox` binary. Th
2222
1. **Clone the Repository:**
2323

2424
```bash
25-
git clone https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit.git
26-
cd data-cloud-ai-dev-kit
25+
git clone https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack.git
26+
cd data-agent-kit-starter-pack
2727
```
2828

2929
2. **Download the Toolbox Binary:** The required version of the `toolbox` binary
@@ -48,7 +48,7 @@ The core logic for this extension is handled by a pre-built `toolbox` binary. Th
4848
The CLI will prompt you to confirm the installation. Accept it to proceed.
4949

5050
4. **Testing Changes:** After installation, start the Gemini CLI (`gemini`).
51-
You can now interact with the `data-cloud-ai-dev-kit` tools to manually test your changes
51+
You can now interact with the `data-agent-kit-starter-pack` tools to manually test your changes
5252
against your connected database.
5353

5454
### Adding or Updating Skills

README.md

Lines changed: 93 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,49 +1,115 @@
1-
# Data Cloud AI Dev Kit
1+
# Data Agent Kit Starter Pack
22

33
> [!NOTE]
44
> This extension is currently in beta (pre-v1.0), and may see breaking changes until the first stable release (v1.0).
55
6-
This plugin provides a specialized suite of skills and MCP tools for data engineers and database practitioners working on Google Cloud. It acts as an expert assistant, allowing you to use natural language prompts in your preferred coding agent to architect complex data pipelines, transform data with dbt, write Spark and BigQuery SQL notebooks, and orchestrate end-to-end workflows across GCP's data ecosystem (BigQuery, Spanner, BigLake, Dataproc, etc.).
6+
This plugin provides a specialized suite of skills and MCP tools for data engineers and database practitioners working on Google Cloud. It acts as an expert assistant, allowing you to use natural language prompts in your preferred coding agent to architect complex data pipelines, transform data with dbt, write Spark and BigQuery SQL notebooks, and orchestrate end-to-end workflows across the Google Cloud data ecosystem (BigQuery, Spanner, BigLake, Dataproc, etc.).
77

88
> [!IMPORTANT]
99
> **We Want Your Feedback!**
1010
> Please share your thoughts with us by opening an issue on GitHub. Your input is invaluable and helps us improve the project for everyone.
1111
12-
## Why Use the Data Cloud AI Dev Kit?
12+
## Why Use the Data Agent Kit Starter Pack?
1313

14-
* **Seamless Workflow:** Brings Google Cloud data engineering expertise directly into your terminal or IDE via Gemini CLI, Claude Code, or Codex.
15-
* **End-to-End Data Pipelines:** Effortlessly generate code that reads raw data from GCS, processes it with Spark or BigQuery, transforms it through medallion architectures (bronze, silver, gold) using dbt, and exports it to serving layers like Cloud Spanner.
16-
* **Ecosystem Integration:** Work across boundaries—generate BigLake Iceberg catalog tables, train BigQuery ML models (XGBoost, KMEANS), and create interactive Streamlit dashboards or LookML models all from natural language.
14+
* **Seamless Workflow:** Bring Google Cloud data engineering expertise directly into your terminal or IDE via Gemini CLI, Claude Code, or Codex.
15+
* **End-to-End Data Pipelines:** Effortlessly generate code that reads raw data from Cloud Storage, processes it with Spark or BigQuery, transform it through medallion architectures (bronze, silver, gold) using dbt, and export it to serving layers like Spanner.
16+
* **Ecosystem Integration:** Work across boundaries—generate BigLake Iceberg catalog tables, train BigQuery ML models (XGBoost, KMEANS), and create interactive Streamlit dashboards or LookML models, all from natural language.
1717
* **Workflow Orchestration:** Automatically create and schedule orchestration pipelines that tie your notebooks and dbt models together into robust, scheduled jobs.
1818

1919
## Prerequisites
2020

21-
Ensure you have one of the following coding agents installed:
22-
* [Gemini CLI](https://github.com/google-gemini/gemini-cli) (v0.6.0+)
23-
* [Claude Code](https://docs.anthropic.com/en/docs/agents-and-tools/claude-code/overview)
24-
* Codex CLI
21+
Ensure you have the following installed:
22+
* **Node.js and npm** (Latest version recommended)
23+
* One of the following coding agents:
24+
* [Gemini CLI](https://github.com/google-gemini/gemini-cli) (v0.6.0+)
25+
* [Claude Code](https://docs.anthropic.com/en/docs/agents-and-tools/claude-code/overview)
26+
* Codex CLI
27+
* **(Optional) IDE Extension:** [Google Cloud Data Agent Kit](https://docs.cloud.google.com/data-cloud-extension/vs-code/install).
2528

2629
## Getting Started
2730

2831
### Installation
2932

30-
Choose the installation method for your preferred coding agent.
33+
Choose the installation method for your preferred coding agent. Run the commands in terminal
3134

32-
#### Gemini CLI
35+
<details>
36+
<summary><b>Gemini CLI and Gemini Code Assist</b></summary>
37+
38+
```bash
39+
gemini extensions install https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack
40+
```
41+
</details>
42+
43+
<details>
44+
<summary><b>Claude Code</b></summary>
45+
46+
Run the `claude` command to start the agent, then run:
47+
48+
```bash
49+
/plugin marketplace add https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack
50+
/plugin install data-agent-kit-starter-pack@data-agent-kit-starter-pack-marketplace
51+
```
52+
</details>
53+
54+
<details>
55+
<summary><b>Codex</b></summary>
56+
57+
Run the following commands in your terminal:
58+
59+
**macOS / Linux:**
3360
```bash
34-
gemini extensions install https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit
61+
curl -sSL https://raw.githubusercontent.com/gemini-cli-extensions/data-agent-kit-starter-pack/main/codex-install.sh | bash
3562
```
3663

37-
#### Claude Code
64+
**Windows:**
65+
```powershell
66+
irm https://raw.githubusercontent.com/gemini-cli-extensions/data-agent-kit-starter-pack/main/codex-install.ps1 | iex
67+
```
68+
69+
After running the installation script, run the `codex` command to start the agent, then run:
70+
3871
```bash
39-
/plugin marketplace add https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit
40-
/plugin install data-cloud-ai-dev-kit@data-cloud-ai-dev-kit-marketplace
72+
/plugins
4173
```
4274

43-
#### Codex (One-Liner)
75+
Use the interactive options to install the extension with the name `Data Agent Kit Starter Pack`.
76+
</details>
77+
78+
### Configuration
79+
80+
MCP toolboxes are added to the respective agent configuration files. You must configure the MCP toolboxes in your agent's configuration files for them to start successfully.
81+
82+
In all cases, you must restart the agent after updating the configuration files.
83+
84+
<details>
85+
<summary><b>Gemini CLI and Gemini Code Assist</b></summary>
86+
87+
Edit the configuration file:
88+
`~/.gemini/extensions/data-agent-kit-starter-pack/gemini-extension.json`
89+
</details>
90+
91+
<details>
92+
<summary><b>Claude Code</b></summary>
93+
94+
1. Edit the configuration file:
95+
`~/.claude/plugins/cache/data-agent-kit-starter-pack-marketplace/data-agent-kit-starter-pack/<version>/.mcp.json`
96+
97+
2. Reinstall the plugin:
98+
Run `/plugin` and use interactive options to uninstall `data-agent-kit-starter-pack`. Then run `/plugin install` to add it back.
99+
</details>
100+
101+
<details>
102+
<summary><b>Codex</b></summary>
103+
104+
1. Edit the configuration file:
105+
`~/.agents/plugins/data-agent-kit-starter-pack/.mcp.json`
106+
107+
2. Use interactive options to uninstall and install the extension:
44108
```bash
45-
curl -sSL https://raw.githubusercontent.com/gemini-cli-extensions/data-cloud-ai-dev-kit/main/codex-install.sh | bash
109+
/plugins
46110
```
111+
Install with name: `Data Agent Kit Starter Pack`
112+
</details>
47113

48114
## Usage Examples
49115

@@ -67,14 +133,15 @@ Interact with your coding agent using natural language prompts to perform comple
67133

68134
* **Plugin Not Found:** Ensure you have restarted your agent (e.g., Gemini CLI or Codex) after installation.
69135
* **Authentication Errors:** Many GCP skills require an active authenticated session. Ensure you have run `gcloud auth login` and `gcloud auth application-default login` on your machine.
136+
* **MCP Connection Issues:** Update the MCP server configurations such as project, region etc. needed by MCP toolboxes in order to connect successfully to them.
70137

71138
## Security Reminder: Agent Environment Hardening
72139

73-
Your agent has the power to
74-
execute tools and commands on your behalf. Protect your GCP resources by
75-
enforcing **Strict Least Privilege** across all CLIs, MCP servers and other
76-
resources available to your agents. For example, use scoped Service Accounts
77-
(read more
78-
[here](https://docs.cloud.google.com/docs/authentication/use-service-account-impersonation))
79-
for tasks accessing your cloud resources and conducting regular permission and
80-
agent settings audits to minimize your attack surface.
140+
Your agent has the power to execute tools and commands on your behalf. Protect your GCP resources by enforcing **Strict Least Privilege** across all CLIs, MCP servers and other resources available to your agents.
141+
142+
* Use [service accounts](https://docs.cloud.google.com/docs/authentication/use-service-account-impersonation) for accessing your cloud resources.
143+
* Assign the service account a role with [limited permissions](https://docs.cloud.google.com/iam/docs/roles-overview).
144+
* Prevent unwanted cross-org agent access by utilizing **Principal Access Boundaries** to scope your agent to [projects](https://docs.cloud.google.com/iam/docs/principal-access-boundary-policies#use-case-one-project) you intend the agent to access.
145+
146+
> [!NOTE]
147+
> The Principal Access Boundary condition should bind the policy to the service accounts you intend to restrict.

codex-install.ps1

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
1414

1515
$ErrorActionPreference = "Stop"
1616

17-
$pluginName = "data-cloud-ai-dev-kit"
18-
$repoUrl = "https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit"
17+
$pluginName = "data-agent-kit-starter-pack"
18+
$repoUrl = "https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack"
1919
$pluginsRoot = Join-Path $HOME ".agents\plugins"
2020
$installDir = Join-Path $pluginsRoot $pluginName
2121
$marketplaceFile = Join-Path $pluginsRoot "marketplace.json"
@@ -85,7 +85,7 @@ $pluginJson = @"
8585
{
8686
"name": "$pluginName",
8787
"interface": {
88-
"displayName": "Google Data Cloud AI Dev Kit"
88+
"displayName": "Data Agent Kit Starter Pack"
8989
},
9090
"source": {
9191
"source": "local",

codex-install.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515

1616
set -e
1717

18-
PLUGIN_NAME="data-cloud-ai-dev-kit"
19-
REPO_URL="https://github.com/gemini-cli-extensions/data-cloud-ai-dev-kit"
18+
PLUGIN_NAME="data-agent-kit-starter-pack"
19+
REPO_URL="https://github.com/gemini-cli-extensions/data-agent-kit-starter-pack"
2020
INSTALL_DIR="$HOME/.agents/plugins/$PLUGIN_NAME"
2121
MARKETPLACE_FILE="$HOME/.agents/plugins/marketplace.json"
2222

@@ -53,7 +53,7 @@ data.plugins = data.plugins || [];
5353
data.plugins = data.plugins.filter(p => p.name !== '${PLUGIN_NAME}');
5454
data.plugins.push({
5555
name: '${PLUGIN_NAME}',
56-
interface: { displayName: 'Google Data Cloud AI Dev Kit' },
56+
interface: { displayName: 'Data Agent Kit Starter Pack' },
5757
source: { source: 'local', path: './.agents/plugins/${PLUGIN_NAME}' },
5858
policy: { installation: 'AVAILABLE', authentication: 'ON_INSTALL' },
5959
category: 'Productivity'

0 commit comments

Comments
 (0)