@agi-cli/database 0.1.55

package/drizzle/meta/0003_snapshot.json

@@ -0,0 +1,444 @@
+{
+	"version": "6",
+	"dialect": "sqlite",
+	"id": "dcfa8932-b15c-4c9b-9846-07331968bb2c",
+	"prevId": "c0260f78-cf4f-4305-a086-c1def673eebb",
+	"tables": {
+		"artifacts": {
+			"name": "artifacts",
+			"columns": {
+				"id": {
+					"name": "id",
+					"type": "text",
+					"primaryKey": true,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"message_part_id": {
+					"name": "message_part_id",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"kind": {
+					"name": "kind",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"path": {
+					"name": "path",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"mime": {
+					"name": "mime",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"size": {
+					"name": "size",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"sha256": {
+					"name": "sha256",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				}
+			},
+			"indexes": {
+				"artifacts_message_part_id_unique": {
+					"name": "artifacts_message_part_id_unique",
+					"columns": ["message_part_id"],
+					"isUnique": true
+				}
+			},
+			"foreignKeys": {
+				"artifacts_message_part_id_message_parts_id_fk": {
+					"name": "artifacts_message_part_id_message_parts_id_fk",
+					"tableFrom": "artifacts",
+					"tableTo": "message_parts",
+					"columnsFrom": ["message_part_id"],
+					"columnsTo": ["id"],
+					"onDelete": "cascade",
+					"onUpdate": "no action"
+				}
+			},
+			"compositePrimaryKeys": {},
+			"uniqueConstraints": {},
+			"checkConstraints": {}
+		},
+		"message_parts": {
+			"name": "message_parts",
+			"columns": {
+				"id": {
+					"name": "id",
+					"type": "text",
+					"primaryKey": true,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"message_id": {
+					"name": "message_id",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"index": {
+					"name": "index",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"step_index": {
+					"name": "step_index",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"type": {
+					"name": "type",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"content": {
+					"name": "content",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"agent": {
+					"name": "agent",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"provider": {
+					"name": "provider",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"model": {
+					"name": "model",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"started_at": {
+					"name": "started_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"completed_at": {
+					"name": "completed_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"tool_name": {
+					"name": "tool_name",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"tool_call_id": {
+					"name": "tool_call_id",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"tool_duration_ms": {
+					"name": "tool_duration_ms",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				}
+			},
+			"indexes": {},
+			"foreignKeys": {
+				"message_parts_message_id_messages_id_fk": {
+					"name": "message_parts_message_id_messages_id_fk",
+					"tableFrom": "message_parts",
+					"tableTo": "messages",
+					"columnsFrom": ["message_id"],
+					"columnsTo": ["id"],
+					"onDelete": "cascade",
+					"onUpdate": "no action"
+				}
+			},
+			"compositePrimaryKeys": {},
+			"uniqueConstraints": {},
+			"checkConstraints": {}
+		},
+		"messages": {
+			"name": "messages",
+			"columns": {
+				"id": {
+					"name": "id",
+					"type": "text",
+					"primaryKey": true,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"session_id": {
+					"name": "session_id",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"role": {
+					"name": "role",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"status": {
+					"name": "status",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"agent": {
+					"name": "agent",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"provider": {
+					"name": "provider",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"model": {
+					"name": "model",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"created_at": {
+					"name": "created_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"completed_at": {
+					"name": "completed_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"latency_ms": {
+					"name": "latency_ms",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"prompt_tokens": {
+					"name": "prompt_tokens",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"completion_tokens": {
+					"name": "completion_tokens",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"total_tokens": {
+					"name": "total_tokens",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"error": {
+					"name": "error",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"error_type": {
+					"name": "error_type",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"error_details": {
+					"name": "error_details",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"is_aborted": {
+					"name": "is_aborted",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				}
+			},
+			"indexes": {},
+			"foreignKeys": {
+				"messages_session_id_sessions_id_fk": {
+					"name": "messages_session_id_sessions_id_fk",
+					"tableFrom": "messages",
+					"tableTo": "sessions",
+					"columnsFrom": ["session_id"],
+					"columnsTo": ["id"],
+					"onDelete": "cascade",
+					"onUpdate": "no action"
+				}
+			},
+			"compositePrimaryKeys": {},
+			"uniqueConstraints": {},
+			"checkConstraints": {}
+		},
+		"sessions": {
+			"name": "sessions",
+			"columns": {
+				"id": {
+					"name": "id",
+					"type": "text",
+					"primaryKey": true,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"title": {
+					"name": "title",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"agent": {
+					"name": "agent",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"provider": {
+					"name": "provider",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"model": {
+					"name": "model",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"project_path": {
+					"name": "project_path",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"created_at": {
+					"name": "created_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": true,
+					"autoincrement": false
+				},
+				"last_active_at": {
+					"name": "last_active_at",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"total_input_tokens": {
+					"name": "total_input_tokens",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"total_output_tokens": {
+					"name": "total_output_tokens",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"total_tool_time_ms": {
+					"name": "total_tool_time_ms",
+					"type": "integer",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				},
+				"tool_counts_json": {
+					"name": "tool_counts_json",
+					"type": "text",
+					"primaryKey": false,
+					"notNull": false,
+					"autoincrement": false
+				}
+			},
+			"indexes": {},
+			"foreignKeys": {},
+			"compositePrimaryKeys": {},
+			"uniqueConstraints": {},
+			"checkConstraints": {}
+		}
+	},
+	"views": {},
+	"enums": {},
+	"_meta": {
+		"schemas": {},
+		"tables": {},
+		"columns": {}
+	},
+	"internal": {
+		"indexes": {}
+	}
+}

package/drizzle/meta/_journal.json

@@ -0,0 +1,34 @@
+{
+	"version": "7",
+	"dialect": "sqlite",
+	"entries": [
+		{
+			"idx": 0,
+			"version": "6",
+			"when": 1757903506317,
+			"tag": "0000_tense_shadow_king",
+			"breakpoints": true
+		},
+		{
+			"idx": 1,
+			"version": "6",
+			"when": 1757926724962,
+			"tag": "0001_past_kabuki",
+			"breakpoints": true
+		},
+		{
+			"idx": 2,
+			"version": "6",
+			"when": 1758380042084,
+			"tag": "0002_vengeful_warlock",
+			"breakpoints": true
+		},
+		{
+			"idx": 3,
+			"version": "6",
+			"when": 1759590176611,
+			"tag": "0003_pale_violations",
+			"breakpoints": true
+		}
+	]
+}

package/drizzle.config.ts

@@ -0,0 +1,14 @@
+import { defineConfig } from 'drizzle-kit';
+import { getLocalDataDir } from '@agi-cli/sdk';
+
+const dataDir = getLocalDataDir(process.cwd());
+const dbPath = `${dataDir}/agi.db`;
+
+export default defineConfig({
+	dialect: 'sqlite',
+	schema: './src/schema.ts',
+	out: './drizzle',
+	dbCredentials: {
+		url: dbPath,
+	},
+});

package/package.json

@@ -0,0 +1,33 @@
+{
+	"name": "@agi-cli/database",
+	"version": "0.1.55",
+	"description": "Database and persistence layer for AGI CLI",
+	"type": "module",
+	"main": "./src/index.ts",
+	"types": "./src/index.ts",
+	"exports": {
+		".": {
+			"import": "./src/index.ts",
+			"types": "./src/index.ts"
+		},
+		"./schema": {
+			"import": "./src/schema.ts",
+			"types": "./src/schema.ts"
+		}
+	},
+	"scripts": {
+		"dev": "bun run src/index.ts",
+		"test": "bun test",
+		"db:generate": "drizzle-kit generate --config=drizzle.config.ts",
+		"typecheck": "tsc --noEmit"
+	},
+	"dependencies": {
+		"@agi-cli/sdk": "0.1.55",
+		"drizzle-orm": "^0.44.5"
+	},
+	"devDependencies": {
+		"@types/bun": "latest",
+		"drizzle-kit": "^0.31.4",
+		"typescript": "^5"
+	}
+}

package/src/index.ts

@@ -0,0 +1,75 @@
+import { Database } from 'bun:sqlite';
+import { drizzle, type BunSQLiteDatabase } from 'drizzle-orm/bun-sqlite';
+import { loadConfig } from '@agi-cli/sdk';
+import * as schema from './schema/index.ts';
+import { bundledMigrations } from './runtime/migrations-bundled.ts';
+
+const dbCache: Map<string, BunSQLiteDatabase<typeof schema>> = new Map();
+const migratedPaths = new Set<string>();
+
+export async function getDb(projectRootInput?: string) {
+	const cfg = await loadConfig(projectRootInput);
+	const dbPath = cfg.paths.dbPath;
+	// Data dir is ensured by loadConfig() already.
+
+	const key = dbPath;
+	const cached = dbCache.get(key);
+	if (cached) return cached;
+
+	const sqlite = new Database(dbPath, { create: true });
+	const db = drizzle(sqlite, { schema });
+
+	// Run migrations once per db path (apply any not yet applied)
+	if (!migratedPaths.has(dbPath)) {
+		try {
+			// Ensure migrations tracking table exists
+			sqlite.exec(
+				'CREATE TABLE IF NOT EXISTS agi_migrations (name TEXT PRIMARY KEY, applied_at INTEGER NOT NULL)',
+			);
+
+			// Read applied migrations
+			const appliedRows = sqlite
+				.query('SELECT name FROM agi_migrations')
+				.all() as Array<{ name: string }>;
+			const applied = new Set(appliedRows.map((r) => r.name));
+
+			for (const m of bundledMigrations) {
+				if (applied.has(m.name)) continue;
+				try {
+					sqlite.exec('BEGIN TRANSACTION');
+					sqlite.exec(m.content);
+					sqlite.exec('COMMIT');
+					sqlite
+						.query(
+							'INSERT INTO agi_migrations (name, applied_at) VALUES (?, ?)',
+						)
+						.run(m.name, Date.now());
+				} catch (err) {
+					// If migration fails due to already-applied schema (e.g., table exists / duplicate column), mark as applied and continue.
+					sqlite.exec('ROLLBACK');
+					const msg = String((err as Error)?.message ?? err);
+					const benign =
+						msg.includes('already exists') || msg.includes('duplicate column');
+					if (benign) {
+						sqlite
+							.query(
+								'INSERT OR IGNORE INTO agi_migrations (name, applied_at) VALUES (?, ?)',
+							)
+							.run(m.name, Date.now());
+						continue;
+					}
+					throw err;
+				}
+			}
+			migratedPaths.add(dbPath);
+		} catch (error) {
+			console.error('❌ Local database migration failed:', error);
+			throw error;
+		}
+	}
+	dbCache.set(key, db);
+	return db;
+}
+
+export type DB = Awaited<ReturnType<typeof getDb>>;
+export * as dbSchema from './schema/index.ts';

package/src/runtime/migrations-bundled.ts

@@ -0,0 +1,20 @@
+// Bundled Drizzle migrations for single-binary builds
+// These imports ensure Bun embeds the SQL files into the executable.
+// Order matters: keep in incremental order.
+import mig0000 from '../../drizzle/0000_tense_shadow_king.sql' with {
+	type: 'text',
+};
+import mig0001 from '../../drizzle/0001_past_kabuki.sql' with { type: 'text' };
+import mig0002 from '../../drizzle/0002_vengeful_warlock.sql' with {
+	type: 'text',
+};
+import mig0003 from '../../drizzle/0003_pale_violations.sql' with {
+	type: 'text',
+};
+
+export const bundledMigrations: Array<{ name: string; content: string }> = [
+	{ name: '0000_tense_shadow_king.sql', content: mig0000 },
+	{ name: '0001_past_kabuki.sql', content: mig0001 },
+	{ name: '0002_vengeful_warlock.sql', content: mig0002 },
+	{ name: '0003_pale_violations.sql', content: mig0003 },
+];

package/src/schema/artifacts.ts

@@ -0,0 +1,14 @@
+import { sqliteTable, text, integer } from 'drizzle-orm/sqlite-core';
+import { messageParts } from './message-parts.ts';
+
+export const artifacts = sqliteTable('artifacts', {
+	id: text('id').primaryKey(),
+	messagePartId: text('message_part_id')
+		.unique()
+		.references(() => messageParts.id, { onDelete: 'cascade' }),
+	kind: text('kind').notNull(), // 'file' | 'file_diff' | ...
+	path: text('path'),
+	mime: text('mime'),
+	size: integer('size'),
+	sha256: text('sha256'),
+});

package/src/schema/index.ts

@@ -0,0 +1,36 @@
+import { relations } from 'drizzle-orm';
+export { sessions } from './sessions.ts';
+export { messages } from './messages.ts';
+export { messageParts } from './message-parts.ts';
+export { artifacts } from './artifacts.ts';
+
+import { sessions } from './sessions.ts';
+import { messages } from './messages.ts';
+import { messageParts } from './message-parts.ts';
+import { artifacts } from './artifacts.ts';
+
+export const sessionsRelations = relations(sessions, ({ many }) => ({
+	messages: many(messages),
+}));
+
+export const messagesRelations = relations(messages, ({ one, many }) => ({
+	session: one(sessions, {
+		fields: [messages.sessionId],
+		references: [sessions.id],
+	}),
+	parts: many(messageParts),
+}));
+
+export const messagePartsRelations = relations(messageParts, ({ one }) => ({
+	message: one(messages, {
+		fields: [messageParts.messageId],
+		references: [messages.id],
+	}),
+}));
+
+export const artifactsRelations = relations(artifacts, ({ one }) => ({
+	part: one(messageParts, {
+		fields: [artifacts.messagePartId],
+		references: [messageParts.id],
+	}),
+}));

package/src/schema/message-parts.ts

@@ -0,0 +1,23 @@
+import { sqliteTable, text, integer } from 'drizzle-orm/sqlite-core';
+import { messages } from './messages.ts';
+
+export const messageParts = sqliteTable('message_parts', {
+	id: text('id').primaryKey(),
+	messageId: text('message_id')
+		.notNull()
+		.references(() => messages.id, { onDelete: 'cascade' }),
+	index: integer('index').notNull(),
+	stepIndex: integer('step_index'),
+	type: text('type').notNull(), // 'text' | 'tool_call' | 'tool_result' | 'image' | 'error'
+	content: text('content').notNull(), // JSON string
+	agent: text('agent').notNull(),
+	provider: text('provider').notNull(),
+	model: text('model').notNull(),
+	// Timestamps
+	startedAt: integer('started_at', { mode: 'number' }),
+	completedAt: integer('completed_at', { mode: 'number' }),
+	// Tool metadata (for tool_call/tool_result)
+	toolName: text('tool_name'),
+	toolCallId: text('tool_call_id'),
+	toolDurationMs: integer('tool_duration_ms'),
+});