diff --git a/packages/tools/src/tools/wiki.ts b/packages/tools/src/tools/wikipedia.ts similarity index 73% rename from packages/tools/src/tools/wiki.ts rename to packages/tools/src/tools/wikipedia.ts index e5eff4718e2d654d5cde1d0ec44ab54c329ac551..c3d3cbd86041ca855e188f0f4f5bb2492c55d764 100644 --- a/packages/tools/src/tools/wiki.ts +++ b/packages/tools/src/tools/wikipedia.ts @@ -1,5 +1,5 @@ import { tool } from "@llamaindex/core/tools"; -import { default as wikipedia } from "wikipedia"; +import { default as wiki } from "wikipedia"; import { z } from "zod"; export type WikiToolOutput = { @@ -7,7 +7,7 @@ export type WikiToolOutput = { content: string; }; -export const wiki = () => { +export const wikipedia = () => { return tool({ name: "wikipedia", description: "Use this function to search Wikipedia", @@ -16,11 +16,11 @@ export const wiki = () => { lang: z.string().describe("The language to search in").default("en"), }), execute: async ({ query, lang }): Promise<WikiToolOutput> => { - wikipedia.setLang(lang); - const searchResult = await wikipedia.search(query); + wiki.setLang(lang); + const searchResult = await wiki.search(query); const pageTitle = searchResult?.results[0]?.title; if (!pageTitle) return { title: "No search results.", content: "" }; - const result = await wikipedia.page(pageTitle, { autoSuggest: false }); + const result = await wiki.page(pageTitle, { autoSuggest: false }); return { title: pageTitle, content: await result.content() }; }, }); diff --git a/packages/tools/tests/wiki.test.ts b/packages/tools/tests/wikipedia.test.ts similarity index 79% rename from packages/tools/tests/wiki.test.ts rename to packages/tools/tests/wikipedia.test.ts index 0ac22913680432b6014d0e166833ec994c929441..e1c40f024375f1bc750c5b80405d57b2b4f910da 100644 --- a/packages/tools/tests/wiki.test.ts +++ b/packages/tools/tests/wikipedia.test.ts @@ -1,9 +1,9 @@ import { describe, expect, test } from "vitest"; -import { wiki } from "../src/tools/wiki"; +import { wikipedia } from "../src/tools/wikipedia"; describe("Wikipedia Tool", () => { test("wiki tool returns content for valid query", async () => { - const wikipediaTool = wiki(); + const wikipediaTool = wikipedia(); const result = await wikipediaTool.call({ query: "Albert Einstein", lang: "en",