mirror of
https://github.com/pacnpal/Roo-Code.git
synced 2025-12-21 04:41:16 -05:00
2067 lines
83 KiB
TypeScript
2067 lines
83 KiB
TypeScript
import { Anthropic } from "@anthropic-ai/sdk"
|
|
import cloneDeep from "clone-deep"
|
|
import delay from "delay"
|
|
import fs from "fs/promises"
|
|
import os from "os"
|
|
import pWaitFor from "p-wait-for"
|
|
import * as path from "path"
|
|
import { serializeError } from "serialize-error"
|
|
import * as vscode from "vscode"
|
|
import { ApiHandler, buildApiHandler } from "../api"
|
|
import { ApiStream } from "../api/transform/stream"
|
|
import { DiffViewProvider } from "../integrations/editor/DiffViewProvider"
|
|
import { formatContentBlockToMarkdown } from "../integrations/misc/export-markdown"
|
|
import { extractTextFromFile } from "../integrations/misc/extract-text"
|
|
import { TerminalManager } from "../integrations/terminal/TerminalManager"
|
|
import { UrlContentFetcher } from "../services/browser/UrlContentFetcher"
|
|
import { listFiles } from "../services/glob/list-files"
|
|
import { regexSearchFiles } from "../services/ripgrep"
|
|
import { parseSourceCodeForDefinitionsTopLevel } from "../services/tree-sitter"
|
|
import { ApiConfiguration } from "../shared/api"
|
|
import { findLastIndex } from "../shared/array"
|
|
import { combineApiRequests } from "../shared/combineApiRequests"
|
|
import { combineCommandSequences } from "../shared/combineCommandSequences"
|
|
import {
|
|
ClaudeApiReqCancelReason,
|
|
ClaudeApiReqInfo,
|
|
ClaudeAsk,
|
|
ClaudeMessage,
|
|
ClaudeSay,
|
|
ClaudeSayTool,
|
|
} from "../shared/ExtensionMessage"
|
|
import { getApiMetrics } from "../shared/getApiMetrics"
|
|
import { HistoryItem } from "../shared/HistoryItem"
|
|
import { ToolName } from "../shared/Tool"
|
|
import { ClaudeAskResponse } from "../shared/WebviewMessage"
|
|
import { calculateApiCost } from "../utils/cost"
|
|
import { fileExistsAtPath } from "../utils/fs"
|
|
import { arePathsEqual, getReadablePath } from "../utils/path"
|
|
import { parseMentions } from "./mentions"
|
|
import {
|
|
AssistantMessageContent,
|
|
TextContent,
|
|
ToolParamName,
|
|
toolParamNames,
|
|
ToolUse,
|
|
ToolUseName,
|
|
toolUseNames,
|
|
} from "./prompts/AssistantMessage"
|
|
import { formatResponse } from "./prompts/responses"
|
|
import { addCustomInstructions, SYSTEM_PROMPT } from "./prompts/system"
|
|
import { truncateHalfConversation } from "./sliding-window"
|
|
import { ClaudeDevProvider, GlobalFileNames } from "./webview/ClaudeDevProvider"
|
|
|
|
const cwd =
|
|
vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath).at(0) ?? path.join(os.homedir(), "Desktop") // may or may not exist but fs checking existence would immediately ask for permission which would be bad UX, need to come up with a better solution
|
|
|
|
type ToolResponse = string | Array<Anthropic.TextBlockParam | Anthropic.ImageBlockParam>
|
|
type UserContent = Array<
|
|
Anthropic.TextBlockParam | Anthropic.ImageBlockParam | Anthropic.ToolUseBlockParam | Anthropic.ToolResultBlockParam
|
|
>
|
|
|
|
export class ClaudeDev {
|
|
readonly taskId: string
|
|
api: ApiHandler
|
|
private terminalManager: TerminalManager
|
|
private urlContentFetcher: UrlContentFetcher
|
|
private didEditFile: boolean = false
|
|
customInstructions?: string
|
|
alwaysAllowReadOnly: boolean
|
|
apiConversationHistory: Anthropic.MessageParam[] = []
|
|
claudeMessages: ClaudeMessage[] = []
|
|
private askResponse?: ClaudeAskResponse
|
|
private askResponseText?: string
|
|
private askResponseImages?: string[]
|
|
private lastMessageTs?: number
|
|
private consecutiveMistakeCount: number = 0
|
|
private providerRef: WeakRef<ClaudeDevProvider>
|
|
private abort: boolean = false
|
|
didFinishAborting = false
|
|
private diffViewProvider: DiffViewProvider
|
|
|
|
// streaming
|
|
private currentStreamingContentIndex = 0
|
|
private assistantMessageContent: AssistantMessageContent[] = []
|
|
private presentAssistantMessageLocked = false
|
|
private presentAssistantMessageHasPendingUpdates = false
|
|
private userMessageContent: (Anthropic.TextBlockParam | Anthropic.ImageBlockParam)[] = []
|
|
private userMessageContentReady = false
|
|
private didRejectTool = false
|
|
private didCompleteReadingStream = false
|
|
|
|
constructor(
|
|
provider: ClaudeDevProvider,
|
|
apiConfiguration: ApiConfiguration,
|
|
customInstructions?: string,
|
|
alwaysAllowReadOnly?: boolean,
|
|
task?: string,
|
|
images?: string[],
|
|
historyItem?: HistoryItem
|
|
) {
|
|
this.providerRef = new WeakRef(provider)
|
|
this.api = buildApiHandler(apiConfiguration)
|
|
this.terminalManager = new TerminalManager()
|
|
this.urlContentFetcher = new UrlContentFetcher(provider.context)
|
|
this.diffViewProvider = new DiffViewProvider(cwd)
|
|
this.customInstructions = customInstructions
|
|
this.alwaysAllowReadOnly = alwaysAllowReadOnly ?? false
|
|
|
|
if (historyItem) {
|
|
this.taskId = historyItem.id
|
|
this.resumeTaskFromHistory()
|
|
} else if (task || images) {
|
|
this.taskId = Date.now().toString()
|
|
this.startTask(task, images)
|
|
} else {
|
|
throw new Error("Either historyItem or task/images must be provided")
|
|
}
|
|
}
|
|
|
|
// Storing task to disk for history
|
|
|
|
private async ensureTaskDirectoryExists(): Promise<string> {
|
|
const globalStoragePath = this.providerRef.deref()?.context.globalStorageUri.fsPath
|
|
if (!globalStoragePath) {
|
|
throw new Error("Global storage uri is invalid")
|
|
}
|
|
const taskDir = path.join(globalStoragePath, "tasks", this.taskId)
|
|
await fs.mkdir(taskDir, { recursive: true })
|
|
return taskDir
|
|
}
|
|
|
|
private async getSavedApiConversationHistory(): Promise<Anthropic.MessageParam[]> {
|
|
const filePath = path.join(await this.ensureTaskDirectoryExists(), GlobalFileNames.apiConversationHistory)
|
|
const fileExists = await fileExistsAtPath(filePath)
|
|
if (fileExists) {
|
|
return JSON.parse(await fs.readFile(filePath, "utf8"))
|
|
}
|
|
return []
|
|
}
|
|
|
|
private async addToApiConversationHistory(message: Anthropic.MessageParam) {
|
|
this.apiConversationHistory.push(message)
|
|
await this.saveApiConversationHistory()
|
|
}
|
|
|
|
private async overwriteApiConversationHistory(newHistory: Anthropic.MessageParam[]) {
|
|
this.apiConversationHistory = newHistory
|
|
await this.saveApiConversationHistory()
|
|
}
|
|
|
|
private async saveApiConversationHistory() {
|
|
try {
|
|
const filePath = path.join(await this.ensureTaskDirectoryExists(), GlobalFileNames.apiConversationHistory)
|
|
await fs.writeFile(filePath, JSON.stringify(this.apiConversationHistory))
|
|
} catch (error) {
|
|
// in the off chance this fails, we don't want to stop the task
|
|
console.error("Failed to save API conversation history:", error)
|
|
}
|
|
}
|
|
|
|
private async getSavedClaudeMessages(): Promise<ClaudeMessage[]> {
|
|
const filePath = path.join(await this.ensureTaskDirectoryExists(), GlobalFileNames.claudeMessages)
|
|
const fileExists = await fileExistsAtPath(filePath)
|
|
if (fileExists) {
|
|
return JSON.parse(await fs.readFile(filePath, "utf8"))
|
|
}
|
|
return []
|
|
}
|
|
|
|
private async addToClaudeMessages(message: ClaudeMessage) {
|
|
this.claudeMessages.push(message)
|
|
await this.saveClaudeMessages()
|
|
}
|
|
|
|
private async overwriteClaudeMessages(newMessages: ClaudeMessage[]) {
|
|
this.claudeMessages = newMessages
|
|
await this.saveClaudeMessages()
|
|
}
|
|
|
|
private async saveClaudeMessages() {
|
|
try {
|
|
const filePath = path.join(await this.ensureTaskDirectoryExists(), GlobalFileNames.claudeMessages)
|
|
await fs.writeFile(filePath, JSON.stringify(this.claudeMessages))
|
|
// combined as they are in ChatView
|
|
const apiMetrics = getApiMetrics(combineApiRequests(combineCommandSequences(this.claudeMessages.slice(1))))
|
|
const taskMessage = this.claudeMessages[0] // first message is always the task say
|
|
const lastRelevantMessage =
|
|
this.claudeMessages[
|
|
findLastIndex(
|
|
this.claudeMessages,
|
|
(m) => !(m.ask === "resume_task" || m.ask === "resume_completed_task")
|
|
)
|
|
]
|
|
await this.providerRef.deref()?.updateTaskHistory({
|
|
id: this.taskId,
|
|
ts: lastRelevantMessage.ts,
|
|
task: taskMessage.text ?? "",
|
|
tokensIn: apiMetrics.totalTokensIn,
|
|
tokensOut: apiMetrics.totalTokensOut,
|
|
cacheWrites: apiMetrics.totalCacheWrites,
|
|
cacheReads: apiMetrics.totalCacheReads,
|
|
totalCost: apiMetrics.totalCost,
|
|
})
|
|
} catch (error) {
|
|
console.error("Failed to save claude messages:", error)
|
|
}
|
|
}
|
|
|
|
// Communicate with webview
|
|
|
|
// partial has three valid states true (partial message), false (completion of partial message), undefined (individual complete message)
|
|
async ask(
|
|
type: ClaudeAsk,
|
|
text?: string,
|
|
partial?: boolean
|
|
): Promise<{ response: ClaudeAskResponse; text?: string; images?: string[] }> {
|
|
// If this ClaudeDev instance was aborted by the provider, then the only thing keeping us alive is a promise still running in the background, in which case we don't want to send its result to the webview as it is attached to a new instance of ClaudeDev now. So we can safely ignore the result of any active promises, and this class will be deallocated. (Although we set claudeDev = undefined in provider, that simply removes the reference to this instance, but the instance is still alive until this promise resolves or rejects.)
|
|
if (this.abort) {
|
|
throw new Error("ClaudeDev instance aborted")
|
|
}
|
|
let askTs: number
|
|
if (partial !== undefined) {
|
|
const lastMessage = this.claudeMessages.at(-1)
|
|
const isUpdatingPreviousPartial =
|
|
lastMessage && lastMessage.partial && lastMessage.type === "ask" && lastMessage.ask === type
|
|
if (partial) {
|
|
if (isUpdatingPreviousPartial) {
|
|
// existing partial message, so update it
|
|
lastMessage.text = text
|
|
lastMessage.partial = partial
|
|
// todo be more efficient about saving and posting only new data or one whole message at a time so ignore partial for saves, and only post parts of partial message instead of whole array in new listener
|
|
// await this.saveClaudeMessages()
|
|
// await this.providerRef.deref()?.postStateToWebview()
|
|
await this.providerRef
|
|
.deref()
|
|
?.postMessageToWebview({ type: "partialMessage", partialMessage: lastMessage })
|
|
throw new Error("Current ask promise was ignored 1")
|
|
} else {
|
|
// this is a new partial message, so add it with partial state
|
|
// this.askResponse = undefined
|
|
// this.askResponseText = undefined
|
|
// this.askResponseImages = undefined
|
|
askTs = Date.now()
|
|
this.lastMessageTs = askTs
|
|
await this.addToClaudeMessages({ ts: askTs, type: "ask", ask: type, text, partial })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
throw new Error("Current ask promise was ignored 2")
|
|
}
|
|
} else {
|
|
// partial=false means its a complete version of a previously partial message
|
|
if (isUpdatingPreviousPartial) {
|
|
// this is the complete version of a previously partial message, so replace the partial with the complete version
|
|
this.askResponse = undefined
|
|
this.askResponseText = undefined
|
|
this.askResponseImages = undefined
|
|
|
|
/*
|
|
Bug for the history books:
|
|
In the webview we use the ts as the chatrow key for the virtuoso list. Since we would update this ts right at the end of streaming, it would cause the view to flicker. The key prop has to be stable otherwise react has trouble reconciling items between renders, causing unmounting and remounting of components (flickering).
|
|
The lesson here is if you see flickering when rendering lists, it's likely because the key prop is not stable.
|
|
So in this case we must make sure that the message ts is never altered after first setting it.
|
|
*/
|
|
askTs = lastMessage.ts
|
|
this.lastMessageTs = askTs
|
|
// lastMessage.ts = askTs
|
|
lastMessage.text = text
|
|
lastMessage.partial = false
|
|
await this.saveClaudeMessages()
|
|
// await this.providerRef.deref()?.postStateToWebview()
|
|
await this.providerRef
|
|
.deref()
|
|
?.postMessageToWebview({ type: "partialMessage", partialMessage: lastMessage })
|
|
} else {
|
|
// this is a new partial=false message, so add it like normal
|
|
this.askResponse = undefined
|
|
this.askResponseText = undefined
|
|
this.askResponseImages = undefined
|
|
askTs = Date.now()
|
|
this.lastMessageTs = askTs
|
|
await this.addToClaudeMessages({ ts: askTs, type: "ask", ask: type, text })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
}
|
|
} else {
|
|
// this is a new non-partial message, so add it like normal
|
|
// const lastMessage = this.claudeMessages.at(-1)
|
|
this.askResponse = undefined
|
|
this.askResponseText = undefined
|
|
this.askResponseImages = undefined
|
|
askTs = Date.now()
|
|
this.lastMessageTs = askTs
|
|
await this.addToClaudeMessages({ ts: askTs, type: "ask", ask: type, text })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
|
|
await pWaitFor(() => this.askResponse !== undefined || this.lastMessageTs !== askTs, { interval: 100 })
|
|
if (this.lastMessageTs !== askTs) {
|
|
throw new Error("Current ask promise was ignored") // could happen if we send multiple asks in a row i.e. with command_output. It's important that when we know an ask could fail, it is handled gracefully
|
|
}
|
|
const result = { response: this.askResponse!, text: this.askResponseText, images: this.askResponseImages }
|
|
this.askResponse = undefined
|
|
this.askResponseText = undefined
|
|
this.askResponseImages = undefined
|
|
return result
|
|
}
|
|
|
|
async handleWebviewAskResponse(askResponse: ClaudeAskResponse, text?: string, images?: string[]) {
|
|
this.askResponse = askResponse
|
|
this.askResponseText = text
|
|
this.askResponseImages = images
|
|
}
|
|
|
|
async say(type: ClaudeSay, text?: string, images?: string[], partial?: boolean): Promise<undefined> {
|
|
if (this.abort) {
|
|
throw new Error("ClaudeDev instance aborted")
|
|
}
|
|
|
|
if (partial !== undefined) {
|
|
const lastMessage = this.claudeMessages.at(-1)
|
|
const isUpdatingPreviousPartial =
|
|
lastMessage && lastMessage.partial && lastMessage.type === "say" && lastMessage.say === type
|
|
if (partial) {
|
|
if (isUpdatingPreviousPartial) {
|
|
// existing partial message, so update it
|
|
lastMessage.text = text
|
|
lastMessage.images = images
|
|
lastMessage.partial = partial
|
|
await this.providerRef
|
|
.deref()
|
|
?.postMessageToWebview({ type: "partialMessage", partialMessage: lastMessage })
|
|
} else {
|
|
// this is a new partial message, so add it with partial state
|
|
const sayTs = Date.now()
|
|
this.lastMessageTs = sayTs
|
|
await this.addToClaudeMessages({ ts: sayTs, type: "say", say: type, text, images, partial })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
} else {
|
|
// partial=false means its a complete version of a previously partial message
|
|
if (isUpdatingPreviousPartial) {
|
|
// this is the complete version of a previously partial message, so replace the partial with the complete version
|
|
this.lastMessageTs = lastMessage.ts
|
|
// lastMessage.ts = sayTs
|
|
lastMessage.text = text
|
|
lastMessage.images = images
|
|
lastMessage.partial = false
|
|
|
|
// instead of streaming partialMessage events, we do a save and post like normal to persist to disk
|
|
await this.saveClaudeMessages()
|
|
// await this.providerRef.deref()?.postStateToWebview()
|
|
await this.providerRef
|
|
.deref()
|
|
?.postMessageToWebview({ type: "partialMessage", partialMessage: lastMessage }) // more performant than an entire postStateToWebview
|
|
} else {
|
|
// this is a new partial=false message, so add it like normal
|
|
const sayTs = Date.now()
|
|
this.lastMessageTs = sayTs
|
|
await this.addToClaudeMessages({ ts: sayTs, type: "say", say: type, text, images })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
}
|
|
} else {
|
|
// this is a new non-partial message, so add it like normal
|
|
const sayTs = Date.now()
|
|
this.lastMessageTs = sayTs
|
|
await this.addToClaudeMessages({ ts: sayTs, type: "say", say: type, text, images })
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
}
|
|
|
|
async sayAndCreateMissingParamError(toolName: ToolName, paramName: string, relPath?: string) {
|
|
await this.say(
|
|
"error",
|
|
`Claude tried to use ${toolName}${
|
|
relPath ? ` for '${relPath.toPosix()}'` : ""
|
|
} without value for required parameter '${paramName}'. Retrying...`
|
|
)
|
|
return formatResponse.toolError(formatResponse.missingToolParameterError(paramName))
|
|
}
|
|
|
|
// Task lifecycle
|
|
|
|
private async startTask(task?: string, images?: string[]): Promise<void> {
|
|
// conversationHistory (for API) and claudeMessages (for webview) need to be in sync
|
|
// if the extension process were killed, then on restart the claudeMessages might not be empty, so we need to set it to [] when we create a new ClaudeDev client (otherwise webview would show stale messages from previous session)
|
|
this.claudeMessages = []
|
|
this.apiConversationHistory = []
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
|
|
await this.say("text", task, images)
|
|
|
|
let imageBlocks: Anthropic.ImageBlockParam[] = formatResponse.imageBlocks(images)
|
|
await this.initiateTaskLoop([
|
|
{
|
|
type: "text",
|
|
text: `<task>\n${task}\n</task>`,
|
|
},
|
|
...imageBlocks,
|
|
])
|
|
}
|
|
|
|
private async resumeTaskFromHistory() {
|
|
const modifiedClaudeMessages = await this.getSavedClaudeMessages()
|
|
|
|
// Remove any resume messages that may have been added before
|
|
const lastRelevantMessageIndex = findLastIndex(
|
|
modifiedClaudeMessages,
|
|
(m) => !(m.ask === "resume_task" || m.ask === "resume_completed_task")
|
|
)
|
|
if (lastRelevantMessageIndex !== -1) {
|
|
modifiedClaudeMessages.splice(lastRelevantMessageIndex + 1)
|
|
}
|
|
|
|
// since we don't use api_req_finished anymore, we need to check if the last api_req_started has a cost value, if it doesn't and no cancellation reason to present, then we remove it since it indicates an api request without any partial content streamed
|
|
const lastApiReqStartedIndex = findLastIndex(
|
|
modifiedClaudeMessages,
|
|
(m) => m.type === "say" && m.say === "api_req_started"
|
|
)
|
|
if (lastApiReqStartedIndex !== -1) {
|
|
const lastApiReqStarted = modifiedClaudeMessages[lastApiReqStartedIndex]
|
|
const { cost, cancelReason }: ClaudeApiReqInfo = JSON.parse(lastApiReqStarted.text || "{}")
|
|
if (cost === undefined || cancelReason === undefined) {
|
|
modifiedClaudeMessages.splice(lastApiReqStartedIndex, 1)
|
|
}
|
|
}
|
|
|
|
await this.overwriteClaudeMessages(modifiedClaudeMessages)
|
|
this.claudeMessages = await this.getSavedClaudeMessages()
|
|
|
|
// Now present the claude messages to the user and ask if they want to resume
|
|
|
|
const lastClaudeMessage = this.claudeMessages
|
|
.slice()
|
|
.reverse()
|
|
.find((m) => !(m.ask === "resume_task" || m.ask === "resume_completed_task")) // could be multiple resume tasks
|
|
// const lastClaudeMessage = this.claudeMessages[lastClaudeMessageIndex]
|
|
// could be a completion result with a command
|
|
// const secondLastClaudeMessage = this.claudeMessages
|
|
// .slice()
|
|
// .reverse()
|
|
// .find(
|
|
// (m, index) =>
|
|
// index !== lastClaudeMessageIndex && !(m.ask === "resume_task" || m.ask === "resume_completed_task")
|
|
// )
|
|
// (lastClaudeMessage?.ask === "command" && secondLastClaudeMessage?.ask === "completion_result")
|
|
|
|
let askType: ClaudeAsk
|
|
if (lastClaudeMessage?.ask === "completion_result") {
|
|
askType = "resume_completed_task"
|
|
} else {
|
|
askType = "resume_task"
|
|
}
|
|
|
|
const { response, text, images } = await this.ask(askType) // calls poststatetowebview
|
|
let responseText: string | undefined
|
|
let responseImages: string[] | undefined
|
|
if (response === "messageResponse") {
|
|
await this.say("user_feedback", text, images)
|
|
responseText = text
|
|
responseImages = images
|
|
}
|
|
|
|
// need to make sure that the api conversation history can be resumed by the api, even if it goes out of sync with claude messages
|
|
|
|
// if the last message is an assistant message, we need to check if there's tool use since every tool use has to have a tool response
|
|
// if there's no tool use and only a text block, then we can just add a user message
|
|
// (note this isn't relevant anymore since we use custom tool prompts instead of tool use blocks, but this is here for legacy purposes in case users resume old tasks)
|
|
|
|
// if the last message is a user message, we can need to get the assistant message before it to see if it made tool calls, and if so, fill in the remaining tool responses with 'interrupted'
|
|
|
|
const existingApiConversationHistory: Anthropic.Messages.MessageParam[] =
|
|
await this.getSavedApiConversationHistory()
|
|
|
|
let modifiedOldUserContent: UserContent // either the last message if its user message, or the user message before the last (assistant) message
|
|
let modifiedApiConversationHistory: Anthropic.Messages.MessageParam[] // need to remove the last user message to replace with new modified user message
|
|
if (existingApiConversationHistory.length > 0) {
|
|
const lastMessage = existingApiConversationHistory[existingApiConversationHistory.length - 1]
|
|
|
|
if (lastMessage.role === "assistant") {
|
|
const content = Array.isArray(lastMessage.content)
|
|
? lastMessage.content
|
|
: [{ type: "text", text: lastMessage.content }]
|
|
const hasToolUse = content.some((block) => block.type === "tool_use")
|
|
|
|
if (hasToolUse) {
|
|
const toolUseBlocks = content.filter(
|
|
(block) => block.type === "tool_use"
|
|
) as Anthropic.Messages.ToolUseBlock[]
|
|
const toolResponses: Anthropic.ToolResultBlockParam[] = toolUseBlocks.map((block) => ({
|
|
type: "tool_result",
|
|
tool_use_id: block.id,
|
|
content: "Task was interrupted before this tool call could be completed.",
|
|
}))
|
|
modifiedApiConversationHistory = [...existingApiConversationHistory] // no changes
|
|
modifiedOldUserContent = [...toolResponses]
|
|
} else {
|
|
modifiedApiConversationHistory = [...existingApiConversationHistory]
|
|
modifiedOldUserContent = []
|
|
}
|
|
} else if (lastMessage.role === "user") {
|
|
const previousAssistantMessage: Anthropic.Messages.MessageParam | undefined =
|
|
existingApiConversationHistory[existingApiConversationHistory.length - 2]
|
|
|
|
const existingUserContent: UserContent = Array.isArray(lastMessage.content)
|
|
? lastMessage.content
|
|
: [{ type: "text", text: lastMessage.content }]
|
|
if (previousAssistantMessage && previousAssistantMessage.role === "assistant") {
|
|
const assistantContent = Array.isArray(previousAssistantMessage.content)
|
|
? previousAssistantMessage.content
|
|
: [{ type: "text", text: previousAssistantMessage.content }]
|
|
|
|
const toolUseBlocks = assistantContent.filter(
|
|
(block) => block.type === "tool_use"
|
|
) as Anthropic.Messages.ToolUseBlock[]
|
|
|
|
if (toolUseBlocks.length > 0) {
|
|
const existingToolResults = existingUserContent.filter(
|
|
(block) => block.type === "tool_result"
|
|
) as Anthropic.ToolResultBlockParam[]
|
|
|
|
const missingToolResponses: Anthropic.ToolResultBlockParam[] = toolUseBlocks
|
|
.filter(
|
|
(toolUse) => !existingToolResults.some((result) => result.tool_use_id === toolUse.id)
|
|
)
|
|
.map((toolUse) => ({
|
|
type: "tool_result",
|
|
tool_use_id: toolUse.id,
|
|
content: "Task was interrupted before this tool call could be completed.",
|
|
}))
|
|
|
|
modifiedApiConversationHistory = existingApiConversationHistory.slice(0, -1) // removes the last user message
|
|
modifiedOldUserContent = [...existingUserContent, ...missingToolResponses]
|
|
} else {
|
|
modifiedApiConversationHistory = existingApiConversationHistory.slice(0, -1)
|
|
modifiedOldUserContent = [...existingUserContent]
|
|
}
|
|
} else {
|
|
modifiedApiConversationHistory = existingApiConversationHistory.slice(0, -1)
|
|
modifiedOldUserContent = [...existingUserContent]
|
|
}
|
|
} else {
|
|
throw new Error("Unexpected: Last message is not a user or assistant message")
|
|
}
|
|
} else {
|
|
throw new Error("Unexpected: No existing API conversation history")
|
|
}
|
|
|
|
let newUserContent: UserContent = [...modifiedOldUserContent]
|
|
|
|
const agoText = (() => {
|
|
const timestamp = lastClaudeMessage?.ts ?? Date.now()
|
|
const now = Date.now()
|
|
const diff = now - timestamp
|
|
const minutes = Math.floor(diff / 60000)
|
|
const hours = Math.floor(minutes / 60)
|
|
const days = Math.floor(hours / 24)
|
|
|
|
if (days > 0) {
|
|
return `${days} day${days > 1 ? "s" : ""} ago`
|
|
}
|
|
if (hours > 0) {
|
|
return `${hours} hour${hours > 1 ? "s" : ""} ago`
|
|
}
|
|
if (minutes > 0) {
|
|
return `${minutes} minute${minutes > 1 ? "s" : ""} ago`
|
|
}
|
|
return "just now"
|
|
})()
|
|
|
|
newUserContent.push({
|
|
type: "text",
|
|
text:
|
|
`Task resumption: This autonomous coding task was interrupted ${agoText}. It may or may not be complete, so please reassess the task context. Be aware that the project state may have changed since then. The current working directory is now '${cwd.toPosix()}'. If the task has not been completed, retry the last step before interruption and proceed with completing the task. Note: If you previously attempted a tool use that the user did not provide a result for, you should assume the tool use was not successful and assess whether you should retry.` +
|
|
(responseText
|
|
? `\n\nNew instructions for task continuation:\n<user_message>\n${responseText}\n</user_message>`
|
|
: ""),
|
|
})
|
|
|
|
if (responseImages && responseImages.length > 0) {
|
|
newUserContent.push(...formatResponse.imageBlocks(responseImages))
|
|
}
|
|
|
|
await this.overwriteApiConversationHistory(modifiedApiConversationHistory)
|
|
await this.initiateTaskLoop(newUserContent)
|
|
}
|
|
|
|
private async initiateTaskLoop(userContent: UserContent): Promise<void> {
|
|
let nextUserContent = userContent
|
|
let includeFileDetails = true
|
|
while (!this.abort) {
|
|
const didEndLoop = await this.recursivelyMakeClaudeRequests(nextUserContent, includeFileDetails)
|
|
includeFileDetails = false // we only need file details the first time
|
|
|
|
// The way this agentic loop works is that claude will be given a task that he then calls tools to complete. unless there's an attempt_completion call, we keep responding back to him with his tool's responses until he either attempt_completion or does not use anymore tools. If he does not use anymore tools, we ask him to consider if he's completed the task and then call attempt_completion, otherwise proceed with completing the task.
|
|
// There is a MAX_REQUESTS_PER_TASK limit to prevent infinite requests, but Claude is prompted to finish the task as efficiently as he can.
|
|
|
|
//const totalCost = this.calculateApiCost(totalInputTokens, totalOutputTokens)
|
|
if (didEndLoop) {
|
|
// For now a task never 'completes'. This will only happen if the user hits max requests and denies resetting the count.
|
|
//this.say("task_completed", `Task completed. Total API usage cost: ${totalCost}`)
|
|
break
|
|
} else {
|
|
// this.say(
|
|
// "tool",
|
|
// "Claude responded with only text blocks but has not called attempt_completion yet. Forcing him to continue with task..."
|
|
// )
|
|
nextUserContent = [
|
|
{
|
|
type: "text",
|
|
text: formatResponse.noToolsUsed(),
|
|
},
|
|
]
|
|
this.consecutiveMistakeCount++
|
|
}
|
|
}
|
|
}
|
|
|
|
abortTask() {
|
|
this.abort = true // will stop any autonomously running promises
|
|
this.terminalManager.disposeAll()
|
|
this.urlContentFetcher.closeBrowser()
|
|
}
|
|
|
|
// Tools
|
|
|
|
async executeCommandTool(
|
|
command: string,
|
|
returnEmptyStringOnSuccess: boolean = false
|
|
): Promise<[boolean, ToolResponse]> {
|
|
const terminalInfo = await this.terminalManager.getOrCreateTerminal(cwd)
|
|
terminalInfo.terminal.show() // weird visual bug when creating new terminals (even manually) where there's an empty space at the top.
|
|
const process = this.terminalManager.runCommand(terminalInfo, command)
|
|
|
|
let userFeedback: { text?: string; images?: string[] } | undefined
|
|
let didContinue = false
|
|
const sendCommandOutput = async (line: string): Promise<void> => {
|
|
try {
|
|
const { response, text, images } = await this.ask("command_output", line)
|
|
if (response === "yesButtonTapped") {
|
|
// proceed while running
|
|
} else {
|
|
userFeedback = { text, images }
|
|
}
|
|
didContinue = true
|
|
process.continue() // continue past the await
|
|
} catch {
|
|
// This can only happen if this ask promise was ignored, so ignore this error
|
|
}
|
|
}
|
|
|
|
let result = ""
|
|
process.on("line", (line) => {
|
|
result += line + "\n"
|
|
if (!didContinue) {
|
|
sendCommandOutput(line)
|
|
} else {
|
|
this.say("command_output", line)
|
|
}
|
|
})
|
|
|
|
let completed = false
|
|
process.once("completed", () => {
|
|
completed = true
|
|
})
|
|
|
|
process.once("no_shell_integration", async () => {
|
|
await this.say("shell_integration_warning")
|
|
})
|
|
|
|
await process
|
|
|
|
// Wait for a short delay to ensure all messages are sent to the webview
|
|
// This delay allows time for non-awaited promises to be created and
|
|
// for their associated messages to be sent to the webview, maintaining
|
|
// the correct order of messages (although the webview is smart about
|
|
// grouping command_output messages despite any gaps anyways)
|
|
await delay(50)
|
|
|
|
result = result.trim()
|
|
|
|
if (userFeedback) {
|
|
await this.say("user_feedback", userFeedback.text, userFeedback.images)
|
|
return [
|
|
true,
|
|
formatResponse.toolResult(
|
|
`Command is still running in the user's terminal.${
|
|
result.length > 0 ? `\nHere's the output so far:\n${result}` : ""
|
|
}\n\nThe user provided the following feedback:\n<feedback>\n${userFeedback.text}\n</feedback>`,
|
|
userFeedback.images
|
|
),
|
|
]
|
|
}
|
|
|
|
// for attemptCompletion, we don't want to return the command output
|
|
if (returnEmptyStringOnSuccess) {
|
|
return [false, ""]
|
|
}
|
|
if (completed) {
|
|
return [false, `Command executed.${result.length > 0 ? `\nOutput:\n${result}` : ""}`]
|
|
} else {
|
|
return [
|
|
false,
|
|
`Command is still running in the user's terminal.${
|
|
result.length > 0 ? `\nHere's the output so far:\n${result}` : ""
|
|
}\n\nYou will be updated on the terminal status and new output in the future.`,
|
|
]
|
|
}
|
|
}
|
|
|
|
async *attemptApiRequest(previousApiReqIndex: number): ApiStream {
|
|
try {
|
|
let systemPrompt = await SYSTEM_PROMPT(cwd, this.api.getModel().info.supportsImages ?? false)
|
|
if (this.customInstructions && this.customInstructions.trim()) {
|
|
// altering the system prompt mid-task will break the prompt cache, but in the grand scheme this will not change often so it's better to not pollute user messages with it the way we have to with <potentially relevant details>
|
|
systemPrompt += addCustomInstructions(this.customInstructions)
|
|
}
|
|
|
|
// If the previous API request's total token usage is close to the context window, truncate the conversation history to free up space for the new request
|
|
if (previousApiReqIndex >= 0) {
|
|
const previousRequest = this.claudeMessages[previousApiReqIndex]
|
|
if (previousRequest && previousRequest.text) {
|
|
const { tokensIn, tokensOut, cacheWrites, cacheReads }: ClaudeApiReqInfo = JSON.parse(
|
|
previousRequest.text
|
|
)
|
|
const totalTokens = (tokensIn || 0) + (tokensOut || 0) + (cacheWrites || 0) + (cacheReads || 0)
|
|
const contextWindow = this.api.getModel().info.contextWindow || 128_000
|
|
const maxAllowedSize = Math.max(contextWindow - 40_000, contextWindow * 0.8)
|
|
if (totalTokens >= maxAllowedSize) {
|
|
const truncatedMessages = truncateHalfConversation(this.apiConversationHistory)
|
|
await this.overwriteApiConversationHistory(truncatedMessages)
|
|
}
|
|
}
|
|
}
|
|
|
|
const stream = this.api.createMessage(systemPrompt, this.apiConversationHistory)
|
|
const iterator = stream[Symbol.asyncIterator]()
|
|
// awaiting first chunk to see if it will throw an error
|
|
const firstChunk = await iterator.next()
|
|
yield firstChunk.value
|
|
// no error, so we can continue to yield all remaining chunks
|
|
// this delegates to another generator or iterable object. In this case, it's saying "yield all remaining values from this iterator". This effectively passes along all subsequent chunks from the original stream.
|
|
yield* iterator
|
|
} catch (error) {
|
|
// note that this api_req_failed ask is unique in that we only present this option if the api hasn't streamed any content yet (ie it fails on the first chunk due), as it would allow them to hit a retry button. However if the api failed mid-stream, it could be in any arbitrary state where some tools may have executed, so that error is handled differently and requires cancelling the task entirely.
|
|
const { response } = await this.ask(
|
|
"api_req_failed",
|
|
error.message ?? JSON.stringify(serializeError(error), null, 2)
|
|
)
|
|
if (response !== "yesButtonTapped") {
|
|
// this will never happen since if noButtonTapped, we will clear current task, aborting this instance
|
|
throw new Error("API request failed")
|
|
}
|
|
await this.say("api_req_retried")
|
|
// delegate generator output from the recursive call
|
|
yield* this.attemptApiRequest(previousApiReqIndex)
|
|
}
|
|
}
|
|
|
|
async presentAssistantMessage() {
|
|
if (this.abort) {
|
|
throw new Error("ClaudeDev instance aborted")
|
|
}
|
|
|
|
if (this.presentAssistantMessageLocked) {
|
|
this.presentAssistantMessageHasPendingUpdates = true
|
|
return
|
|
}
|
|
this.presentAssistantMessageLocked = true
|
|
this.presentAssistantMessageHasPendingUpdates = false
|
|
|
|
if (this.currentStreamingContentIndex >= this.assistantMessageContent.length) {
|
|
// this may happen if the last content block was completed before streaming could finish. if streaming is finished, and we're out of bounds then this means we already presented/executed the last content block and are ready to continue to next request
|
|
if (this.didCompleteReadingStream) {
|
|
this.userMessageContentReady = true
|
|
}
|
|
// console.log("no more content blocks to stream! this shouldn't happen?")
|
|
this.presentAssistantMessageLocked = false
|
|
return
|
|
//throw new Error("No more content blocks to stream! This shouldn't happen...") // remove and just return after testing
|
|
}
|
|
|
|
const block = cloneDeep(this.assistantMessageContent[this.currentStreamingContentIndex]) // need to create copy bc while stream is updating the array, it could be updating the reference block properties too
|
|
switch (block.type) {
|
|
case "text": {
|
|
let content = block.content
|
|
if (block.partial && content) {
|
|
// Remove partial XML tag at the very end of the content
|
|
const lastOpenBracketIndex = content.lastIndexOf("<")
|
|
if (lastOpenBracketIndex !== -1) {
|
|
const possibleTag = content.slice(lastOpenBracketIndex)
|
|
// Check if there's a '>' after the last '<' (i.e., if the tag is complete)
|
|
const hasCloseBracket = possibleTag.includes(">")
|
|
if (!hasCloseBracket) {
|
|
// Extract the potential tag name
|
|
const tagContent = possibleTag.slice(1).trim()
|
|
// Check if tagContent is likely an incomplete tag name (letters and underscores only)
|
|
const isLikelyTagName = /^[a-zA-Z_]+$/.test(tagContent)
|
|
// If the tag is incomplete and at the end, remove it from the content
|
|
if (isLikelyTagName) {
|
|
content = content.slice(0, lastOpenBracketIndex).trim()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
await this.say("text", content, undefined, block.partial)
|
|
break
|
|
}
|
|
case "tool_use":
|
|
const toolDescription = () => {
|
|
switch (block.name) {
|
|
case "execute_command":
|
|
return `[${block.name} for '${block.params.command}']`
|
|
case "read_file":
|
|
return `[${block.name} for '${block.params.path}']`
|
|
case "write_to_file":
|
|
return `[${block.name} for '${block.params.path}']`
|
|
case "search_files":
|
|
return `[${block.name} for '${block.params.regex}'${
|
|
block.params.file_pattern ? ` in '${block.params.file_pattern}'` : ""
|
|
}]`
|
|
case "list_files":
|
|
return `[${block.name} for '${block.params.path}']`
|
|
case "list_code_definition_names":
|
|
return `[${block.name} for '${block.params.path}']`
|
|
case "inspect_site":
|
|
return `[${block.name} for '${block.params.url}']`
|
|
case "ask_followup_question":
|
|
return `[${block.name} for '${block.params.question}']`
|
|
case "attempt_completion":
|
|
return `[${block.name}]`
|
|
}
|
|
}
|
|
|
|
if (this.didRejectTool) {
|
|
// ignore any tool content after user has rejected tool once
|
|
// we'll fill it in with a rejection message when the message is complete
|
|
if (!block.partial) {
|
|
this.userMessageContent.push({
|
|
type: "text",
|
|
text: `Skipping tool ${toolDescription()} due to user rejecting a previous tool.`,
|
|
})
|
|
}
|
|
break
|
|
}
|
|
|
|
const pushToolResult = (content: ToolResponse) => {
|
|
this.userMessageContent.push({
|
|
type: "text",
|
|
text: `${toolDescription()} Result:`,
|
|
})
|
|
if (typeof content === "string") {
|
|
this.userMessageContent.push({
|
|
type: "text",
|
|
text: content || "(tool did not return anything)",
|
|
})
|
|
} else {
|
|
this.userMessageContent.push(...content)
|
|
}
|
|
}
|
|
|
|
const askApproval = async (type: ClaudeAsk, partialMessage?: string) => {
|
|
const { response, text, images } = await this.ask(type, partialMessage, false)
|
|
if (response !== "yesButtonTapped") {
|
|
if (response === "messageResponse") {
|
|
await this.say("user_feedback", text, images)
|
|
pushToolResult(
|
|
formatResponse.toolResult(formatResponse.toolDeniedWithFeedback(text), images)
|
|
)
|
|
// this.userMessageContent.push({
|
|
// type: "text",
|
|
// text: `${toolDescription()}`,
|
|
// })
|
|
// this.toolResults.push({
|
|
// type: "tool_result",
|
|
// tool_use_id: toolUseId,
|
|
// content: this.formatToolResponseWithImages(
|
|
// await this.formatToolDeniedFeedback(text),
|
|
// images
|
|
// ),
|
|
// })
|
|
this.didRejectTool = true
|
|
return false
|
|
}
|
|
pushToolResult(formatResponse.toolDenied())
|
|
// this.toolResults.push({
|
|
// type: "tool_result",
|
|
// tool_use_id: toolUseId,
|
|
// content: await this.formatToolDenied(),
|
|
// })
|
|
this.didRejectTool = true
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
const handleError = async (action: string, error: Error) => {
|
|
const errorString = `Error ${action}: ${JSON.stringify(serializeError(error))}`
|
|
await this.say(
|
|
"error",
|
|
`Error ${action}:\n${error.message ?? JSON.stringify(serializeError(error), null, 2)}`
|
|
)
|
|
// this.toolResults.push({
|
|
// type: "tool_result",
|
|
// tool_use_id: toolUseId,
|
|
// content: await this.formatToolError(errorString),
|
|
// })
|
|
pushToolResult(formatResponse.toolError(errorString))
|
|
}
|
|
|
|
// If block is partial, remove partial closing tag so its not presented to user
|
|
const removeClosingTag = (tag: ToolParamName, text?: string) => {
|
|
if (!block.partial) {
|
|
return text || ""
|
|
}
|
|
if (!text) {
|
|
return ""
|
|
}
|
|
// This regex dynamically constructs a pattern to match the closing tag:
|
|
// - Optionally matches whitespace before the tag
|
|
// - Matches '<' or '</' optionally followed by any subset of characters from the tag name
|
|
const tagRegex = new RegExp(
|
|
`\\s?<\/?${tag
|
|
.split("")
|
|
.map((char) => `(?:${char})?`)
|
|
.join("")}$`,
|
|
"g"
|
|
)
|
|
return text.replace(tagRegex, "")
|
|
}
|
|
|
|
switch (block.name) {
|
|
case "write_to_file": {
|
|
const relPath: string | undefined = block.params.path
|
|
let newContent: string | undefined = block.params.content
|
|
if (!relPath || !newContent) {
|
|
// checking for newContent ensure relPath is complete
|
|
// wait so we can determine if it's a new file or editing an existing file
|
|
break
|
|
}
|
|
// Check if file exists using cached map or fs.access
|
|
let fileExists: boolean
|
|
if (this.diffViewProvider.editType !== undefined) {
|
|
fileExists = this.diffViewProvider.editType === "modify"
|
|
} else {
|
|
const absolutePath = path.resolve(cwd, relPath)
|
|
fileExists = await fileExistsAtPath(absolutePath)
|
|
this.diffViewProvider.editType = fileExists ? "modify" : "create"
|
|
}
|
|
|
|
// pre-processing newContent for cases where weaker models might add artifacts like markdown codeblock markers (deepseek/llama) or extra escape characters (gemini)
|
|
if (newContent.startsWith("```")) {
|
|
// this handles cases where it includes language specifiers like ```python ```js
|
|
newContent = newContent.split("\n").slice(1).join("\n").trim()
|
|
}
|
|
if (newContent.endsWith("```")) {
|
|
newContent = newContent.split("\n").slice(0, -1).join("\n").trim()
|
|
}
|
|
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: fileExists ? "editedExistingFile" : "newFileCreated",
|
|
path: getReadablePath(cwd, removeClosingTag("path", relPath)),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
// update gui message
|
|
const partialMessage = JSON.stringify(sharedMessageProps)
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
// update editor
|
|
if (!this.diffViewProvider.isEditing) {
|
|
// open the editor and prepare to stream content in
|
|
await this.diffViewProvider.open(relPath)
|
|
}
|
|
// editor is open, stream content in
|
|
await this.diffViewProvider.update(newContent, false)
|
|
break
|
|
} else {
|
|
if (!relPath) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("write_to_file", "path"))
|
|
await this.diffViewProvider.reset()
|
|
break
|
|
}
|
|
if (!newContent) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("write_to_file", "content"))
|
|
await this.diffViewProvider.reset()
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
|
|
// if isEditingFile false, that means we have the full contents of the file already.
|
|
// it's important to note how this function works, you can't make the assumption that the block.partial conditional will always be called since it may immediately get complete, non-partial data. So this part of the logic will always be called.
|
|
// in other words, you must always repeat the block.partial logic here
|
|
if (!this.diffViewProvider.isEditing) {
|
|
await this.diffViewProvider.open(relPath)
|
|
}
|
|
await this.diffViewProvider.update(newContent, true)
|
|
await delay(300) // wait for diff view to update
|
|
this.diffViewProvider.scrollToFirstDiff()
|
|
|
|
const completeMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: fileExists ? undefined : newContent,
|
|
diff: fileExists
|
|
? formatResponse.createPrettyPatch(
|
|
relPath,
|
|
this.diffViewProvider.originalContent,
|
|
newContent
|
|
)
|
|
: undefined,
|
|
} satisfies ClaudeSayTool)
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
await this.diffViewProvider.revertChanges()
|
|
break
|
|
}
|
|
const { newProblemsMessage, userEdits } = await this.diffViewProvider.saveChanges()
|
|
this.didEditFile = true // used to determine if we should wait for busy terminal to update before sending api request
|
|
if (userEdits) {
|
|
await this.say(
|
|
"user_feedback_diff",
|
|
JSON.stringify({
|
|
tool: fileExists ? "editedExistingFile" : "newFileCreated",
|
|
path: getReadablePath(cwd, relPath),
|
|
diff: userEdits,
|
|
} satisfies ClaudeSayTool)
|
|
)
|
|
pushToolResult(
|
|
`The user made the following updates to your content:\n\n${userEdits}\n\nThe updated content, which includes both your original modifications and the user's additional edits, has been successfully saved to ${relPath.toPosix()}. (Note this does not mean you need to re-write the file with the user's changes, as they have already been applied to the file.)${newProblemsMessage}`
|
|
)
|
|
} else {
|
|
pushToolResult(
|
|
`The content was successfully saved to ${relPath.toPosix()}.${newProblemsMessage}`
|
|
)
|
|
}
|
|
await this.diffViewProvider.reset()
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("writing file", error)
|
|
await this.diffViewProvider.reset()
|
|
break
|
|
}
|
|
}
|
|
case "read_file": {
|
|
const relPath: string | undefined = block.params.path
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: "readFile",
|
|
path: getReadablePath(cwd, removeClosingTag("path", relPath)),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
const partialMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: undefined,
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", partialMessage, undefined, block.partial)
|
|
} else {
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
}
|
|
break
|
|
} else {
|
|
if (!relPath) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("read_file", "path"))
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const absolutePath = path.resolve(cwd, relPath)
|
|
const completeMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: absolutePath,
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", completeMessage, undefined, false) // need to be sending partialValue bool, since undefined has its own purpose in that the message is treated neither as a partial or completion of a partial, but as a single complete message
|
|
} else {
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
}
|
|
// now execute the tool like normal
|
|
const content = await extractTextFromFile(absolutePath)
|
|
pushToolResult(content)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("reading file", error)
|
|
break
|
|
}
|
|
}
|
|
case "list_files": {
|
|
const relDirPath: string | undefined = block.params.path
|
|
const recursiveRaw: string | undefined = block.params.recursive
|
|
const recursive = recursiveRaw?.toLowerCase() === "true"
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: !recursive ? "listFilesTopLevel" : "listFilesRecursive",
|
|
path: getReadablePath(cwd, removeClosingTag("path", relDirPath)),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
const partialMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: "",
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", partialMessage, undefined, block.partial)
|
|
} else {
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
}
|
|
break
|
|
} else {
|
|
if (!relDirPath) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("list_files", "path"))
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const absolutePath = path.resolve(cwd, relDirPath)
|
|
const [files, didHitLimit] = await listFiles(absolutePath, recursive, 200)
|
|
const result = formatResponse.formatFilesList(absolutePath, files, didHitLimit)
|
|
const completeMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: result,
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", completeMessage, undefined, false)
|
|
} else {
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
}
|
|
pushToolResult(result)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("listing files", error)
|
|
break
|
|
}
|
|
}
|
|
case "list_code_definition_names": {
|
|
const relDirPath: string | undefined = block.params.path
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: "listCodeDefinitionNames",
|
|
path: getReadablePath(cwd, removeClosingTag("path", relDirPath)),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
const partialMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: "",
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", partialMessage, undefined, block.partial)
|
|
} else {
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
}
|
|
break
|
|
} else {
|
|
if (!relDirPath) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(
|
|
await this.sayAndCreateMissingParamError("list_code_definition_names", "path")
|
|
)
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const absolutePath = path.resolve(cwd, relDirPath)
|
|
const result = await parseSourceCodeForDefinitionsTopLevel(absolutePath)
|
|
const completeMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: result,
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", completeMessage, undefined, false)
|
|
} else {
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
}
|
|
pushToolResult(result)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("parsing source code definitions", error)
|
|
break
|
|
}
|
|
}
|
|
case "search_files": {
|
|
const relDirPath: string | undefined = block.params.path
|
|
const regex: string | undefined = block.params.regex
|
|
const filePattern: string | undefined = block.params.file_pattern
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: "searchFiles",
|
|
path: getReadablePath(cwd, removeClosingTag("path", relDirPath)),
|
|
regex: removeClosingTag("regex", regex),
|
|
filePattern: removeClosingTag("file_pattern", filePattern),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
const partialMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: "",
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", partialMessage, undefined, block.partial)
|
|
} else {
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
}
|
|
break
|
|
} else {
|
|
if (!relDirPath) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("search_files", "path"))
|
|
break
|
|
}
|
|
if (!regex) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("search_files", "regex"))
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const absolutePath = path.resolve(cwd, relDirPath)
|
|
const results = await regexSearchFiles(cwd, absolutePath, regex, filePattern)
|
|
const completeMessage = JSON.stringify({
|
|
...sharedMessageProps,
|
|
content: results,
|
|
} satisfies ClaudeSayTool)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", completeMessage, undefined, false)
|
|
} else {
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
}
|
|
pushToolResult(results)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("searching files", error)
|
|
break
|
|
}
|
|
}
|
|
case "inspect_site": {
|
|
const url: string | undefined = block.params.url
|
|
const sharedMessageProps: ClaudeSayTool = {
|
|
tool: "inspectSite",
|
|
path: removeClosingTag("url", url),
|
|
}
|
|
try {
|
|
if (block.partial) {
|
|
const partialMessage = JSON.stringify(sharedMessageProps)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", partialMessage, undefined, block.partial)
|
|
} else {
|
|
await this.ask("tool", partialMessage, block.partial).catch(() => {})
|
|
}
|
|
break
|
|
} else {
|
|
if (!url) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(await this.sayAndCreateMissingParamError("inspect_site", "url"))
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const completeMessage = JSON.stringify(sharedMessageProps)
|
|
if (this.alwaysAllowReadOnly) {
|
|
await this.say("tool", completeMessage, undefined, false)
|
|
} else {
|
|
const didApprove = await askApproval("tool", completeMessage)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
}
|
|
|
|
// execute tool
|
|
// NOTE: it's okay that we call this message since the partial inspect_site is finished streaming. The only scenario we have to avoid is sending messages WHILE a partial message exists at the end of the messages array. For example the api_req_finished message would interfere with the partial message, so we needed to remove that.
|
|
await this.say("inspect_site_result", "") // no result, starts the loading spinner waiting for result
|
|
await this.urlContentFetcher.launchBrowser()
|
|
let result: {
|
|
screenshot: string
|
|
logs: string
|
|
}
|
|
try {
|
|
result = await this.urlContentFetcher.urlToScreenshotAndLogs(url)
|
|
} finally {
|
|
await this.urlContentFetcher.closeBrowser()
|
|
}
|
|
const { screenshot, logs } = result
|
|
await this.say("inspect_site_result", logs, [screenshot])
|
|
|
|
pushToolResult(
|
|
formatResponse.toolResult(
|
|
`The site has been visited, with console logs captured and a screenshot taken for your analysis.\n\nConsole logs:\n${
|
|
logs || "(No logs)"
|
|
}`,
|
|
[screenshot]
|
|
)
|
|
)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("inspecting site", error)
|
|
break
|
|
}
|
|
}
|
|
case "execute_command": {
|
|
const command: string | undefined = block.params.command
|
|
try {
|
|
if (block.partial) {
|
|
await this.ask("command", removeClosingTag("command", command), block.partial).catch(
|
|
() => {}
|
|
)
|
|
break
|
|
} else {
|
|
if (!command) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(
|
|
await this.sayAndCreateMissingParamError("execute_command", "command")
|
|
)
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const didApprove = await askApproval("command", command)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
const [userRejected, result] = await this.executeCommandTool(command)
|
|
if (userRejected) {
|
|
this.didRejectTool = true // test whats going on here
|
|
}
|
|
pushToolResult(result)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("inspecting site", error)
|
|
break
|
|
}
|
|
}
|
|
|
|
case "ask_followup_question": {
|
|
const question: string | undefined = block.params.question
|
|
try {
|
|
if (block.partial) {
|
|
await this.ask("followup", removeClosingTag("question", question), block.partial).catch(
|
|
() => {}
|
|
)
|
|
break
|
|
} else {
|
|
if (!question) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(
|
|
await this.sayAndCreateMissingParamError("ask_followup_question", "question")
|
|
)
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
const { text, images } = await this.ask("followup", question, false)
|
|
await this.say("user_feedback", text ?? "", images)
|
|
pushToolResult(formatResponse.toolResult(`<answer>\n${text}\n</answer>`, images))
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("asking question", error)
|
|
break
|
|
}
|
|
}
|
|
case "attempt_completion": {
|
|
/*
|
|
this.consecutiveMistakeCount = 0
|
|
let resultToSend = result
|
|
if (command) {
|
|
await this.say("completion_result", resultToSend)
|
|
// TODO: currently we don't handle if this command fails, it could be useful to let claude know and retry
|
|
const [didUserReject, commandResult] = await this.executeCommand(command, true)
|
|
// if we received non-empty string, the command was rejected or failed
|
|
if (commandResult) {
|
|
return [didUserReject, commandResult]
|
|
}
|
|
resultToSend = ""
|
|
}
|
|
const { response, text, images } = await this.ask("completion_result", resultToSend) // this prompts webview to show 'new task' button, and enable text input (which would be the 'text' here)
|
|
if (response === "yesButtonTapped") {
|
|
return [false, ""] // signals to recursive loop to stop (for now this never happens since yesButtonTapped will trigger a new task)
|
|
}
|
|
await this.say("user_feedback", text ?? "", images)
|
|
return [
|
|
*/
|
|
const result: string | undefined = block.params.result
|
|
const command: string | undefined = block.params.command
|
|
try {
|
|
const lastMessage = this.claudeMessages.at(-1)
|
|
if (block.partial) {
|
|
if (command) {
|
|
// the attempt_completion text is done, now we're getting command
|
|
// remove the previous partial attempt_completion ask, replace with say, post state to webview, then stream command
|
|
|
|
// const secondLastMessage = this.claudeMessages.at(-2)
|
|
if (lastMessage && lastMessage.ask === "command") {
|
|
// update command
|
|
await this.ask(
|
|
"command",
|
|
removeClosingTag("command", command),
|
|
block.partial
|
|
).catch(() => {})
|
|
} else {
|
|
// last message is completion_result
|
|
// we have command string, which means we have the result as well, so finish it (doesnt have to exist yet)
|
|
await this.say(
|
|
"completion_result",
|
|
removeClosingTag("result", result),
|
|
undefined,
|
|
false
|
|
)
|
|
await this.ask(
|
|
"command",
|
|
removeClosingTag("command", command),
|
|
block.partial
|
|
).catch(() => {})
|
|
}
|
|
} else {
|
|
// no command, still outputting partial result
|
|
await this.say(
|
|
"completion_result",
|
|
removeClosingTag("result", result),
|
|
undefined,
|
|
block.partial
|
|
)
|
|
}
|
|
break
|
|
} else {
|
|
if (!result) {
|
|
this.consecutiveMistakeCount++
|
|
pushToolResult(
|
|
await this.sayAndCreateMissingParamError("attempt_completion", "result")
|
|
)
|
|
break
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
|
|
if (command) {
|
|
if (lastMessage && lastMessage.ask !== "command") {
|
|
// havent sent a command message yet so first send completion_result then command
|
|
await this.say("completion_result", result, undefined, false)
|
|
}
|
|
|
|
// complete command message
|
|
const didApprove = await askApproval("command", command)
|
|
if (!didApprove) {
|
|
break
|
|
}
|
|
const [userRejected, commandResult] = await this.executeCommandTool(command!, true)
|
|
if (commandResult) {
|
|
if (userRejected) {
|
|
this.didRejectTool = true // test whats going on here
|
|
}
|
|
pushToolResult(commandResult)
|
|
break
|
|
}
|
|
} else {
|
|
await this.say("completion_result", result, undefined, false)
|
|
}
|
|
|
|
// we already sent completion_result says, an empty string asks relinquishes control over button and field
|
|
const { response, text, images } = await this.ask("completion_result", "", false)
|
|
if (response === "yesButtonTapped") {
|
|
pushToolResult("") // signals to recursive loop to stop (for now this never happens since yesButtonTapped will trigger a new task)
|
|
break
|
|
}
|
|
await this.say("user_feedback", text ?? "", images)
|
|
pushToolResult(
|
|
formatResponse.toolResult(
|
|
`The user has provided feedback on the results. Consider their input to continue the task, and then attempt completion again.\n<feedback>\n${text}\n</feedback>`,
|
|
images
|
|
)
|
|
)
|
|
break
|
|
}
|
|
} catch (error) {
|
|
await handleError("inspecting site", error)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
break
|
|
}
|
|
|
|
/*
|
|
Seeing out of bounds is fine, it means that the next too call is being built up and ready to add to assistantMessageContent to present.
|
|
When you see the UI inactive during this, it means that a tool is breaking without presenting any UI. For example the write_to_file tool was breaking when relpath was undefined, and for invalid relpath it never presented UI.
|
|
*/
|
|
this.presentAssistantMessageLocked = false // this needs to be placed here, if not then calling this.presentAssistantMessage below would fail (sometimes) since it's locked
|
|
if (!block.partial) {
|
|
// block is finished streaming and executing
|
|
if (this.currentStreamingContentIndex === this.assistantMessageContent.length - 1) {
|
|
// its okay that we increment if !didCompleteReadingStream, it'll just return bc out of bounds and as streaming continues it will call presentAssitantMessage if a new block is ready. if streaming is finished then we set userMessageContentReady to true when out of bounds. This gracefully allows the stream to continue on and all potential content blocks be presented.
|
|
// last block is complete and it is finished executing
|
|
this.userMessageContentReady = true // will allow pwaitfor to continue
|
|
}
|
|
|
|
// call next block if it exists (if not then read stream will call it when its ready)
|
|
this.currentStreamingContentIndex++ // need to increment regardless, so when read stream calls this function again it will be streaming the next block
|
|
|
|
if (this.currentStreamingContentIndex < this.assistantMessageContent.length) {
|
|
// there are already more content blocks to stream, so we'll call this function ourselves
|
|
// await this.presentAssistantContent()
|
|
|
|
this.presentAssistantMessage()
|
|
return
|
|
}
|
|
}
|
|
// block is partial, but the read stream may have finished
|
|
if (this.presentAssistantMessageHasPendingUpdates) {
|
|
this.presentAssistantMessage()
|
|
}
|
|
}
|
|
|
|
parseAssistantMessage(assistantMessage: string) {
|
|
let textContent: TextContent = {
|
|
type: "text",
|
|
content: "",
|
|
partial: true,
|
|
}
|
|
let toolUses: ToolUse[] = []
|
|
let currentToolUse: ToolUse | undefined = undefined
|
|
let currentToolUseStartIndex = 0
|
|
let currentParamName: ToolParamName | undefined = undefined
|
|
let currentParamValueStartIndex = 0
|
|
let accumulator = ""
|
|
|
|
for (let i = 0; i < assistantMessage.length; i++) {
|
|
const char = assistantMessage[i]
|
|
accumulator += char
|
|
|
|
// there should not be a param without a tool use
|
|
if (currentToolUse && currentParamName) {
|
|
const currentParamValue = accumulator.slice(currentParamValueStartIndex)
|
|
const paramClosingTag = `</${currentParamName}>`
|
|
if (currentParamValue.endsWith(paramClosingTag)) {
|
|
// end of param value
|
|
currentToolUse.params[currentParamName] = currentParamValue.slice(0, -paramClosingTag.length).trim()
|
|
currentParamName = undefined
|
|
continue
|
|
} else {
|
|
// partial param value is accumulating
|
|
continue
|
|
}
|
|
}
|
|
|
|
// no currentParamName
|
|
|
|
if (currentToolUse) {
|
|
const currentToolValue = accumulator.slice(currentToolUseStartIndex)
|
|
const toolUseClosingTag = `</${currentToolUse.name}>`
|
|
if (currentToolValue.endsWith(toolUseClosingTag)) {
|
|
// end of a tool use
|
|
currentToolUse.partial = false
|
|
toolUses.push(currentToolUse)
|
|
currentToolUse = undefined
|
|
continue
|
|
} else {
|
|
const possibleParamOpeningTags = toolParamNames.map((name) => `<${name}>`)
|
|
for (const paramOpeningTag of possibleParamOpeningTags) {
|
|
if (accumulator.endsWith(paramOpeningTag)) {
|
|
// start of a new parameter
|
|
currentParamName = paramOpeningTag.slice(1, -1) as ToolParamName
|
|
currentParamValueStartIndex = accumulator.length
|
|
break
|
|
}
|
|
}
|
|
|
|
// there's no current param, and not starting a new param
|
|
|
|
// special case for write_to_file where file contents could contain the closing tag, in which case the param would have closed and we end up with the rest of the file contents here. To work around this, we get the string between the starting content tag and the LAST content tag.
|
|
const contentParamName: ToolParamName = "content"
|
|
if (currentToolUse.name === "write_to_file" && accumulator.endsWith(`</${contentParamName}>`)) {
|
|
const toolContent = accumulator.slice(currentToolUseStartIndex)
|
|
const contentStartTag = `<${contentParamName}>`
|
|
const contentEndTag = `</${contentParamName}>`
|
|
const contentStartIndex = toolContent.indexOf(contentStartTag) + contentStartTag.length
|
|
const contentEndIndex = toolContent.lastIndexOf(contentEndTag)
|
|
if (contentStartIndex !== -1 && contentEndIndex !== -1 && contentEndIndex > contentStartIndex) {
|
|
currentToolUse.params[contentParamName] = toolContent
|
|
.slice(contentStartIndex, contentEndIndex)
|
|
.trim()
|
|
}
|
|
}
|
|
|
|
// partial tool value is accumulating
|
|
continue
|
|
}
|
|
}
|
|
|
|
// no currentToolUse
|
|
|
|
const possibleToolUseOpeningTags = toolUseNames.map((name) => `<${name}>`)
|
|
for (const toolUseOpeningTag of possibleToolUseOpeningTags) {
|
|
if (accumulator.endsWith(toolUseOpeningTag)) {
|
|
// start of a new tool use
|
|
currentToolUse = {
|
|
type: "tool_use",
|
|
name: toolUseOpeningTag.slice(1, -1) as ToolUseName,
|
|
params: {},
|
|
partial: true,
|
|
}
|
|
currentToolUseStartIndex = accumulator.length
|
|
// this also indicates the end of the text content
|
|
textContent.partial = false
|
|
// remove the partially accumulated tool use tag from the end of text (<tool)
|
|
textContent.content = textContent.content.slice(0, -toolUseOpeningTag.slice(0, -1).length).trim()
|
|
break
|
|
}
|
|
}
|
|
|
|
// only add text content if we haven't started a tool yet
|
|
if (textContent.partial) {
|
|
textContent.content = accumulator.trim()
|
|
}
|
|
}
|
|
|
|
if (currentToolUse) {
|
|
// stream did not complete tool call, add it as partial
|
|
if (currentParamName) {
|
|
// tool call has a parameter that was not completed
|
|
currentToolUse.params[currentParamName] = accumulator.slice(currentParamValueStartIndex).trim()
|
|
}
|
|
toolUses.push(currentToolUse)
|
|
}
|
|
|
|
const prevLength = this.assistantMessageContent.length
|
|
this.assistantMessageContent = [textContent, ...toolUses]
|
|
if (this.assistantMessageContent.length > prevLength) {
|
|
this.userMessageContentReady = false // new content we need to present, reset to false in case previous content set this to true
|
|
}
|
|
}
|
|
|
|
async recursivelyMakeClaudeRequests(
|
|
userContent: UserContent,
|
|
includeFileDetails: boolean = false
|
|
): Promise<boolean> {
|
|
if (this.abort) {
|
|
throw new Error("ClaudeDev instance aborted")
|
|
}
|
|
|
|
if (this.consecutiveMistakeCount >= 3) {
|
|
const { response, text, images } = await this.ask(
|
|
"mistake_limit_reached",
|
|
this.api.getModel().id.includes("claude")
|
|
? `This may indicate a failure in his thought process or inability to use a tool properly, which can be mitigated with some user guidance (e.g. "Try breaking down the task into smaller steps").`
|
|
: "Claude Dev uses complex prompts and iterative task execution that may be challenging for less capable models. For best results, it's recommended to use Claude 3.5 Sonnet for its advanced agentic coding capabilities."
|
|
)
|
|
if (response === "messageResponse") {
|
|
userContent.push(
|
|
...[
|
|
{
|
|
type: "text",
|
|
text: formatResponse.tooManyMistakes(text),
|
|
} as Anthropic.Messages.TextBlockParam,
|
|
...formatResponse.imageBlocks(images),
|
|
]
|
|
)
|
|
}
|
|
this.consecutiveMistakeCount = 0
|
|
}
|
|
|
|
// get previous api req's index to check token usage and determine if we need to truncate conversation history
|
|
const previousApiReqIndex = findLastIndex(this.claudeMessages, (m) => m.say === "api_req_started")
|
|
|
|
// getting verbose details is an expensive operation, it uses globby to top-down build file structure of project which for large projects can take a few seconds
|
|
// for the best UX we show a placeholder api_req_started message with a loading spinner as this happens
|
|
await this.say(
|
|
"api_req_started",
|
|
JSON.stringify({
|
|
request:
|
|
userContent.map((block) => formatContentBlockToMarkdown(block)).join("\n\n") + "\n\nLoading...",
|
|
})
|
|
)
|
|
|
|
const [parsedUserContent, environmentDetails] = await this.loadContext(userContent, includeFileDetails)
|
|
userContent = parsedUserContent
|
|
// add environment details as its own text block, separate from tool results
|
|
userContent.push({ type: "text", text: environmentDetails })
|
|
|
|
await this.addToApiConversationHistory({ role: "user", content: userContent })
|
|
|
|
// since we sent off a placeholder api_req_started message to update the webview while waiting to actually start the API request (to load potential details for example), we need to update the text of that message
|
|
const lastApiReqIndex = findLastIndex(this.claudeMessages, (m) => m.say === "api_req_started")
|
|
this.claudeMessages[lastApiReqIndex].text = JSON.stringify({
|
|
request: userContent.map((block) => formatContentBlockToMarkdown(block)).join("\n\n"),
|
|
} satisfies ClaudeApiReqInfo)
|
|
await this.saveClaudeMessages()
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
|
|
try {
|
|
let cacheWriteTokens = 0
|
|
let cacheReadTokens = 0
|
|
let inputTokens = 0
|
|
let outputTokens = 0
|
|
let totalCost: number | undefined
|
|
|
|
// update api_req_started. we can't use api_req_finished anymore since it's a unique case where it could come after a streaming message (ie in the middle of being updated or executed)
|
|
// fortunately api_req_finished was always parsed out for the gui anyways, so it remains solely for legacy purposes to keep track of prices in tasks from history
|
|
// (it's worth removing a few months from now)
|
|
const updateApiReqMsg = (cancelReason?: ClaudeApiReqCancelReason, streamingFailedMessage?: string) => {
|
|
this.claudeMessages[lastApiReqIndex].text = JSON.stringify({
|
|
...JSON.parse(this.claudeMessages[lastApiReqIndex].text || "{}"),
|
|
tokensIn: inputTokens,
|
|
tokensOut: outputTokens,
|
|
cacheWrites: cacheWriteTokens,
|
|
cacheReads: cacheReadTokens,
|
|
cost:
|
|
totalCost ??
|
|
calculateApiCost(
|
|
this.api.getModel().info,
|
|
inputTokens,
|
|
outputTokens,
|
|
cacheWriteTokens,
|
|
cacheReadTokens
|
|
),
|
|
cancelReason,
|
|
streamingFailedMessage,
|
|
} satisfies ClaudeApiReqInfo)
|
|
}
|
|
|
|
const abortStream = async (cancelReason: ClaudeApiReqCancelReason, streamingFailedMessage?: string) => {
|
|
if (this.diffViewProvider.isEditing) {
|
|
await this.diffViewProvider.revertChanges() // closes diff view
|
|
}
|
|
|
|
// if last message is a partial we need to update and save it
|
|
const lastMessage = this.claudeMessages.at(-1)
|
|
if (lastMessage && lastMessage.partial) {
|
|
// lastMessage.ts = Date.now() DO NOT update ts since it is used as a key for virtuoso list
|
|
lastMessage.partial = false
|
|
// instead of streaming partialMessage events, we do a save and post like normal to persist to disk
|
|
console.log("updating partial message", lastMessage)
|
|
// await this.saveClaudeMessages()
|
|
}
|
|
|
|
// Let assistant know their response was interrupted for when task is resumed
|
|
await this.addToApiConversationHistory({
|
|
role: "assistant",
|
|
content: [
|
|
{
|
|
type: "text",
|
|
text:
|
|
assistantMessage +
|
|
`\n\n[${
|
|
cancelReason === "streaming_failed"
|
|
? "Response interrupted by API Error"
|
|
: "Response interrupted by user"
|
|
}]`,
|
|
},
|
|
],
|
|
})
|
|
|
|
// update api_req_started to have cancelled and cost, so that we can display the cost of the partial stream
|
|
updateApiReqMsg(cancelReason, streamingFailedMessage)
|
|
await this.saveClaudeMessages()
|
|
|
|
// signals to provider that it can retrieve the saved messages from disk, as abortTask can not be awaited on in nature
|
|
this.didFinishAborting = true
|
|
}
|
|
|
|
// reset streaming state
|
|
this.currentStreamingContentIndex = 0
|
|
this.assistantMessageContent = []
|
|
this.didCompleteReadingStream = false
|
|
this.userMessageContent = []
|
|
this.userMessageContentReady = false
|
|
this.didRejectTool = false
|
|
this.presentAssistantMessageLocked = false
|
|
this.presentAssistantMessageHasPendingUpdates = false
|
|
await this.diffViewProvider.reset()
|
|
|
|
const stream = this.attemptApiRequest(previousApiReqIndex) // yields only if the first chunk is successful, otherwise will allow the user to retry the request (most likely due to rate limit error, which gets thrown on the first chunk)
|
|
let assistantMessage = ""
|
|
try {
|
|
for await (const chunk of stream) {
|
|
switch (chunk.type) {
|
|
case "usage":
|
|
inputTokens += chunk.inputTokens
|
|
outputTokens += chunk.outputTokens
|
|
cacheWriteTokens += chunk.cacheWriteTokens ?? 0
|
|
cacheReadTokens += chunk.cacheReadTokens ?? 0
|
|
totalCost = chunk.totalCost
|
|
break
|
|
case "text":
|
|
assistantMessage += chunk.text
|
|
this.parseAssistantMessage(assistantMessage)
|
|
this.presentAssistantMessage()
|
|
break
|
|
}
|
|
|
|
if (this.abort) {
|
|
console.log("aborting stream...")
|
|
await abortStream("user_cancelled")
|
|
break // aborts the stream
|
|
}
|
|
|
|
if (this.didRejectTool) {
|
|
// userContent has a tool rejection, so interrupt the assistant's response to present the user's feedback
|
|
assistantMessage += "\n\n[Response interrupted by user feedback]"
|
|
this.userMessageContentReady = true
|
|
break
|
|
}
|
|
}
|
|
} catch (error) {
|
|
this.abortTask() // if the stream failed, there's various states the task could be in (i.e. could have streamed some tools the user may have executed), so we just resort to replicating a cancel task
|
|
await abortStream("streaming_failed", error.message ?? JSON.stringify(serializeError(error), null, 2))
|
|
const history = await this.providerRef.deref()?.getTaskWithId(this.taskId)
|
|
if (history) {
|
|
await this.providerRef.deref()?.initClaudeDevWithHistoryItem(history.historyItem)
|
|
// await this.providerRef.deref()?.postStateToWebview()
|
|
}
|
|
}
|
|
|
|
// need to call here in case the stream was aborted
|
|
if (this.abort) {
|
|
throw new Error("ClaudeDev instance aborted")
|
|
}
|
|
|
|
this.didCompleteReadingStream = true
|
|
|
|
// in case no tool calls were made or tool call wasn't closed properly, set partial to false
|
|
// should not do this if text block is not the last block, since presentAssistantMessage presents the last block
|
|
if (this.assistantMessageContent.length === 1 && this.assistantMessageContent[0].partial) {
|
|
// this.assistantMessageContent.forEach((e) => (e.partial = false)) // cant just do this bc a tool could be in the middle of executing
|
|
// this was originally intended just to update text content in case no tools were called
|
|
this.assistantMessageContent[0].partial = false
|
|
this.presentAssistantMessage() // if there is content to update then it will complete and update this.userMessageContentReady to true, which we pwaitfor before making the next request
|
|
}
|
|
|
|
updateApiReqMsg()
|
|
await this.saveClaudeMessages()
|
|
await this.providerRef.deref()?.postStateToWebview()
|
|
|
|
// now add to apiconversationhistory
|
|
// need to save assistant responses to file before proceeding to tool use since user can exit at any moment and we wouldn't be able to save the assistant's response
|
|
let didEndLoop = false
|
|
if (assistantMessage.length > 0) {
|
|
await this.addToApiConversationHistory({
|
|
role: "assistant",
|
|
content: [{ type: "text", text: assistantMessage }],
|
|
})
|
|
|
|
// in case the content blocks finished
|
|
// it may be the api stream finished after the last parsed content block was executed, so we are able to detect out of bounds and set userMessageContentReady to true (not you should not call presentAssistantMessage since if the last block is completed it will be presented again)
|
|
const completeBlocks = this.assistantMessageContent.filter((block) => !block.partial) // if there are any partial blocks after the stream ended we can consider them invalid
|
|
if (this.currentStreamingContentIndex >= completeBlocks.length) {
|
|
this.userMessageContentReady = true
|
|
}
|
|
|
|
await pWaitFor(() => this.userMessageContentReady)
|
|
|
|
// if the model did not tool use, then we need to tell it to either use a tool or attempt_completion
|
|
const didToolUse = this.assistantMessageContent.some((block) => block.type === "tool_use")
|
|
if (!didToolUse) {
|
|
this.userMessageContent.push({
|
|
type: "text",
|
|
text: formatResponse.noToolsUsed(),
|
|
})
|
|
this.consecutiveMistakeCount++
|
|
}
|
|
|
|
const recDidEndLoop = await this.recursivelyMakeClaudeRequests(this.userMessageContent)
|
|
didEndLoop = recDidEndLoop
|
|
} else {
|
|
// if there's no assistant_responses, that means we got no text or tool_use content blocks from API which we should assume is an error
|
|
await this.say(
|
|
"error",
|
|
"Unexpected API Response: The language model did not provide any assistant messages. This may indicate an issue with the API or the model's output."
|
|
)
|
|
await this.addToApiConversationHistory({
|
|
role: "assistant",
|
|
content: [{ type: "text", text: "Failure: I did not provide a response." }],
|
|
})
|
|
}
|
|
|
|
return didEndLoop // will always be false for now
|
|
} catch (error) {
|
|
// this should never happen since the only thing that can throw an error is the attemptApiRequest, which is wrapped in a try catch that sends an ask where if noButtonTapped, will clear current task and destroy this instance. However to avoid unhandled promise rejection, we will end this loop which will end execution of this instance (see startTask)
|
|
return true
|
|
}
|
|
}
|
|
|
|
async loadContext(userContent: UserContent, includeFileDetails: boolean = false) {
|
|
return await Promise.all([
|
|
// Process userContent array, which contains various block types:
|
|
// TextBlockParam, ImageBlockParam, ToolUseBlockParam, and ToolResultBlockParam.
|
|
// We need to apply parseMentions() to:
|
|
// 1. All TextBlockParam's text (first user message with task)
|
|
// 2. ToolResultBlockParam's content/context text arrays if it contains "<feedback>" (see formatToolDeniedFeedback, attemptCompletion, executeCommand, and consecutiveMistakeCount >= 3) or "<answer>" (see askFollowupQuestion), we place all user generated content in these tags so they can effectively be used as markers for when we should parse mentions)
|
|
Promise.all(
|
|
userContent.map(async (block) => {
|
|
if (block.type === "text") {
|
|
return {
|
|
...block,
|
|
text: await parseMentions(block.text, cwd, this.urlContentFetcher),
|
|
}
|
|
} else if (block.type === "tool_result") {
|
|
const isUserMessage = (text: string) => text.includes("<feedback>") || text.includes("<answer>")
|
|
if (typeof block.content === "string" && isUserMessage(block.content)) {
|
|
return {
|
|
...block,
|
|
content: await parseMentions(block.content, cwd, this.urlContentFetcher),
|
|
}
|
|
} else if (Array.isArray(block.content)) {
|
|
const parsedContent = await Promise.all(
|
|
block.content.map(async (contentBlock) => {
|
|
if (contentBlock.type === "text" && isUserMessage(contentBlock.text)) {
|
|
return {
|
|
...contentBlock,
|
|
text: await parseMentions(contentBlock.text, cwd, this.urlContentFetcher),
|
|
}
|
|
}
|
|
return contentBlock
|
|
})
|
|
)
|
|
return {
|
|
...block,
|
|
content: parsedContent,
|
|
}
|
|
}
|
|
}
|
|
return block
|
|
})
|
|
),
|
|
this.getEnvironmentDetails(includeFileDetails),
|
|
])
|
|
}
|
|
|
|
async getEnvironmentDetails(includeFileDetails: boolean = false) {
|
|
let details = ""
|
|
|
|
// It could be useful for claude to know if the user went from one or no file to another between messages, so we always include this context
|
|
details += "\n\n# VSCode Visible Files"
|
|
const visibleFiles = vscode.window.visibleTextEditors
|
|
?.map((editor) => editor.document?.uri?.fsPath)
|
|
.filter(Boolean)
|
|
.map((absolutePath) => path.relative(cwd, absolutePath).toPosix())
|
|
.join("\n")
|
|
if (visibleFiles) {
|
|
details += `\n${visibleFiles}`
|
|
} else {
|
|
details += "\n(No visible files)"
|
|
}
|
|
|
|
details += "\n\n# VSCode Open Tabs"
|
|
const openTabs = vscode.window.tabGroups.all
|
|
.flatMap((group) => group.tabs)
|
|
.map((tab) => (tab.input as vscode.TabInputText)?.uri?.fsPath)
|
|
.filter(Boolean)
|
|
.map((absolutePath) => path.relative(cwd, absolutePath).toPosix())
|
|
.join("\n")
|
|
if (openTabs) {
|
|
details += `\n${openTabs}`
|
|
} else {
|
|
details += "\n(No open tabs)"
|
|
}
|
|
|
|
const busyTerminals = this.terminalManager.getTerminals(true)
|
|
const inactiveTerminals = this.terminalManager.getTerminals(false)
|
|
// const allTerminals = [...busyTerminals, ...inactiveTerminals]
|
|
|
|
if (busyTerminals.length > 0 && this.didEditFile) {
|
|
// || this.didEditFile
|
|
await delay(300) // delay after saving file to let terminals catch up
|
|
}
|
|
|
|
// let terminalWasBusy = false
|
|
if (busyTerminals.length > 0) {
|
|
// wait for terminals to cool down
|
|
// terminalWasBusy = allTerminals.some((t) => this.terminalManager.isProcessHot(t.id))
|
|
await pWaitFor(() => busyTerminals.every((t) => !this.terminalManager.isProcessHot(t.id)), {
|
|
interval: 100,
|
|
timeout: 15_000,
|
|
}).catch(() => {})
|
|
}
|
|
|
|
// we want to get diagnostics AFTER terminal cools down for a few reasons: terminal could be scaffolding a project, dev servers (compilers like webpack) will first re-compile and then send diagnostics, etc
|
|
/*
|
|
let diagnosticsDetails = ""
|
|
const diagnostics = await this.diagnosticsMonitor.getCurrentDiagnostics(this.didEditFile || terminalWasBusy) // if claude ran a command (ie npm install) or edited the workspace then wait a bit for updated diagnostics
|
|
for (const [uri, fileDiagnostics] of diagnostics) {
|
|
const problems = fileDiagnostics.filter((d) => d.severity === vscode.DiagnosticSeverity.Error)
|
|
if (problems.length > 0) {
|
|
diagnosticsDetails += `\n## ${path.relative(cwd, uri.fsPath)}`
|
|
for (const diagnostic of problems) {
|
|
// let severity = diagnostic.severity === vscode.DiagnosticSeverity.Error ? "Error" : "Warning"
|
|
const line = diagnostic.range.start.line + 1 // VSCode lines are 0-indexed
|
|
const source = diagnostic.source ? `[${diagnostic.source}] ` : ""
|
|
diagnosticsDetails += `\n- ${source}Line ${line}: ${diagnostic.message}`
|
|
}
|
|
}
|
|
}
|
|
*/
|
|
this.didEditFile = false // reset, this lets us know when to wait for saved files to update terminals
|
|
|
|
// waiting for updated diagnostics lets terminal output be the most up-to-date possible
|
|
let terminalDetails = ""
|
|
if (busyTerminals.length > 0) {
|
|
// terminals are cool, let's retrieve their output
|
|
terminalDetails += "\n\n# Active Terminals"
|
|
for (const busyTerminal of busyTerminals) {
|
|
terminalDetails += `\n## ${busyTerminal.lastCommand}`
|
|
const newOutput = this.terminalManager.getUnretrievedOutput(busyTerminal.id)
|
|
if (newOutput) {
|
|
terminalDetails += `\n### New Output\n${newOutput}`
|
|
} else {
|
|
// details += `\n(Still running, no new output)` // don't want to show this right after running the command
|
|
}
|
|
}
|
|
}
|
|
// only show inactive terminals if there's output to show
|
|
if (inactiveTerminals.length > 0) {
|
|
const inactiveTerminalOutputs = new Map<number, string>()
|
|
for (const inactiveTerminal of inactiveTerminals) {
|
|
const newOutput = this.terminalManager.getUnretrievedOutput(inactiveTerminal.id)
|
|
if (newOutput) {
|
|
inactiveTerminalOutputs.set(inactiveTerminal.id, newOutput)
|
|
}
|
|
}
|
|
if (inactiveTerminalOutputs.size > 0) {
|
|
terminalDetails += "\n\n# Inactive Terminals"
|
|
for (const [terminalId, newOutput] of inactiveTerminalOutputs) {
|
|
const inactiveTerminal = inactiveTerminals.find((t) => t.id === terminalId)
|
|
if (inactiveTerminal) {
|
|
terminalDetails += `\n## ${inactiveTerminal.lastCommand}`
|
|
terminalDetails += `\n### New Output\n${newOutput}`
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// details += "\n\n# VSCode Workspace Errors"
|
|
// if (diagnosticsDetails) {
|
|
// details += diagnosticsDetails
|
|
// } else {
|
|
// details += "\n(No errors detected)"
|
|
// }
|
|
|
|
if (terminalDetails) {
|
|
details += terminalDetails
|
|
}
|
|
|
|
if (includeFileDetails) {
|
|
details += `\n\n# Current Working Directory (${cwd.toPosix()}) Files\n`
|
|
const isDesktop = arePathsEqual(cwd, path.join(os.homedir(), "Desktop"))
|
|
if (isDesktop) {
|
|
// don't want to immediately access desktop since it would show permission popup
|
|
details += "(Desktop files not shown automatically. Use list_files to explore if needed.)"
|
|
} else {
|
|
const [files, didHitLimit] = await listFiles(cwd, true, 200)
|
|
const result = formatResponse.formatFilesList(cwd, files, didHitLimit)
|
|
details += result
|
|
}
|
|
}
|
|
|
|
return `<environment_details>\n${details.trim()}\n</environment_details>`
|
|
}
|
|
}
|