Skip to content

Commit 4bfd070

Browse files
authored
fix: pass tool protocol parameter to lineCountTruncationError (#9358)
1 parent 9538d82 commit 4bfd070

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/core/tools/WriteToFileTool.ts

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import { EXPERIMENT_IDS, experiments } from "../../shared/experiments"
1818
import { convertNewFileToUnifiedDiff, computeDiffStats, sanitizeUnifiedDiff } from "../diff/stats"
1919
import { BaseTool, ToolCallbacks } from "./BaseTool"
2020
import type { ToolUse } from "../../shared/tools"
21+
import { resolveToolProtocol } from "../../utils/resolveToolProtocol"
2122

2223
interface WriteToFileParams {
2324
path: string
@@ -109,6 +110,8 @@ export class WriteToFileTool extends BaseTool<"write_to_file"> {
109110
const actualLineCount = newContent.split("\n").length
110111
const isNewFile = !fileExists
111112
const diffStrategyEnabled = !!task.diffStrategy
113+
const modelInfo = task.api.getModel().info
114+
const toolProtocol = resolveToolProtocol(task.apiConfiguration, modelInfo)
112115

113116
await task.say(
114117
"error",
@@ -119,7 +122,12 @@ export class WriteToFileTool extends BaseTool<"write_to_file"> {
119122

120123
pushToolResult(
121124
formatResponse.toolError(
122-
formatResponse.lineCountTruncationError(actualLineCount, isNewFile, diffStrategyEnabled),
125+
formatResponse.lineCountTruncationError(
126+
actualLineCount,
127+
isNewFile,
128+
diffStrategyEnabled,
129+
toolProtocol,
130+
),
123131
),
124132
)
125133
await task.diffViewProvider.revertChanges()

0 commit comments

Comments
 (0)