@mastra/lance 0.3.10 → 1.0.0-beta.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.cjs CHANGED
@@ -5,131 +5,15 @@ var error = require('@mastra/core/error');
5
5
  var storage = require('@mastra/core/storage');
6
6
  var agent = require('@mastra/core/agent');
7
7
  var apacheArrow = require('apache-arrow');
8
- var scores = require('@mastra/core/scores');
8
+ var evals = require('@mastra/core/evals');
9
9
  var vector = require('@mastra/core/vector');
10
10
  var filter = require('@mastra/core/vector/filter');
11
11
 
12
12
  // src/storage/index.ts
13
- var StoreLegacyEvalsLance = class extends storage.LegacyEvalsStorage {
14
- client;
15
- constructor({ client }) {
16
- super();
17
- this.client = client;
18
- }
19
- async getEvalsByAgentName(agentName, type) {
20
- try {
21
- const table = await this.client.openTable(storage.TABLE_EVALS);
22
- const query = table.query().where(`agent_name = '${agentName}'`);
23
- const records = await query.toArray();
24
- let filteredRecords = records;
25
- if (type === "live") {
26
- filteredRecords = records.filter((record) => record.test_info === null);
27
- } else if (type === "test") {
28
- filteredRecords = records.filter((record) => record.test_info !== null);
29
- }
30
- return filteredRecords.map((record) => {
31
- return {
32
- id: record.id,
33
- input: record.input,
34
- output: record.output,
35
- agentName: record.agent_name,
36
- metricName: record.metric_name,
37
- result: JSON.parse(record.result),
38
- instructions: record.instructions,
39
- testInfo: record.test_info ? JSON.parse(record.test_info) : null,
40
- globalRunId: record.global_run_id,
41
- runId: record.run_id,
42
- createdAt: new Date(record.created_at).toString()
43
- };
44
- });
45
- } catch (error$1) {
46
- throw new error.MastraError(
47
- {
48
- id: "LANCE_STORE_GET_EVALS_BY_AGENT_NAME_FAILED",
49
- domain: error.ErrorDomain.STORAGE,
50
- category: error.ErrorCategory.THIRD_PARTY,
51
- details: { agentName }
52
- },
53
- error$1
54
- );
55
- }
56
- }
57
- async getEvals(options) {
58
- try {
59
- const table = await this.client.openTable(storage.TABLE_EVALS);
60
- const conditions = [];
61
- if (options.agentName) {
62
- conditions.push(`agent_name = '${options.agentName}'`);
63
- }
64
- if (options.type === "live") {
65
- conditions.push("test_info IS NULL");
66
- } else if (options.type === "test") {
67
- conditions.push("test_info IS NOT NULL");
68
- }
69
- const startDate = options.dateRange?.start || options.fromDate;
70
- const endDate = options.dateRange?.end || options.toDate;
71
- if (startDate) {
72
- conditions.push(`\`created_at\` >= ${startDate.getTime()}`);
73
- }
74
- if (endDate) {
75
- conditions.push(`\`created_at\` <= ${endDate.getTime()}`);
76
- }
77
- let total = 0;
78
- if (conditions.length > 0) {
79
- total = await table.countRows(conditions.join(" AND "));
80
- } else {
81
- total = await table.countRows();
82
- }
83
- const query = table.query();
84
- if (conditions.length > 0) {
85
- const whereClause = conditions.join(" AND ");
86
- query.where(whereClause);
87
- }
88
- const records = await query.toArray();
89
- const evals = records.sort((a, b) => b.created_at - a.created_at).map((record) => {
90
- return {
91
- id: record.id,
92
- input: record.input,
93
- output: record.output,
94
- agentName: record.agent_name,
95
- metricName: record.metric_name,
96
- result: JSON.parse(record.result),
97
- instructions: record.instructions,
98
- testInfo: record.test_info ? JSON.parse(record.test_info) : null,
99
- globalRunId: record.global_run_id,
100
- runId: record.run_id,
101
- createdAt: new Date(record.created_at).toISOString()
102
- };
103
- });
104
- const page = options.page || 0;
105
- const perPage = options.perPage || 10;
106
- const pagedEvals = evals.slice(page * perPage, (page + 1) * perPage);
107
- return {
108
- evals: pagedEvals,
109
- total,
110
- page,
111
- perPage,
112
- hasMore: total > (page + 1) * perPage
113
- };
114
- } catch (error$1) {
115
- throw new error.MastraError(
116
- {
117
- id: "LANCE_STORE_GET_EVALS_FAILED",
118
- domain: error.ErrorDomain.STORAGE,
119
- category: error.ErrorCategory.THIRD_PARTY,
120
- details: { agentName: options.agentName ?? "" }
121
- },
122
- error$1
123
- );
124
- }
125
- }
126
- };
127
13
  function getPrimaryKeys(tableName) {
128
14
  let primaryId = ["id"];
129
15
  if (tableName === storage.TABLE_WORKFLOW_SNAPSHOT) {
130
16
  primaryId = ["workflow_name", "run_id"];
131
- } else if (tableName === storage.TABLE_EVALS) {
132
- primaryId = ["agent_name", "metric_name", "run_id"];
133
17
  }
134
18
  return primaryId;
135
19
  }
@@ -249,6 +133,10 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
249
133
  this.client = client;
250
134
  this.operations = operations;
251
135
  }
136
+ // Utility to escape single quotes in SQL strings
137
+ escapeSql(str) {
138
+ return str.replace(/'/g, "''");
139
+ }
252
140
  async getThreadById({ threadId }) {
253
141
  try {
254
142
  const thread = await this.operations.load({ tableName: storage.TABLE_THREADS, keys: { id: threadId } });
@@ -271,26 +159,6 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
271
159
  );
272
160
  }
273
161
  }
