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)

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

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

diff --git a/src/client/views/nodes/chatbot/tools/SearchTool.ts b/src/client/views/nodes/chatbot/tools/SearchTool.ts
new file mode 100644
index 000000000..b926cbadc
--- /dev/null
+++ b/src/client/views/nodes/chatbot/tools/SearchTool.ts
@@ -0,0 +1,54 @@
+import { max } from 'lodash';
+import { Networking } from '../../../../Network';
+import { BaseTool } from './BaseTool';
+import { v4 as uuidv4 } from 'uuid';
+
+export class SearchTool extends BaseTool<{ query: string | string[] }> {
+    private _addLinkedUrlDoc: (url: string, id: string) => void;
+    private _max_results: number;
+    constructor(addLinkedUrlDoc: (url: string, id: string) => void, max_results: number = 5) {
+        super(
+            'searchTool',
+            'Search the web to find a wide range of websites related to a query or multiple queries',
+            {
+                query: {
+                    type: 'string',
+                    description: 'The search query or queries to use for finding websites',
+                    required: 'true',
+                    max_inputs: '3',
+                },
+            },
+            'Provide up to 3 search queries to find a broad range of websites. This tool is intended to help you identify relevant websites, but not to be used for providing the final answer. Use this information to determine which specific website to investigate further.',
+            'Returns a list of websites and their overviews based on the search queries, helping to identify which websites might contain relevant information.'
+        );
+        this._addLinkedUrlDoc = addLinkedUrlDoc;
+        this._max_results = max_results;
+    }
+
+    async execute(args: { query: string | string[] }): Promise<any> {
+        const queries = Array.isArray(args.query) ? args.query : [args.query];
+        const allResults = [];
+
+        for (const query of queries) {
+            try {
+                const { results } = await Networking.PostToServer('/getWebSearchResults', { query, max_results: this._max_results });
+                const data: { type: string; text: string }[] = results.map((result: { url: string; snippet: string }) => {
+                    const id = uuidv4();
+                    return {
+                        type: 'text',
+                        text: `<chunk chunk_id="${id}" chunk_type="text">
+                                <url>${result.url}</url>
+                                <overview>${result.snippet}</overview>
+                            </chunk>`,
+                    };
+                });
+                allResults.push(...data);
+            } catch (error) {
+                console.log(error);
+                allResults.push({ type: 'text', text: `An error occurred while performing the web search for query: ${query}` });
+            }
+        }
+
+        return allResults;
+    }
+}
-- 
cgit v1.2.3-70-g09d2