From c47ed4e8d895ade3cd327841ab536702902b054d Mon Sep 17 00:00:00 2001
From: thucpn <thucsh2@gmail.com>
Date: Wed, 19 Mar 2025 13:50:11 +0700
Subject: [PATCH] revert wiki (update in another PR)
---
packages/tools/src/index.ts | 2 +-
packages/tools/src/tools/{wikipedia.ts => wiki.ts} | 10 +++++-----
packages/tools/tests/wikipedia.test.ts | 6 +++---
3 files changed, 9 insertions(+), 9 deletions(-)
rename packages/tools/src/tools/{wikipedia.ts => wiki.ts} (73%)
diff --git a/packages/tools/src/index.ts b/packages/tools/src/index.ts
index d13b32694..5c49d8c69 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 c3d3cbd86..e5eff4718 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 e1c40f024..1587e0323 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",
});
--
GitLab