274
- async getThreadsByResourceId({ resourceId }) {
275
- try {
276
- const table = await this.client.openTable(storage.TABLE_THREADS);
277
- const query = table.query().where(`\`resourceId\` = '${resourceId}'`);
278
- const records = await query.toArray();
279
- return processResultWithTypeConversion(
280
- records,
281
- await getTableSchema({ tableName: storage.TABLE_THREADS, client: this.client })
282
- );
283
- } catch (error$1) {
284
- throw new error.MastraError(
285
- {
286
- id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_FAILED",
287
- domain: error.ErrorDomain.STORAGE,
288
- category: error.ErrorCategory.THIRD_PARTY
289
- },
290
- error$1
291
- );
292
- }
293
- }
294
162
  /**
295
163
  * Saves a thread to the database. This function doesn't overwrite existing threads.
296
164
  * @param thread - The thread to save
@@ -395,100 +263,174 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
395
263
  })() : message.content
396
264
  };
397
265
  }
398
- async getMessages({
399
- threadId,
400
- resourceId,
401
- selectBy,
402
- format,
403
- threadConfig
404
- }) {
266
+ async listMessagesById({ messageIds }) {
267
+ if (messageIds.length === 0) return { messages: [] };
405
268
  try {
406
- if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
407
- if (threadConfig) {
408
- throw new Error("ThreadConfig is not supported by LanceDB storage");
409
- }
410
- const limit = storage.resolveMessageLimit({ last: selectBy?.last, defaultLimit: Number.MAX_SAFE_INTEGER });
411
269
  const table = await this.client.openTable(storage.TABLE_MESSAGES);
412
- let allRecords = [];
413
- if (selectBy?.include && selectBy.include.length > 0) {
414
- const threadIds = [...new Set(selectBy.include.map((item) => item.threadId))];
415
- for (const threadId2 of threadIds) {
416
- const threadQuery = table.query().where(`thread_id = '${threadId2}'`);
417
- let threadRecords = await threadQuery.toArray();
418
- allRecords.push(...threadRecords);
419
- }
420
- } else {
421
- let query = table.query().where(`\`thread_id\` = '${threadId}'`);
422
- allRecords = await query.toArray();
423
- }
424
- allRecords.sort((a, b) => {
425
- const dateA = new Date(a.createdAt).getTime();
426
- const dateB = new Date(b.createdAt).getTime();
427
- return dateA - dateB;
428
- });
429
- if (selectBy?.include && selectBy.include.length > 0) {
430
- allRecords = this.processMessagesWithContext(allRecords, selectBy.include);
431
- }
432
- if (limit !== Number.MAX_SAFE_INTEGER) {
433
- allRecords = allRecords.slice(-limit);
434
- }
270
+ const quotedIds = messageIds.map((id) => `'${id}'`).join(", ");
271
+ const allRecords = await table.query().where(`id IN (${quotedIds})`).toArray();
435
272
  const messages = processResultWithTypeConversion(
436
273
  allRecords,
437
274
  await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
438
275
  );
439
- const list = new agent.MessageList({ threadId, resourceId }).add(messages.map(this.normalizeMessage), "memory");
440
- if (format === "v2") return list.get.all.v2();
441
- return list.get.all.v1();
276
+ const list = new agent.MessageList().add(
277
+ messages.map(this.normalizeMessage),
278
+ "memory"
279
+ );
280
+ return { messages: list.get.all.db() };
442
281
  } catch (error$1) {
443
282
  throw new error.MastraError(
444
283
  {
445
- id: "LANCE_STORE_GET_MESSAGES_FAILED",
284
+ id: "LANCE_STORE_LIST_MESSAGES_BY_ID_FAILED",
446
285
  domain: error.ErrorDomain.STORAGE,
447
286
  category: error.ErrorCategory.THIRD_PARTY,
448
287
  details: {
449
- threadId,
450
- resourceId: resourceId ?? ""
288
+ messageIds: JSON.stringify(messageIds)
451
289
  }
452
290
  },
453
291
  error$1
454
292
  );
455
293
  }
456
294
  }
457
- async getMessagesById({
458
- messageIds,
459
- format
460
- }) {
461
- if (messageIds.length === 0) return [];
295
+ async listMessages(args) {
296
+ const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
297
+ if (!threadId.trim()) {
298
+ throw new error.MastraError(
299
+ {
300
+ id: "STORAGE_LANCE_LIST_MESSAGES_INVALID_THREAD_ID",
301
+ domain: error.ErrorDomain.STORAGE,
302
+ category: error.ErrorCategory.THIRD_PARTY,
303
+ details: { threadId }
304
+ },
305
+ new Error("threadId must be a non-empty string")
306
+ );
307
+ }
308
+ const perPage = storage.normalizePerPage(perPageInput, 40);
309
+ const { offset, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
462
310
  try {
311
+ if (page < 0) {
312
+ throw new error.MastraError(
313
+ {
314
+ id: "STORAGE_LANCE_LIST_MESSAGES_INVALID_PAGE",
315
+ domain: error.ErrorDomain.STORAGE,
316
+ category: error.ErrorCategory.USER,
317
+ details: { page }
318
+ },
319
+ new Error("page must be >= 0")
320
+ );
321
+ }
322
+ const { field, direction } = this.parseOrderBy(orderBy, "ASC");
463
323
  const table = await this.client.openTable(storage.TABLE_MESSAGES);
464
- const quotedIds = messageIds.map((id) => `'${id}'`).join(", ");
465
- const allRecords = await table.query().where(`id IN (${quotedIds})`).toArray();
466
- const messages = processResultWithTypeConversion(
467
- allRecords,
468
- await getTableSchema({ tableName: storage.TABLE_MESSAGES, client: this.client })
469
- );
470
- const list = new agent.MessageList().add(messages.map(this.normalizeMessage), "memory");
471
- if (format === `v1`) return list.get.all.v1();
472
- return list.get.all.v2();
324
+ const conditions = [`thread_id = '${this.escapeSql(threadId)}'`];
325
+ if (resourceId) {
326
+ conditions.push(`\`resourceId\` = '${this.escapeSql(resourceId)}'`);
327
+ }
328
+ if (filter?.dateRange?.start) {
329
+ const startTime = filter.dateRange.start instanceof Date ? filter.dateRange.start.getTime() : new Date(filter.dateRange.start).getTime();
330
+ conditions.push(`\`createdAt\` >= ${startTime}`);
331
+ }
332
+ if (filter?.dateRange?.end) {
333
+ const endTime = filter.dateRange.end instanceof Date ? filter.dateRange.end.getTime() : new Date(filter.dateRange.end).getTime();
334
+ conditions.push(`\`createdAt\` <= ${endTime}`);
335
+ }
336
+ const whereClause = conditions.join(" AND ");
337
+ const total = await table.countRows(whereClause);
338
+ const query = table.query().where(whereClause);
339
+ let allRecords = await query.toArray();
340
+ allRecords.sort((a, b) => {
341
+ const aValue = field === "createdAt" ? a.createdAt : a[field];
342
+ const bValue = field === "createdAt" ? b.createdAt : b[field];
343
+ if (aValue == null && bValue == null) return 0;
344
+ if (aValue == null) return direction === "ASC" ? -1 : 1;
345
+ if (bValue == null) return direction === "ASC" ? 1 : -1;
346
+ if (typeof aValue === "string" && typeof bValue === "string") {
347
+ return direction === "ASC" ? aValue.localeCompare(bValue) : bValue.localeCompare(aValue);
348
+ }
349
+ return direction === "ASC" ? aValue - bValue : bValue - aValue;
350
+ });
351
+ const paginatedRecords = allRecords.slice(offset, offset + perPage);
352
+ const messages = paginatedRecords.map((row) => this.normalizeMessage(row));
353
+ if (total === 0 && messages.length === 0 && (!include || include.length === 0)) {
354
+ return {
355
+ messages: [],
356
+ total: 0,
357
+ page,
358
+ perPage: perPageForResponse,
359
+ hasMore: false
360
+ };
361
+ }
362
+ const messageIds = new Set(messages.map((m) => m.id));
363
+ if (include && include.length > 0) {
364
+ const threadIds = [...new Set(include.map((item) => item.threadId || threadId))];
365
+ const allThreadMessages = [];
366
+ for (const tid of threadIds) {
367
+ const threadQuery = table.query().where(`thread_id = '${tid}'`);
368
+ let threadRecords = await threadQuery.toArray();
369
+ allThreadMessages.push(...threadRecords);
370
+ }
371
+ allThreadMessages.sort((a, b) => a.createdAt - b.createdAt);
372
+ const contextMessages = this.processMessagesWithContext(allThreadMessages, include);
373
+ const includedMessages = contextMessages.map((row) => this.normalizeMessage(row));
374
+ for (const includeMsg of includedMessages) {
375
+ if (!messageIds.has(includeMsg.id)) {
376
+ messages.push(includeMsg);
377
+ messageIds.add(includeMsg.id);
378
+ }
379
+ }
380
+ }
381
+ const list = new agent.MessageList().add(messages, "memory");
382
+ let finalMessages = list.get.all.db();
383
+ finalMessages = finalMessages.sort((a, b) => {
384
+ const aValue = field === "createdAt" ? new Date(a.createdAt).getTime() : a[field];
385
+ const bValue = field === "createdAt" ? new Date(b.createdAt).getTime() : b[field];
386
+ if (aValue == null && bValue == null) return 0;
387
+ if (aValue == null) return direction === "ASC" ? -1 : 1;
388
+ if (bValue == null) return direction === "ASC" ? 1 : -1;
389
+ if (typeof aValue === "string" && typeof bValue === "string") {
390
+ return direction === "ASC" ? aValue.localeCompare(bValue) : bValue.localeCompare(aValue);
391
+ }
392
+ return direction === "ASC" ? aValue - bValue : bValue - aValue;
393
+ });
394
+ const returnedThreadMessageIds = new Set(finalMessages.filter((m) => m.threadId === threadId).map((m) => m.id));
395
+ const allThreadMessagesReturned = returnedThreadMessageIds.size >= total;
396
+ const fetchedAll = perPageInput === false || allThreadMessagesReturned;
397
+ const hasMore = !fetchedAll && offset + perPage < total;
398
+ return {
399
+ messages: finalMessages,
400
+ total,
401
+ page,
402
+ perPage: perPageForResponse,
403
+ hasMore
404
+ };
473
405
  } catch (error$1) {
474
- throw new error.MastraError(
406
+ const mastraError = new error.MastraError(
475
407
  {
476
- id: "LANCE_STORE_GET_MESSAGES_BY_ID_FAILED",
408
+ id: "LANCE_STORE_LIST_MESSAGES_FAILED",
477
409
  domain: error.ErrorDomain.STORAGE,
478
410
  category: error.ErrorCategory.THIRD_PARTY,
479
411
  details: {
480
- messageIds: JSON.stringify(messageIds)
412
+ threadId,
413
+ resourceId: resourceId ?? ""
481
414
  }
482
415
  },
483
416
  error$1
484
417
  );
418
+ this.logger?.error?.(mastraError.toString());
419
+ this.logger?.trackException?.(mastraError);
420
+ return {
421
+ messages: [],
422
+ total: 0,
423
+ page,
424
+ perPage: perPageForResponse,
425
+ hasMore: false
426
+ };
485
427
  }
486
428
  }
487
429
  async saveMessages(args) {
488
430
  try {
489
- const { messages, format = "v1" } = args;
431
+ const { messages } = args;
490
432
  if (messages.length === 0) {
491
- return [];
433
+ return { messages: [] };
492
434
  }
493
435
  const threadId = messages[0]?.threadId;
494
436
  if (!threadId) {
@@ -524,8 +466,7 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
524
466
  const updateRecord = { id: threadId, updatedAt: currentTime };
525
467
  await threadsTable.mergeInsert("id").whenMatchedUpdateAll().whenNotMatchedInsertAll().execute([updateRecord]);
526
468
  const list = new agent.MessageList().add(messages, "memory");
527
- if (format === `v2`) return list.get.all.v2();
528
- return list.get.all.v1();
469
+ return { messages: list.get.all.db() };
529
470
  } catch (error$1) {
530
471
  throw new error.MastraError(
531
472
  {
@@ -537,32 +478,54 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
537
478
  );
538
479
  }
539
480
  }
540
- async getThreadsByResourceIdPaginated(args) {
481
+ async listThreadsByResourceId(args) {
541
482
  try {
542
- const { resourceId, page = 0, perPage = 10 } = args;
543
- const table = await this.client.openTable(storage.TABLE_THREADS);
544
- const total = await table.countRows(`\`resourceId\` = '${resourceId}'`);
545
- const query = table.query().where(`\`resourceId\` = '${resourceId}'`);
546
- const offset = page * perPage;
547
- query.limit(perPage);
548
- if (offset > 0) {
549
- query.offset(offset);
483
+ const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
484
+ const perPage = storage.normalizePerPage(perPageInput, 100);
485
+ if (page < 0) {
486
+ throw new error.MastraError(
487
+ {
488
+ id: "STORAGE_LANCE_LIST_THREADS_BY_RESOURCE_ID_INVALID_PAGE",
489
+ domain: error.ErrorDomain.STORAGE,
490
+ category: error.ErrorCategory.USER,
491
+ details: { page }
492
+ },
493
+ new Error("page must be >= 0")
494
+ );
550
495
  }
496
+ const { offset, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
497
+ const { field, direction } = this.parseOrderBy(orderBy);
498
+ const table = await this.client.openTable(storage.TABLE_THREADS);
499
+ const total = await table.countRows(`\`resourceId\` = '${this.escapeSql(resourceId)}'`);
500
+ const query = table.query().where(`\`resourceId\` = '${this.escapeSql(resourceId)}'`);
551
501
  const records = await query.toArray();
552
- records.sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime());
502
+ records.sort((a, b) => {
503
+ const aValue = ["createdAt", "updatedAt"].includes(field) ? new Date(a[field]).getTime() : a[field];
504
+ const bValue = ["createdAt", "updatedAt"].includes(field) ? new Date(b[field]).getTime() : b[field];
505
+ if (aValue == null && bValue == null) return 0;
506
+ if (aValue == null) return direction === "ASC" ? -1 : 1;
507
+ if (bValue == null) return direction === "ASC" ? 1 : -1;
508
+ if (typeof aValue === "string" && typeof bValue === "string") {
509
+ return direction === "ASC" ? aValue.localeCompare(bValue) : bValue.localeCompare(aValue);
510
+ }
511
+ return direction === "ASC" ? aValue - bValue : bValue - aValue;
512
+ });
513
+ const paginatedRecords = records.slice(offset, offset + perPage);
553
514
  const schema = await getTableSchema({ tableName: storage.TABLE_THREADS, client: this.client });
554
- const threads = records.map((record) => processResultWithTypeConversion(record, schema));
515
+ const threads = paginatedRecords.map(
516
+ (record) => processResultWithTypeConversion(record, schema)
517
+ );
555
518
  return {
556
519
  threads,
557
520
  total,
558
521
  page,
559
- perPage,
560
- hasMore: total > (page + 1) * perPage
522
+ perPage: perPageForResponse,
523
+ hasMore: offset + perPage < total
561
524
  };
562
525
  } catch (error$1) {
563
526
  throw new error.MastraError(
564
527
  {
565
- id: "LANCE_STORE_GET_THREADS_BY_RESOURCE_ID_PAGINATED_FAILED",
528
+ id: "LANCE_STORE_LIST_THREADS_BY_RESOURCE_ID_FAILED",
566
529
  domain: error.ErrorDomain.STORAGE,
567
530
  category: error.ErrorCategory.THIRD_PARTY
568
531
  },
@@ -618,132 +581,8 @@ var StoreMemoryLance = class extends storage.MemoryStorage {
618
581
  });
619
582
  return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
620
583
  }
621
- async getMessagesPaginated(args) {
622
- const { threadId, resourceId, selectBy, format = "v1" } = args;
623
- const page = selectBy?.pagination?.page ?? 0;
624
- const perPage = selectBy?.pagination?.perPage ?? 10;
625
- try {
626
- if (!threadId.trim()) throw new Error("threadId must be a non-empty string");
627
- const dateRange = selectBy?.pagination?.dateRange;
628
- const fromDate = dateRange?.start;
629
- const toDate = dateRange?.end;
630
- const table = await this.client.openTable(storage.TABLE_MESSAGES);
631
- const messages = [];
632
- if (selectBy?.include && Array.isArray(selectBy.include)) {
633
- const threadIds = [...new Set(selectBy.include.map((item) => item.threadId))];
634
- const allThreadMessages = [];
635
- for (const threadId2 of threadIds) {
636
- const threadQuery = table.query().where(`thread_id = '${threadId2}'`);
637
- let threadRecords = await threadQuery.toArray();
638
- if (fromDate) threadRecords = threadRecords.filter((m) => m.createdAt >= fromDate.getTime());
639
- if (toDate) threadRecords = threadRecords.filter((m) => m.createdAt <= toDate.getTime());
640
- allThreadMessages.push(...threadRecords);
641
- }
642
- allThreadMessages.sort((a, b) => a.createdAt - b.createdAt);
643
- const contextMessages = this.processMessagesWithContext(allThreadMessages, selectBy.include);
644
- messages.push(...contextMessages);
645
- }
646
- const conditions = [`thread_id = '${threadId}'`];
647
- if (resourceId) {
648
- conditions.push(`\`resourceId\` = '${resourceId}'`);
649
- }
650
- if (fromDate) {
651
- conditions.push(`\`createdAt\` >= ${fromDate.getTime()}`);
652
- }
653
- if (toDate) {
654
- conditions.push(`\`createdAt\` <= ${toDate.getTime()}`);
655
- }
656
- let total = 0;
657
- if (conditions.length > 0) {
658
- total = await table.countRows(conditions.join(" AND "));
659
- } else {
660
- total = await table.countRows();
661
- }
662
- if (total === 0 && messages.length === 0) {
663
- return {
664
- messages: [],
665
- total: 0,
666
- page,
667
- perPage,
668
- hasMore: false
669
- };
670
- }
671
- const excludeIds = messages.map((m) => m.id);
672
- let selectedMessages = [];
673
- if (selectBy?.last && selectBy.last > 0) {
674
- const query = table.query();
675
- if (conditions.length > 0) {
676
- query.where(conditions.join(" AND "));
677
- }
678
- let records = await query.toArray();
679
- records = records.sort((a, b) => a.createdAt - b.createdAt);
680
- if (excludeIds.length > 0) {
681
- records = records.filter((m) => !excludeIds.includes(m.id));
682
- }
683
- selectedMessages = records.slice(-selectBy.last);
684
- } else {
685
- const query = table.query();
686
- if (conditions.length > 0) {
687
- query.where(conditions.join(" AND "));
688
- }
689
- let records = await query.toArray();
690
- records = records.sort((a, b) => a.createdAt - b.createdAt);
691
- if (excludeIds.length > 0) {
692
- records = records.filter((m) => !excludeIds.includes(m.id));
693
- }
694
- selectedMessages = records.slice(page * perPage, (page + 1) * perPage);
695
- }
696
- const allMessages = [...messages, ...selectedMessages];
697
- const seen = /* @__PURE__ */ new Set();
698
- const dedupedMessages = allMessages.filter((m) => {
699
- const key = `${m.id}:${m.thread_id}`;
700
- if (seen.has(key)) return false;
701
- seen.add(key);
702
- return true;
703
- });
704
- const formattedMessages = dedupedMessages.map((msg) => {
705
- const { thread_id, ...rest } = msg;
706
- return {
707
- ...rest,
708
- threadId: thread_id,
709
- content: typeof msg.content === "string" ? (() => {
710
- try {
711
- return JSON.parse(msg.content);
712
- } catch {
713
- return msg.content;
714
- }
715
- })() : msg.content
716
- };
717
- });
718
- const list = new agent.MessageList().add(formattedMessages, "memory");
719
- return {
720
- messages: format === "v2" ? list.get.all.v2() : list.get.all.v1(),
721
- total,
722
- // Total should be the count of messages matching the filters
723
- page,
724
- perPage,
725
- hasMore: total > (page + 1) * perPage
726
- };
727
- } catch (error$1) {
728
- const mastraError = new error.MastraError(
729
- {
730
- id: "LANCE_STORE_GET_MESSAGES_PAGINATED_FAILED",
731
- domain: error.ErrorDomain.STORAGE,
732
- category: error.ErrorCategory.THIRD_PARTY,
733
- details: {
734
- threadId,
735
- resourceId: resourceId ?? ""
736
- }
737
- },
738
- error$1
739
- );
740
- this.logger?.trackException?.(mastraError);
741
- this.logger?.error?.(mastraError.toString());
742
- return { messages: [], total: 0, page, perPage, hasMore: false };
743
- }
744
- }
745
584
  /**
746
- * Parse message data from LanceDB record format to MastraMessageV2 format
585
+ * Parse message data from LanceDB record format to MastraDBMessage format
747
586
  */
