🐛 Vote not working with new messages (#249)

* fix retry not working after vote changes

* once assistant answer is received, invalidate the messages so we get the update id

* pass response message id from client to server

* remove url dependency

src/lib/components/chat/ChatMessage.svelte

@@ -182,7 +182,7 @@
 						class="cursor-pointer rounded-lg border border-gray-100 p-1 text-xs text-gray-400 group-hover:block hover:text-gray-500 dark:border-gray-800 dark:text-gray-400 dark:hover:text-gray-300 md:hidden lg:-right-2"
 						title="Retry"
 						type="button"
-						on:click={() => dispatch("retry")}
+						on:click={() => dispatch("retry", { content: message.content, id: message.id })}
 					>
 						<CarbonRotate360 />
 					</button>

src/routes/conversation/[id]/+page.svelte

@@ -31,6 +31,7 @@
 
 	async function getTextGenerationStream(inputs: string, messageId: string, isRetry = false) {
 		const conversationId = $page.params.id;
+		const responseId = randomUUID();
 
 		const response = textGenerationStream(
 			{
@@ -43,6 +44,7 @@
 			},
 			{
 				id: messageId,
+				response_id: responseId,
 				is_retry: isRetry,
 				use_cache: false,
 			} as Options
@@ -89,7 +91,7 @@
 					messages = [
 						...messages,
 						// id doesn't match the backend id but it's not important for assistant messages
-						{ from: "assistant", content: output.token.text.trimStart(), id: randomUUID() },
+						{ from: "assistant", content: output.token.text.trimStart(), id: responseId },
 					];
 				} else {
 					lastMessage.content += output.token.text;

src/routes/conversation/[id]/+server.ts

@@ -38,12 +38,13 @@ export async function POST({ request, fetch, locals, params }) {
 	const json = await request.json();
 	const {
 		inputs: newPrompt,
-		options: { id: messageId, is_retry },
+		options: { id: messageId, is_retry, response_id: responseId },
 	} = z
 		.object({
 			inputs: z.string().trim().min(1),
 			options: z.object({
 				id: z.optional(z.string().uuid()),
+				response_id: z.optional(z.string().uuid()),
 				is_retry: z.optional(z.boolean()),
 			}),
 		})
@@ -110,7 +111,11 @@ export async function POST({ request, fetch, locals, params }) {
 			}
 		}
 
-		messages.push({ from: "assistant", content: generated_text, id: crypto.randomUUID() });
+		messages.push({
+			from: "assistant",
+			content: generated_text,
+			id: (responseId as Message["id"]) || crypto.randomUUID(),
+		});
 
 		await collections.conversations.updateOne(
 			{