fix: gptCustom icon should show as same in model and message

This commit is contained in:
Wentao Lyu 2023-03-15 14:42:39 +08:00
parent 5d0b849930
commit 45ca0a8713
4 changed files with 15 additions and 10 deletions

View file

@ -2,7 +2,6 @@ import React from 'react';
import ModelItem from './ModelItem';
export default function MenuItems({ models, onSelect }) {
console.log(models)
return (
<>
{models.map((modelItem) => (
@ -11,6 +10,7 @@ export default function MenuItems({ models, onSelect }) {
id={modelItem._id}
modelName={modelItem.name}
value={modelItem.value}
model={modelItem.model || 'chatgptCustom'}
onSelect={onSelect}
chatGptLabel={modelItem.chatGptLabel}
promptPrefix={modelItem.promptPrefix}

View file

@ -9,7 +9,7 @@ import TrashIcon from '../svg/TrashIcon';
import manualSWR from '~/utils/fetchers';
import { getIconOfModel } from '../../utils';
export default function ModelItem({ modelName, value, onSelect, id, chatGptLabel, promptPrefix }) {
export default function ModelItem({ modelName, value, model, onSelect, id, chatGptLabel, promptPrefix }) {
const dispatch = useDispatch();
const { customModel } = useSelector((state) => state.submit);
const { initial } = useSelector((state) => state.models);
@ -28,7 +28,7 @@ export default function ModelItem({ modelName, value, onSelect, id, chatGptLabel
dispatch(setModels(fetchedModels));
});
const icon = getIconOfModel({ size: 16, sender: modelName, isCreatedByUser: false, model: value, chatGptLabel, promptPrefix, error: false, className: "mr-2" });
const icon = getIconOfModel({ size: 16, sender: modelName, isCreatedByUser: false, model, chatGptLabel, promptPrefix, error: false, className: "mr-2" });
if (value === 'chatgptCustom') {
return (