Skip to content

Commit 1e87618

Browse files
test: hardcode tool_use
1 parent 8c3805c commit 1e87618

File tree

3 files changed

+8
-9
lines changed

3 files changed

+8
-9
lines changed

extensions/vscode/e2e/test-continue-yaml/config.yaml

+2
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ models:
4343
- chat
4444
- edit
4545
- apply
46+
capabilities:
47+
- tool_use
4648

4749
- name: SYSTEM MESSAGE MOCK LLM
4850
provider: mock

extensions/vscode/e2e/test-continue/config.json

+3
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@
1717
"provider": "mock",
1818
"title": "TOOL MOCK LLM",
1919
"model": "claude-3-5-sonnet-latest",
20+
"capabilities": {
21+
"tools": true
22+
},
2023
"requestOptions": {
2124
"extraBodyProperties": {
2225
"chatStream": [

extensions/vscode/e2e/tests/GUI.test.ts

+3-9
Original file line numberDiff line numberDiff line change
@@ -240,10 +240,12 @@ describe("GUI Test", () => {
240240
});
241241

242242
describe("Agent with tools", () => {
243-
it("should render tool call", async () => {
243+
beforeEach(async () => {
244244
await GUIActions.selectModelFromDropdown(view, "TOOL MOCK LLM");
245245
await GUIActions.selectModeFromDropdown(view, "Agent");
246+
});
246247

248+
it("should render tool call", async () => {
247249
const [messageInput] = await GUISelectors.getMessageInputFields(view);
248250
await messageInput.sendKeys("Hello");
249251
await messageInput.sendKeys(Key.ENTER);
@@ -261,10 +263,6 @@ describe("GUI Test", () => {
261263
it("should call tool after approval", async () => {
262264
await GUIActions.toggleToolPolicy(view, "builtin_view_diff", 2);
263265

264-
await TestUtils.waitForSuccess(() =>
265-
GUIActions.selectModelFromDropdown(view, "TOOL MOCK LLM"),
266-
);
267-
268266
const [messageInput] = await GUISelectors.getMessageInputFields(view);
269267
await messageInput.sendKeys("Hello");
270268
await messageInput.sendKeys(Key.ENTER);
@@ -286,10 +284,6 @@ describe("GUI Test", () => {
286284
it("should cancel tool", async () => {
287285
await GUIActions.toggleToolPolicy(view, "builtin_view_diff", 2);
288286

289-
await TestUtils.waitForSuccess(() =>
290-
GUIActions.selectModelFromDropdown(view, "TOOL MOCK LLM"),
291-
);
292-
293287
const [messageInput] = await GUISelectors.getMessageInputFields(view);
294288
await messageInput.sendKeys("Hello");
295289
await messageInput.sendKeys(Key.ENTER);

0 commit comments

Comments
 (0)