Skip to content

Commit 87540b7

Browse files
committed
Fix Novita e2e mock completion fixture
1 parent b757163 commit 87540b7

1 file changed

Lines changed: 13 additions & 4 deletions

File tree

apps/vscode-e2e/src/fixtures/novita.ts

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,22 @@
1-
import { LLMock } from "@copilotkit/aimock"
1+
import { LLMock, type ChatCompletionRequest } from "@copilotkit/aimock"
22

3-
import { toolResultContains } from "./tool-result"
3+
function hasMarkerToolResult(req: ChatCompletionRequest) {
4+
const messages = Array.isArray(req?.messages) ? req.messages : []
5+
6+
return messages.some((message) => {
7+
if (message?.role !== "tool" && message?.role !== "user") {
8+
return false
9+
}
10+
11+
return JSON.stringify(message.content ?? "").includes("NOVITA_E2E_MARKER")
12+
})
13+
}
414

515
export function addNovitaFixtures(mock: InstanceType<typeof LLMock>) {
616
mock.addFixture({
717
match: {
818
model: "moonshotai/kimi-k2.7-code",
9-
toolCallId: "call_novita_read",
10-
predicate: (req) => toolResultContains(req, "call_novita_read", ["NOVITA_E2E_MARKER"]),
19+
predicate: hasMarkerToolResult,
1120
},
1221
response: {
1322
toolCalls: [

0 commit comments

Comments
 (0)