Revert a bit

langchain-ai · Dec 18, 2024 · c4fa105 · c4fa105
1 parent 0b4ed0f
commit c4fa105
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 125 deletions.
diff --git a/langchain-core/src/messages/developer.ts b/langchain-core/src/messages/developer.ts
diff --git a/langchain-core/src/messages/index.ts b/langchain-core/src/messages/index.ts
@@ -4,7 +4,6 @@ export * from "./chat.js";
 export * from "./function.js";
 export * from "./human.js";
 export * from "./system.js";
-export * from "./developer.js";
 export * from "./utils.js";
 export * from "./transformers.js";
 export * from "./modifier.js";

diff --git a/langchain-core/src/messages/system.ts b/langchain-core/src/messages/system.ts
@@ -57,3 +57,14 @@ export function isSystemMessageChunk(
 ): x is SystemMessageChunk {
   return x._getType() === "system";
 }
+
+/**
+ * Represents a developer message in a conversation.
+ */
+export class DeveloperMessage extends SystemMessage {}
+
+/**
+ * Represents a chunk of a developer message, which can be concatenated with
+ * other developer message chunks.
+ */
+export class DeveloperMessageChunk extends SystemMessageChunk {}
diff --git a/langchain-core/src/messages/transformers.ts b/langchain-core/src/messages/transformers.ts
@@ -13,7 +13,6 @@ import {
   ChatMessageChunk,
   ChatMessageFieldsWithRole,
 } from "./chat.js";
