feat: download model

2025-03-19 20:09:43 +08:00
parent 254dd75cd0
commit 2ceefab18c
10 changed files with 971 additions and 317 deletions
@@ -0,0 +1,234 @@
+import IconFont from '@/components/icon-font';
+import HotKeys from '@/config/hotkeys';
+import {
+  DeleteOutlined,
+  EditOutlined,
+  ExperimentOutlined
+} from '@ant-design/icons';
+import _ from 'lodash';
+import React from 'react';
+import { modelCategoriesMap, modelSourceMap } from './index';
+
+const icons = {
+  EditOutlined: React.createElement(EditOutlined),
+  ExperimentOutlined: React.createElement(ExperimentOutlined),
+  DeleteOutlined: React.createElement(DeleteOutlined),
+  Stop: React.createElement(IconFont, { type: 'icon-stop1' }),
+  Play: React.createElement(IconFont, { type: 'icon-outline-play' }),
+  Catalog: React.createElement(IconFont, { type: 'icon-catalog' }),
+  HF: React.createElement(IconFont, { type: 'icon-huggingface' }),
+  Ollama: React.createElement(IconFont, { type: 'icon-ollama' }),
+  ModelScope: React.createElement(IconFont, { type: 'icon-tu2' }),
+  LocalPath: React.createElement(IconFont, { type: 'icon-hard-disk' })
+};
+
+export const modalConfig: Record<
+  string,
+  { show: boolean; width: string | number; source: any }
+> = {
+  huggingface: {
+    show: true,
+    width: 'calc(100vw - 220px)',
+    source: modelSourceMap.huggingface_value
+  },
+  ollama_library: {
+    show: true,
+    width: 600,
+    source: modelSourceMap.ollama_library_value
+  },
+  modelscope: {
+    show: true,
+    width: 'calc(100vw - 220px)',
+    source: modelSourceMap.modelscope_value
+  },
+  local_path: {
+    show: true,
+    width: 600,
+    source: modelSourceMap.local_path_value
+  }
+};
+
+interface ActionItem {
+  label: string;
+  key: string;
+  icon: React.ReactNode;
+  props?: {
+    danger?: boolean;
+  };
+}
+
+export const ActionList: ActionItem[] = [
+  {
+    label: 'common.button.edit',
+    key: 'edit',
+    icon: icons.EditOutlined
+  },
+  {
+    label: 'models.openinplayground',
+    key: 'chat',
+    icon: icons.ExperimentOutlined
+  },
+  {
+    label: 'common.button.stop',
+    key: 'stop',
+    icon: icons.Stop
+  },
+  {
+    label: 'common.button.start',
+    key: 'start',
+    icon: icons.Play
+  },
+  {
+    label: 'common.button.delete',
+    key: 'delete',
+    props: {
+      danger: true
+    },
+    icon: icons.DeleteOutlined
+  }
+];
+
+export const ButtonList = [
+  {
+    label: 'common.button.start',
+    key: 'start',
+    icon: icons.Play
+  },
+  {
+    label: 'common.button.stop',
+    key: 'stop',
+    icon: icons.Stop
+  },
+  {
+    label: 'common.button.delete',
+    key: 'delete',
+    icon: icons.DeleteOutlined,
+    props: {
+      danger: true
+    }
+  }
+];
+
+export const onLineSourceOptions = [
+  {
+    label: 'Hugging Face',
+    locale: false,
+    value: modelSourceMap.huggingface_value,
+    key: 'huggingface',
+    icon: icons.HF
+  },
+  {
+    label: 'Ollama Library',
+    locale: false,
+    value: modelSourceMap.ollama_library_value,
+    key: 'ollama_library',
+    icon: icons.Ollama
+  },
+  {
+    label: 'ModelScope',
+    locale: false,
+    value: modelSourceMap.modelscope_value,
+    key: 'modelscope',
+    icon: icons.ModelScope
+  }
+];
+
+export const sourceOptions = [
+  {
+    label: 'menu.models.modelCatalog',
+    locale: true,
+    value: 'catalog',
+    key: 'catalog',
+    icon: icons.Catalog
+  },
+  ...onLineSourceOptions,
+  {
+    label: 'models.form.localPath',
+    locale: true,
+    value: modelSourceMap.local_path_value,
+    key: 'local_path',
+    icon: icons.LocalPath
+  }
+];
+
+export const generateSource = (record: any) => {
+  if (record.source === modelSourceMap.modelscope_value) {
+    return `${modelSourceMap.modelScope}/${record.model_scope_model_id}`;
+  }
+  if (record.source === modelSourceMap.huggingface_value) {
+    return `${modelSourceMap.huggingface}/${record.huggingface_repo_id}`;
+  }
+  if (record.source === modelSourceMap.local_path_value) {
+    return `${record.local_path}`;
+  }
+  if (record.source === modelSourceMap.ollama_library_value) {
+    return `${modelSourceMap.ollama_library}/${record.ollama_library_model_name}`;
+  }
+  return '';
+};
+
+export const setModelActionList = (record: any) => {
+  return _.filter(ActionList, (action: any) => {
+    if (action.key === 'chat') {
+      return record.ready_replicas > 0;
+    }
+    if (action.key === 'start') {
+      return record.replicas === 0;
+    }
+
+    if (action.key === 'stop') {
+      return record.replicas > 0;
+    }
+
+    return true;
+  });
+};
+
+export const categoryToPathMap: Record<string, string> = {
+  [modelCategoriesMap.llm]: '/playground/chat',
+  [modelCategoriesMap.image]: '/playground/text-to-image',
+  [modelCategoriesMap.text_to_speech]: '/playground/speech?type=tts',
+  [modelCategoriesMap.speech_to_text]: '/playground/speech?type=stt',
+  [modelCategoriesMap.reranker]: '/playground/rerank',
+  [modelCategoriesMap.embedding]: '/playground/embedding'
+};
+
+export const hotkeyConfigs = [
+  {
+    keys: HotKeys.NEW1,
+    width: 'calc(100vw - 220px)',
+    source: modelSourceMap.huggingface_value
+  },
+  {
+    keys: HotKeys.NEW3,
+    width: 'calc(100vw - 220px)',
+    source: modelSourceMap.modelscope_value
+  },
+  {
+    keys: HotKeys.NEW2,
+    width: 600,
+    source: modelSourceMap.ollama_library_value
+  },
+  { keys: HotKeys.NEW4, width: 600, source: modelSourceMap.local_path_value }
+];
+
+/**
+ * hotkeyConfigs.map(({ keys, width, source }) =>
+       useHotkeys(
+         keys.join(','),
+         () => {
+           setOpenDeployModal({
+             show: true,
+             width,
+             source,
+             gpuOptions: gpuDeviceList.current
+           });
+         },
+         {
+           preventDefault: true,
+           enabled: !openAddModal && !openDeployModal.show && !openLogModal
+         }
+       )
+     );
+ * 
+ */