AlloyDB for PostgreSQL Agent Skills

June 1, 2026 · View on GitHub

Note

Currently in beta (pre-v1.0), and may see breaking changes until the first stable release (v1.0).

This repository provides a set of agent skills to interact with AlloyDB for PostgreSQL instances. These skills can be used with various AI agents, including Antigravity, Claude Code and Codex, to manage your databases, execute queries, explore schemas, and troubleshoot issues using natural language prompts.

Important

We Want Your Feedback! Please share your thoughts with us by filling out our feedback form. Your input is invaluable and helps us improve the project for everyone.

Table of Contents

Why Use AlloyDB for PostgreSQL Agent Skills?

  • Seamless Workflow: Integrates seamlessly into your AI agent's environment. No need to constantly switch contexts for common database tasks.
  • Natural Language Queries: Stop wrestling with complex commands. Explore schemas and query data by describing what you want in plain English.
  • Full Lifecycle Control: Manage the entire lifecycle of your database, from creating instances to exploring schemas and running queries.
  • Code Generation: Accelerate development by asking your agent to generate data classes and other code snippets based on your table schemas.

Prerequisites

Before you begin, ensure you have the following:

  • One of these AI agents installed
  • A Google Cloud project with the AlloyDB API enabled.
  • Ensure Application Default Credentials are available in your environment.
  • IAM Permissions:
    • AlloyDB Client (roles/alloydb.client)
    • AlloyDB Admin (roles/alloydb.admin)

Getting Started

Configuration

Please keep these env vars handy during the installation process:

  • ALLOYDB_POSTGRES_PROJECT: The GCP project ID.
  • ALLOYDB_POSTGRES_REGION: The region of your AlloyDB instance.
  • ALLOYDB_POSTGRES_CLUSTER: The ID of your AlloyDB cluster.
  • ALLOYDB_POSTGRES_INSTANCE: The ID of your AlloyDB instance.
  • ALLOYDB_POSTGRES_DATABASE: The name of the database to connect to.
  • ALLOYDB_POSTGRES_USER: (Optional) The database username.
  • ALLOYDB_POSTGRES_PASSWORD: (Optional) The password for the database user.
  • ALLOYDB_POSTGRES_IP_TYPE: (Optional) Type of the IP address: PUBLIC, PRIVATE, or PSC. Defaults to PUBLIC.

Note

  • Ensure Application Default Credentials are available in your environment.
  • If your AlloyDB instance uses private IPs, you must run your agent in the same Virtual Private Cloud (VPC) network.

Installation & Usage

To start interacting with your database, install the skills for your preferred AI agent, then launch the agent and use natural language to ask questions or perform tasks.

For the latest version, check the releases page.

Antigravity

You can use either of these two agents for Antigravity:

💡 Tip — Migrating from Gemini CLI?
If you previously installed this extension with gemini extensions install, you can convert it to an Antigravity plugin instead of reinstalling from scratch:
  • On first launch of Antigravity CLI, accept the Migration Options prompt to automatically convert your installed Gemini CLI extensions to Antigravity plugins.
  • Or, from your terminal, run: 
    agy plugin import gemini
See Migrating from Gemini CLI for details on plugins, context files (GEMINI.md / AGENTS.md), and MCP server config differences.

Antigravity 2.0 (IDE)

1. Clone the Repo:

git clone --branch 0.2.0 https://github.com/gemini-cli-extensions/alloydb.git

2. Install the skills:

Choose a location for the skills:

  • Global (all workspaces): ~/.gemini/antigravity/skills/
  • Workspace-specific: <workspace-root>/.agents/skills/

Copy the skill folders from the cloned repository's skills/ directory to your chosen location:

cp -R alloydb/skills/* ~/.gemini/antigravity/skills/

3. Set env vars: Set your environment vars as described in the configuration section.

(Tip: Antigravity 2.0 automatically discovers skills in these directories at the start of a session. You can verify they are active by running the /skills command in your active session.)

Antigravity CLI

1. Clone the Repo:

git clone --branch 0.2.0 https://github.com/gemini-cli-extensions/alloydb.git

2. Install the skills:

Choose a location for the skills:

  • Global (all workspaces): ~/.gemini/antigravity-cli/skills/
  • Workspace-specific: <workspace-root>/.agents/skills/

Copy the skill folders from the cloned repository's skills/ directory to your chosen location:

cp -R alloydb/skills/* ~/.gemini/antigravity-cli/skills/

3. Set env vars: Set your environment vars as described in the configuration section.

(Tip: Antigravity CLI automatically discovers skills in these directories at the start of a session. You can verify they are active by running the /skills command in your active session.)

Claude Code

1. Set env vars: In your terminal, set your environment vars as described in the configuration section.

2. Start the agent:

claude

3. Install the plugin:

/plugin install alloydb@claude-plugins-official

(Tip: Run /plugin list inside Claude Code to verify the plugin is active, or /reload-plugins if you just installed it.)

Codex

1. Clone the Repo:

git clone --branch 0.2.0 git@github.com:gemini-cli-extensions/alloydb.git

2. Install the plugin:

mkdir -p ~/.codex/plugins
cp -R /absolute/path/to/alloydb ~/.codex/plugins/alloydb

3. Set env vars: Enter your environment vars as described in the configuration section.

4. Create or update marketplace.json: ~/.agents/plugins/marketplace.json

{
  "name": "my-data-cloud-google-marketplace",
  "interface": {
    "displayName": "Google Data Cloud Skills"
  },
  "plugins": [
    {
      "name": "alloydb",
      "source": {
        "source": "local",
        "path": "./plugins/alloydb"
      },
      "policy": {
        "installation": "AVAILABLE",
        "authentication": "ON_INSTALL"
      },
      "category": "Database"
    }
  ]
}

(Tip: Run codex plugin list or use the /plugins interactive menu to verify your installed plugins.)

## Installing using [open agent skills tool](https://github.com/vercel-labs/skills)

You can install skills using the npx skills command.

Run the following command in your terminal to automatically download and register the skills:

npx skills add https://github.com/gemini-cli-extensions/alloydb/tree/0.2.0

For detailed info check out the Skills npm package.

2. Set env vars: Set your environment vars as described in the configuration section.

Usage Examples

Interact with AlloyDB using natural language right from your agent:

  • Provision Infrastructure:
    • "Create a new AlloyDB cluster named 'e-commerce-prod' in project 'my-gcp-project'."
    • "Add a read-only instance to my 'e-commerce-prod' cluster."
  • Explore Schemas and Data:
    • "Show me all tables in the 'orders' database."
    • "What are the columns in the 'products' table?"
    • "How many orders were placed in the last 30 days?"
  • Generate Code:
    • "Generate a Python dataclass to represent the 'customers' table."

Supported Skills

This repository includes the following skill sets:

Additional Agent Skills

Find additional skills to support your entire software development lifecycle at github.com/gemini-cli-extensions, including:

Troubleshooting

Use the debug mode of your agent (e.g., gemini --debug) to enable debugging.

Common issues:

  • "failed to find default credentials: google: could not find default credentials.": Ensure Application Default Credentials are available in your environment. See Set up Application Default Credentials for more information.
  • "✖ Error during discovery for server: MCP error -32000: Connection closed": The database connection has not been established. Ensure your configuration is set via environment variables.
  • "✖ MCP ERROR: Error: spawn .../toolbox ENOENT": The Toolbox binary did not download correctly. Ensure you are using the latest version of your agent.
  • "cannot execute binary file": The Toolbox binary did not download correctly. Ensure the correct binary for your OS/Architecture has been downloaded. See Installing the server for more information.