better

1 files changed, 9 insertions, 8 deletions

diff --git a/src/client/views/nodes/chatbot/tools/WikipediaTool.ts b/src/client/views/nodes/chatbot/tools/WikipediaTool.ts
index f2dbf3cfd..ee815532a 100644
--- a/src/client/views/nodes/chatbot/tools/WikipediaTool.ts
+++ b/src/client/views/nodes/chatbot/tools/WikipediaTool.ts
@@ -2,7 +2,7 @@ import { v4 as uuidv4 } from 'uuid';
 import { Networking } from '../../../../Network';
 import { BaseTool } from './BaseTool';
 import { Observation } from '../types/types';
-import { ParametersType } from '../types/tool_types';
+import { ParametersType, ToolInfo } from '../types/tool_types';
 
 const wikipediaToolParams = [
     {
@@ -15,17 +15,18 @@ const wikipediaToolParams = [
 
 type WikipediaToolParamsType = typeof wikipediaToolParams;
 
+const wikipediaToolInfo: ToolInfo<WikipediaToolParamsType> = {
+    name: 'wikipedia',
+    citationRules: 'No citation needed.',
+    parameterRules: wikipediaToolParams,
+    description: 'Returns a summary from searching an article title on Wikipedia.',
+};
+
 export class WikipediaTool extends BaseTool<WikipediaToolParamsType> {
     private _addLinkedUrlDoc: (url: string, id: string) => void;
 
     constructor(addLinkedUrlDoc: (url: string, id: string) => void) {
-        super(
-            'wikipedia',
-            'Search Wikipedia and return a summary',
-            wikipediaToolParams,
-            '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'
-        );
+        super(wikipediaToolInfo);
         this._addLinkedUrlDoc = addLinkedUrlDoc;
     }