summaryrefslogtreecommitdiffhomepage
path: root/js/src/session
diff options
context:
space:
mode:
Diffstat (limited to 'js/src/session')
-rw-r--r--js/src/session/session.ts12
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();