bug: fix file loading 500 error when write is in progress (#2037)

Author: ibuildthecloud

ui/user/src/lib/components/navbar/Clone.svelte

@@ -15,7 +15,6 @@
 	async function copy() {
 		dialog?.close();
 		const newProject = await ChatService.copyProject(project.assistantID, project.id);
-		console.log('old', project.id, 'new', newProject.id);
 		await goto(`/o/${newProject.id}`);
 	}
 </script>

ui/user/src/lib/services/chat/messages.ts

@@ -17,8 +17,19 @@ function toMessageFromInput(s: string): string {
 	return s;
 }
 
-function setFileContent(items: EditorItem[], name: string, content: string, full: boolean = false) {
-	const existing = items.find((f) => f.id === name);
+function setFileContent(
+	items: EditorItem[],
+	name: string,
+	content: string,
+	full: boolean = false,
+	opts: {
+		threadID?: string;
+		taskID?: string;
+		runID?: string;
+	} = {}
+) {
+	const id = opts.runID ? `${opts.taskID}/${opts.runID}/${name}` : `${opts.threadID}/${name}`;
+	const existing = items.find((f) => f.id === id);
 	if (existing && existing.file) {
 		if (full) {
 			existing.file.contents = content;
@@ -29,7 +40,7 @@ function setFileContent(items: EditorItem[], name: string, content: string, full
 		}
 	} else {
 		items.push({
-			id: name,
+			id: id,
 			name: name,
 			file: {
 				contents: content,
@@ -107,7 +118,16 @@ function getFilenameAndContent(content: string) {
 	};
 }
 
-function reformatWriteMessage(items: EditorItem[], msg: Message, last: boolean) {
+function reformatWriteMessage(
+	items: EditorItem[],
+	msg: Message,
+	last: boolean,
+	opts: {
+		threadID?: string;
+		taskID?: string;
+		runID?: string;
+	} = {}
+) {
 	msg.icon = 'Pencil';
 	msg.done = !last || msg.toolCall !== undefined;
 	msg.sourceName = msg.done ? 'Wrote to Workspace' : 'Writing to Workspace';
@@ -129,17 +149,25 @@ function reformatWriteMessage(items: EditorItem[], msg: Message, last: boolean)
 	}
 
 	if (last && msg.file?.filename && msg.file?.content) {
-		setFileContent(items, msg.file.filename, msg.file.content, msg.toolCall !== undefined);
+		setFileContent(items, msg.file.filename, msg.file.content, msg.toolCall !== undefined, opts);
 	}
 }
 
-export function buildMessagesFromProgress(items: EditorItem[], progresses: Progress[]): Messages {
+export function buildMessagesFromProgress(
+	items: EditorItem[],
+	progresses: Progress[],
+	opts: {
+		threadID?: string;
+		taskID?: string;
+		runID?: string;
+	}
+): Messages {
 	const messages = toMessages(progresses);
 
 	// Post Process for much more better-ness
 	messages.messages.forEach((item, i) => {
 		if (item.tool && item.sourceName == 'workspace_write') {
-			reformatWriteMessage(items, item, i == messages.messages.length - 1);
+			reformatWriteMessage(items, item, i == messages.messages.length - 1, opts);
 			return;
 		} else if (item.sent) {
 			reformatInputMessage(item);

ui/user/src/lib/services/chat/thread.svelte.ts

@@ -154,14 +154,27 @@ export class Thread {
 		}
 
 		for (const [stepID, msgs] of newMessages) {
-			this.onStepMessages(stepID, buildMessagesFromProgress(this.#items, msgs));
+			this.onStepMessages(
+				stepID,
+				buildMessagesFromProgress(this.#items, msgs, {
+					taskID: this.#task?.id,
+					runID: this.#runID,
+					threadID: this.threadID
+				})
+			);
 		}
 	}
 
 	#onProgress(progress: Progress) {
 		this.#progresses.push(progress);
 		if (this.replayComplete) {
-			this.onMessages(buildMessagesFromProgress(this.#items, this.#progresses));
+			this.onMessages(
+				buildMessagesFromProgress(this.#items, this.#progresses, {
+					taskID: this.#task?.id,
+					runID: this.#runID,
+					threadID: this.threadID
+				})
+			);
 			this.#handleSteps();
 		}
 	}

ui/user/src/lib/services/editor/index.svelte.ts

@@ -118,6 +118,8 @@ async function loadFile(
 		let fileID = file;
 		if (opts?.taskID && opts?.runID) {
 			fileID = `${opts.taskID}/${opts.runID}/${file}`;
+		} else if (opts?.threadID) {
+			fileID = `${opts.threadID}/${file}`;
 		}
 		const targetFile: EditorItem = {
 			id: fileID,