Browse Source

refac: feedback base_models

Timothy J. Baek 6 months ago
parent
commit
09935d191f
1 changed files with 15 additions and 3 deletions
  1. 15 3
      src/lib/components/chat/Messages/ResponseMessage.svelte

+ 15 - 3
src/lib/components/chat/Messages/ResponseMessage.svelte

@@ -362,8 +362,6 @@
 
 		const messages = createMessagesList(history, message.id);
 
-		const baseModelId = $models.find((m) => m.id === message.model)?.info?.base_model_id ?? null;
-
 		let feedbackItem = {
 			type: 'rating',
 			data: {
@@ -380,7 +378,6 @@
 			meta: {
 				arena: message ? message.arena : false,
 				model_id: message.model,
-				...(baseModelId ? { base_model_id: baseModelId } : {}),
 				message_id: message.id,
 				message_index: messages.length,
 				chat_id: chatId
@@ -390,6 +387,21 @@
 			}
 		};
 
+		const baseModels = [
+			feedbackItem.data.model_id,
+			...(feedbackItem.data.sibling_model_ids ?? [])
+		].reduce((acc, modelId) => {
+			const model = $models.find((m) => m.id === modelId);
+			if (model) {
+				acc[model.id] = model?.info?.base_model_id ?? null;
+			} else {
+				// Log or handle cases where corresponding model is not found
+				console.warn(`Model with ID ${modelId} not found`);
+			}
+			return acc;
+		}, {});
+		feedbackItem.meta.base_models = baseModels;
+
 		let feedback = null;
 		if (message?.feedbackId) {
 			feedback = await updateFeedbackById(