import { DatabaseService } from "@agentdb/sdk";
import fs from "fs";
import "dotenv/config";
const agentdb = new DatabaseService("https://api.agentdb.dev", process.env.AGENTDB_API_KEY, true);
// Generate a new UUID for the database token
const newDatabaseToken = crypto.randomUUID();
console.log(`Generated new database token: ${newDatabaseToken}`);
const connection = agentdb.connect(newDatabaseToken, "users_db", "sqlite");
const main = async () => {
try {
console.log("Creating users table...");
// Create the users table
await connection.execute({
sql: `CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
email TEXT UNIQUE NOT NULL,
age INTEGER,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
)`
});
console.log("Populating users table with sample data...");
// Insert sample data
const sampleUsers = [
{ name: "Alice Johnson", email: "alice@example.com", age: 28 },
{ name: "Bob Smith", email: "bob@example.com", age: 34 },
{ name: "Carol Davis", email: "carol@example.com", age: 25 },
{ name: "David Wilson", email: "david@example.com", age: 42 },
{ name: "Eve Brown", email: "eve@example.com", age: 31 }
];
for (const user of sampleUsers) {
await connection.execute({
sql: "INSERT INTO users (name, email, age) VALUES (?, ?, ?)",
params: [user.name, user.email, user.age]
});
}
console.log("Querying all users from the database...");
// Query all users
const allUsers = await connection.execute({
sql: "SELECT * FROM users ORDER BY created_at"
});
console.log("All users:");
console.log(JSON.stringify(allUsers, null, 2));
console.log("Querying users older than 30...");
// Query users with a condition
const olderUsers = await connection.execute({
sql: "SELECT name, email, age FROM users WHERE age > ? ORDER BY age DESC",
params: [30]
});
console.log("Users older than 30:");
console.log(JSON.stringify(olderUsers, null, 2));
// Download the sqlite database
const download = await agentdb.getDownloadUrl(newDatabaseToken, "users_db", "sqlite");
const database = await fetch(download.downloadUrl).then((res) => res.arrayBuffer());
// Write the database to a file using node
fs.writeFileSync(download.fileName, Buffer.from(database));
console.log("Database operations completed successfully!");
} catch (error) {
console.error("Error during database operations:", error);
}
};
main();
# Execute SQL queries with ease
curl -X POST 'https://api.agentdb.dev/db/api/execute' \
-H 'Authorization: Bearer your-api-key' \
-H 'Content-Type: application/json' \
-d '{
"token": "your-uuid-token",
"dbName": "users_db",
"dbType": "sqlite",
"statements": [
{
"sql": "SELECT * FROM users WHERE age > ?",
"params": [25]
}
]
}'
# Vector search example
curl -X POST 'https://api.agentdb.dev/db/api/execute' \
-H 'Authorization: Bearer your-api-key' \
-H 'Content-Type: application/json' \
-d '{
"token": "your-uuid-token",
"dbName": "documents_db",
"dbType": "sqlite",
"statements": [
{
"sql": "SELECT id, text, vec_distance_cosine(embedding, ?) as distance FROM documents ORDER BY distance LIMIT 5",
"params": [[0.1, 0.2, 0.3, 0.4]]
}
]
}'