Implement ensemble into final model structure

This commit is contained in:
William Jeynes
2026-03-24 19:07:24 +00:00
parent 624d45bc53
commit 8f939d54c4
9 changed files with 71 additions and 347 deletions
+17
View File
@@ -0,0 +1,17 @@
import { GraphNode } from "@langchain/langgraph";
import { MessagesState } from "../state";
import { AIMessage } from "@langchain/core/messages";
import { evaluateWithEnsemble } from "../tools/ensembleCall";
export function createEnsembleNode(title: string, method: string): GraphNode<typeof MessagesState> {
return async (state) => {
const answer = state.proposedTriggerEvent[state.proposedTriggerEventIndex].Event
const result = await evaluateWithEnsemble({ answer, method })
const score = result.validProb - result.invalidProb;
return {
messages: [new AIMessage(title + ":" + score)]
};
};
};