748
587
  parseMessageData(data) {
749
588
  const { thread_id, ...rest } = data;
@@ -1401,7 +1240,7 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1401
1240
  async saveScore(score) {
1402
1241
  let validatedScore;
1403
1242
  try {
1404
- validatedScore = scores.saveScorePayloadSchema.parse(score);
1243
+ validatedScore = evals.saveScorePayloadSchema.parse(score);
1405
1244
  } catch (error$1) {
1406
1245
  throw new error.MastraError(
1407
1246
  {
@@ -1466,7 +1305,7 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1466
1305
  );
1467
1306
  }
1468
1307
  }
1469
- async getScoresByScorerId({
1308
+ async listScoresByScorerId({
1470
1309
  scorerId,
1471
1310
  pagination,
1472
1311
  entityId,
@@ -1474,9 +1313,10 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1474
1313
  source
1475
1314
  }) {
1476
1315
  try {
1316
+ const { page, perPage: perPageInput } = pagination;
1317
+ const perPage = storage.normalizePerPage(perPageInput, 100);
1318
+ const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
1477
1319
  const table = await this.client.openTable(storage.TABLE_SCORERS);
1478
- const { page = 0, perPage = 10 } = pagination || {};
1479
- const offset = page * perPage;
1480
1320
  let query = table.query().where(`\`scorerId\` = '${scorerId}'`);
1481
1321
  if (source) {
1482
1322
  query = query.where(`\`source\` = '${source}'`);
@@ -1487,23 +1327,32 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1487
1327
  if (entityType) {
1488
1328
  query = query.where(`\`entityType\` = '${entityType}'`);
1489
1329
  }
1490
- query = query.limit(perPage);
1491
- if (offset > 0) query.offset(offset);
1492
- const records = await query.toArray();
1493
- const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
1494
- const scores = processResultWithTypeConversion(records, schema);
1495
1330
  let totalQuery = table.query().where(`\`scorerId\` = '${scorerId}'`);
1496
1331
  if (source) {
1497
1332
  totalQuery = totalQuery.where(`\`source\` = '${source}'`);
1498
1333
  }
1334
+ if (entityId) {
1335
+ totalQuery = totalQuery.where(`\`entityId\` = '${entityId}'`);
1336
+ }
1337
+ if (entityType) {
1338
+ totalQuery = totalQuery.where(`\`entityType\` = '${entityType}'`);
1339
+ }
1499
1340
  const allRecords = await totalQuery.toArray();
1500
1341
  const total = allRecords.length;
1342
+ const end = perPageInput === false ? total : start + perPage;
1343
+ if (perPageInput !== false) {
1344
+ query = query.limit(perPage);
1345
+ if (start > 0) query = query.offset(start);
1346
+ }
1347
+ const records = await query.toArray();
1348
+ const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
1349
+ const scores = processResultWithTypeConversion(records, schema);
1501
1350
  return {
1502
1351
  pagination: {
1503
1352
  page,
1504
- perPage,
1353
+ perPage: perPageForResponse,
1505
1354
  total,
1506
- hasMore: offset + scores.length < total
1355
+ hasMore: end < total
1507
1356
  },
1508
1357
  scores
1509
1358
  };
@@ -1520,27 +1369,32 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1520
1369
  );
1521
1370
  }
1522
1371
  }
1523
- async getScoresByRunId({
1372
+ async listScoresByRunId({
1524
1373
  runId,
1525
1374
  pagination
1526
1375
  }) {
1527
1376
  try {
1377
+ const { page, perPage: perPageInput } = pagination;
1378
+ const perPage = storage.normalizePerPage(perPageInput, 100);
1379
+ const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
1528
1380
  const table = await this.client.openTable(storage.TABLE_SCORERS);
1529
- const { page = 0, perPage = 10 } = pagination || {};
1530
- const offset = page * perPage;
1531
- const query = table.query().where(`\`runId\` = '${runId}'`).limit(perPage);
1532
- if (offset > 0) query.offset(offset);
1381
+ const allRecords = await table.query().where(`\`runId\` = '${runId}'`).toArray();
1382
+ const total = allRecords.length;
1383
+ const end = perPageInput === false ? total : start + perPage;
1384
+ let query = table.query().where(`\`runId\` = '${runId}'`);
1385
+ if (perPageInput !== false) {
1386
+ query = query.limit(perPage);
1387
+ if (start > 0) query = query.offset(start);
1388
+ }
1533
1389
  const records = await query.toArray();
1534
1390
  const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
1535
1391
  const scores = processResultWithTypeConversion(records, schema);
1536
- const allRecords = await table.query().where(`\`runId\` = '${runId}'`).toArray();
1537
- const total = allRecords.length;
1538
1392
  return {
1539
1393
  pagination: {
1540
1394
  page,
1541
- perPage,
1395
+ perPage: perPageForResponse,
1542
1396
  total,
1543
- hasMore: offset + scores.length < total
1397
+ hasMore: end < total
1544
1398
  },
1545
1399
  scores
1546
1400
  };
@@ -1557,28 +1411,33 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1557
1411
  );
1558
1412
  }
1559
1413
  }
1560
- async getScoresByEntityId({
1414
+ async listScoresByEntityId({
1561
1415
  entityId,
1562
1416
  entityType,
1563
1417
  pagination
1564
1418
  }) {
1565
1419
  try {
1420
+ const { page, perPage: perPageInput } = pagination;
1421
+ const perPage = storage.normalizePerPage(perPageInput, 100);
1422
+ const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
1566
1423
  const table = await this.client.openTable(storage.TABLE_SCORERS);
1567
- const { page = 0, perPage = 10 } = pagination || {};
1568
- const offset = page * perPage;
1569
- const query = table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).limit(perPage);
1570
- if (offset > 0) query.offset(offset);
1424
+ const allRecords = await table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).toArray();
1425
+ const total = allRecords.length;
1426
+ const end = perPageInput === false ? total : start + perPage;
1427
+ let query = table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`);
1428
+ if (perPageInput !== false) {
1429
+ query = query.limit(perPage);
1430
+ if (start > 0) query = query.offset(start);
1431
+ }
1571
1432
  const records = await query.toArray();
1572
1433
  const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
1573
1434
  const scores = processResultWithTypeConversion(records, schema);
1574
- const allRecords = await table.query().where(`\`entityId\` = '${entityId}' AND \`entityType\` = '${entityType}'`).toArray();
1575
- const total = allRecords.length;
1576
1435
  return {
1577
1436
  pagination: {
1578
1437
  page,
1579
- perPage,
1438
+ perPage: perPageForResponse,
1580
1439
  total,
1581
- hasMore: offset + scores.length < total
1440
+ hasMore: end < total
1582
1441
  },
1583
1442
  scores
1584
1443
  };
@@ -1595,28 +1454,33 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1595
1454
  );
1596
1455
  }
1597
1456
  }
1598
- async getScoresBySpan({
1457
+ async listScoresBySpan({
1599
1458
  traceId,
1600
1459
  spanId,
1601
1460
  pagination
1602
1461
  }) {
1603
1462
  try {
1463
+ const { page, perPage: perPageInput } = pagination;
1464
+ const perPage = storage.normalizePerPage(perPageInput, 100);
1465
+ const { offset: start, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
1604
1466
  const table = await this.client.openTable(storage.TABLE_SCORERS);
1605
- const { page = 0, perPage = 10 } = pagination || {};
1606
- const offset = page * perPage;
1607
- const query = table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).limit(perPage);
1608
- if (offset > 0) query.offset(offset);
1467
+ const allRecords = await table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).toArray();
1468
+ const total = allRecords.length;
1469
+ const end = perPageInput === false ? total : start + perPage;
1470
+ let query = table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`);
1471
+ if (perPageInput !== false) {
1472
+ query = query.limit(perPage);
1473
+ if (start > 0) query = query.offset(start);
1474
+ }
1609
1475
  const records = await query.toArray();
1610
1476
  const schema = await getTableSchema({ tableName: storage.TABLE_SCORERS, client: this.client });
1611
1477
  const scores = processResultWithTypeConversion(records, schema);
1612
- const allRecords = await table.query().where(`\`traceId\` = '${traceId}' AND \`spanId\` = '${spanId}'`).toArray();
1613
- const total = allRecords.length;
1614
1478
  return {
1615
1479
  pagination: {
1616
1480
  page,
1617
- perPage,
1481
+ perPage: perPageForResponse,
1618
1482
  total,
1619
- hasMore: offset + scores.length < total
1483
+ hasMore: end < total
1620
1484
  },
1621
1485
  scores
1622
1486
  };
@@ -1634,198 +1498,6 @@ var StoreScoresLance = class extends storage.ScoresStorage {
1634
1498
  }
1635
1499
  }
1636
1500
  };
1637
- var StoreTracesLance = class extends storage.TracesStorage {
1638
- client;
1639
- operations;
1640
- constructor({ client, operations }) {
1641
- super();
1642
- this.client = client;
1643
- this.operations = operations;
1644
- }
1645
- async saveTrace({ trace }) {
1646
- try {
1647
- const table = await this.client.openTable(storage.TABLE_TRACES);
1648
- const record = {
1649
- ...trace,
1650
- attributes: JSON.stringify(trace.attributes),
1651
- status: JSON.stringify(trace.status),
1652
- events: JSON.stringify(trace.events),
1653
- links: JSON.stringify(trace.links),
1654
- other: JSON.stringify(trace.other)
1655
- };
1656
- await table.add([record], { mode: "append" });
1657
- return trace;
1658
- } catch (error$1) {
1659
- throw new error.MastraError(
1660
- {
1661
- id: "LANCE_STORE_SAVE_TRACE_FAILED",
1662
- domain: error.ErrorDomain.STORAGE,
1663
- category: error.ErrorCategory.THIRD_PARTY
1664
- },
1665
- error$1
1666
- );
1667
- }
1668
- }
1669
- async getTraceById({ traceId }) {
1670
- try {
1671
- const table = await this.client.openTable(storage.TABLE_TRACES);
1672
- const query = table.query().where(`id = '${traceId}'`);
1673
- const records = await query.toArray();
1674
- return records[0];
1675
- } catch (error$1) {
1676
- throw new error.MastraError(
1677
- {
1678
- id: "LANCE_STORE_GET_TRACE_BY_ID_FAILED",
1679
- domain: error.ErrorDomain.STORAGE,
1680
- category: error.ErrorCategory.THIRD_PARTY
1681
- },
1682
- error$1
1683
- );
1684
- }
1685
- }
1686
- async getTraces({
1687
- name,
1688
- scope,
1689
- page = 1,
1690
- perPage = 10,
1691
- attributes
1692
- }) {
1693
- try {
1694
- const table = await this.client.openTable(storage.TABLE_TRACES);
1695
- const query = table.query();
1696
- if (name) {
1697
- query.where(`name = '${name}'`);
1698
- }
1699
- if (scope) {
1700
- query.where(`scope = '${scope}'`);
1701
- }
1702
- if (attributes) {
1703
- query.where(`attributes = '${JSON.stringify(attributes)}'`);
1704
- }
1705
- const offset = (page - 1) * perPage;
1706
- query.limit(perPage);
1707
- if (offset > 0) {
1708
- query.offset(offset);
1709
- }
1710
- const records = await query.toArray();
1711
- return records.map((record) => {
1712
- const processed = {
1713
- ...record,
1714
- attributes: record.attributes ? JSON.parse(record.attributes) : {},
1715
- status: record.status ? JSON.parse(record.status) : {},
1716
- events: record.events ? JSON.parse(record.events) : [],
1717
- links: record.links ? JSON.parse(record.links) : [],
1718
- other: record.other ? JSON.parse(record.other) : {},
1719
- startTime: new Date(record.startTime),
1720
- endTime: new Date(record.endTime),
1721
- createdAt: new Date(record.createdAt)
1722
- };
1723
- if (processed.parentSpanId === null || processed.parentSpanId === void 0) {
1724
- processed.parentSpanId = "";
1725
- } else {
1726
- processed.parentSpanId = String(processed.parentSpanId);
1727
- }
1728
- return processed;
1729
- });
1730
- } catch (error$1) {
1731
- throw new error.MastraError(
1732
- {
1733
- id: "LANCE_STORE_GET_TRACES_FAILED",
1734
- domain: error.ErrorDomain.STORAGE,
1735
- category: error.ErrorCategory.THIRD_PARTY,
1736
- details: { name: name ?? "", scope: scope ?? "" }
1737
- },
1738
- error$1
1739
- );
1740
- }
1741
- }
1742
- async getTracesPaginated(args) {
1743
- try {
1744
- const table = await this.client.openTable(storage.TABLE_TRACES);
1745
- const query = table.query();
1746
- const conditions = [];
1747
- if (args.name) {
1748
- conditions.push(`name = '${args.name}'`);
1749
- }
1750
- if (args.scope) {
1751
- conditions.push(`scope = '${args.scope}'`);
1752
- }
1753
- if (args.attributes) {
1754
- const attributesStr = JSON.stringify(args.attributes);
1755
- conditions.push(`attributes LIKE '%${attributesStr.replace(/"/g, '\\"')}%'`);
1756
- }
1757
- if (args.dateRange?.start) {
1758
- conditions.push(`\`createdAt\` >= ${args.dateRange.start.getTime()}`);
1759
- }
1760
- if (args.dateRange?.end) {
1761
- conditions.push(`\`createdAt\` <= ${args.dateRange.end.getTime()}`);
1762
- }
1763
- if (conditions.length > 0) {
1764
- const whereClause = conditions.join(" AND ");
1765
- query.where(whereClause);
1766
- }
1767
- let total = 0;
1768
- if (conditions.length > 0) {
1769
- const countQuery = table.query().where(conditions.join(" AND "));
1770
- const allRecords = await countQuery.toArray();
1771
- total = allRecords.length;
1772
- } else {
1773
- total = await table.countRows();
1774
- }
1775
- const page = args.page || 0;
1776
- const perPage = args.perPage || 10;
1777
- const offset = page * perPage;
1778
- query.limit(perPage);
1779
- if (offset > 0) {
1780
- query.offset(offset);
1781
- }
1782
- const records = await query.toArray();
1783
- const traces = records.map((record) => {
1784
- const processed = {
1785
- ...record,
1786
- attributes: record.attributes ? JSON.parse(record.attributes) : {},
1787
- status: record.status ? JSON.parse(record.status) : {},
1788
- events: record.events ? JSON.parse(record.events) : [],
1789
- links: record.links ? JSON.parse(record.links) : [],
1790
- other: record.other ? JSON.parse(record.other) : {},
1791
- startTime: new Date(record.startTime),
1792
- endTime: new Date(record.endTime),
1793
- createdAt: new Date(record.createdAt)
1794
- };
1795
- if (processed.parentSpanId === null || processed.parentSpanId === void 0) {
1796
- processed.parentSpanId = "";
1797
- } else {
1798
- processed.parentSpanId = String(processed.parentSpanId);
1799
- }
1800
- return processed;
1801
- });
1802
- return {
1803
- traces,
1804
- total,
1805
- page,
1806
- perPage,
1807
- hasMore: total > (page + 1) * perPage
1808
- };
1809
- } catch (error$1) {
1810
- throw new error.MastraError(
1811
- {
1812
- id: "LANCE_STORE_GET_TRACES_PAGINATED_FAILED",
1813
- domain: error.ErrorDomain.STORAGE,
1814
- category: error.ErrorCategory.THIRD_PARTY,
1815
- details: { name: args.name ?? "", scope: args.scope ?? "" }
1816
- },
1817
- error$1
1818
- );
1819
- }
1820
- }
1821
- async batchTraceInsert({ records }) {
1822
- this.logger.debug("Batch inserting traces", { count: records.length });
1823
- await this.operations.batchInsert({
1824
- tableName: storage.TABLE_TRACES,
1825
- records
1826
- });
1827
- }
1828
- };
1829
1501
  function parseWorkflowRun(row) {
1830
1502
  let parsedSnapshot = row.snapshot;
1831
1503
  if (typeof parsedSnapshot === "string") {
@@ -1855,7 +1527,7 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
1855
1527
  // runId,
1856
1528
  // stepId,
1857
1529
  // result,
1858
- // runtimeContext,
1530
+ // requestContext,
1859
1531
  }) {
1860
1532
  throw new Error("Method not implemented.");
1861
1533
  }
@@ -1949,7 +1621,7 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
1949
1621
  );
1950
1622
  }
1951
1623
  }
