UNPKG

@langgraph-js/pro

Version:

The Pro SDK for LangGraph - seamlessly integrate your AI agents with frontend interfaces and build complex AI workflows

30 lines (29 loc) 944 B
import { SystemMessage, HumanMessage, AIMessageChunk, ToolMessage } from "@langchain/core/messages"; export const createChatTemplate = (template) => { return template.map((i) => { switch (i.role) { case "system": /** @ts-ignore */ return new SystemMessage(i.content); case "human": return new HumanMessage(i); case "ai": return new AIMessageChunk(i); case "tool": return new ToolMessage(i); default: return new HumanMessage(i.content); } }); }; export const messagesToTemplate = (messages) => { return messages.map((i) => { return { role: i.getType(), content: i.content, tool_calls: i.tool_calls, tool_call_id: i.tool_call_id, invalid_tool_calls: i.invalid_tool_calls, }; }); };