aboutsummaryrefslogtreecommitdiff
path: root/src/client/views/nodes/ChatBox/tools/WikipediaTool.ts
diff options
context:
space:
mode:
authorA.J. Shulman <Shulman.aj@gmail.com>2024-09-07 12:43:05 -0400
committerA.J. Shulman <Shulman.aj@gmail.com>2024-09-07 12:43:05 -0400
commit4791cd23af08da70895204a3a7fbaf889d9af2d5 (patch)
treec4c2534e64724d62bae9152763f1a74cd5a963e0 /src/client/views/nodes/ChatBox/tools/WikipediaTool.ts
parent210f8f5f1cd19e9416a12524cce119b273334fd3 (diff)
completely restructured, added comments, and significantly reduced the length of the prompt (~72% shorter and cheaper)
Diffstat (limited to 'src/client/views/nodes/ChatBox/tools/WikipediaTool.ts')
-rw-r--r--src/client/views/nodes/ChatBox/tools/WikipediaTool.ts37
1 files changed, 0 insertions, 37 deletions
diff --git a/src/client/views/nodes/ChatBox/tools/WikipediaTool.ts b/src/client/views/nodes/ChatBox/tools/WikipediaTool.ts
deleted file mode 100644
index 143d91d80..000000000
--- a/src/client/views/nodes/ChatBox/tools/WikipediaTool.ts
+++ /dev/null
@@ -1,37 +0,0 @@
-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.' }];
- }
- }
-}