1 files changed, 6 insertions, 13 deletions
diff --git a/src/client/views/nodes/ChatBox/tools/BaseTool.ts b/src/client/views/nodes/ChatBox/tools/BaseTool.ts
index c7942e359..2e2267653 100644
--- a/src/client/views/nodes/ChatBox/tools/BaseTool.ts
+++ b/src/client/views/nodes/ChatBox/tools/BaseTool.ts
@@ -5,26 +5,19 @@ export abstract class BaseTool<T extends Record<string, any> = Record<string, an
         public name: string,
         public description: string,
         public parameters: Record<string, any>,
-        public useRules: string,
+        public citationRules: string,
         public briefSummary: string
     ) {}
 
     abstract execute(args: T): Promise<any>;
 
-    getActionRule(isCurrentTool: boolean): Record<string, any> {
-        if (isCurrentTool) {
-            return {
-                [this.name]: {
-                    name: this.name,
-                    useRules: this.useRules,
-                    description: this.description,
-                    parameters: this.parameters,
-                },
-            };
-        }
+    getActionRule(): Record<string, any> {
         return {
             [this.name]: {
-                description: 'This tool is not currently selected.',
+                name: this.name,
+                citationRules: this.citationRules,
+                description: this.description,
+                parameters: this.parameters,
             },
         };
     }