support sequence2txt and tts model in Xinference

infiniflow · Sep 30, 2024 · 7b1428c · 7b1428c
1 parent d2fb3f5
commit 7b1428c
Showing 1 changed file with 25 additions and 12 deletions.
diff --git a/web/src/pages/user-setting/setting-model/ollama-modal/index.tsx b/web/src/pages/user-setting/setting-model/ollama-modal/index.tsx
@@ -53,6 +53,26 @@ const OllamaModal = ({
   const url =
     llmFactoryToUrlMap[llmFactory as LlmFactory] ||
     'https://github.com/infiniflow/ragflow/blob/main/docs/guides/deploy_local_llm.mdx';
+  const optionsMap = {
+    HuggingFace: [{ value: 'embedding', label: 'embedding' }],
+    Xinference: [
+      { value: 'chat', label: 'chat' },
+      { value: 'embedding', label: 'embedding' },
+      { value: 'rerank', label: 'rerank' },
+      { value: 'image2text', label: 'image2text' },
+      { value: 'speech2text', label: 'sequence2text' },
+      { value: 'tts', label: 'tts' },
+    ],
+    Default: [
+      { value: 'chat', label: 'chat' },
+      { value: 'embedding', label: 'embedding' },
+      { value: 'rerank', label: 'rerank' },
+      { value: 'image2text', label: 'image2text' },
+    ],
+  };
+  const getOptions = (factory: string) => {
+    return optionsMap[factory as keyof typeof optionsMap] || optionsMap.Default;
+  };
   return (
     <Modal
       title={t('addLlmTitle', { name: llmFactory })}
@@ -85,18 +105,11 @@ const OllamaModal = ({
           rules={[{ required: true, message: t('modelTypeMessage') }]}
         >
           <Select placeholder={t('modelTypeMessage')}>
-            {llmFactory === 'HuggingFace' ? (
-              <Option value="embedding">embedding</Option>
-            ) : (
-              <>
-                <Option value="chat">chat</Option>
-                <Option value="embedding">embedding</Option>
-                <Option value="rerank">rerank</Option>
-                <Option value="image2text">image2text</Option>
-                <Option value="speech2text">sequence2text</Option>
-                <Option value="tts">tts</Option>
-              </>
-            )}
+            {getOptions(llmFactory).map((option) => (
+              <Option key={option.value} value={option.value}>
+                {option.label}
+              </Option>
+            ))}
           </Select>
         </Form.Item>
         <Form.Item<FieldType>