enh: arena model rate to reveal

This commit is contained in:
Timothy J. Baek
2024-10-22 18:05:30 -07:00
parent 6d52f913d2
commit 00e9362c2c
3 changed files with 32 additions and 2 deletions
+6
View File
@@ -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 ?? '',