William Jeynes
32 lines
939 B
TypeScript
32 lines
939 B
TypeScript
import { ConditionalEdgeRouter, END } from "@langchain/langgraph";
|
|
import { MessagesState } from "../state";
|
|
import { AIMessage } from "@langchain/core/messages";
|
|
|
|
export function createToolConditional(a: String, b: String): ConditionalEdgeRouter<typeof MessagesState, String> {
|
|
// @ts-expect-error
|
|
var genericToolConditional: ConditionalEdgeRouter<typeof MessagesState, String> = (state) => {
|
|
const lastMessage = state.messages.at(-1);
|
|
|
|
//STARTTEMP
|
|
if (lastMessage?.content?.toString().indexOf("qwe") != -1) {
|
|
return a
|
|
}
|
|
//ENDTEMP
|
|
|
|
// Check if it's an AIMessage before accessing tool_calls
|
|
if (!lastMessage || !AIMessage.isInstance(lastMessage)) {
|
|
return b;
|
|
}
|
|
|
|
// If the LLM makes a tool call, then perform an action
|
|
if (lastMessage.tool_calls?.length) {
|
|
return a;
|
|
}
|
|
|
|
// Otherwise, we stop (reply to the user)
|
|
return b;
|
|
};
|
|
return genericToolConditional
|
|
}
|
|
|