1952
- async getWorkflowRuns(args) {
1624
+ async listWorkflowRuns(args) {
1953
1625
  try {
1954
1626
  const table = await this.client.openTable(storage.TABLE_WORKFLOW_SNAPSHOT);
1955
1627
  let query = table.query();
@@ -1973,11 +1645,22 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
1973
1645
  } else {
1974
1646
  total = await table.countRows();
1975
1647
  }
1976
- if (args?.limit) {
1977
- query.limit(args.limit);
1978
- }
1979
- if (args?.offset) {
1980
- query.offset(args.offset);
1648
+ if (args?.perPage !== void 0 && args?.page !== void 0) {
1649
+ const normalizedPerPage = storage.normalizePerPage(args.perPage, Number.MAX_SAFE_INTEGER);
1650
+ if (args.page < 0 || !Number.isInteger(args.page)) {
1651
+ throw new error.MastraError(
1652
+ {
1653
+ id: "LANCE_STORE_INVALID_PAGINATION_PARAMS",
1654
+ domain: error.ErrorDomain.STORAGE,
1655
+ category: error.ErrorCategory.USER,
1656
+ details: { page: args.page, perPage: args.perPage }
1657
+ },
1658
+ new Error(`Invalid pagination parameters: page=${args.page}, perPage=${args.perPage}`)
1659
+ );
1660
+ }
1661
+ const offset = args.page * normalizedPerPage;
1662
+ query.limit(normalizedPerPage);
1663
+ query.offset(offset);
1981
1664
  }
1982
1665
  const records = await query.toArray();
1983
1666
  return {
@@ -1987,10 +1670,10 @@ var StoreWorkflowsLance = class extends storage.WorkflowsStorage {
1987
1670
  } catch (error$1) {
1988
1671
  throw new error.MastraError(
1989
1672
  {
1990
- id: "LANCE_STORE_GET_WORKFLOW_RUNS_FAILED",
1673
+ id: "LANCE_STORE_LIST_WORKFLOW_RUNS_FAILED",
1991
1674
  domain: error.ErrorDomain.STORAGE,
1992
1675
  category: error.ErrorCategory.THIRD_PARTY,
1993
- details: { namespace: args?.namespace ?? "", workflowName: args?.workflowName ?? "" }
1676
+ details: { resourceId: args?.resourceId ?? "", workflowName: args?.workflowName ?? "" }
1994
1677
  },
1995
1678
  error$1
1996
1679
  );
@@ -2004,6 +1687,8 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2004
1687
  lanceClient;
2005
1688
  /**
2006
1689
  * Creates a new instance of LanceStorage
1690
+ * @param id The unique identifier for this storage instance
1691
+ * @param name The name for this storage instance
2007
1692
  * @param uri The URI to connect to LanceDB
2008
1693
  * @param options connection options
2009
1694
  *
@@ -2011,31 +1696,29 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2011
1696
  *
2012
1697
  * Connect to a local database
2013
1698
  * ```ts
2014
- * const store = await LanceStorage.create('/path/to/db');
1699
+ * const store = await LanceStorage.create('my-storage-id', 'MyStorage', '/path/to/db');
2015
1700
  * ```
2016
1701
  *
2017
1702
  * Connect to a LanceDB cloud database
2018
1703
  * ```ts
2019
- * const store = await LanceStorage.create('db://host:port');
1704
+ * const store = await LanceStorage.create('my-storage-id', 'MyStorage', 'db://host:port');
2020
1705
  * ```
2021
1706
  *
2022
1707
  * Connect to a cloud database
2023
1708
  * ```ts
2024
- * const store = await LanceStorage.create('s3://bucket/db', { storageOptions: { timeout: '60s' } });
1709
+ * const store = await LanceStorage.create('my-storage-id', 'MyStorage', 's3://bucket/db', { storageOptions: { timeout: '60s' } });
2025
1710
  * ```
2026
1711
  */
2027
- static async create(name, uri, options) {
2028
- const instance = new _LanceStorage(name);
1712
+ static async create(id, name, uri, options) {
1713
+ const instance = new _LanceStorage(id, name);
2029
1714
  try {
2030
1715
  instance.lanceClient = await lancedb.connect(uri, options);
2031
1716
  const operations = new StoreOperationsLance({ client: instance.lanceClient });
2032
1717
  instance.stores = {
2033
1718
  operations: new StoreOperationsLance({ client: instance.lanceClient }),
2034
1719
  workflows: new StoreWorkflowsLance({ client: instance.lanceClient }),
2035
- traces: new StoreTracesLance({ client: instance.lanceClient, operations }),
2036
1720
  scores: new StoreScoresLance({ client: instance.lanceClient }),
2037
- memory: new StoreMemoryLance({ client: instance.lanceClient, operations }),
2038
- legacyEvals: new StoreLegacyEvalsLance({ client: instance.lanceClient })
1721
+ memory: new StoreMemoryLance({ client: instance.lanceClient, operations })
2039
1722
  };
2040
1723
  return instance;
2041
1724
  } catch (e) {
@@ -2055,15 +1738,13 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2055
1738
  * @internal
2056
1739
  * Private constructor to enforce using the create factory method
2057
1740
  */
2058
- constructor(name) {
2059
- super({ name });
1741
+ constructor(id, name) {
1742
+ super({ id, name });
2060
1743
  const operations = new StoreOperationsLance({ client: this.lanceClient });
2061
1744
  this.stores = {
2062
1745
  operations: new StoreOperationsLance({ client: this.lanceClient }),
2063
1746
  workflows: new StoreWorkflowsLance({ client: this.lanceClient }),
2064
- traces: new StoreTracesLance({ client: this.lanceClient, operations }),
2065
1747
  scores: new StoreScoresLance({ client: this.lanceClient }),
2066
- legacyEvals: new StoreLegacyEvalsLance({ client: this.lanceClient }),
2067
1748
  memory: new StoreMemoryLance({ client: this.lanceClient, operations })
2068
1749
  };
2069
1750
  }
@@ -2098,9 +1779,6 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2098
1779
  async getThreadById({ threadId }) {
2099
1780
  return this.stores.memory.getThreadById({ threadId });
2100
1781
  }
2101
- async getThreadsByResourceId({ resourceId }) {
2102
- return this.stores.memory.getThreadsByResourceId({ resourceId });
2103
- }
2104
1782
  /**
2105
1783
  * Saves a thread to the database. This function doesn't overwrite existing threads.
2106
1784
  * @param thread - The thread to save
@@ -2126,7 +1804,7 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2126
1804
  hasColumn: true,
2127
1805
  createTable: true,
2128
1806
  deleteMessages: false,
2129
- getScoresBySpan: true
1807
+ listScoresBySpan: true
2130
1808
  };
2131
1809
  }
2132
1810
  async getResourceById({ resourceId }) {
@@ -2190,50 +1868,17 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2190
1868
  });
2191
1869
  return Array.from(allIndices).sort((a, b) => a - b).map((index) => records[index]);
2192
1870
  }
2193
- async getMessages({
2194
- threadId,
2195
- resourceId,
2196
- selectBy,
2197
- format,
2198
- threadConfig
2199
- }) {
2200
- return this.stores.memory.getMessages({ threadId, resourceId, selectBy, format, threadConfig });
2201
- }
2202
- async getMessagesById({
2203
- messageIds,
2204
- format
2205
- }) {
2206
- return this.stores.memory.getMessagesById({ messageIds, format });
1871
+ async listMessagesById({ messageIds }) {
1872
+ return this.stores.memory.listMessagesById({ messageIds });
2207
1873
  }
2208
1874
  async saveMessages(args) {
2209
1875
  return this.stores.memory.saveMessages(args);
2210
1876
  }
2211
- async getThreadsByResourceIdPaginated(args) {
2212
- return this.stores.memory.getThreadsByResourceIdPaginated(args);
2213
- }
2214
- async getMessagesPaginated(args) {
2215
- return this.stores.memory.getMessagesPaginated(args);
2216
- }
2217
1877
  async updateMessages(_args) {
2218
1878
  return this.stores.memory.updateMessages(_args);
2219
1879
  }
2220
- async getTraceById(args) {
2221
- return this.stores.traces.getTraceById(args);
2222
- }
2223
- async getTraces(args) {
2224
- return this.stores.traces.getTraces(args);
2225
- }
2226
- async getTracesPaginated(args) {
2227
- return this.stores.traces.getTracesPaginated(args);
2228
- }
2229
- async getEvalsByAgentName(agentName, type) {
2230
- return this.stores.legacyEvals.getEvalsByAgentName(agentName, type);
2231
- }
2232
- async getEvals(options) {
2233
- return this.stores.legacyEvals.getEvals(options);
2234
- }
2235
- async getWorkflowRuns(args) {
2236
- return this.stores.workflows.getWorkflowRuns(args);
1880
+ async listWorkflowRuns(args) {
1881
+ return this.stores.workflows.listWorkflowRuns(args);
2237
1882
  }
2238
1883
  async getWorkflowRunById(args) {
2239
1884
  return this.stores.workflows.getWorkflowRunById(args);
@@ -2243,9 +1888,9 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2243
1888
  runId,
2244
1889
  stepId,
2245
1890
  result,
2246
- runtimeContext
1891
+ requestContext
2247
1892
  }) {
2248
- return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, runtimeContext });
1893
+ return this.stores.workflows.updateWorkflowResults({ workflowName, runId, stepId, result, requestContext });
2249
1894
  }
2250
1895
  async updateWorkflowState({
2251
1896
  workflowName,
@@ -2271,37 +1916,37 @@ var LanceStorage = class _LanceStorage extends storage.MastraStorage {
2271
1916
  async getScoreById({ id: _id }) {
2272
1917
  return this.stores.scores.getScoreById({ id: _id });
2273
1918
  }
2274
- async getScoresByScorerId({
1919
+ async listScoresByScorerId({
2275
1920
  scorerId,
2276
1921
  source,
2277
1922
  entityId,
2278
1923
  entityType,
2279
1924
  pagination
2280
1925
  }) {
2281
- return this.stores.scores.getScoresByScorerId({ scorerId, source, pagination, entityId, entityType });
1926
+ return this.stores.scores.listScoresByScorerId({ scorerId, source, pagination, entityId, entityType });
2282
1927
  }
2283
1928
  async saveScore(_score) {
2284
1929
  return this.stores.scores.saveScore(_score);
2285
1930
  }
2286
- async getScoresByRunId({
1931
+ async listScoresByRunId({
2287
1932
  runId,
2288
1933
  pagination
2289
1934
  }) {
2290
- return this.stores.scores.getScoresByRunId({ runId, pagination });
1935
+ return this.stores.scores.listScoresByRunId({ runId, pagination });
2291
1936
  }
2292
- async getScoresByEntityId({
1937
+ async listScoresByEntityId({
2293
1938
  entityId,
2294
1939
  entityType,
2295
1940
  pagination
2296
1941
  }) {
2297
- return this.stores.scores.getScoresByEntityId({ entityId, entityType, pagination });
1942
+ return this.stores.scores.listScoresByEntityId({ entityId, entityType, pagination });
2298
1943
  }
2299
- async getScoresBySpan({
1944
+ async listScoresBySpan({
2300
1945
  traceId,
2301
1946
  spanId,
2302
1947
  pagination
2303
1948
  }) {
2304
- return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
1949
+ return this.stores.scores.listScoresBySpan({ traceId, spanId, pagination });
2305
1950
  }
2306
1951
  };
2307
1952
  var LanceFilterTranslator = class extends filter.BaseFilterTranslator {
@@ -2650,7 +2295,7 @@ var LanceVectorStore = class _LanceVectorStore extends vector.MastraVector {
2650
2295
  * ```
2651
2296
  */
2652
2297
  static async create(uri, options) {
2653
- const instance = new _LanceVectorStore();
2298
+ const instance = new _LanceVectorStore(options?.id || crypto.randomUUID());
2654
2299
  try {
2655
2300
  instance.lanceClient = await lancedb.connect(uri, options);
2656
2301
  return instance;
@@ -2670,8 +2315,8 @@ var LanceVectorStore = class _LanceVectorStore extends vector.MastraVector {
2670
2315
  * @internal
2671
2316
  * Private constructor to enforce using the create factory method
2672
2317
  */
2673
- constructor() {
2674
- super();
2318
+ constructor(id) {
2319
+ super({ id });
2675
2320
  }
2676
2321
  close() {
2677
2322
  if (this.lanceClient) {