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
+7 -1
View File
@@ -1591,7 +1591,7 @@
const textStream = await createOpenAITextStream(res.body, $settings.splitLargeChunks);
for await (const update of textStream) {
const { value, done, citations, error, usage } = update;
const { value, done, citations, selectedModelId, error, usage } = update;
if (error) {
await handleOpenAIError(error, null, model, responseMessage);
break;
@@ -1611,6 +1611,12 @@
responseMessage.info = { ...usage, openai: true, usage };
}
if (selectedModelId) {
responseMessage.selectedModelId = selectedModelId;
responseMessage.arena = true;
continue;
}
if (citations) {
responseMessage.citations = citations;
// Only remove status if it was initially set