@mastra/lance 1.1.1-alpha.0 → 1.1.2-alpha.0

package/dist/index.js

@@ -1639,6 +1639,16 @@ var StoreMemoryLance = class extends MemoryStorage {
     throw new Error("Unexpected end of retry loop");
   }
 };
+function lanceStringLiteral(value) {
+  return `'${value.replaceAll("'", "''")}'`;
+}
+function tenancyClause(filters) {
+  const parts = [];
+  if (filters?.organizationId !== void 0)
+    parts.push(`\`organizationId\` = ${lanceStringLiteral(filters.organizationId)}`);
+  if (filters?.projectId !== void 0) parts.push(`\`projectId\` = ${lanceStringLiteral(filters.projectId)}`);
+  return parts.length ? ` AND ${parts.join(" AND ")}` : "";
+}
 var StoreScoresLance = class extends ScoresStorage {
   client;
   #db;
@@ -1653,7 +1663,7 @@ var StoreScoresLance = class extends ScoresStorage {
     await this.#db.alterTable({
       tableName: TABLE_SCORERS,
       schema: SCORERS_SCHEMA,
-      ifNotExists: ["spanId", "requestContext"]
+      ifNotExists: ["spanId", "requestContext", "organizationId", "projectId", "batchId", "datasetId", "datasetItemId"]
     });
   }
   async dangerouslyClearAll() {
@@ -1719,7 +1729,7 @@ var StoreScoresLance = class extends ScoresStorage {
   async getScoreById({ id }) {
     try {
       const table = await this.client.openTable(TABLE_SCORERS);
-      const query = table.query().where(`id = '${id}'`).limit(1);
+      const query = table.query().where(`id = ${lanceStringLiteral(id)}`).limit(1);
       const records = await query.toArray();
       if (records.length === 0) return null;
       return await this.transformScoreRow(records[0]);
@@ -1757,33 +1767,33 @@ var StoreScoresLance = class extends ScoresStorage {
     pagination,
     entityId,
     entityType,
-    source
+    source,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(TABLE_SCORERS);
-      let query = table.query().where(`\`scorerId\` = '${scorerId}'`);
+      const conditions = [`\`scorerId\` = ${lanceStringLiteral(scorerId)}`];
       if (source) {
-        query = query.where(`\`source\` = '${source}'`);
+        conditions.push(`\`source\` = ${lanceStringLiteral(source)}`);
       }
       if (entityId) {
-        query = query.where(`\`entityId\` = '${entityId}'`);
+        conditions.push(`\`entityId\` = ${lanceStringLiteral(entityId)}`);
       }
       if (entityType) {
-        query = query.where(`\`entityType\` = '${entityType}'`);
-      }
-      let totalQuery = table.query().where(`\`scorerId\` = '${scorerId}'`);
-      if (source) {
-        totalQuery = totalQuery.where(`\`source\` = '${source}'`);
+        conditions.push(`\`entityType\` = ${lanceStringLiteral(entityType)}`);
       }
-      if (entityId) {
-        totalQuery = totalQuery.where(`\`entityId\` = '${entityId}'`);
+      if (filters?.organizationId !== void 0) {
+        conditions.push(`\`organizationId\` = ${lanceStringLiteral(filters.organizationId)}`);
       }
-      if (entityType) {
-        totalQuery = totalQuery.where(`\`entityType\` = '${entityType}'`);
+      if (filters?.projectId !== void 0) {
+        conditions.push(`\`projectId\` = ${lanceStringLiteral(filters.projectId)}`);
       }
+      const whereClause = conditions.join(" AND ");
+      let query = table.query().where(whereClause);
+      const totalQuery = table.query().where(whereClause);
       const allRecords = await totalQuery.toArray();
       const total = allRecords.length;
       const end = perPageInput === false ? total : start + perPage;
@@ -1817,17 +1827,18 @@ var StoreScoresLance = class extends ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(TABLE_SCORERS);
-      const allRecords = await table.query().where(`\`runId\` = '${runId}'`).toArray();
+      const allRecords = await table.query().where(`\`runId\` = ${lanceStringLiteral(runId)}${tenancyClause(filters)}`).toArray();
       const total = allRecords.length;
       const end = perPageInput === false ? total : start + perPage;
-      let query = table.query().where(`\`runId\` = '${runId}'`);
+      let query = table.query().where(`\`runId\` = ${lanceStringLiteral(runId)}${tenancyClause(filters)}`);
       if (perPageInput !== false) {
         query = query.limit(perPage);
         if (start > 0) query = query.offset(start);
@@ -1859,17 +1870,22 @@ var StoreScoresLance = class extends ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(TABLE_SCORERS);
-      const allRecords = await table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).toArray();
+      const allRecords = await table.query().where(
+        `\`entityId\` = ${lanceStringLiteral(entityId)} AND \`entityType\` = ${lanceStringLiteral(entityType)}${tenancyClause(filters)}`
+      ).toArray();
       const total = allRecords.length;
       const end = perPageInput === false ? total : start + perPage;
-      let query = table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`);
+      let query = table.query().where(
+        `\`entityId\` = ${lanceStringLiteral(entityId)} AND \`entityType\` = ${lanceStringLiteral(entityType)}${tenancyClause(filters)}`
+      );
       if (perPageInput !== false) {
         query = query.limit(perPage);
         if (start > 0) query = query.offset(start);
@@ -1901,17 +1917,22 @@ var StoreScoresLance = class extends ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(TABLE_SCORERS);
-      const allRecords = await table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).toArray();
+      const allRecords = await table.query().where(
+        `\`traceId\` = ${lanceStringLiteral(traceId)} AND \`spanId\` = ${lanceStringLiteral(spanId)}${tenancyClause(filters)}`
+      ).toArray();
       const total = allRecords.length;
       const end = perPageInput === false ? total : start + perPage;
-      let query = table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`);
+      let query = table.query().where(
+        `\`traceId\` = ${lanceStringLiteral(traceId)} AND \`spanId\` = ${lanceStringLiteral(spanId)}${tenancyClause(filters)}`
+      );
       if (perPageInput !== false) {
         query = query.limit(perPage);
         if (start > 0) query = query.offset(start);