diff --git a/src/components/icon-font/index.tsx b/src/components/icon-font/index.tsx index d36ad3dc..0cb33cb0 100644 --- a/src/components/icon-font/index.tsx +++ b/src/components/icon-font/index.tsx @@ -1,7 +1,7 @@ import { createFromIconfontCN } from '@ant-design/icons'; const IconFont = createFromIconfontCN({ - scriptUrl: '//at.alicdn.com/t/c/font_4613488_68kh94h3e2r.js' + scriptUrl: '//at.alicdn.com/t/c/font_4613488_cpb5yo1pk77.js' }); export default IconFont; diff --git a/src/pages/llmodels/apis/index.ts b/src/pages/llmodels/apis/index.ts index 6c570cad..91bbbf81 100644 --- a/src/pages/llmodels/apis/index.ts +++ b/src/pages/llmodels/apis/index.ts @@ -1,4 +1,4 @@ -import { listFiles } from '@huggingface/hub'; +import { listFiles, listModels } from '@huggingface/hub'; import { request } from '@umijs/max'; import { FormData, @@ -116,6 +116,23 @@ export async function callHuggingfaceQuickSearch(params: any) { }); } +export async function queryHuggingfaceModels(params: { + search: { + query: string; + tags: string[]; + }; +}) { + const result = []; + for await (const model of listModels({ + ...params, + limit: 100, + additionalFields: ['author'] + })) { + result.push(model); + } + return result; +} + export async function queryHuggingfaceModelFiles(params: { repo: string }) { const result = []; for await (const fileInfo of listFiles(params)) { diff --git a/src/pages/llmodels/components/add-modal.tsx b/src/pages/llmodels/components/add-modal.tsx index 509b3af9..77a11350 100644 --- a/src/pages/llmodels/components/add-modal.tsx +++ b/src/pages/llmodels/components/add-modal.tsx @@ -10,10 +10,7 @@ import { useIntl } from '@umijs/max'; import { Form, Input, Modal } from 'antd'; import _ from 'lodash'; import { memo, useEffect, useState } from 'react'; -import { - callHuggingfaceQuickSearch, - queryHuggingfaceModelFiles -} from '../apis'; +import { queryHuggingfaceModelFiles, queryHuggingfaceModels } from '../apis'; import { ollamaModelOptions } from '../config'; import { FormData, ListItem } from '../config/types'; @@ -106,14 +103,17 @@ const AddModal: React.FC = (props) => { const handleOnSearchRepo = async (text: string) => { try { const params = { - q: `${text}`, - type: 'model' + search: { + query: text, + tags: ['gguf'] + } }; - const res = await callHuggingfaceQuickSearch(params); - const list = _.map(res.models || [], (item: any) => { + const models = await queryHuggingfaceModels(params); + const list = _.map(models || [], (item: any) => { return { - value: item.id, - label: item.id + ...item, + value: item.name, + label: item.name }; }); setRepoOptions(list); diff --git a/src/pages/playground/components/chat-footer.tsx b/src/pages/playground/components/chat-footer.tsx index 700bd130..0d841725 100644 --- a/src/pages/playground/components/chat-footer.tsx +++ b/src/pages/playground/components/chat-footer.tsx @@ -88,7 +88,7 @@ const ChatFooter: React.FC = (props) => { style={{ textAlign: 'right' }} > - + */} {!disabled ? ( + + + ]} + className="playground-container" + >
- +
- + > */}