Build an Agent (SQLite Store)

import { DatabaseSync } from 'node:sqlite';

import { Sqlite } from '@deepagents/text2sql/sqlite';
import * as sqlite from '@deepagents/text2sql/sqlite';

const db = new DatabaseSync('./your-database.db', { readOnly: true });
const adapter = new Sqlite({
  execute: (sql) => db.prepare(sql).all(),
  grounding: [sqlite.tables(), sqlite.info()],
});

import pg from 'pg';

import { Postgres } from '@deepagents/text2sql/postgres';
import * as postgres from '@deepagents/text2sql/postgres';

const pool = new pg.Pool({ connectionString: process.env.DATABASE_URL });
const adapter = new Postgres({
  execute: async (sql) => (await pool.query(sql)).rows,
  grounding: [postgres.tables(), postgres.info()],
});

import sql from 'mssql';

import { SqlServer } from '@deepagents/text2sql/sqlserver';
import * as sqlserver from '@deepagents/text2sql/sqlserver';

const pool = await sql.connect({
  server: process.env.DB_HOST,
  database: process.env.DB_NAME,
  user: process.env.DB_USER,
  password: process.env.DB_PASSWORD,
  options: { encrypt: true, trustServerCertificate: true },
});
const adapter = new SqlServer({
  execute: async (query) => (await pool.request().query(query)).recordset,
  grounding: [sqlserver.tables(), sqlserver.info()],
});

import mysql from 'mysql2/promise';

import { Mysql } from '@deepagents/text2sql/mysql';
import * as mysqlGrounding from '@deepagents/text2sql/mysql';

const pool = mysql.createPool({
  host: process.env.DB_HOST,
  user: process.env.DB_USER,
  password: process.env.DB_PASSWORD,
  database: process.env.DB_NAME,
});
const adapter = new Mysql({
  execute: async (sql) => {
    const [rows] = await pool.query(sql);
    return rows;
  },
  grounding: [mysqlGrounding.tables(), mysqlGrounding.info()],
});

import { BigQuery as BigQueryClient } from '@google-cloud/bigquery';

import { BigQuery } from '@deepagents/text2sql/bigquery';
import * as bigquery from '@deepagents/text2sql/bigquery';

const projectId = process.env.BQ_PROJECT_ID!;
const client = new BigQueryClient({ projectId });
const adapter = new BigQuery({
  projectId,
  datasets: ['analytics'],
  execute: async (sql) => {
    const [job] = await client.createQueryJob({ query: sql });
    const [rows] = await job.getQueryResults();
    return rows;
  },
  validate: async (sql) => {
    await client.createQueryJob({ query: sql, dryRun: true, useQueryCache: false });
  },
  grounding: [bigquery.tables(), bigquery.info()],
});