William Jeynes
31 lines
1.2 KiB
TypeScript
31 lines
1.2 KiB
TypeScript
import { GraphNode } from "@langchain/langgraph";
|
|
import { MessagesState, ProposedTriggerEventArray } from "../state";
|
|
import { logger } from "../utils/logger";
|
|
import { queryScraper } from "../tools/webSearch";
|
|
import { rankAndDisplayData } from "../tools/triggerEventTools";
|
|
|
|
export const verificationSetup: GraphNode<typeof MessagesState> = async (state) => {
|
|
//this is kinda doing two things, but having two nodes for it seems overkill
|
|
|
|
if (state.proposedTriggerEvent == undefined) {
|
|
logger.warn("No trigger events in memory, parsing")
|
|
|
|
let genResponse = state.messages.at(-1)?.content.toString() ?? "";
|
|
const parsed = ProposedTriggerEventArray.parse(JSON.parse(genResponse));
|
|
|
|
for (let i = 0; i < parsed.length; i++) {
|
|
const search = parsed[i].SearchQuery
|
|
const data = await queryScraper(search);
|
|
const output = await rankAndDisplayData(data, search);
|
|
|
|
parsed[i].context = output;
|
|
}
|
|
|
|
return { proposedTriggerEvent: parsed, proposedTriggerEventIndex: 0 };
|
|
}
|
|
else {
|
|
logger.info("Trigger event index %s", state.proposedTriggerEventIndex+1)
|
|
|
|
return { proposedTriggerEvent: state.proposedTriggerEvent, proposedTriggerEventIndex: state.proposedTriggerEventIndex+1 };
|
|
}
|
|
}; |