From 4791cd23af08da70895204a3a7fbaf889d9af2d5 Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Sat, 7 Sep 2024 12:43:05 -0400
Subject: completely restructured, added comments, and significantly reduced
 the length of the prompt (~72% shorter and cheaper)

---
 .../views/nodes/chatbot/tools/WikipediaTool.ts     | 37 ++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/client/views/nodes/chatbot/tools/WikipediaTool.ts

(limited to 'src/client/views/nodes/chatbot/tools/WikipediaTool.ts')

diff --git a/src/client/views/nodes/chatbot/tools/WikipediaTool.ts b/src/client/views/nodes/chatbot/tools/WikipediaTool.ts
new file mode 100644
index 000000000..143d91d80
--- /dev/null
+++ b/src/client/views/nodes/chatbot/tools/WikipediaTool.ts
@@ -0,0 +1,37 @@
+import { title } from 'process';
+import { Networking } from '../../../../Network';
+import { BaseTool } from './BaseTool';
+import axios from 'axios';
+import { v4 as uuidv4 } from 'uuid';
+
+export class WikipediaTool extends BaseTool<{ title: string }> {
+    private _addLinkedUrlDoc: (url: string, id: string) => void;
+    constructor(addLinkedUrlDoc: (url: string, id: string) => void) {
+        super(
+            'wikipedia',
+            'Search Wikipedia and return a summary',
+            {
+                title: {
+                    type: 'string',
+                    description: 'The title of the Wikipedia article to search',
+                    required: true,
+                },
+            },
+            'Provide simply the title you want to search on Wikipedia and nothing more. If re-using this tool, try a different title for different information.',
+            'Returns a summary from searching an article title on Wikipedia'
+        );
+        this._addLinkedUrlDoc = addLinkedUrlDoc;
+    }
+
+    async execute(args: { title: string }): Promise<any> {
+        try {
+            const { text } = await Networking.PostToServer('/getWikipediaSummary', { title: args.title });
+            const id = uuidv4();
+            const url = `https://en.wikipedia.org/wiki/${args.title.replace(/ /g, '_')}`;
+            this._addLinkedUrlDoc(url, id);
+            return [{ type: 'text', text: `<chunk chunk_id=${id} chunk_type=csv}> ${text} </chunk>` }];
+        } catch (error) {
+            return [{ type: 'text', text: 'An error occurred while fetching the article.' }];
+        }
+    }
+}
-- 
cgit v1.2.3-70-g09d2