William Jeynes
48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
import { tool } from "@langchain/core/tools";
|
|
import * as z from "zod";
|
|
import { queryScraper } from "./webSearch";
|
|
import { extractWebpageContent } from "./webpageFetch";
|
|
import { rankDynamically } from "./retreiveExamples";
|
|
|
|
|
|
export async function rankAndDisplayData(data: string[], context: string):Promise<string> {
|
|
let index = 0;
|
|
let ranked = await rankDynamically(context, data.map(irm => ({ id: index++, rawtext: irm })))
|
|
return ranked.map(itm => itm.rawtext).join("\n")
|
|
}
|
|
|
|
// Define tools
|
|
const webSearch = tool(
|
|
async ({ a }) => {
|
|
const data = await queryScraper(a);
|
|
return await rankAndDisplayData(data, a);
|
|
},
|
|
{
|
|
name: "WebSearch",
|
|
description: "Search DuckDuckGo for the provided query",
|
|
schema: z.object({
|
|
a: z.string().describe("Search term"),
|
|
}),
|
|
}
|
|
);
|
|
|
|
const openWebpage = tool(
|
|
async ({ a }) => {
|
|
const data = await extractWebpageContent(a);
|
|
return data;
|
|
},
|
|
{
|
|
name: "WebRetreiveUrl",
|
|
description: "Fetches and reads the FULL contents of a webpage. REQUIRED when detailed or accurate information is needed.",
|
|
schema: z.object({
|
|
a: z.url().describe("URL")
|
|
}),
|
|
}
|
|
);
|
|
|
|
// Augment the LLM with tools
|
|
export const triggerEventToolsByName = {
|
|
[webSearch.name]: webSearch,
|
|
[openWebpage.name]: openWebpage
|
|
};
|