diff options
Diffstat (limited to 'js/src/session')
| -rw-r--r-- | js/src/session/session.ts | 12 |
1 files changed, 3 insertions, 9 deletions
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(); |
