github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-16 06:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): Support model icon

Browse Source
This commit is contained in:
Fangyin Cheng
2025-03-19 11:51:26 +08:00
parent 88bbd695d0
commit 4489f74e8d
474 changed files with 15535 additions and 7832 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
web/pages/construct/models/index.tsx
View File

@@ -3,7 +3,7 @@ import ModelForm from '@/components/model/model-form';
import BlurredCard, { InnerDropdown } from '@/new-components/common/blurredCard';
import ConstructLayout from '@/new-components/layout/Construct';
import { IModelData } from '@/types/model';
import { MODEL_ICON_DICT } from '@/utils/constants';
import { getModelIcon } from '@/utils/constants';
import { PlusOutlined } from '@ant-design/icons';
import { Button, Modal, Tag, message } from 'antd';
import moment from 'moment';
@@ -97,15 +97,7 @@ function Models() {
// ).run;
const returnLogo = (name: string) => {
const formatterModal = name?.replaceAll('-', '_').split('_')[0];
const dict = Object.keys(MODEL_ICON_DICT);
for (let i = 0; i < dict.length; i++) {
const element = dict[i];
if (formatterModal?.includes(element)) {
return MODEL_ICON_DICT[element];
}
}
return '/pictures/model.png';
return getModelIcon(name);
};
return (