fix(tools): correctly infer input type for DynamicTool

Signed-off-by: Tomas Dvorak <[email protected]>
i-am-bee · Nov 14, 2024 · 90db143 · 90db143
1 parent 5e1de24
commit 90db143
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/tools/base.ts b/src/tools/base.ts
@@ -177,7 +177,7 @@ export abstract class Tool<
   abstract description: string;
 
   public readonly cache: BaseCache<Task<TOutput>>;
-  protected readonly options: TOptions;
+  public readonly options: TOptions;
 
   public readonly emitter = Emitter.root.child<ToolCallbacks<ToolInput<this>, TOutput>>({
     namespace: ["tool"],
@@ -400,10 +400,10 @@ export class DynamicTool<
 
   declare name: string;
   declare description: string;
-  private readonly _inputSchema: AnyToolSchemaLike;
+  private readonly _inputSchema: TInputSchema;
   private readonly handler;
 
-  inputSchema() {
+  inputSchema(): TInputSchema {
     return this._inputSchema;
   }