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

Allow multiple source CSV files for normalisation. Implement real model node. Add normalizarion prompt. Implement normalization setup. Start on RAG retreival functions

Browse Source
This commit is contained in:
William Jeynes
2026-02-09 16:32:40 +00:00
parent 8eaa7bfbff
commit 02eac0f553
9 changed files with 311 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agent/agent.ts
+6 -4
View File
@@ -8,14 +8,16 @@ 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 dummyNormalisationModel = createDummyModelNode("Normalised");
const dummyVerificationModel = createDummyModelNode("verification of");
const normalisationModel = createModelNode([], "normalization.txt");
const triggerEventToolConditional = createToolConditional("triggerEventToolNode", verificationSetup.name);
const verificationToolConditional = createToolConditional("verificationToolNode", produceRanking.name);
@@ -25,7 +27,7 @@ const agent = new StateGraph(MessagesState)
//NODES
.addNode(normalizationSetup.name, normalizationSetup)
.addNode("dummyNormalisationModel", dummyNormalisationModel)
.addNode("normalisationModel", normalisationModel)
.addNode("triggerEventToolNode", triggerEventToolNode)
.addNode("dummyTriggerEventModel", dummyTriggerEventModel)
@@ -37,8 +39,8 @@ const agent = new StateGraph(MessagesState)
.addNode(produceRanking.name, produceRanking)
.addEdge(START, normalizationSetup.name)
.addEdge(normalizationSetup.name, "dummyNormalisationModel")
.addEdge("dummyNormalisationModel", "dummyTriggerEventModel")
.addEdge(normalizationSetup.name, "normalisationModel")
.addEdge("normalisationModel", "dummyTriggerEventModel")
// @ts-expect-error
.addConditionalEdges("dummyTriggerEventModel", triggerEventToolConditional, ["triggerEventToolNode", verificationSetup.name])