William Jeynes
14 lines
667 B
TypeScript
14 lines
667 B
TypeScript
import { GraphNode } from "@langchain/langgraph";
|
|
import { MessagesState } from "../state";
|
|
import { AIMessage, BaseMessage, HumanMessage } from "@langchain/core/messages";
|
|
import { rankNormalizedClaims } from "../tools/retreiveExamples";
|
|
|
|
export const normalizationSetup: GraphNode<typeof MessagesState> = async (state) => {
|
|
let similarityResults = await rankNormalizedClaims(state.disinformationTitle)
|
|
|
|
let messages : BaseMessage[] = similarityResults.map((item) => {
|
|
return new AIMessage(`Original Claim: ${item.rawtext}. \n\n Normalised Claim: ${item.cleantext}`)
|
|
})
|
|
|
|
return { messages: messages, disinformationTitle: state.disinformationTitle };
|
|
}; |