Fix completion result and text without tool calls

This commit is contained in:
Saoud Rizwan
2024-09-27 22:18:52 -04:00
parent 68b87e6056
commit 103f82145e

View File

@@ -1770,21 +1770,24 @@ ${this.customInstructions.trim()}
break break
} }
this.consecutiveMistakeCount = 0 this.consecutiveMistakeCount = 0
if (lastMessage && lastMessage.ask === "command") { if (lastMessage && lastMessage.ask === "command") {
// complete command message // complete command message
const didApprove = await askApproval("command", command) const didApprove = await askApproval("command", command)
if (!didApprove) { if (!didApprove) {
break break
} }
const [userRejected, result] = await this.executeCommandTool(command!) const [userRejected, commandResult] = await this.executeCommandTool(command!, true)
if (commandResult) {
if (userRejected) { if (userRejected) {
this.didRejectTool = true // test whats going on here this.didRejectTool = true // test whats going on here
} }
pushToolResult(result) pushToolResult(commandResult)
break break
} else { }
// last message is completion_result, not command so it wasn't completed, need to complete it }
// empty string makes it invisible and just shows new task button
// 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) const { response, text, images } = await this.ask("completion_result", "", false)
if (response === "yesButtonTapped") { if (response === "yesButtonTapped") {
pushToolResult("") // signals to recursive loop to stop (for now this never happens since yesButtonTapped will trigger a new task) pushToolResult("") // signals to recursive loop to stop (for now this never happens since yesButtonTapped will trigger a new task)
@@ -1799,7 +1802,6 @@ ${this.customInstructions.trim()}
) )
break break
} }
}
} catch (error) { } catch (error) {
await handleError("inspecting site", error) await handleError("inspecting site", error)
break break
@@ -2081,6 +2083,11 @@ ${this.customInstructions.trim()}
} }
} }
// in case no tool calls were made or tool call wasn't closed properly, set partial to false
if (this.assistantMessageContent.some((e) => e.partial)) {
this.assistantMessageContent.forEach((e) => (e.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
}
this.didCompleteReadingStream = true this.didCompleteReadingStream = true
console.log("contentBlocks", apiContentBlocks) console.log("contentBlocks", apiContentBlocks)