Add removing of duplicates from pipeline. Add to sort step. Move score logic to robertaMetrics node.

2026-03-13 14:51:14 +00:00
parent d5c6cb444d
commit 0a7bb114d2
6 changed files with 105 additions and 7 deletions
@@ -8,7 +8,12 @@ export const robertaMetrics: GraphNode<typeof MessagesState> = async (state) =>
  
  const result = await evaluateWithRoberta({answer})
  
+  let score = 0;
+  if (result.validProb > result.invalidProb) {
+    score = 0.7 + ((result.validProb - result.invalidProb)*0.3);
+  }
+  
  return {
-    messages: [ new AIMessage("ROBERTA:" + result)]
+    messages: [ new AIMessage("ROBERTA:" + score)]
  };
 };