# Introducing new AI functions for AlloyDB: ai.classify and ai.extract

> Source: <https://cloud.google.com/blog/products/databases/ai-classify-and-ai-extract-functions-in-alloydb-ai/>
> Published: 2026-11-11 17:00:00+00:00

Transactional databases store vast amounts of unstructured text — clinical notes, customer reviews, support tickets, invoices, and property descriptions. Historically, working with this text meant building complex application code: You had to export data from the database, call external model endpoints, write custom prompt wrappers, parse unpredictable responses, manage retry logic, and update database records. This application-side loop creates operational friction. It introduces query latency, bloats application code, increases model token costs, and creates data compliance risks by moving sensitive information across multiple network boundaries.

We built [AlloyDB AI](https://cloud.google.com/alloydb/ai) to bring Gemini world knowledge directly into your operational database. Rather than treating artificial intelligence as a disconnected external service, we integrated model intelligence directly into the database engine. Today, we are announcing the preview of two new SQL functions in the AlloyDB AI query engine — `ai.classify` and `ai.extract` — available in the `google_ml_integration` extension (version 1.6.1 or later) on PostgreSQL 18, allowing you to perform classification and structured extraction directly inside your database using standard SQL. Supported by smart batching for AI functions (previously known as AI Function Acceleration), these functions deliver production scalability and performance. 

Now, let’s take a look at these new functions and how to use them.

The `ai.classify` function categorizes text into user-defined classes without requiring manual prompt engineering. You can specify categories as a simple list of labels or provide a list containing label names and descriptions to guide model context.

Key capabilities include:

Single-label classification (`output_mode => 'single'`) to assign the single best category

Multi-label classification (`output_mode => 'multi'`) to return all matching categories

Category descriptions using 2D arrays to provide explicit context for each label

Few-shot prompting by supplying inline examples

Flexible execution modes for individual rows (`TEXT`), arrays (` TEXT[]`), and cursors (` REFCURSOR`)

Here’s a quick example showing how you can classify customer feedback directly within a SQL query:

The `ai.extract` function transforms unstructured text into structured JSON data matching your target schema — specified as a JSON schema or a text list of fields. It handles system instructions internally, saving you from writing prompt templates or response parsers.

Schema flexibility supporting standard JSON schemas or plain text field lists

Automatic transformation of unstructured text into machine-readable JSON

Overloaded function signatures for single rows, arrays, and cursor batching

Here’s an example extracting clinical fields from free-form medical notes:

Running AI evaluations row-by-row can slow down query execution. AlloyDB addresses this with smart batching for AI functions. When you use `ai.classify` or `ai.extract` in SQL `WHERE` clause expressions, the database engine automatically groups inputs into optimized micro-batches before sending them to Gemini Enterprise Agent Platform models.

Smart batching delivers significant performance and efficiency improvements:

Faster inference for query execution compared to row-at-a-time evaluations

Reduces total model token spend by optimizing input prompt overhead

Native classification and extraction functions support operational workflows across key industry sectors:

**Healthcare:** Securely convert free-text clinical notes into structured records containing patient names, symptoms, and prescribed medications, keeping patient data safely within the confines of the database.

**Financial services:** Extract vendor details, invoice dates, and line-item totals from scanned document text.

**Legal technology:** Pull governing law, liability caps, and termination notice periods from contract documents.

**Real estate:** Extract property types, bedroom counts, and price ranges from descriptive listing text.

**Knowledge graphs:** Extract entities and relationships from documents to populate PostgreSQL Property Graph (PGQ) structures.

We are continuing to expand in-database AI capabilities in AlloyDB, with upcoming enhancements planned for advanced cost optimization, line-level provenance tracking, and query pushdowns beyond `WHERE` clauses.

To start using `ai.classify` and `ai.extract` in your database:

Review the [AlloyDB AI Query Engine documentation](https://docs.cloud.google.com/alloydb/docs/ai).

Explore the [AlloyDB release notes](https://docs.cloud.google.com/alloydb/docs/release-notes).