Merge remote-tracking branch 'origin/ajs-finalagent'

1 files changed, 3 insertions, 2 deletions

diff --git a/src/client/views/nodes/chatbot/tools/BaseTool.ts b/src/client/views/nodes/chatbot/tools/BaseTool.ts
index 58cd514d9..8efba2d28 100644
--- a/src/client/views/nodes/chatbot/tools/BaseTool.ts
+++ b/src/client/views/nodes/chatbot/tools/BaseTool.ts
@@ -1,5 +1,5 @@
 import { Observation } from '../types/types';
-import { Parameter, Tool, ParametersType } from './ToolTypes';
+import { Parameter, ParametersType } from '../types/tool_types';
 
 /**
  * @file BaseTool.ts
@@ -14,7 +14,7 @@ import { Parameter, Tool, ParametersType } from './ToolTypes';
  * It is generic over a type parameter `P`, which extends `ReadonlyArray<Parameter>`.
  * This means `P` is a readonly array of `Parameter` objects that cannot be modified (immutable).
  */
-export abstract class BaseTool<P extends ReadonlyArray<Parameter>> implements Tool<P> {
+export abstract class BaseTool<P extends ReadonlyArray<Parameter>> {
     // The name of the tool (e.g., "calculate", "searchTool")
     name: string;
     // A description of the tool's functionality
@@ -59,6 +59,7 @@ export abstract class BaseTool<P extends ReadonlyArray<Parameter>> implements To
         return {
             tool: this.name,
             description: this.description,
+            citationRules: this.citationRules,
             parameters: this.parameterRules.reduce(
                 (acc, param) => {
                     // Build an object for each parameter without the 'name' property, since it's used as the key