genai-toolbox/docs/en/resources/tools/bigtable-sql.md at a087280fe2e022e34aea9603ade2bcc4108dba97

mirror of https://github.com/googleapis/genai-toolbox.git synced 2026-02-12 08:05:06 -05:00

Files

An Nguyen ae53b8eeff feat: Add Bigtable source and tool (#418 )

# Add Bigtable support

A `bigtable` source can be added as the following example

```
sources:
  test-bigtable-source:
    kind: "bigtable"
    project: "sample-project"
    instance: "sample-instance"
```

A `bigtable` tool can be added as below

```
tools:
  get-test-tool-data:
    kind: bigtable-sql
    source: test-bigtable-source
    description: Some description
    statement: SELECT * FROM `test-table` WHERE address['state'] = @state;
    parameters:
      - name: state
        type: string
        description: Filter by state
```

---------

Co-authored-by: Yuan <45984206+Yuan325@users.noreply.github.com>
Co-authored-by: Kurtis Van Gent <31518063+kurtisvg@users.noreply.github.com>
Co-authored-by: Wenxin Du <117315983+duwenxin99@users.noreply.github.com>

2025-04-18 10:48:28 -06:00

3.4 KiB

Raw Blame History

title, type, weight, description

title	type	weight	description
bigtable-sql	docs	1	A "bigtable-sql" tool executes a pre-defined SQL statement against a Google Cloud Bigtable instance.

About

A bigtable-sql tool executes a pre-defined SQL statement against a Bigtable instance. It's compatible with any of the following sources:

bigtable

GoogleSQL

Bigtable supports SQL queries. The integration with Toolbox supports googlesql dialect, the specified SQL statement is executed as a data manipulation language (DML) statements, and specified parameters will inserted according to their name: e.g. @name.

Example

tools:
 search_user_by_id_or_name:
    kind: bigtable-sql
    source: my-bigtable-instance
    statement: |
      SELECT 
        TO_INT64(cf[ 'id' ]) as id, 
        CAST(cf[ 'name' ] AS string) as name, 
      FROM 
        % s 
      WHERE 
        TO_INT64(cf[ 'id' ]) = @id 
        OR CAST(cf[ 'name' ] AS string) = @name;
    description: |
      Use this tool to get information for a specific user.
      Takes an id number or a name and returns info on the user.

      Example:
      {{
          "id": 123,
          "name": "Alice",
      }}
    parameters:
      - name: id
        type: integer
        description: User ID
      - name: name
        type: string
        description: Name of the user

Reference

field	type	required	description
kind	string	true	Must be "bigtable-sql".
source	string	true	Name of the source the SQL should execute on.
description	string	true	Description of the tool that is passed to the LLM.
statement	string	true	SQL statement to execute on.
parameters	parameters	false	List of parameters that will be inserted into the SQL statement.

Tips

Bigtable Studio is a useful to explore and manage your Bigtable data. If you're unfamiliar with the query syntax, Query Builder lets you build a query, run it against a table, and then view the results in the console.
Some Python libraries limit the use of underscore columns such as _key. A workaround would be to leverage Bigtable Logical Views to rename the columns.

3.4 KiB Raw Blame History