From 6f9b8f9b393d411a17f7954b6cc36618efe698e2 Mon Sep 17 00:00:00 2001
From: "A.J. Shulman" <Shulman.aj@gmail.com>
Date: Thu, 15 Aug 2024 13:16:32 -0400
Subject: implemented search tool and other tools but scraping doesn't work

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

(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
new file mode 100644
index 000000000..91ecc71ff
--- /dev/null
+++ b/src/client/views/nodes/ChatBox/tools/SearchTool.ts
@@ -0,0 +1,47 @@
+import { Networking } from '../../../../Network';
+import { BaseTool } from './BaseTool';
+import { v4 as uuidv4 } from 'uuid';
+
+export class SearchTool extends BaseTool<{ query: string }> {
+    private _addLinkedUrlDoc: (url: string, id: string) => void;
+
+    constructor(addLinkedUrlDoc: (url: string, id: string) => void) {
+        super(
+            'searchTool',
+            'Search the web to find a wide range of websites related to a query',
+            {
+                query: {
+                    type: 'string',
+                    description: 'The search query to use for finding websites',
+                    required: true,
+                },
+            },
+            '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.'
+        );
+        this._addLinkedUrlDoc = addLinkedUrlDoc;
+    }
+
+    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);
+                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.' }];
+        }
+    }
+}
-- 
cgit v1.2.3-70-g09d2