Quellcode durchsuchen

feat: add xinference rerank model (#1619)

zxhlyh vor 1 Jahr
Ursprung
Commit
bb98f5756a

+ 4 - 4
web/app/components/app/chat/more-info/index.tsx

@@ -14,10 +14,10 @@ export type IMoreInfoProps = {
 const MoreInfo: FC<IMoreInfoProps> = ({ more, isQuestion, className }) => {
   const { t } = useTranslation()
   return (<div className={`mt-1 w-full text-xs text-gray-400 ${isQuestion ? 'mr-2 text-right ' : 'pl-2 text-left float-right'} ${className}`}>
-    <span>{`${t('appLog.detail.timeConsuming')} ${more.latency}${t('appLog.detail.second')}`}</span>
-    <span>{`${t('appLog.detail.tokenCost')} ${formatNumber(more.tokens)}`}</span>
-    <span>· </span>
-    <span>{more.time} </span>
+    <span className='mr-2'>{`${t('appLog.detail.timeConsuming')} ${more.latency}${t('appLog.detail.second')}`}</span>
+    <span className='mr-2'>{`${t('appLog.detail.tokenCost')} ${formatNumber(more.tokens)}`}</span>
+    <span className='mr-2'>·</span>
+    <span>{more.time}</span>
   </div>)
 }
 export default React.memo(MoreInfo)

+ 7 - 0
web/app/components/header/account-setting/model-page/configs/xinference.tsx

@@ -80,6 +80,13 @@ const config: ProviderConfig = {
               'zh-Hans': 'Embeddings',
             },
           },
+          {
+            key: 'reranking',
+            label: {
+              'en': 'Rerank',
+              'zh-Hans': 'Rerank',
+            },
+          },
         ],
       },
       {

+ 2 - 1
web/app/components/header/account-setting/model-page/model-modal/Form.tsx

@@ -150,10 +150,11 @@ const Form: FC<FormProps> = ({
 
     if (field.type === 'radio') {
       const options = typeof field.options === 'function' ? field.options(value) : field.options
+
       return (
         <div key={field.key} className='py-3'>
           <div className={nameClassName}>{field.label[locale]}</div>
-          <div className='grid grid-cols-2 gap-3'>
+          <div className={`grid grid-cols-${options?.length} gap-3`}>
             {
               options?.map(option => (
                 <div