-import { DeveloperMessage, DeveloperMessageChunk } from "./developer.js";
 import {
   FunctionMessage,
   FunctionMessageChunk,
@@ -908,8 +907,8 @@ const _MSG_CHUNK_MAP: Record<
     messageChunk: SystemMessageChunk,
   },
   developer: {
-    message: DeveloperMessage,
-    messageChunk: DeveloperMessageChunk,
+    message: SystemMessage,
+    messageChunk: SystemMessageChunk,
   },
   tool: {
     message: ToolMessage,
@@ -984,9 +983,9 @@ function _switchTypeToMessage(
       break;
     case "developer":
       if (returnChunk) {
-        chunk = new DeveloperMessageChunk(fields);
+        chunk = new SystemMessageChunk(fields);
       } else {
-        msg = new DeveloperMessage(fields);
+        msg = new SystemMessage(fields);
       }
       break;
     case "tool":

diff --git a/langchain-core/src/messages/utils.ts b/langchain-core/src/messages/utils.ts
@@ -16,7 +16,6 @@ import {
   ChatMessageFieldsWithRole,
   ChatMessageChunk,
 } from "./chat.js";
-import { DeveloperMessage, DeveloperMessageChunk } from "./developer.js";
 import {
   FunctionMessage,
   FunctionMessageFieldsWithName,
@@ -88,11 +87,6 @@ function _constructMessageFromParams(
       className === "SystemMessageChunk"
     ) {
       type = "system";
-    } else if (
-      className === "DeveloperMessage" ||
-      className === "DeveloperMessageChunk"
-    ) {
-      type = "developer";
     } else {
       type = "unknown";
     }
@@ -114,7 +108,7 @@ function _constructMessageFromParams(
   } else if (type === "system") {
     return new SystemMessage(rest);
   } else if (type === "developer") {
-    return new DeveloperMessage(rest);
+    return new SystemMessage(rest);
   } else if (type === "tool" && "tool_call_id" in rest) {
     return new ToolMessage({
       ...rest,
@@ -125,7 +119,7 @@ function _constructMessageFromParams(
   } else {
     const error = addLangChainErrorFields(
       new Error(
-        `Unable to coerce message from array: only human, AI, system, or tool message coercion is currently supported.\n\nReceived: ${JSON.stringify(
+        `Unable to coerce message from array: only human, AI, system, developer, or tool message coercion is currently supported.\n\nReceived: ${JSON.stringify(
           params,
           null,
           2
@@ -174,8 +168,6 @@ export function getBufferString(
       role = aiPrefix;
     } else if (m._getType() === "system") {
       role = "System";
-    } else if (m._getType() === "developer") {
-      role = "Developer";
     } else if (m._getType() === "function") {
       role = "Function";
     } else if (m._getType() === "tool") {
@@ -231,8 +223,6 @@ export function mapStoredMessageToChatMessage(message: StoredMessage) {
       return new AIMessage(storedMessage.data);
     case "system":
       return new SystemMessage(storedMessage.data);
-    case "developer":
-      return new DeveloperMessage(storedMessage.data);
     case "function":
       if (storedMessage.data.name === undefined) {
         throw new Error("Name must be defined for function messages");
@@ -309,9 +299,6 @@ export function convertToChunk(message: BaseMessage) {
   } else if (type === "system") {
     // eslint-disable-next-line @typescript-eslint/no-use-before-define
     return new SystemMessageChunk({ ...message });
-  } else if (type === "developer") {
-    // eslint-disable-next-line @typescript-eslint/no-use-before-define
-    return new DeveloperMessageChunk({ ...message });
   } else if (type === "function") {
     // eslint-disable-next-line @typescript-eslint/no-use-before-define
     return new FunctionMessageChunk({ ...message });

diff --git a/langchain-core/src/prompts/chat.ts b/langchain-core/src/prompts/chat.ts
@@ -13,7 +13,6 @@ import {
   isBaseMessage,
   MessageContent,
   MessageContentComplex,
-  DeveloperMessage,
 } from "../messages/index.js";
 import {
   type ChatPromptValueInterface,
@@ -705,33 +704,6 @@ export class SystemMessagePromptTemplate<
   }
 }
 
-/**
- * Class that represents a developer message prompt template. It extends the
- * BaseMessageStringPromptTemplate.
- * @example
- * ```typescript
- * const message = DeveloperMessagePromptTemplate.fromTemplate("{text}");
- * const formatted = await message.format({ text: "Hello world!" });
- *
- * const chatPrompt = ChatPromptTemplate.fromMessages([message]);
- * const formattedChatPrompt = await chatPrompt.invoke({
- *   text: "Hello world!",
- * });
- * ```
- */
-export class DeveloperMessagePromptTemplate<
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  RunInput extends InputValues = any
-> extends _StringImageMessagePromptTemplate<RunInput> {
-  static _messageClass(): typeof SystemMessage {
-    return DeveloperMessage;
-  }
-
-  static lc_name() {
-    return "DeveloperMessagePromptTemplate";
-  }
-}
-
 /**
  * Interface for the input of a ChatPromptTemplate.
  */
@@ -842,8 +814,6 @@ function _coerceMessagePromptTemplateLike<
     return AIMessagePromptTemplate.fromTemplate(templateData, extra);
   } else if (message._getType() === "system") {
     return SystemMessagePromptTemplate.fromTemplate(templateData, extra);
-  } else if (message._getType() === "developer") {
-    return DeveloperMessagePromptTemplate.fromTemplate(templateData, extra);
   } else if (ChatMessage.isInstance(message)) {
     return ChatMessagePromptTemplate.fromTemplate(
       message.content as string,

diff --git a/langchain-core/src/runnables/remote.ts b/langchain-core/src/runnables/remote.ts
@@ -14,8 +14,6 @@ import {
   AIMessageChunk,
   ChatMessage,
   ChatMessageChunk,
-  DeveloperMessage,
-  DeveloperMessageChunk,
   FunctionMessage,
   FunctionMessageChunk,
   HumanMessage,
@@ -72,11 +70,6 @@ function revive(obj: any): any {
           content: obj.content,
         });
       }
-      if (obj.type === "DeveloperMessage" || obj.type === "developer") {
-        return new DeveloperMessage({
-          content: obj.content,
-        });
-      }
       if (obj.type === "ChatMessage" || obj.type === "generic") {
         return new ChatMessage({
           content: obj.content,
@@ -111,11 +104,6 @@ function revive(obj: any): any {
           content: obj.content,
         });
       }
-      if (obj.type === "DeveloperMessageChunk") {
-        return new DeveloperMessageChunk({
-          content: obj.content,
-        });
-      }
       if (obj.type === "ChatMessageChunk") {
         return new ChatMessageChunk({
           content: obj.content,