William Jeynes
60 lines
2.5 KiB
TypeScript
60 lines
2.5 KiB
TypeScript
import { END, START, StateGraph } from "@langchain/langgraph";
|
|
import { MessagesState } from "./state";
|
|
import { createToolNode } from "./nodes/tool";
|
|
import { createToolConditional } from "./conditionals/tool_end";
|
|
import { normalizationSetup } from "./nodes/normalizationSetup";
|
|
import { arithmeticToolsByName } from "./tools/arithmetic"
|
|
import { createDummyModelNode } from "./nodes/dummyModel";
|
|
import { verificationSetup } from "./nodes/verificationSetup";
|
|
import { dummyRagasMetrics } from "./nodes/dummyRagasMetrics";
|
|
import { produceRanking } from "./nodes/produceRanking";
|
|
import { createModelNode } from "./nodes/model";
|
|
|
|
const triggerEventToolNode = createToolNode(arithmeticToolsByName);
|
|
const verificationToolNode = createToolNode(arithmeticToolsByName);
|
|
|
|
const dummyTriggerEventModel = createDummyModelNode("Trigger Events of");
|
|
const dummyVerificationModel = createDummyModelNode("verification of");
|
|
|
|
const normalisationModel = createModelNode([], "normalization.txt");
|
|
|
|
const triggerEventToolConditional = createToolConditional("triggerEventToolNode", verificationSetup.name);
|
|
const verificationToolConditional = createToolConditional("verificationToolNode", produceRanking.name);
|
|
|
|
|
|
const agent = new StateGraph(MessagesState)
|
|
|
|
//NODES
|
|
|
|
.addNode(normalizationSetup.name, normalizationSetup)
|
|
.addNode("normalisationModel", normalisationModel)
|
|
|
|
.addNode("triggerEventToolNode", triggerEventToolNode)
|
|
.addNode("dummyTriggerEventModel", dummyTriggerEventModel)
|
|
|
|
.addNode(verificationSetup.name, verificationSetup)
|
|
.addNode("dummyVerificationModel", dummyVerificationModel)
|
|
.addNode(dummyRagasMetrics.name, dummyRagasMetrics)
|
|
.addNode("verificationToolNode", verificationToolNode)
|
|
.addNode(produceRanking.name, produceRanking)
|
|
|
|
.addEdge(START, normalizationSetup.name)
|
|
.addEdge(normalizationSetup.name, "normalisationModel")
|
|
.addEdge("normalisationModel", "dummyTriggerEventModel")
|
|
|
|
// @ts-expect-error
|
|
.addConditionalEdges("dummyTriggerEventModel", triggerEventToolConditional, ["triggerEventToolNode", verificationSetup.name])
|
|
.addEdge("triggerEventToolNode", "dummyTriggerEventModel")
|
|
|
|
.addEdge(verificationSetup.name, "dummyVerificationModel")
|
|
.addEdge(verificationSetup.name, dummyRagasMetrics.name)
|
|
|
|
// @ts-expect-error
|
|
.addConditionalEdges("dummyVerificationModel", verificationToolConditional, ["verificationToolNode", produceRanking.name])
|
|
.addEdge("verificationToolNode", "dummyVerificationModel")
|
|
|
|
.addEdge(dummyRagasMetrics.name, produceRanking.name)
|
|
|
|
.compile();
|
|
|
|
export {agent} |