@falai/agent 0.1.0-alpha2

package/dist/src/adapters/OpenSearchAdapter.js

@@ -0,0 +1,454 @@
+/**
+ * OpenSearch Persistence Adapter
+ *
+ * Provides persistence for sessions and messages using OpenSearch.
+ * Also compatible with Elasticsearch 7.x (not tested with newer versions).
+ *
+ * @example
+ * ```typescript
+ * import { Client } from '@opensearch-project/opensearch';
+ * import { OpenSearchAdapter } from '@falai/agent';
+ *
+ * const client = new Client({
+ *   node: 'https://localhost:9200',
+ *   auth: {
+ *     username: 'admin',
+ *     password: 'admin'
+ *   }
+ * });
+ *
+ * const adapter = new OpenSearchAdapter(client, {
+ *   indices: {
+ *     sessions: 'agent_sessions',
+ *     messages: 'agent_messages'
+ *   },
+ *   autoCreateIndices: true
+ * });
+ *
+ * const agent = new Agent({
+ *   model: provider,
+ *   persistence: { adapter }
+ * });
+ * ```
+ */
+import { cloneDeep } from "../utils/clone";
+/**
+ * OpenSearch persistence adapter
+ *
+ * Stores sessions and messages as documents in OpenSearch indices.
+ * Compatible with OpenSearch 1.x, 2.x and Elasticsearch 7.x.
+ */
+export class OpenSearchAdapter {
+    constructor(client, options = {}) {
+        this.client = client;
+        this.sessionIndex = options.indices?.sessions || "agent_sessions";
+        this.messageIndex = options.indices?.messages || "agent_messages";
+        this.autoCreateIndices = options.autoCreateIndices ?? true;
+        this.refresh = options.refresh ?? false;
+        this.sessionRepository = new OpenSearchSessionRepository(this.client, this.sessionIndex, this.refresh);
+        this.messageRepository = new OpenSearchMessageRepository(this.client, this.messageIndex, this.refresh);
+    }
+    async initialize() {
+        if (!this.autoCreateIndices) {
+            return;
+        }
+        // Create sessions index with mappings
+        const sessionExists = await this.client.indices.exists({
+            index: this.sessionIndex,
+        });
+        if (!sessionExists.body) {
+            await this.client.indices.create({
+                index: this.sessionIndex,
+                body: {
+                    mappings: {
+                        properties: {
+                            id: { type: "keyword" },
+                            userId: { type: "keyword" },
+                            agentName: { type: "keyword" },
+                            status: { type: "keyword" },
+                            currentRoute: { type: "keyword" },
+                            currentStep: { type: "keyword" },
+                            collectedData: { type: "object", enabled: false },
+                            messageCount: { type: "integer" },
+                            createdAt: { type: "date" },
+                            updatedAt: { type: "date" },
+                            lastMessageAt: { type: "date" },
+                            completedAt: { type: "date" },
+                        },
+                    },
+                },
+            });
+        }
+        // Create messages index with mappings
+        const messageExists = await this.client.indices.exists({
+            index: this.messageIndex,
+        });
+        if (!messageExists.body) {
+            await this.client.indices.create({
+                index: this.messageIndex,
+                body: {
+                    mappings: {
+                        properties: {
+                            id: { type: "keyword" },
+                            sessionId: { type: "keyword" },
+                            userId: { type: "keyword" },
+                            role: { type: "keyword" },
+                            content: { type: "text" },
+                            route: { type: "keyword" },
+                            step: { type: "keyword" },
+                            toolCalls: { type: "object", enabled: false },
+                            event: { type: "object", enabled: false },
+                            createdAt: { type: "date" },
+                        },
+                    },
+                },
+            });
+        }
+    }
+    async disconnect() {
+        // OpenSearch client doesn't have a close method like some other clients
+        // Connection pooling is managed automatically
+        await Promise.resolve();
+    }
+}
+/**
+ * OpenSearch-based session repository implementation
+ */
+class OpenSearchSessionRepository {
+    constructor(client, index, refresh) {
+        this.client = client;
+        this.index = index;
+        this.refresh = refresh;
+    }
+    async create(data) {
+        const id = data.id || `session_${Date.now()}_${Math.random().toString(36).slice(2)}`;
+        const now = new Date();
+        const session = {
+            ...data,
+            id,
+            status: data.status || "active",
+            messageCount: data.messageCount || 0,
+            createdAt: now,
+            updatedAt: now,
+        };
+        const clonedSession = cloneDeep(session);
+        await this.client.index({
+            index: this.index,
+            id,
+            body: clonedSession,
+            refresh: true, // Refresh to make the document immediately available for search
+        });
+        return session;
+    }
+    async findById(id) {
+        try {
+            const response = await this.client.get({
+                index: this.index,
+                id,
+            });
+            return this.deserializeSession(response.body._source);
+        }
+        catch (error) {
+            if (this.isNotFoundError(error)) {
+                return null;
+            }
+            throw error;
+        }
+    }
+    async findActiveByUserId(userId) {
+        const response = await this.client.search({
+            index: this.index,
+            body: {
+                query: {
+                    bool: {
+                        must: [{ term: { userId } }, { term: { status: "active" } }],
+                    },
+                },
+                sort: [{ createdAt: { order: "desc" } }],
+                size: 1,
+            },
+        });
+        const hits = response.body.hits.hits;
+        if (hits.length === 0) {
+            return null;
+        }
+        return this.deserializeSession(hits[0]._source);
+    }
+    async findByUserId(userId, limit = 100) {
+        const response = await this.client.search({
+            index: this.index,
+            body: {
+                query: {
+                    term: { userId },
+                },
+                sort: [{ createdAt: { order: "desc" } }],
+                size: limit,
+            },
+        });
+        return response.body.hits.hits.map((hit) => this.deserializeSession(hit._source));
+    }
+    async update(id, updates) {
+        const doc = {
+            ...updates,
+            updatedAt: new Date().toISOString(),
+        };
+        // Serialize dates
+        if (updates.completedAt) {
+            doc.completedAt = updates.completedAt.toISOString();
+        }
+        if (updates.lastMessageAt) {
+            doc.lastMessageAt = updates.lastMessageAt.toISOString();
+        }
+        await this.client.update({
+            index: this.index,
+            id,
+            body: { doc },
+            refresh: this.refresh,
+        });
+        return await this.findById(id);
+    }
+    async updateStatus(id, status, completedAt) {
+        const doc = {
+            status,
+            updatedAt: new Date().toISOString(),
+        };
+        if (completedAt) {
+            doc.completedAt = completedAt.toISOString();
+        }
+        await this.client.update({
+            index: this.index,
+            id,
+            body: { doc },
+            refresh: this.refresh,
+        });
+        return await this.findById(id);
+    }
+    async updateCollectedData(id, collectedData) {
+        await this.client.update({
+            index: this.index,
+            id,
+            body: {
+                doc: {
+                    collectedData,
+                    updatedAt: new Date().toISOString(),
+                },
+            },
+            refresh: this.refresh,
+        });
+        return await this.findById(id);
+    }
+    async updateRouteStep(id, route, step) {
+        const doc = {
+            updatedAt: new Date().toISOString(),
+        };
+        if (route !== undefined) {
+            doc.currentRoute = route;
+        }
+        if (step !== undefined) {
+            doc.currentStep = step;
+        }
+        await this.client.update({
+            index: this.index,
+            id,
+            body: { doc },
+            refresh: this.refresh,
+        });
+        return await this.findById(id);
+    }
+    async incrementMessageCount(id) {
+        const session = await this.findById(id);
+        if (!session) {
+            return null;
+        }
+        const newCount = (session.messageCount || 0) + 1;
+        await this.client.update({
+            index: this.index,
+            id,
+            body: {
+                doc: {
+                    messageCount: newCount,
+                    lastMessageAt: new Date().toISOString(),
+                    updatedAt: new Date().toISOString(),
+                },
+            },
+            refresh: this.refresh,
+        });
+        return await this.findById(id);
+    }
+    async delete(id) {
+        try {
+            await this.client.delete({
+                index: this.index,
+                id,
+                refresh: this.refresh,
+            });
+            return true;
+        }
+        catch (error) {
+            if (this.isNotFoundError(error)) {
+                return false;
+            }
+            throw error;
+        }
+    }
+    deserializeSession(doc) {
+        return {
+            id: doc.id,
+            userId: doc.userId,
+            agentName: doc.agentName,
+            status: doc.status,
+            currentRoute: doc.currentRoute,
+            currentStep: doc.currentStep,
+            collectedData: doc.collectedData,
+            messageCount: doc.messageCount || 0,
+            createdAt: new Date(doc.createdAt),
+            updatedAt: new Date(doc.updatedAt),
+            lastMessageAt: doc.lastMessageAt
+                ? new Date(doc.lastMessageAt)
+                : undefined,
+            completedAt: doc.completedAt
+                ? new Date(doc.completedAt)
+                : undefined,
+        };
+    }
+    isNotFoundError(error) {
+        return (typeof error === "object" &&
+            error !== null &&
+            "statusCode" in error &&
+            error.statusCode === 404);
+    }
+}
+/**
+ * OpenSearch-based message repository implementation
+ */
+class OpenSearchMessageRepository {
+    constructor(client, index, refresh) {
+        this.client = client;
+        this.index = index;
+        this.refresh = refresh;
+    }
+    async create(data) {
+        const id = `msg_${Date.now()}_${Math.random().toString(36).slice(2)}`;
+        const now = new Date();
+        const message = {
+            id,
+            ...data,
+            createdAt: now,
+        };
+        await this.client.index({
+            index: this.index,
+            id,
+            body: this.serializeMessage(message),
+            refresh: this.refresh,
+        });
+        return message;
+    }
+    async findById(id) {
+        try {
+            const response = await this.client.get({
+                index: this.index,
+                id,
+            });
+            return this.deserializeMessage(response.body._source);
+        }
+        catch (error) {
+            if (this.isNotFoundError(error)) {
+                return null;
+            }
+            throw error;
+        }
+    }
+    async findBySessionId(sessionId, limit = 1000) {
+        const response = await this.client.search({
+            index: this.index,
+            body: {
+                query: {
+                    term: { sessionId },
+                },
+                sort: [{ createdAt: { order: "asc" } }],
+                size: limit,
+            },
+        });
+        return response.body.hits.hits.map((hit) => this.deserializeMessage(hit._source));
+    }
+    async findByUserId(userId, limit = 100) {
+        const response = await this.client.search({
+            index: this.index,
+            body: {
+                query: {
+                    term: { userId },
+                },
+                sort: [{ createdAt: { order: "desc" } }],
+                size: limit,
+            },
+        });
+        return response.body.hits.hits.map((hit) => this.deserializeMessage(hit._source));
+    }
+    async delete(id) {
+        try {
+            await this.client.delete({
+                index: this.index,
+                id,
+                refresh: this.refresh,
+            });
+            return true;
+        }
+        catch (error) {
+            if (this.isNotFoundError(error)) {
+                return false;
+            }
+            throw error;
+        }
+    }
+    async deleteBySessionId(sessionId) {
+        const response = await this.client.deleteByQuery({
+            index: this.index,
+            body: {
+                query: {
+                    term: { sessionId },
+                },
+            },
+            refresh: this.refresh === "wait_for" ? false : this.refresh,
+        });
+        return response.body.deleted;
+    }
+    async deleteByUserId(userId) {
+        const response = await this.client.deleteByQuery({
+            index: this.index,
+            body: {
+                query: {
+                    term: { userId },
+                },
+            },
+            refresh: this.refresh === "wait_for" ? false : this.refresh,
+        });
+        return response.body.deleted;
+    }
+    serializeMessage(message) {
+        return {
+            ...message,
+            createdAt: message.createdAt.toISOString(),
+        };
+    }
+    deserializeMessage(doc) {
+        return {
+            id: doc.id,
+            sessionId: doc.sessionId,
+            userId: doc.userId,
+            role: doc.role,
+            content: doc.content,
+            route: doc.route,
+            step: doc.step,
+            toolCalls: doc.toolCalls,
+            event: doc.event,
+            createdAt: new Date(doc.createdAt),
+        };
+    }
+    isNotFoundError(error) {
+        return (typeof error === "object" &&
+            error !== null &&
+            "statusCode" in error &&
+            error.statusCode === 404);
+    }
+}
+//# sourceMappingURL=OpenSearchAdapter.js.map

