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