🏷️ fix: Ensure modelLabel Field Usage for ModelSpecs/GPTPlugins (#4224)

This commit is contained in:
Danny Avila 2024-09-24 07:27:11 -04:00 committed by GitHub
parent 321260e3c7
commit 6f498eee0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 20 additions and 10 deletions

View file

@ -220,13 +220,16 @@ export const getResponseSender = (endpointOption: t.TEndpointOption): string =>
model: _m,
endpoint,
endpointType,
modelDisplayLabel,
chatGptLabel,
modelLabel,
modelDisplayLabel: _mdl,
chatGptLabel: _cgl,
modelLabel: _ml,
jailbreak,
} = endpointOption;
const model = _m ?? '';
const modelDisplayLabel = _mdl ?? '';
const chatGptLabel = _cgl ?? '';
const modelLabel = _ml ?? '';
if (
[
EModelEndpoint.openAI,
@ -238,6 +241,8 @@ export const getResponseSender = (endpointOption: t.TEndpointOption): string =>
) {
if (chatGptLabel) {
return chatGptLabel;
} else if (modelLabel) {
return modelLabel;
} else if (model && /\bo1\b/i.test(model)) {
return 'o1';
} else if (model && model.includes('gpt-3')) {
@ -257,11 +262,11 @@ export const getResponseSender = (endpointOption: t.TEndpointOption): string =>
}
if (endpoint === EModelEndpoint.anthropic) {
return modelLabel ?? 'Claude';
return modelLabel || 'Claude';
}
if (endpoint === EModelEndpoint.bedrock) {
return modelLabel ?? alternateName[endpoint];
return modelLabel || alternateName[endpoint];
}
if (endpoint === EModelEndpoint.google) {

View file

@ -1018,6 +1018,7 @@ export const compactChatGPTSchema = tConversationSchema
export const compactPluginsSchema = tConversationSchema
.pick({
model: true,
modelLabel: true,
chatGptLabel: true,
promptPrefix: true,
temperature: true,
@ -1033,6 +1034,9 @@ export const compactPluginsSchema = tConversationSchema
})
.transform((obj) => {
const newObj: Partial<TConversation> = { ...obj };
if (newObj.modelLabel === null) {
delete newObj.modelLabel;
}
if (newObj.chatGptLabel === null) {
delete newObj.chatGptLabel;
}