	In Browser
	StumbleUpon
	del.icio.us
	Google
	Google Buzz
	reddit
	LinkedIn

	Facebook
	Twitter
	Linkedin
	E-Mail

Generative AI > Google Gemini API > ADK with AlloyDB for Enterprise Data Queries

ADK with AlloyDB for Enterprise Data Queries

Author: Venkata Sudhakar

AlloyDB is a fully managed PostgreSQL-compatible database built for enterprise workloads. It delivers 4x faster queries than standard PostgreSQL and is ideal for ADK agents that need real-time access to transactional data. ShopMax India uses AlloyDB to store order management data and exposes it to their fulfilment agent through a secure database tool.

The integration uses the standard psycopg2 PostgreSQL driver. The ADK tool wraps parameterised queries - never string interpolation - to prevent SQL injection. Credentials are injected from Secret Manager at startup.

The below example shows how ShopMax India builds an order lookup tool using AlloyDB for the fulfilment agent.

import psycopg2
import os
from google.adk.tools import FunctionTool

# Connection pool for AlloyDB
def get_connection():
    return psycopg2.connect(
        host=os.environ["ALLOYDB_HOST"],
        port=5432,
        database="shopmax_orders",
        user=os.environ["ALLOYDB_USER"],
        password=os.environ["ALLOYDB_PASSWORD"],
        sslmode="require"
    )

def get_order_details(order_id: str) -> dict:
    """Retrieve full order details from AlloyDB by order ID."""
    conn = get_connection()
    try:
        with conn.cursor() as cur:
            cur.execute(
                """SELECT o.order_id, o.customer_name, o.city,
                          o.total_amount, o.status, o.created_at,
                          array_agg(i.product_name) as items
                   FROM orders o
                   JOIN order_items i ON o.order_id = i.order_id
                   WHERE o.order_id = %s
                   GROUP BY o.order_id""",
                (order_id,)
            )
            row = cur.fetchone()
            if not row:
                return {"error": f"Order {order_id} not found"}
            return {
                "order_id": row[0], "customer": row[1], "city": row[2],
                "total": float(row[3]), "status": row[4],
                "created_at": str(row[5]), "items": row[6]
            }
    finally:
        conn.close()

order_tool = FunctionTool(func=get_order_details)

Running the fulfilment agent with the AlloyDB order tool,

It gives the following output,

Order ORD-2024-88712 Details:
Customer: Rajesh Kumar, Hyderabad
Items: Samsung Galaxy S24, Boat Airdopes 141
Total: Rs 64,490
Status: OUT_FOR_DELIVERY
Ordered: 2024-11-14 09:23:15

Your order is out for delivery today. Expected delivery by 6 PM IST.

For production AlloyDB access from Cloud Run, use the AlloyDB Auth Proxy sidecar container instead of direct TCP connections. The proxy handles IAM-based authentication and TLS automatically, eliminating the need to manage database passwords. Grant the Cloud Run service account the roles/alloydb.client IAM role.

Send your comments, suggestions or queries regarding this site to [email protected].