plugins: add wiki search (labring#2886)

* plugins: add wiki search

* 扁平化代码

* fix: url error

Author: Patrickill

packages/plugins/package.json

@@ -3,11 +3,12 @@
   "version": "1.0.0",
   "type": "module",
   "dependencies": {
-    "duck-duck-scrape": "^2.2.5",
-    "lodash": "^4.17.21",
     "axios": "^1.5.1",
+    "duck-duck-scrape": "^2.2.5",
+    "echarts": "5.4.1",
     "expr-eval": "^2.0.2",
-    "echarts": "5.4.1"
+    "lodash": "^4.17.21",
+    "wikijs": "^6.4.1"
   },
   "devDependencies": {
     "@fastgpt/global": "workspace:*",

packages/plugins/register.ts

@@ -24,7 +24,8 @@ const packagePluginList = [
   'duckduckgo/searchNews',
   'duckduckgo/searchVideo',
   'drawing',
-  'drawing/baseChart'
+  'drawing/baseChart',
+  'wiki'
 ];
 
 export const list = [...staticPluginList, ...packagePluginList];

packages/plugins/src/wiki/index.ts

@@ -0,0 +1,43 @@
+import { getErrText } from '@fastgpt/global/common/error/utils';
+import { addLog } from '@fastgpt/service/common/system/log';
+import { delay } from '@fastgpt/global/common/system/utils';
+import wiki from 'wikijs';
+
+type Props = {
+  query: string;
+};
+
+// Response type same as HTTP outputs
+type Response = Promise<{
+  result: string;
+}>;
+
+const main = async (props: Props, retry = 3): Response => {
+  const { query } = props;
+
+  try {
+    const searchResults = await wiki({ apiUrl: 'https://zh.wikipedia.org/w/api.php' })
+      .page(query)
+      .then((page) => {
+        return page.summary();
+      });
+
+    return {
+      result: searchResults
+    };
+  } catch (error) {
+    console.log(error);
+
+    if (retry <= 0) {
+      addLog.warn('search wiki error', { error });
+      return {
+        result: getErrText(error, 'Failed to fetch data from wiki')
+      };
+    }
+
+    await delay(Math.random() * 5000);
+    return main(props, retry - 1);
+  }
+};
+
+export default main;