@mastra/libsql 0.0.4 → 0.1.0-alpha.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/libsql",
-  "version": "0.0.4",
+  "version": "0.1.0-alpha.1",
   "description": "Libsql provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -20,8 +20,7 @@
   },
   "license": "MIT",
   "dependencies": {
-    "@libsql/client": "^0.15.4",
-    "@mastra/core": "^0.9.4"
+    "@libsql/client": "^0.15.4"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",
@@ -31,7 +30,11 @@
     "typescript": "^5.8.3",
     "vitest": "^3.1.2",
     "@internal/lint": "0.0.5",
-    "@internal/storage-test-utils": "0.0.2-alpha.5"
+    "@internal/storage-test-utils": "0.0.2-alpha.5",
+    "@mastra/core": "0.10.0-alpha.1"
+  },
+  "peerDependencies": {
+    "@mastra/core": "^0.9.4"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",

package/src/storage/index.ts

@@ -18,6 +18,7 @@ import type {
   WorkflowRun,
   WorkflowRuns,
 } from '@mastra/core/storage';
+import { parseSqlIdentifier } from '@mastra/core/utils';
 import type { WorkflowRunState } from '@mastra/core/workflows';
 
 function safelyParseJSON(jsonString: string): any {
@@ -48,7 +49,9 @@ export class LibSQLStore extends MastraStorage {
   }
 
   private getCreateTableSQL(tableName: TABLE_NAMES, schema: Record<string, StorageColumn>): string {
+    const parsedTableName = parseSqlIdentifier(tableName, 'table name');
     const columns = Object.entries(schema).map(([name, col]) => {
+      const parsedColumnName = parseSqlIdentifier(name, 'column name');
       let type = col.type.toUpperCase();
       if (type === 'TEXT') type = 'TEXT';
       if (type === 'TIMESTAMP') type = 'TEXT'; // Store timestamps as ISO strings
@@ -57,19 +60,19 @@ export class LibSQLStore extends MastraStorage {
       const nullable = col.nullable ? '' : 'NOT NULL';
       const primaryKey = col.primaryKey ? 'PRIMARY KEY' : '';
 
-      return `${name} ${type} ${nullable} ${primaryKey}`.trim();
+      return `${parsedColumnName} ${type} ${nullable} ${primaryKey}`.trim();
     });
 
     // For workflow_snapshot table, create a composite primary key
     if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
-      const stmnt = `CREATE TABLE IF NOT EXISTS ${tableName} (
+      const stmnt = `CREATE TABLE IF NOT EXISTS ${parsedTableName} (
                 ${columns.join(',\n')},
                 PRIMARY KEY (workflow_name, run_id)
             )`;
       return stmnt;
     }
 
-    return `CREATE TABLE IF NOT EXISTS ${tableName} (${columns.join(', ')})`;
+    return `CREATE TABLE IF NOT EXISTS ${parsedTableName} (${columns.join(', ')})`;
   }
 
   async createTable({
@@ -90,8 +93,9 @@ export class LibSQLStore extends MastraStorage {
   }
 
   async clearTable({ tableName }: { tableName: TABLE_NAMES }): Promise<void> {
+    const parsedTableName = parseSqlIdentifier(tableName, 'table name');
     try {
-      await this.client.execute(`DELETE FROM ${tableName}`);
+      await this.client.execute(`DELETE FROM ${parsedTableName}`);
     } catch (e) {
       if (e instanceof Error) {
         this.logger.error(e.message);
@@ -103,7 +107,8 @@ export class LibSQLStore extends MastraStorage {
     sql: string;
     args: InValue[];
   } {
-    const columns = Object.keys(record);
+    const parsedTableName = parseSqlIdentifier(tableName, 'table name');
+    const columns = Object.keys(record).map(col => parseSqlIdentifier(col, 'column name'));
     const values = Object.values(record).map(v => {
       if (typeof v === `undefined`) {
         // returning an undefined value will cause libsql to throw
@@ -117,7 +122,7 @@ export class LibSQLStore extends MastraStorage {
     const placeholders = values.map(() => '?').join(', ');
 
     return {
-      sql: `INSERT OR REPLACE INTO ${tableName} (${columns.join(', ')}) VALUES (${placeholders})`,
+      sql: `INSERT OR REPLACE INTO ${parsedTableName} (${columns.join(', ')}) VALUES (${placeholders})`,
       args: values,
     };
   }
@@ -149,13 +154,15 @@ export class LibSQLStore extends MastraStorage {
   }
 
   async load<R>({ tableName, keys }: { tableName: TABLE_NAMES; keys: Record<string, string> }): Promise<R | null> {
-    const conditions = Object.entries(keys)
-      .map(([key]) => `${key} = ?`)
-      .join(' AND ');
+    const parsedTableName = parseSqlIdentifier(tableName, 'table name');
+
+    const parsedKeys = Object.keys(keys).map(key => parseSqlIdentifier(key, 'column name'));
+
+    const conditions = parsedKeys.map(key => `${key} = ?`).join(' AND ');
     const values = Object.values(keys);
 
     const result = await this.client.execute({
-      sql: `SELECT * FROM ${tableName} WHERE ${conditions} ORDER BY createdAt DESC LIMIT 1`,
+      sql: `SELECT * FROM ${parsedTableName} WHERE ${conditions} ORDER BY createdAt DESC LIMIT 1`,
       args: values,
     });
 
@@ -506,6 +513,7 @@ export class LibSQLStore extends MastraStorage {
     if (toDate) {
       conditions.push('createdAt <= ?');
     }
+
     const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';
 
     if (name) {