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

Add relation model. Add calculate score initial version

Browse Source
This commit is contained in:
William Jeynes
2026-02-12 23:26:59 +00:00
parent c89f73e138
commit b06c08daab
5 changed files with 60 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agent/agent.ts
+5 -1
View File
@@ -15,13 +15,13 @@ const triggerEventToolNode = createToolNode(triggerEventToolsByName);
const normalisationModel = createModelNode([], "normalization.txt");
const triggerEventModel = createModelNode(triggerEventToolsByName, "trigger.txt");
const verificationModel = createModelNode([], "verify.txt");
const relationModel = createModelNode([], "relation.txt");
const triggerEventToolConditional = createToolConditional("triggerEventToolNode", verificationSetup.name);
const agent = new StateGraph(MessagesState)
//NODES
.addNode(normalizationSetup.name, normalizationSetup)
.addNode("normalisationModel", normalisationModel)
@@ -31,6 +31,8 @@ const agent = new StateGraph(MessagesState)
.addNode(verificationSetup.name, verificationSetup)
.addNode("verificationModel", verificationModel)
.addNode(ragasMetrics.name, ragasMetrics)
.addNode("relationModel", relationModel)
.addNode(produceRanking.name, produceRanking)
.addEdge(START, normalizationSetup.name)
@@ -43,9 +45,11 @@ const agent = new StateGraph(MessagesState)
.addEdge(verificationSetup.name, "verificationModel")
.addEdge(verificationSetup.name, ragasMetrics.name)
.addEdge(verificationSetup.name, "relationModel")
.addEdge(ragasMetrics.name, produceRanking.name)
.addEdge("verificationModel", produceRanking.name)
.addEdge("relationModel", produceRanking.name)
// @ts-expect-error
.addConditionalEdges(produceRanking.name, loopEndConditional, [verificationSetup.name, END])