fix: some fields value are null after select model

src/pages/llmodels/hooks/index.ts

@@ -8,11 +8,15 @@ import _ from 'lodash';
 import { useEffect, useRef, useState } from 'react';
 import { evaluationsModelSpec, queryGPUList } from '../apis';
 import {
+  HuggingFaceTaskMap,
+  ModelscopeTaskMap,
   backendOptionsMap,
   getSourceRepoConfigValue,
   modelSourceMap,
+  modelTaskMap,
   setSourceRepoConfigValue
 } from '../config';
+import { identifyModelTask } from '../config/audio-catalog';
 import {
   EvaluateResult,
   FormData,
@@ -317,7 +321,6 @@ export const useCheckCompatibility = () => {
     return {
       show: !compatible || hasClaim,
       type: !compatible ? 'warning' : 'success',
-      isHtml: hasClaim,
       ...msgData
     };
   };
@@ -495,3 +498,50 @@ export const useCheckCompatibility = () => {
     submitAnyway
   };
 };
+
+export const useSelectModel = () => {
+  const handleRecognizeAudioModel = (selectModel: any, source: string) => {
+    const modelTaskType = identifyModelTask(source, selectModel.name);
+
+    const modelTask =
+      HuggingFaceTaskMap.audio.includes(selectModel.task) ||
+      ModelscopeTaskMap.audio.includes(selectModel.task)
+        ? modelTaskMap.audio
+        : '';
+
+    const modelTaskData = {
+      value: selectModel.task,
+      type: modelTaskType || modelTask,
+      text2speech:
+        HuggingFaceTaskMap[modelTaskMap.textToSpeech] === selectModel.task ||
+        ModelscopeTaskMap[modelTaskMap.textToSpeech] === selectModel.task,
+      speech2text:
+        HuggingFaceTaskMap[modelTaskMap.speechToText] === selectModel.task ||
+        ModelscopeTaskMap[modelTaskMap.speechToText] === selectModel.task
+    };
+    return modelTaskData;
+  };
+
+  // just for setting the model name or repo_id, and the backend, Since the model type is fixed.
+  const onSelectModel = (selectModel: any, source: string) => {
+    let name = _.split(selectModel.name, '/').slice(-1)[0];
+    const reg = /(-gguf)$/i;
+    name = _.toLower(name).replace(reg, '');
+
+    const modelTaskData = handleRecognizeAudioModel(selectModel, source);
+    return {
+      repo_id: selectModel.name,
+      name: name,
+      backend:
+        modelTaskData.type === modelTaskMap.audio
+          ? backendOptionsMap.voxBox
+          : selectModel.isGGUF
+            ? backendOptionsMap.llamaBox
+            : backendOptionsMap.vllm
+    };
+  };
+
+  return {
+    onSelectModel
+  };
+};