jill/LLMsForDisinformationAnalysis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add self improvement pattern with two new prompt nodes

Browse Source
This commit is contained in:
William Jeynes
2026-03-26 14:44:48 +00:00
parent 5e374a8bd6
commit a80d433fb6
7 changed files with 123 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agent/agent.ts
+20 -2
View File
@@ -11,13 +11,18 @@ import { loopEndConditional } from "./conditionals/loop_end";
import { sort } from "./nodes/sort";
import { triggerEventSetup } from "./nodes/triggerEventSetup";
import { createEnsembleNode } from "./nodes/ensembleNode";
import { selfEvalSetup } from "./nodes/selfEvalSetup";
const triggerEventToolNode = createToolNode(triggerEventToolsByName);
const peToolNode = createToolNode(triggerEventToolsByName);
const normalisationModel = createModelNode([], "normalization.txt");
const triggerEventModel = createModelNode(triggerEventToolsByName, "trigger.txt");
const evaluationModel = createModelNode([], "eval.txt");
const peModel = createModelNode(triggerEventToolsByName, "posteval.txt");
const triggerEventToolConditional = createToolConditional("triggerEventToolNode", verificationSetup.name);
const triggerEventToolConditional = createToolConditional("triggerEventToolNode", selfEvalSetup.name);
const peToolConditional = createToolConditional("peToolNode", verificationSetup.name);
const roNode = createEnsembleNode("ROBERTA", "roberta");
const flNode = createEnsembleNode("FLAN", "flan");
@@ -33,6 +38,12 @@ const agent = new StateGraph(MessagesState)
.addNode("triggerEventToolNode", triggerEventToolNode)
.addNode("triggerEventModel", triggerEventModel)
.addNode(selfEvalSetup.name, selfEvalSetup)
.addNode("evaluationModel", evaluationModel)
.addNode("peToolNode", peToolNode)
.addNode("peModel", peModel)
.addNode(verificationSetup.name, verificationSetup)
.addNode("roNode", roNode)
@@ -49,9 +60,16 @@ const agent = new StateGraph(MessagesState)
.addEdge(triggerEventSetup.name, "triggerEventModel")
// @ts-expect-error
.addConditionalEdges("triggerEventModel", triggerEventToolConditional, ["triggerEventToolNode", verificationSetup.name])
.addConditionalEdges("triggerEventModel", triggerEventToolConditional, ["triggerEventToolNode", selfEvalSetup.name])
.addEdge("triggerEventToolNode", "triggerEventModel")
.addEdge(selfEvalSetup.name, "evaluationModel")
.addEdge("evaluationModel", "peModel")
// @ts-expect-error
.addConditionalEdges("peModel", peToolConditional, ["peToolNode", verificationSetup.name])
.addEdge("peToolNode", "peModel")
.addEdge(verificationSetup.name, "roNode")
.addEdge(verificationSetup.name, "flNode")
.addEdge(verificationSetup.name, "lrNode")