From dfcb9a4781d9026634cee84ce1a83893e14809d0 Mon Sep 17 00:00:00 2001
From: thucpn <thucsh2@gmail.com>
Date: Wed, 19 Mar 2025 10:21:06 +0700
Subject: [PATCH] wiki -> wikipedia
---
packages/tools/src/tools/{wiki.ts => wikipedia.ts} | 10 +++++-----
.../tools/tests/{wiki.test.ts => wikipedia.test.ts} | 4 ++--
2 files changed, 7 insertions(+), 7 deletions(-)
rename packages/tools/src/tools/{wiki.ts => wikipedia.ts} (73%)
rename packages/tools/tests/{wiki.test.ts => wikipedia.test.ts} (79%)
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 e5eff4718..c3d3cbd86 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 0ac229136..e1c40f024 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",
--
GitLab