@mastra/pg 1.8.5 → 1.8.6

package/dist/index.js

@@ -4333,6 +4333,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
     await this.#addColumnIfNotExists(TABLE_DATASETS, "scorerIds", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "requestContext", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "JSONB");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "expectedTrajectory", "JSONB");
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -4415,6 +4416,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       datasetVersion: row.datasetVersion,
       input: safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
+      expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
       source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -4431,6 +4433,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       isDeleted: Boolean(row.isDeleted),
       input: safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
+      expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
       source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -4700,13 +4703,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         );
         newVersion = row.version;
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
           [
             id,
             args.datasetId,
             newVersion,
             JSON.stringify(args.input),
             jsonbArg(args.groundTruth),
+            jsonbArg(args.expectedTrajectory),
             jsonbArg(args.requestContext),
             jsonbArg(args.metadata),
             jsonbArg(args.source),
@@ -4727,6 +4731,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         datasetVersion: newVersion,
         input: args.input,
         groundTruth: args.groundTruth,
+        expectedTrajectory: args.expectedTrajectory,
         requestContext: args.requestContext,
         metadata: args.metadata,
         source: args.source,
@@ -4773,11 +4778,12 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const versionId = crypto.randomUUID();
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
-      const mergedInput = args.input ?? existing.input;
-      const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
-      const mergedRequestContext = args.requestContext ?? existing.requestContext;
-      const mergedMetadata = args.metadata ?? existing.metadata;
-      const mergedSource = args.source ?? existing.source;
+      const mergedInput = args.input !== void 0 ? args.input : existing.input;
+      const mergedGroundTruth = args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth;
+      const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
+      const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
+      const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
+      const mergedSource = args.source !== void 0 ? args.source : existing.source;
       let newVersion;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
@@ -4790,13 +4796,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
           [newVersion, args.id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
           [
             args.id,
             args.datasetId,
             newVersion,
             JSON.stringify(mergedInput),
             jsonbArg(mergedGroundTruth),
+            jsonbArg(mergedExpectedTrajectory),
             jsonbArg(mergedRequestContext),
             jsonbArg(mergedMetadata),
             jsonbArg(mergedSource),
@@ -4816,6 +4823,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         datasetVersion: newVersion,
         input: mergedInput,
         groundTruth: mergedGroundTruth,
+        expectedTrajectory: mergedExpectedTrajectory,
         requestContext: mergedRequestContext,
         metadata: mergedMetadata,
         source: mergedSource,
@@ -4864,13 +4872,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
           [newVersion, id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
           [
             id,
             datasetId,
             newVersion,
             JSON.stringify(existing.input),
             jsonbArg(existing.groundTruth),
+            jsonbArg(existing.expectedTrajectory),
             jsonbArg(existing.requestContext),
             jsonbArg(existing.metadata),
             jsonbArg(existing.source),
@@ -4927,13 +4936,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         newVersion = row.version;
         for (const { id, input: itemInput } of itemsWithIds) {
           await t.none(
-            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
             [
               id,
               input.datasetId,
               newVersion,
               JSON.stringify(itemInput.input),
               jsonbArg(itemInput.groundTruth),
+              jsonbArg(itemInput.expectedTrajectory),
               jsonbArg(itemInput.requestContext),
               jsonbArg(itemInput.metadata),
               jsonbArg(itemInput.source),
@@ -4955,6 +4965,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         datasetVersion: newVersion,
         input: itemInput.input,
         groundTruth: itemInput.groundTruth,
+        expectedTrajectory: itemInput.expectedTrajectory,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
         source: itemInput.source,
@@ -5012,13 +5023,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
             [newVersion, item.id]
           );
           await t.none(
-            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
             [
               item.id,
               input.datasetId,
               newVersion,
               JSON.stringify(item.input),
               jsonbArg(item.groundTruth),
+              jsonbArg(item.expectedTrajectory),
               jsonbArg(item.requestContext),
               jsonbArg(item.metadata),
               jsonbArg(item.source),
@@ -9034,7 +9046,8 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
         lastObservedAt: input.chunk.lastObservedAt,
         createdAt: /* @__PURE__ */ new Date(),
         suggestedContinuation: input.chunk.suggestedContinuation,
-        currentTask: input.chunk.currentTask
+        currentTask: input.chunk.currentTask,
+        threadTitle: input.chunk.threadTitle
       };
       const lastBufferedAtTime = input.lastBufferedAtTime ? input.lastBufferedAtTime.toISOString() : null;
       const result = await this.#db.client.query(