trino-sql

A “trino-sql” tool executes a pre-defined SQL statement against a Trino database.

About

A trino-sql tool executes a pre-defined SQL statement against a Trino database. It’s compatible with any of the following sources:

The specified SQL statement is executed as a prepared statement, and specified parameters will be inserted according to their position: e.g. $1 will be the first parameter specified, $2 will be the second parameter, and so on. If template parameters are included, they will be resolved before execution of the prepared statement.

Example

Note: This tool uses parameterized queries to prevent SQL injections. Query parameters can be used as substitutes for arbitrary expressions. Parameters cannot be used as substitutes for identifiers, column names, table names, or other parts of the query.

tools:
 search_orders_by_region:
    kind: trino-sql
    source: my-trino-instance
    statement: |
      SELECT * FROM hive.sales.orders
      WHERE region = $1
      AND order_date >= DATE($2)
      LIMIT 10
    description: |
      Use this tool to get information for orders in a specific region.
      Takes a region code and date and returns info on the orders.
      Do NOT use this tool with an order id. Do NOT guess a region code or date.
      A region code is a code for a geographic region consisting of two-character
      region designator and followed by optional subregion.
      For example, if given US-WEST, the region is "US-WEST".
      Another example for this is EU-CENTRAL, the region is "EU-CENTRAL".
      If the tool returns more than one option choose the date closest to today.
      Example:
      {{
          "region": "US-WEST",
          "order_date": "2024-01-01",
      }}
      Example:
      {{
          "region": "EU-CENTRAL",
          "order_date": "2024-01-15",
      }}
    parameters:
      - name: region
        type: string
        description: Region unique identifier
      - name: order_date
        type: string
        description: Order date in YYYY-MM-DD format

Example with Template Parameters

Note: This tool allows direct modifications to the SQL statement, including identifiers, column names, and table names. This makes it more vulnerable to SQL injections. Using basic parameters only (see above) is recommended for performance and safety reasons. For more details, please check templateParameters.

tools:
 list_table:
    kind: trino-sql
    source: my-trino-instance
    statement: |
      SELECT * FROM {{.tableName}}
    description: |
      Use this tool to list all information from a specific table.
      Example:
      {{
          "tableName": "hive.sales.orders",
      }}
    templateParameters:
      - name: tableName
        type: string
        description: Table to select from

Reference

fieldtyperequireddescription
kindstringtrueMust be “trino-sql”.
sourcestringtrueName of the source the SQL should execute on.
descriptionstringtrueDescription of the tool that is passed to the LLM.
statementstringtrueSQL statement to execute on.
parametersparametersfalseList of parameters that will be inserted into the SQL statement.
templateParameterstemplateParametersfalseList of templateParameters that will be inserted into the SQL statement before executing prepared statement.