@axiom-lattice/pg-stores 1.0.1

package/dist/index.d.ts

@@ -0,0 +1,211 @@
+import { PoolConfig, PoolClient, Pool } from 'pg';
+import { ThreadStore, Thread, CreateThreadRequest, AssistantStore, Assistant, CreateAssistantRequest } from '@axiom-lattice/protocols';
+export { Assistant, AssistantStore, CreateAssistantRequest, CreateThreadRequest, Thread, ThreadStore } from '@axiom-lattice/protocols';
+
+/**
+ * PostgreSQL implementation of ThreadStore
+ */
+
+/**
+ * PostgreSQL ThreadStore options
+ */
+interface PostgreSQLThreadStoreOptions {
+    /**
+     * PostgreSQL connection pool configuration
+     * Can be a connection string or PoolConfig object
+     */
+    poolConfig: string | PoolConfig;
+    /**
+     * Whether to run migrations automatically on initialization
+     * @default true
+     */
+    autoMigrate?: boolean;
+}
+/**
+ * PostgreSQL implementation of ThreadStore
+ */
+declare class PostgreSQLThreadStore implements ThreadStore {
+    private pool;
+    private migrationManager;
+    private initialized;
+    private ownsPool;
+    constructor(options: PostgreSQLThreadStoreOptions);
+    /**
+     * Dispose resources and close the connection pool
+     * Should be called when the store is no longer needed
+     */
+    dispose(): Promise<void>;
+    /**
+     * Initialize the store and run migrations
+     */
+    initialize(): Promise<void>;
+    /**
+     * Get all threads for a specific assistant
+     */
+    getThreadsByAssistantId(assistantId: string): Promise<Thread[]>;
+    /**
+     * Get a thread by ID for a specific assistant
+     */
+    getThreadById(assistantId: string, threadId: string): Promise<Thread | undefined>;
+    /**
+     * Create a new thread for an assistant
+     */
+    createThread(assistantId: string, threadId: string, data: CreateThreadRequest): Promise<Thread>;
+    /**
+     * Update an existing thread
+     */
+    updateThread(assistantId: string, threadId: string, updates: Partial<CreateThreadRequest>): Promise<Thread | null>;
+    /**
+     * Delete a thread by ID
+     */
+    deleteThread(assistantId: string, threadId: string): Promise<boolean>;
+    /**
+     * Check if thread exists
+     */
+    hasThread(assistantId: string, threadId: string): Promise<boolean>;
+    /**
+     * Ensure store is initialized
+     */
+    private ensureInitialized;
+    /**
+     * Map database row to Thread object
+     */
+    private mapRowToThread;
+}
+
+/**
+ * PostgreSQL implementation of AssistantStore
+ */
+
+/**
+ * PostgreSQL AssistantStore options
+ */
+interface PostgreSQLAssistantStoreOptions {
+    /**
+     * PostgreSQL connection pool configuration
+     * Can be a connection string or PoolConfig object
+     */
+    poolConfig: string | PoolConfig;
+    /**
+     * Whether to run migrations automatically on initialization
+     * @default true
+     */
+    autoMigrate?: boolean;
+}
+/**
+ * PostgreSQL implementation of AssistantStore
+ */
+declare class PostgreSQLAssistantStore implements AssistantStore {
+    private pool;
+    private migrationManager;
+    private initialized;
+    private ownsPool;
+    constructor(options: PostgreSQLAssistantStoreOptions);
+    /**
+     * Initialize the store and run migrations
+     */
+    initialize(): Promise<void>;
+    /**
+     * Get all assistants
+     */
+    getAllAssistants(): Promise<Assistant[]>;
+    /**
+     * Get assistant by ID
+     */
+    getAssistantById(id: string): Promise<Assistant | null>;
+    /**
+     * Create a new assistant
+     */
+    createAssistant(id: string, data: CreateAssistantRequest): Promise<Assistant>;
+    /**
+     * Update an existing assistant
+     */
+    updateAssistant(id: string, updates: Partial<CreateAssistantRequest>): Promise<Assistant | null>;
+    /**
+     * Delete an assistant by ID
+     */
+    deleteAssistant(id: string): Promise<boolean>;
+    /**
+     * Check if assistant exists
+     */
+    hasAssistant(id: string): Promise<boolean>;
+    /**
+     * Dispose resources and close the connection pool
+     * Should be called when the store is no longer needed
+     */
+    dispose(): Promise<void>;
+    /**
+     * Ensure store is initialized
+     */
+    private ensureInitialized;
+    /**
+     * Map database row to Assistant object
+     */
+    private mapRowToAssistant;
+}
+
+/**
+ * Migration system for database schema management
+ */
+
+/**
+ * Migration definition
+ */
+interface Migration {
+    version: number;
+    name: string;
+    up: (client: PoolClient) => Promise<void>;
+    down?: (client: PoolClient) => Promise<void>;
+}
+/**
+ * Migration manager
+ */
+declare class MigrationManager {
+    private pool;
+    private migrations;
+    constructor(pool: Pool);
+    /**
+     * Register a migration
+     */
+    register(migration: Migration): void;
+    /**
+     * Initialize migrations table if it doesn't exist
+     */
+    private ensureMigrationsTable;
+    /**
+     * Get applied migrations from database
+     */
+    private getAppliedMigrations;
+    /**
+     * Apply pending migrations
+     */
+    migrate(): Promise<void>;
+    /**
+     * Rollback last migration
+     */
+    rollback(): Promise<void>;
+    /**
+     * Get current migration version
+     */
+    getCurrentVersion(): Promise<number>;
+}
+
+/**
+ * Thread table migrations
+ */
+
+/**
+ * Initial migration: Create threads table
+ */
+declare const createThreadsTable: Migration;
+
+/**
+ * Assistant table migrations
+ */
+
+/**
+ * Initial migration: Create assistants table
+ */
+declare const createAssistantsTable: Migration;
+
+export { type Migration, MigrationManager, PostgreSQLAssistantStore, type PostgreSQLAssistantStoreOptions, PostgreSQLThreadStore, type PostgreSQLThreadStoreOptions, createAssistantsTable, createThreadsTable };