utils.d.ts 1.9 KB

12345678910111213141516171819202122232425262728293031
  1. import { AIMessage, AIMessageChunk } from "./ai.js";
  2. import { BaseMessageLike, BaseMessage, StoredMessage } from "./base.js";
  3. import { ChatMessage, ChatMessageChunk } from "./chat.js";
  4. import { FunctionMessage, FunctionMessageChunk } from "./function.js";
  5. import { HumanMessage, HumanMessageChunk } from "./human.js";
  6. import { SystemMessage, SystemMessageChunk } from "./system.js";
  7. import { ToolMessage } from "./tool.js";
  8. export declare function coerceMessageLikeToMessage(messageLike: BaseMessageLike): BaseMessage;
  9. /**
  10. * This function is used by memory classes to get a string representation
  11. * of the chat message history, based on the message content and role.
  12. */
  13. export declare function getBufferString(messages: BaseMessage[], humanPrefix?: string, aiPrefix?: string): string;
  14. export declare function mapStoredMessageToChatMessage(message: StoredMessage): ToolMessage | AIMessage | ChatMessage | FunctionMessage | HumanMessage | SystemMessage;
  15. /**
  16. * Transforms an array of `StoredMessage` instances into an array of
  17. * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
  18. * function to ensure all messages are in the `StoredMessage` format, then
  19. * creates new instances of the appropriate `BaseMessage` subclass based
  20. * on the type of each message. This function is used to prepare stored
  21. * messages for use in a chat context.
  22. */
  23. export declare function mapStoredMessagesToChatMessages(messages: StoredMessage[]): BaseMessage[];
  24. /**
  25. * Transforms an array of `BaseMessage` instances into an array of
  26. * `StoredMessage` instances. It does this by calling the `toDict` method
  27. * on each `BaseMessage`, which returns a `StoredMessage`. This function
  28. * is used to prepare chat messages for storage.
  29. */
  30. export declare function mapChatMessagesToStoredMessages(messages: BaseMessage[]): StoredMessage[];
  31. export declare function convertToChunk(message: BaseMessage): AIMessageChunk | ChatMessageChunk | FunctionMessageChunk | HumanMessageChunk | SystemMessageChunk;