package/dist/src/adapters/OpenSearchAdapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenSearchAdapter.js","sourceRoot":"","sources":["../../../src/adapters/OpenSearchAdapter.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAsG3C;;;;;GAKG;AACH,MAAM,OAAO,iBAAiB;IAY5B,YACE,MAAwB,EACxB,UAAoC,EAAE;QAEtC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,OAAO,EAAE,QAAQ,IAAI,gBAAgB,CAAC;QAClE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,OAAO,EAAE,QAAQ,IAAI,gBAAgB,CAAC;QAClE,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,IAAI,IAAI,CAAC;QAC3D,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC;QAExC,IAAI,CAAC,iBAAiB,GAAG,IAAI,2BAA2B,CACtD,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,OAAO,CACb,CAAC;QAEF,IAAI,CAAC,iBAAiB,GAAG,IAAI,2BAA2B,CACtD,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,YAAY,EACjB,IAAI,CAAC,OAAO,CACb,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,sCAAsC;QACtC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YACrD,KAAK,EAAE,IAAI,CAAC,YAAY;SACzB,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC/B,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,IAAI,EAAE;oBACJ,QAAQ,EAAE;wBACR,UAAU,EAAE;4BACV,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACvB,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC3B,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC3B,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACjC,WAAW,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAChC,aAAa,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE;4BACjD,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACjC,SAAS,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;4BAC3B,SAAS,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;4BAC3B,aAAa,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;4BAC/B,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;yBAC9B;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QAED,sCAAsC;QACtC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YACrD,KAAK,EAAE,IAAI,CAAC,YAAY;SACzB,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC/B,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,IAAI,EAAE;oBACJ,QAAQ,EAAE;wBACR,UAAU,EAAE;4BACV,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACvB,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC9B,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACzB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;4BACzB,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BAC1B,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;4BACzB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE;4BAC7C,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE;4BACzC,SAAS,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;yBAC5B;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,wEAAwE;QACxE,8CAA8C;QAC9C,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,2BAA2B;IAG/B,YACU,MAAwB,EACxB,KAAa,EACb,OAA6B;QAF7B,WAAM,GAAN,MAAM,CAAkB;QACxB,UAAK,GAAL,KAAK,CAAQ;QACb,YAAO,GAAP,OAAO,CAAsB;IACpC,CAAC;IAEJ,KAAK,CAAC,MAAM,CAAC,IAA8B;QACzC,MAAM,EAAE,GACN,IAAI,CAAC,EAAE,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,OAAO,GAAuB;YAClC,GAAG,IAAI;YACP,EAAE;YACF,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,QAAQ;YAC/B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,CAAC;YACpC,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QACF,MAAM,aAAa,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;QAEzC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE,aAAmD;YACzD,OAAO,EAAE,IAAI,EAAE,gEAAgE;SAChF,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;gBACrC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,EAAE;aACH,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAAc;QACrC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE;wBACJ,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC;qBAC7D;iBACF;gBACD,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC;gBACxC,IAAI,EAAE,CAAC;aACR;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;QACrC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,MAAc,EACd,KAAK,GAAG,GAAG;QAEX,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE,EAAE,MAAM,EAAE;iBACjB;gBACD,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC;gBACxC,IAAI,EAAE,KAAK;aACZ;SACF,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACzC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,CACrC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,EAAU,EACV,OAA8D;QAE9D,MAAM,GAAG,GAA4B;YACnC,GAAG,OAAO;YACV,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,kBAAkB;QAClB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,GAAG,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QACtD,CAAC;QACD,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YAC1B,GAAG,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;QAC1D,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE,EAAE,GAAG,EAAE;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,EAAU,EACV,MAAoC,EACpC,WAAkB;QAElB,MAAM,GAAG,GAA4B;YACnC,MAAM;YACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YAChB,GAAG,CAAC,WAAW,GAAG,WAAW,CAAC,WAAW,EAAE,CAAC;QAC9C,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE,EAAE,GAAG,EAAE;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,EAAU,EACV,aAAwC;QAExC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE;gBACJ,GAAG,EAAE;oBACH,aAAa;oBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;aACF;YACD,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,EAAU,EACV,KAAc,EACd,IAAa;QAEb,MAAM,GAAG,GAA4B;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC;QAC3B,CAAC;QACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE,EAAE,GAAG,EAAE;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,EAAU;QACpC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAEjD,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE;gBACJ,GAAG,EAAE;oBACH,YAAY,EAAE,QAAQ;oBACtB,aAAa,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACvC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;aACF;YACD,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,EAAE;gBACF,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,GAA4B;QACrD,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,MAAM,EAAE,GAAG,CAAC,MAA4B;YACxC,SAAS,EAAE,GAAG,CAAC,SAA+B;YAC9C,MAAM,EAAE,GAAG,CAAC,MAAsC;YAClD,YAAY,EAAE,GAAG,CAAC,YAAkC;YACpD,WAAW,EAAE,GAAG,CAAC,WAAiC;YAClD,aAAa,EAAE,GAAG,CAAC,aAAsD;YACzE,YAAY,EAAG,GAAG,CAAC,YAAuB,IAAI,CAAC;YAC/C,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAmB,CAAC;YAC5C,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAmB,CAAC;YAC5C,aAAa,EAAE,GAAG,CAAC,aAAa;gBAC9B,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,aAAuB,CAAC;gBACvC,CAAC,CAAC,SAAS;YACb,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC1B,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,WAAqB,CAAC;gBACrC,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,KAAc;QACpC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,YAAY,IAAI,KAAK;YACrB,KAAK,CAAC,UAAU,KAAK,GAAG,CACzB,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,2BAA2B;IAC/B,YACU,MAAwB,EACxB,KAAa,EACb,OAA6B;QAF7B,WAAM,GAAN,MAAM,CAAkB;QACxB,UAAK,GAAL,KAAK,CAAQ;QACb,YAAO,GAAP,OAAO,CAAsB;IACpC,CAAC;IAEJ,KAAK,CAAC,MAAM,CACV,IAA2C;QAE3C,MAAM,EAAE,GAAG,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,MAAM,OAAO,GAAgB;YAC3B,EAAE;YACF,GAAG,IAAI;YACP,SAAS,EAAE,GAAG;SACf,CAAC;QAEF,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,EAAE;YACF,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;YACpC,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;gBACrC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,EAAE;aACH,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,SAAiB,EACjB,KAAK,GAAG,IAAI;QAEZ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE,EAAE,SAAS,EAAE;iBACpB;gBACD,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;gBACvC,IAAI,EAAE,KAAK;aACZ;SACF,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACzC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,CACrC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,KAAK,GAAG,GAAG;QAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;YACxC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE,EAAE,MAAM,EAAE;iBACjB;gBACD,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC;gBACxC,IAAI,EAAE,KAAK;aACZ;SACF,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACzC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,CACrC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,EAAE;gBACF,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAC/C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE,EAAE,SAAS,EAAE;iBACpB;aACF;YACD,OAAO,EAAE,IAAI,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;SAC5D,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAC/C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,IAAI,EAAE;gBACJ,KAAK,EAAE;oBACL,IAAI,EAAE,EAAE,MAAM,EAAE;iBACjB;aACF;YACD,OAAO,EAAE,IAAI,CAAC,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;SAC5D,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;IAC/B,CAAC;IAEO,gBAAgB,CAAC,OAAoB;QAC3C,OAAO;YACL,GAAG,OAAO;YACV,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE;SAC3C,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,GAA4B;QACrD,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,SAAS,EAAE,GAAG,CAAC,SAAmB;YAClC,MAAM,EAAE,GAAG,CAAC,MAA4B;YACxC,IAAI,EAAE,GAAG,CAAC,IAA2B;YACrC,OAAO,EAAE,GAAG,CAAC,OAAiB;YAC9B,KAAK,EAAE,GAAG,CAAC,KAA2B;YACtC,IAAI,EAAE,GAAG,CAAC,IAA0B;YACpC,SAAS,EAAE,GAAG,CAAC,SAEF;YACb,KAAK,EAAE,GAAG,CAAC,KAAyC;YACpD,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAmB,CAAC;SAC7C,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,KAAc;QACpC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,YAAY,IAAI,KAAK;YACrB,KAAK,CAAC,UAAU,KAAK,GAAG,CACzB,CAAC;IACJ,CAAC;CACF"}

