ToothFairyAI_Public/tf_code
1
0
Fork 0
mirror of https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git synced 2026-03-31 06:12:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

core: keep message part order stable when files resolve asynchronously (#13915)

Browse Source
This commit is contained in:
Aiden Cline
2026-02-16 18:45:11 -06:00
committed by GitHub
parent 0e669b6016
commit e35a4131d0
2 changed files with 62 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
packages/opencode/src/session/prompt.ts
View File

@@ -974,17 +974,22 @@ export namespace SessionPrompt {
}
using _ = defer(() => InstructionPrompt.clear(info.id))
type Draft<T> = T extends MessageV2.Part ? Omit<T, "id"> & { id?: string } : never
const assign = (part: Draft<MessageV2.Part>): MessageV2.Part => ({
...part,
id: part.id ?? Identifier.ascending("part"),
})
const parts = await Promise.all(
input.parts.map(async (part): Promise<MessageV2.Part[]> => {
input.parts.map(async (part): Promise<Draft<MessageV2.Part>[]> => {
if (part.type === "file") {
// before checking the protocol we check if this is an mcp resource because it needs special handling
if (part.source?.type === "resource") {
const { clientName, uri } = part.source
log.info("mcp resource", { clientName, uri, mime: part.mime })
const pieces: MessageV2.Part[] = [
const pieces: Draft<MessageV2.Part>[] = [
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1007,7 +1012,6 @@ export namespace SessionPrompt {
for (const content of contents) {
if ("text" in content && content.text) {
pieces.push({
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1018,7 +1022,6 @@ export namespace SessionPrompt {
// Handle binary content if needed
const mimeType = "mimeType" in content ? content.mimeType : part.mime
pieces.push({
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1030,7 +1033,6 @@ export namespace SessionPrompt {
pieces.push({
...part,
id: part.id ?? Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
})
@@ -1038,7 +1040,6 @@ export namespace SessionPrompt {
log.error("failed to read MCP resource", { error, clientName, uri })
const message = error instanceof Error ? error.message : String(error)
pieces.push({
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1055,7 +1056,6 @@ export namespace SessionPrompt {
if (part.mime === "text/plain") {
return [
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1063,7 +1063,6 @@ export namespace SessionPrompt {
text: `Called the Read tool with the following input: ${JSON.stringify({ filePath: part.filename })}`,
},
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1072,7 +1071,6 @@ export namespace SessionPrompt {
},
{
...part,
id: part.id ?? Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
},
@@ -1129,9 +1127,8 @@ export namespace SessionPrompt {
}
const args = { filePath: filepath, offset, limit }
const pieces: MessageV2.Part[] = [
const pieces: Draft<MessageV2.Part>[] = [
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1155,7 +1152,6 @@ export namespace SessionPrompt {
}
const result = await t.execute(args, readCtx)
pieces.push({
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1166,7 +1162,6 @@ export namespace SessionPrompt {
pieces.push(
...result.attachments.map((attachment) => ({
...attachment,
id: Identifier.ascending("part"),
synthetic: true,
filename: attachment.filename ?? part.filename,
messageID: info.id,
@@ -1176,7 +1171,6 @@ export namespace SessionPrompt {
} else {
pieces.push({
...part,
id: part.id ?? Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
})
@@ -1192,7 +1186,6 @@ export namespace SessionPrompt {
}).toObject(),
})
pieces.push({
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1219,7 +1212,6 @@ export namespace SessionPrompt {
const result = await ReadTool.init().then((t) => t.execute(args, listCtx))
return [
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1227,7 +1219,6 @@ export namespace SessionPrompt {
text: `Called the Read tool with the following input: ${JSON.stringify(args)}`,
},
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1236,7 +1227,6 @@ export namespace SessionPrompt {
},
{
...part,
id: part.id ?? Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
},
@@ -1247,7 +1237,6 @@ export namespace SessionPrompt {
FileTime.read(input.sessionID, filepath)
return [
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1255,7 +1244,7 @@ export namespace SessionPrompt {
synthetic: true,
},
{
id: part.id ?? Identifier.ascending("part"),
id: part.id,
messageID: info.id,
sessionID: input.sessionID,
type: "file",
@@ -1274,13 +1263,11 @@ export namespace SessionPrompt {
const hint = perm.action === "deny" ? " . Invoked by user; guaranteed to exist." : ""
return [
{
id: Identifier.ascending("part"),
...part,
messageID: info.id,
sessionID: input.sessionID,
},
{
id: Identifier.ascending("part"),
messageID: info.id,
sessionID: input.sessionID,
type: "text",
@@ -1297,14 +1284,13 @@ export namespace SessionPrompt {
return [
{
id: Identifier.ascending("part"),
...part,
messageID: info.id,
sessionID: input.sessionID,
},
]
}),
).then((x) => x.flat())
).then((x) => x.flat().map(assign))
await Plugin.trigger(
"chat.message",