From 79e4c4a3fba42b90ffa656db3ca435505f978afe Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Tue, 20 Aug 2024 18:32:08 -0400
Subject: supports multiple inputs

maybe also make it so web results cannot have overlap (no same url in websites returned by search)

Also make sure it will cite multiple websites
---
 src/client/views/nodes/ChatBox/tools/SearchTool.ts | 63 ++++++++++++----------
 1 file changed, 35 insertions(+), 28 deletions(-)

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

diff --git a/src/client/views/nodes/ChatBox/tools/SearchTool.ts b/src/client/views/nodes/ChatBox/tools/SearchTool.ts
index 6523fd11c..b926cbadc 100644
--- a/src/client/views/nodes/ChatBox/tools/SearchTool.ts
+++ b/src/client/views/nodes/ChatBox/tools/SearchTool.ts
@@ -1,47 +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 }> {
+export class SearchTool extends BaseTool<{ query: string | string[] }> {
     private _addLinkedUrlDoc: (url: string, id: string) => void;
-
-    constructor(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',
+            'Search the web to find a wide range of websites related to a query or multiple queries',
             {
                 query: {
                     type: 'string',
-                    description: 'The search query to use for finding websites',
-                    required: true,
+                    description: 'The search query or queries to use for finding websites',
+                    required: 'true',
+                    max_inputs: '3',
                 },
             },
-            'Provide a search query 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 query, helping to identify which website might contain the most relevant information.'
+            '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 }): Promise<any> {
-        try {
-            const { results } = await Networking.PostToServer('/getWebSearchResults', { query: args.query });
-            console.log(results);
-            const data: { type: string; text: string }[] = results.map((result: { url: string; snippet: string }) => {
-                console.log;
-                const id = uuidv4();
-                //this._addLinkedUrlDoc(result.url, id); //not needed right now because it shouldn't use this information to ground responses and should scrape afterwards
-                return {
-                    type: 'text',
-                    text: `<chunk chunk_id="${id}" chunk_type="text">
-                            <url>${result.url}</url>
-                            <overview>${result.snippet}</overview>
-                        </chunk>`,
-                };
-            });
-            return data;
-        } catch (error) {
-            console.log(error);
-            return [{ type: 'text', text: 'An error occurred while performing the web search.' }];
+    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