package/dist/src/adapters/PostgreSQLAdapter.d.ts

@@ -0,0 +1,71 @@
+/**
+ * PostgreSQL adapter for persistence
+ * Raw SQL adapter for PostgreSQL with custom schemas
+ */
+import type { SessionRepository, MessageRepository, PersistenceAdapter } from "../types";
+/**
+ * PostgreSQL query result interface
+ */
+export interface PgQueryResult<T = Record<string, unknown>> {
+    rows: T[];
+    rowCount: number;
+}
+/**
+ * PostgreSQL client interface - matches pg (node-postgres)
+ */
+export interface PgClient {
+    query<T = Record<string, unknown>>(sql: string, values?: unknown[]): Promise<PgQueryResult<T>>;
+    end(): Promise<void>;
+}
+/**
+ * Options for PostgreSQL adapter
+ */
+export interface PostgreSQLAdapterOptions {
+    /**
+     * PostgreSQL client instance (from 'pg' package)
+     */
+    client: PgClient;
+    /**
+     * Table names (default: "agent_sessions" and "agent_messages")
+     */
+    tables?: {
+        sessions?: string;
+        messages?: string;
+    };
+}
+/**
+ * PostgreSQL Adapter - Provider-style API for PostgreSQL persistence
+ *
+ * @example
+ * ```typescript
+ * import { Client } from 'pg';
+ * import { Agent, PostgreSQLAdapter } from '@falai/agent';
+ *
+ * const client = new Client({
+ *   host: 'localhost',
+ *   port: 5432,
+ *   database: 'myapp',
+ *   user: 'postgres',
+ *   password: 'password',
+ * });
+ * await client.connect();
+ *
+ * const agent = new Agent({
+ *   name: "My Agent",
+ *   provider: provider,
+ *   persistence: {
+ *     adapter: new PostgreSQLAdapter({ client }),
+ *     userId: "user_123",
+ *   },
+ * });
+ * ```
+ */
+export declare class PostgreSQLAdapter<TData = Record<string, unknown>> implements PersistenceAdapter<TData> {
+    readonly sessionRepository: SessionRepository<TData>;
+    readonly messageRepository: MessageRepository;
+    private client;
+    constructor(options: PostgreSQLAdapterOptions);
+    initialize(): Promise<void>;
+    disconnect(): Promise<void>;
+}
+//# sourceMappingURL=PostgreSQLAdapter.d.ts.map

package/dist/src/adapters/PostgreSQLAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PostgreSQLAdapter.d.ts","sourceRoot":"","sources":["../../../src/adapters/PostgreSQLAdapter.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAEV,iBAAiB,EACjB,iBAAiB,EAIjB,kBAAkB,EACnB,MAAM,UAAU,CAAC;AAElB;;GAEG;AACH,MAAM,WAAW,aAAa,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACxD,IAAI,EAAE,CAAC,EAAE,CAAC;IACV,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,QAAQ;IACvB,KAAK,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,GAAG,EAAE,MAAM,EACX,MAAM,CAAC,EAAE,OAAO,EAAE,GACjB,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7B,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC;;OAEG;IACH,MAAM,EAAE,QAAQ,CAAC;IAEjB;;OAEG;IACH,MAAM,CAAC,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,qBAAa,iBAAiB,CAAC,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAC5D,YAAW,kBAAkB,CAAC,KAAK,CAAC;IAEpC,SAAgB,iBAAiB,EAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAC5D,SAAgB,iBAAiB,EAAE,iBAAiB,CAAC;IACrD,OAAO,CAAC,MAAM,CAAW;gBAEb,OAAO,EAAE,wBAAwB;IAiBvC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAiD3B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC"}