@mastra/lance 1.1.1 → 1.1.2-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,42 @@
 # @mastra/lance
 
+## 1.1.2-alpha.0
+
+### Patch Changes
+
+- Add optional `batchId`, `datasetId`, and `datasetItemId` fields to persisted scores so saved baseline scores can be grouped as one scoring pass and joined back to the dataset items they came from. ([#18331](https://github.com/mastra-ai/mastra/pull/18331))
+  - `scoreTrace()` accepts top-level `batchId`, `datasetId`, and `datasetItemId` when persisting a score for a stored trace.
+  - `ScoreRowData` and score save payloads now include nullable `batchId`, `datasetId`, and `datasetItemId`.
+  - Built-in stores with explicit score schema or attribute mappings now persist these provenance fields on saved scores.
+  - D1, DSQL, MSSQL, and Upstash score stores now apply additive provenance migrations or deterministic score ordering for persisted score reads.
+
+  ```ts
+  await scoreTrace({
+    storage,
+    scorer,
+    target: { traceId },
+    batchId: 'baseline-batch-1',
+    datasetId,
+    datasetItemId,
+  });
+  ```
+
+- Added optional `organizationId` and `projectId` fields to scores for multi-tenant isolation. Scores can now be saved with tenancy metadata and the `listScoresBy*` methods accept a `filters` option to scope results by organization and project. ([#18331](https://github.com/mastra-ai/mastra/pull/18331))
+
+  ```ts
+  await storage.saveScore({ ...score, organizationId: 'org-a', projectId: 'proj-1' });
+
+  const result = await storage.listScoresByScorerId({
+    scorerId,
+    filters: { organizationId: 'org-a', projectId: 'proj-1' },
+  });
+  ```
+
+  `projectId` identifies the project scope, separate from `resourceId` which continues to mean the agent memory resource.
+
+- Updated dependencies [[`9250acd`](https://github.com/mastra-ai/mastra/commit/9250acd1357f0f1f33d0dcca16f9655084c58eca), [`215f9b0`](https://github.com/mastra-ai/mastra/commit/215f9b0f3f3f6fc165edad360582dd4d3d7ea748), [`c64c2a8`](https://github.com/mastra-ai/mastra/commit/c64c2a8503a50252f9ca6b8e8c54cadee31b92a2), [`06e2680`](https://github.com/mastra-ai/mastra/commit/06e26806b51d2cbd858afdc66daa2b86ff3ba64a), [`1240f05`](https://github.com/mastra-ai/mastra/commit/1240f051c8e5371f1c014448bf37b1a1b9a05e47), [`215f9b0`](https://github.com/mastra-ai/mastra/commit/215f9b0f3f3f6fc165edad360582dd4d3d7ea748), [`24c10d3`](https://github.com/mastra-ai/mastra/commit/24c10d333e6649ac06075903aeeee13a933db3b3), [`24c10d3`](https://github.com/mastra-ai/mastra/commit/24c10d333e6649ac06075903aeeee13a933db3b3), [`24c10d3`](https://github.com/mastra-ai/mastra/commit/24c10d333e6649ac06075903aeeee13a933db3b3), [`24c10d3`](https://github.com/mastra-ai/mastra/commit/24c10d333e6649ac06075903aeeee13a933db3b3), [`215f9b0`](https://github.com/mastra-ai/mastra/commit/215f9b0f3f3f6fc165edad360582dd4d3d7ea748), [`215f9b0`](https://github.com/mastra-ai/mastra/commit/215f9b0f3f3f6fc165edad360582dd4d3d7ea748)]:
+  - @mastra/core@1.49.0-alpha.5
+
 ## 1.1.1
 
 ### Patch Changes

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-lance
 description: Documentation for @mastra/lance. Use when working with @mastra/lance APIs, configuration, or implementation.
 metadata:
   package: "@mastra/lance"
-  version: "1.1.1"
+  version: "1.1.2-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.1.1",
+  "version": "1.1.2-alpha.0",
   "package": "@mastra/lance",
   "exports": {},
   "modules": {}

package/dist/docs/references/docs-rag-vector-databases.md

@@ -1,3 +1,5 @@
+> Discover all available pages from the documentation index: https://mastra.ai/llms.txt
+
 # Storing embeddings in a vector database
 
 After generating embeddings, you need to store them in a database that supports vector similarity search. Mastra provides a consistent interface for storing and querying embeddings across various vector databases.

package/dist/docs/references/reference-storage-lance.md

@@ -1,3 +1,5 @@
+> Discover all available pages from the documentation index: https://mastra.ai/llms.txt
+
 # LanceDB storage
 
 The LanceDB storage implementation provides a high-performance storage solution using the LanceDB database system, which excels at handling both traditional data storage and vector operations.

package/dist/docs/references/reference-vectors-lance.md

@@ -1,3 +1,5 @@
+> Discover all available pages from the documentation index: https://mastra.ai/llms.txt
+
 # Lance vector store
 
 The LanceVectorStore class provides vector search using [LanceDB](https://lancedb.github.io/lancedb/), an embedded vector database built on the Lance columnar format. It offers efficient storage and fast similarity search for both local development and production deployments.

package/dist/index.cjs

@@ -1641,6 +1641,16 @@ var StoreMemoryLance = class extends storage.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 storage.ScoresStorage {
   client;
   #db;
@@ -1655,7 +1665,7 @@ var StoreScoresLance = class extends storage.ScoresStorage {
     await this.#db.alterTable({
       tableName: storage.TABLE_SCORERS,
       schema: storage.SCORERS_SCHEMA,
-      ifNotExists: ["spanId", "requestContext"]
+      ifNotExists: ["spanId", "requestContext", "organizationId", "projectId", "batchId", "datasetId", "datasetItemId"]
     });
   }
   async dangerouslyClearAll() {
@@ -1721,7 +1731,7 @@ var StoreScoresLance = class extends storage.ScoresStorage {
   async getScoreById({ id }) {
     try {
       const table = await this.client.openTable(storage.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]);
@@ -1759,33 +1769,33 @@ var StoreScoresLance = class extends storage.ScoresStorage {
     pagination,
     entityId,
     entityType,
-    source
+    source,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = storage.normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(storage.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;
@@ -1819,17 +1829,18 @@ var StoreScoresLance = class extends storage.ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = storage.normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(storage.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);
@@ -1861,17 +1872,22 @@ var StoreScoresLance = class extends storage.ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = storage.normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(storage.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);
@@ -1903,17 +1919,22 @@ var StoreScoresLance = class extends storage.ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = storage.normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const table = await this.client.openTable(storage.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);