allow promptMessageId on saveMessage

Author: ianmacartney

CHANGELOG.md

@@ -13,6 +13,7 @@
 - Adds a /test entrypoint to make adding tests with the agent easier.
 - Simplifies the types for generating object & text to more closely match the AI
   SDK.
+- Allows passing a `promptMessageId` to `saveMessage` to match `saveMessages`.
 
 ## 0.2.10
 

src/client/index.ts

@@ -798,6 +798,7 @@ export class Agent<
           : [args.message],
       metadata: args.metadata ? [args.metadata] : undefined,
       skipEmbeddings: args.skipEmbeddings,
+      promptMessageId: args.promptMessageId,
       pendingMessageId: args.pendingMessageId,
     });
     const message = messages.at(-1)!;

src/client/messages.ts

@@ -160,6 +160,11 @@ export async function saveMessages(
 export type SaveMessageArgs = {
   threadId: string;
   userId?: string | null;
+  /**
+   * The message that these messages are in response to. They will be
+   * the same "order" as this message, at increasing stepOrder(s).
+   */
+  promptMessageId?: string;
   /**
    * Metadata to save with the messages. Each element corresponds to the
    * message at the same index.
@@ -218,6 +223,7 @@ export async function saveMessage(
     threadId: args.threadId,
     userId: args.userId ?? undefined,
     agentName: args.agentName,
+    promptMessageId: args.promptMessageId,
     pendingMessageId: args.pendingMessageId,
     messages:
       args.prompt !== undefined