From df368bdd3046f00d96ce8f7dabf3e7f9aee51ede Mon Sep 17 00:00:00 2001
From: Josh Rogers <josh@langchain.dev>
Date: Tue, 1 Jul 2025 18:40:13 -0400
Subject: [PATCH] Updating message types to include all base message fields
 (#5298)

---
 libs/sdk-js/src/types.messages.ts | 39 +++++++++++++------------------
 1 file changed, 16 insertions(+), 23 deletions(-)

diff --git a/libs/sdk-js/src/types.messages.ts b/libs/sdk-js/src/types.messages.ts
index 49123cf1a..26b929535 100644
--- a/libs/sdk-js/src/types.messages.ts
+++ b/libs/sdk-js/src/types.messages.ts
@@ -13,16 +13,22 @@ type MessageContent = string | MessageContentComplex[];
  */
 type MessageAdditionalKwargs = Record<string, unknown>;
 
-export type HumanMessage = {
-  type: "human";
-  id?: string | undefined;
+type BaseMessage = {
+  additional_kwargs?: MessageAdditionalKwargs | undefined;
   content: MessageContent;
+  id?: string | undefined;
+  name?: string | undefined;
+  response_metadata?: Record<string, unknown> | undefined;
 };
 
-export type AIMessage = {
+export type HumanMessage = BaseMessage & {
+  type: "human";
+  example?: boolean | undefined;
+};
+
+export type AIMessage = BaseMessage & {
   type: "ai";
-  id?: string | undefined;
-  content: MessageContent;
+  example?: boolean | undefined;
   tool_calls?:
     | {
         name: string;
@@ -57,19 +63,12 @@ export type AIMessage = {
           | undefined;
       }
     | undefined;
-  additional_kwargs?: MessageAdditionalKwargs | undefined;
-  response_metadata?: Record<string, unknown> | undefined;
 };
 
-export type ToolMessage = {
+export type ToolMessage = BaseMessage & {
   type: "tool";
-  name?: string | undefined;
-  id?: string | undefined;
-  content: MessageContent;
   status?: "error" | "success" | undefined;
   tool_call_id: string;
-  additional_kwargs?: MessageAdditionalKwargs | undefined;
-  response_metadata?: Record<string, unknown> | undefined;
   /**
    * Artifact of the Tool execution which is not meant to be sent to the model.
    *
@@ -81,22 +80,16 @@ export type ToolMessage = {
   artifact?: any;
 };
 
-export type SystemMessage = {
+export type SystemMessage = BaseMessage & {
   type: "system";
-  id?: string | undefined;
-  content: MessageContent;
 };
 
-export type FunctionMessage = {
+export type FunctionMessage = BaseMessage & {
   type: "function";
-  id?: string | undefined;
-  content: MessageContent;
 };
 
-export type RemoveMessage = {
+export type RemoveMessage = BaseMessage & {
   type: "remove";
-  id: string;
-  content: MessageContent;
 };
 
 export type Message =