@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
JavaScript
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,
};
});
};