better

1 files changed, 9 insertions, 8 deletions

diff --git a/src/client/views/nodes/chatbot/tools/GetDocsTool.ts b/src/client/views/nodes/chatbot/tools/GetDocsTool.ts
index 4286e7ffe..05482a66e 100644
--- a/src/client/views/nodes/chatbot/tools/GetDocsTool.ts
+++ b/src/client/views/nodes/chatbot/tools/GetDocsTool.ts
@@ -1,5 +1,5 @@
 import { Observation } from '../types/types';
-import { ParametersType } from '../types/tool_types';
+import { ParametersType, ToolInfo } from '../types/tool_types';
 import { BaseTool } from './BaseTool';
 import { DocServer } from '../../../../DocServer';
 import { Docs } from '../../../../documents/Documents';
@@ -24,17 +24,18 @@ const getDocsToolParams = [
 
 type GetDocsToolParamsType = typeof getDocsToolParams;
 
+const getDocsToolInfo: ToolInfo<GetDocsToolParamsType> = {
+    name: 'retrieveDocs',
+    description: 'Retrieves the contents of all Documents that the user is interacting with in Dash.',
+    citationRules: 'No citation needed.',
+    parameterRules: getDocsToolParams,
+};
+
 export class GetDocsTool extends BaseTool<GetDocsToolParamsType> {
     private _docView: DocumentView;
 
     constructor(docView: DocumentView) {
-        super(
-            'retrieveDocs',
-            'Retrieves the contents of all Documents that the user is interacting with in Dash',
-            getDocsToolParams,
-            'No need to provide anything. Just run the tool and it will retrieve the contents of all Documents that the user is interacting with in Dash.',
-            'Returns the documents in Dash in JSON form.'
-        );
+        super(getDocsToolInfo);
         this._docView = docView;
     }