start adding dummy nodes
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user