enh: arena model rate to reveal

2024-10-22 18:05:30 -07:00
parent 6d52f913d2
commit 00e9362c2c
3 changed files with 32 additions and 2 deletions
@@ -7,6 +7,7 @@ type TextStreamUpdate = {
 	// eslint-disable-next-line @typescript-eslint/no-explicit-any
 	citations?: any;
 	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	selectedModelId?: any;
 	error?: any;
 	usage?: ResponseUsage;
 };
@@ -71,6 +72,11 @@ async function* openAIStreamToIterator(
 				continue;
 			}

+			if (parsedData.selected_model_id) {
+				yield { done: false, value: '', selectedModelId: parsedData.selected_model_id };
+				continue;
+			}
+
 			yield {
 				done: false,
 				value: parsedData.choices?.[0]?.delta?.content ?? '',