From e01afb407c4b3eb50e85e7356b0be9489fb26eba Mon Sep 17 00:00:00 2001
From: Dax Raad <d@ironbay.co>
Date: Wed, 21 May 2025 10:30:39 -0400
Subject: add tool tests

---
 js/src/session/session.ts | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

(limited to 'js/src/session')

diff --git a/js/src/session/session.ts b/js/src/session/session.ts
index dfca22a3f..338c8dda9 100644
--- a/js/src/session/session.ts
+++ b/js/src/session/session.ts
@@ -14,9 +14,7 @@ import {
   type UIMessagePart,
 } from "ai";
 import { z } from "zod";
-import { BashTool } from "../tool/bash";
-import { EditTool } from "../tool/edit";
-import { ViewTool } from "../tool/view";
+import * as tools from "../tool";
 
 import ANTHROPIC_PROMPT from "./prompt/anthropic.txt";
 import type { Tool } from "../tool/tool";
@@ -173,11 +171,7 @@ export namespace Session {
       maxSteps: 1000,
       messages: convertToModelMessages(msgs),
       temperature: 0,
-      tools: {
-        ...BashTool,
-        ...ViewTool,
-        ...EditTool,
-      },
+      tools,
       model,
     });
     const next: Message = {
@@ -192,7 +186,7 @@ export namespace Session {
         tool: {},
       },
     };
-    const metadata = next.metadata!;
+
     msgs.push(next);
     let text: TextUIPart | undefined;
     const reader = result.toUIMessageStream().getReader();
-- 
cgit v1.2.3