@agi-cli/database 0.1.55

package/src/schema/messages.ts

@@ -0,0 +1,26 @@
+import { sqliteTable, text, integer } from 'drizzle-orm/sqlite-core';
+import { sessions } from './sessions.ts';
+
+export const messages = sqliteTable('messages', {
+	id: text('id').primaryKey(),
+	sessionId: text('session_id')
+		.notNull()
+		.references(() => sessions.id, { onDelete: 'cascade' }),
+	role: text('role').notNull(), // 'system' | 'user' | 'assistant' | 'tool'
+	status: text('status').notNull(), // 'pending' | 'complete' | 'error'
+	agent: text('agent').notNull(),
+	provider: text('provider').notNull(),
+	model: text('model').notNull(),
+	createdAt: integer('created_at', { mode: 'number' }).notNull(),
+	// Metadata
+	completedAt: integer('completed_at', { mode: 'number' }),
+	latencyMs: integer('latency_ms'),
+	promptTokens: integer('prompt_tokens'),
+	completionTokens: integer('completion_tokens'),
+	totalTokens: integer('total_tokens'),
+	// Error fields
+	error: text('error'),
+	errorType: text('error_type'), // 'api_error', 'abort', 'validation_error', etc.
+	errorDetails: text('error_details'), // JSON string with full error object
+	isAborted: integer('is_aborted', { mode: 'boolean' }), // flag for user aborts
+});

package/src/schema/sessions.ts

@@ -0,0 +1,17 @@
+import { sqliteTable, text, integer } from 'drizzle-orm/sqlite-core';
+
+export const sessions = sqliteTable('sessions', {
+	id: text('id').primaryKey(),
+	title: text('title'),
+	agent: text('agent').notNull(),
+	provider: text('provider').notNull(),
+	model: text('model').notNull(),
+	projectPath: text('project_path').notNull(),
+	createdAt: integer('created_at', { mode: 'number' }).notNull(),
+	// Metadata
+	lastActiveAt: integer('last_active_at', { mode: 'number' }),
+	totalInputTokens: integer('total_input_tokens'),
+	totalOutputTokens: integer('total_output_tokens'),
+	totalToolTimeMs: integer('total_tool_time_ms'),
+	toolCountsJson: text('tool_counts_json'), // JSON object of name->count
+});

package/src/schema.ts

@@ -0,0 +1 @@
+export * from './schema/index.ts';

package/src/types/index.ts

@@ -0,0 +1,8 @@
+import type { InferSelectModel } from 'drizzle-orm';
+import type { sessions } from '../schema/sessions.ts';
+import type { messages } from '../schema/messages.ts';
+import type { messageParts } from '../schema/message-parts.ts';
+
+export type Session = InferSelectModel<typeof sessions>;
+export type Message = InferSelectModel<typeof messages>;
+export type MessagePart = InferSelectModel<typeof messageParts>;

package/src/types/sql-imports.d.ts

@@ -0,0 +1,5 @@
+// Type declarations for .sql file imports (Bun-specific)
+declare module '*.sql' {
+	const content: string;
+	export default content;
+}

package/tsconfig.json

@@ -0,0 +1,7 @@
+{
+	"extends": "../../tsconfig.base.json",
+	"compilerOptions": {
+		"outDir": "./dist"
+	},
+	"include": ["src/**/*"]
+}