chore: display the direct command in add worker

src/app.tsx

@@ -20,7 +20,10 @@ export async function getInitialState(): Promise<{
   const getUpdateCheck = async () => {
     try {
       const data = await updateCheck();
-      setAtomStorage(UpdateCheckAtom, data);
+
+      setAtomStorage(UpdateCheckAtom, {
+        ...data
+      });
       return data;
     } catch (error) {
       console.error('updateCheck error', error);
@@ -45,7 +48,11 @@ export async function getInitialState(): Promise<{
   const getAppVersionInfo = async () => {
     try {
       const data = await queryVersionInfo();
-      setAtomStorage(GPUStackVersionAtom, data);
+      const isProduction = data.version?.indexOf('0.0.0') === -1;
+      setAtomStorage(GPUStackVersionAtom, {
+        ...data,
+        isProduction
+      });
     } catch (error) {
       console.error('queryVersionInfo error', error);
     }

src/assets/styles/driver.less

@@ -0,0 +1,9 @@
+.driver-popover {
+  color: var(--ant-color-text);
+
+  .driver-popover-title {
+    font-size: 16px;
+    font-weight: var(--font-weight-bold);
+    color: var(--ant-color-text);
+  }
+}

src/atoms/user.ts

@@ -6,9 +6,11 @@ export const userAtom = atomWithStorage<any>('userInfo', null);
 export const GPUStackVersionAtom = atom<{
   version: string;
   git_commit: string;
+  isProduction: boolean;
 }>({
   version: '',
-  git_commit: ''
+  git_commit: '',
+  isProduction: false
 });
 
 export const UpdateCheckAtom = atom<{

src/components/radio-buttons/index.less

@@ -4,10 +4,9 @@
     justify-content: center;
     align-items: center;
     font-size: var(--font-size-base);
-    padding: 2px;
     border-radius: var(--border-radius-base);
-    height: 40px;
-    width: 40px;
+    height: 32px;
+    padding: 0 8px;
     border: 1px solid var(--ant-color-border);
     cursor: pointer;
 

src/components/radio-buttons/index.tsx

@@ -16,7 +16,7 @@ const RadioButtons: React.FC<RadioButtonsProps> = (props) => {
       {options.map((option) => (
         <span
           key={option.value}
-          onClick={() => onChange(option.value)}
+          onClick={() => onChange({ target: { value: option.value } } as any)}
           className={classNames('item', { active: value === option.value })}
         >
           {option.label}

src/config/driver-config.ts

@@ -0,0 +1,5 @@
+const driverIDMap = {
+  deployModel: 'deploy-model'
+};
+
+const driverConfig = {};

src/global.less

@@ -1,5 +1,6 @@
 @import url('src/assets/styles/common.less');
 @import url('src/assets/styles/menu.less');
+@import url('src/assets/styles/driver.less');
 
 html {
   --font-family: 'noto sans', sans-serif;

src/hooks/use-driver.ts

@@ -2,7 +2,7 @@ import { useIntl } from '@umijs/max';
 import { driver, type Config } from 'driver.js';
 import { useEffect, useRef } from 'react';
 
-export const useDriver = (config?: Config) => {
+export const useDriver = (config?: Config & { id: string }) => {
   const intl = useIntl();
   const driverRef = useRef<any>(null);
 

src/locales/en-US/common.ts

@@ -212,5 +212,6 @@ export default {
   'common.text.latest': 'Latest',
   'common.text.new': 'New',
   'common.text.changelog': 'Release Notes',
-  'common.button.recreate': 'Recreate'
+  'common.button.recreate': 'Recreate',
+  'common.button.delrecreate': 'Delete (Recreate)'
 };

src/locales/en-US/models.ts

@@ -78,5 +78,13 @@ export default {
   'models.form.backendVersion': 'Backend Version',
   'models.form.backendVersion.tips':
     'Pin a specific version to keep the backend stable across GPUStack upgrades.',
-  'models.form.gpuselector': 'GPU Selector'
+  'models.form.gpuselector': 'GPU Selector',
+  'models.form.backend.llamabox':
+    'llama-box: For GGUF format models, supports Linux, macOS, and Windows.',
+  'models.form.backend.vllm':
+    'vLLM: For non-GGUF format models, supports x86 Linux only.',
+  'models.form.backend.voxbox': 'vox-box: For non-GGUF format audio models.',
+  'models.form.search.gguftips':
+    'GGUF format is required for Mac and Windows compatibility.',
+  'models.form.button.addlabel': 'Add Label'
 };

src/locales/en-US/playground.ts

@@ -97,8 +97,8 @@ export default {
   'playground.audio.button.generate': 'Generate Text Content',
   'playground.multiple.on': 'Enable',
   'playground.multiple.off': 'Disable',
-  'playground.image.params.sampler': 'Sampler',
-  'playground.image.params.schedule': 'Scheduler',
+  'playground.image.params.sampler': 'Sample Method',
+  'playground.image.params.schedule': 'Schedule Method',
   'playground.image.params.samplerSteps': 'Sampling Steps',
   'playground.image.params.seed': 'Seed',
   'playground.image.params.randomseed': 'Random Seed',

src/locales/en-US/resources.ts

@@ -46,5 +46,9 @@ export default {
   'resources.worker.add.step2.tips':
     'Note: <span style="color: #000;font-weight: 600">mytoken</span> is the token obtained in the first step.',
   'resources.worker.add.step3':
-    'Refresh workers list, you will see the new worker.'
+    'After success, refresh the workers list to see the new worker.',
+  'resources.worker.container.supported': 'Do not support windows.',
+  'resources.worker.current.version': 'Current version is {version}.',
+  'resources.worker.select.command':
+    'Select a label to generate the command and copy it using the copy button.'
 };

src/locales/zh-CN/common.ts

@@ -205,5 +205,6 @@ export default {
   'common.text.latest': '最新',
   'common.text.new': '新',
   'common.text.changelog': '更新日志',
-  'common.button.recreate': '重新创建'
+  'common.button.recreate': '重新创建',
+  'common.button.delrecreate': '删除(重建)'
 };

src/locales/zh-CN/models.ts

@@ -75,5 +75,12 @@ export default {
   'models.form.backendVersion': '后端版本',
   'models.form.backendVersion.tips':
     '固定指定版本以保持后端在 GPUStack 升级过程中的稳定性',
-  'models.form.gpuselector': 'GPU 选择器'
+  'models.form.gpuselector': 'GPU 选择器',
+  'models.form.backend.llamabox':
+    'llama-box: 用于 GGUF 格式模型，支持 Linux, macOS 和 Windows',
+  'models.form.backend.vllm': 'vLLM: 用于非 GGUF 格式模型，仅支持 x86 Linux',
+  'models.form.backend.voxbox': 'vox-box: 用于非 GGUF 格式的音频模型',
+  'models.form.search.gguftips':
+    'Mac 和 Windows 需要使用 GGUF 格式以确保兼容性',
+  'models.form.button.addlabel': '添加标签'
 };

src/locales/zh-CN/playground.ts

@@ -95,8 +95,8 @@ export default {
   'playground.multiple.on': '开启',
   'playground.multiple.off': '关闭',
   'playground.image.params.sampler': '采样方法',
-  'playground.image.params.schedule': '调度器',
-  'playground.image.params.samplerSteps': '迭代步数',
+  'playground.image.params.schedule': '调度方法',
+  'playground.image.params.samplerSteps': '采样步数',
   'playground.image.params.seed': '种子',
   'playground.image.params.randomseed': '随机种子',
   'playground.image.params.negativePrompt': '负向提示',

src/locales/zh-CN/resources.ts

@@ -45,5 +45,8 @@ export default {
   'resources.worker.add.step2': '注册 Worker',
   'resources.worker.add.step2.tips':
     '注意：<span style="color: #000;font-weight: 600">mytoken</span> 为第一步获取到的 Token',
-  'resources.worker.add.step3': '刷新 workers 列表，可以看到新添加的 worker'
+  'resources.worker.add.step3': '成功后，刷新 worker 列表即可看到新的 worker',
+  'resources.worker.container.supported': '不支持 Windows',
+  'resources.worker.current.version': '当前版本为 {version}',
+  'resources.worker.select.command': '选择一个标签生成命令并使用复制按钮复制'
 };

src/pages/llmodels/components/advance-config.tsx

@@ -153,6 +153,21 @@ const AdvanceConfig: React.FC<AdvanceConfigProps> = (props) => {
   const collapseItems = useMemo(() => {
     const children = (
       <>
+        {/* <Form.Item<FormData> name="labels">
+          <ListInput
+            placeholder={
+              backendParamsHolderTips[backend]
+                ? intl.formatMessage({
+                    id: backendParamsHolderTips[backend].holder
+                  })
+                : ''
+            }
+            btnText="models.form.button.addlabel"
+            label="Labels"
+            dataList={form.getFieldValue('backend_parameters') || []}
+            options={modelLabels}
+          ></ListInput>
+        </Form.Item> */}
         <Form.Item name="scheduleType">
           <SealSelect
             label={intl.formatMessage({ id: 'models.form.scheduletype' })}

src/pages/llmodels/components/data-form.tsx

@@ -508,38 +508,54 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
           min={0}
         ></SealInput.Number>
       </Form.Item>
-      <Form.Item name="backend" id="backend-field">
-        <SealSelect
-          onChange={handleBackendChange}
-          label={intl.formatMessage({ id: 'models.form.backend' })}
-          options={[
-            {
-              label: `llama-box`,
-              value: backendOptionsMap.llamaBox,
-              disabled:
-                props.source === modelSourceMap.local_path_value
-                  ? false
-                  : !isGGUF
-            },
-            {
-              label: 'vLLM',
-              value: backendOptionsMap.vllm,
-              disabled:
-                props.source === modelSourceMap.local_path_value
-                  ? false
-                  : isGGUF
-            },
-            {
-              label: 'vox-box',
-              value: backendOptionsMap.voxBox,
-              disabled: props.source === modelSourceMap.ollama_library_value
+      <Form.Item name="backend">
+        <div id="backend-field">
+          <SealSelect
+            onChange={handleBackendChange}
+            label={intl.formatMessage({ id: 'models.form.backend' })}
+            description={
+              <div>
+                <div>
+                  1.{' '}
+                  {intl.formatMessage({ id: 'models.form.backend.llamabox' })}
+                </div>
+                <div>
+                  2. {intl.formatMessage({ id: 'models.form.backend.vllm' })}
+                </div>
+                <div>
+                  3. {intl.formatMessage({ id: 'models.form.backend.voxbox' })}
+                </div>
+              </div>
             }
-          ]}
-          disabled={
-            action === PageAction.EDIT &&
-            props.source !== modelSourceMap.local_path_value
-          }
-        ></SealSelect>
+            options={[
+              {
+                label: `llama-box`,
+                value: backendOptionsMap.llamaBox,
+                disabled:
+                  props.source === modelSourceMap.local_path_value
+                    ? false
+                    : !isGGUF
+              },
+              {
+                label: 'vLLM',
+                value: backendOptionsMap.vllm,
+                disabled:
+                  props.source === modelSourceMap.local_path_value
+                    ? false
+                    : isGGUF
+              },
+              {
+                label: 'vox-box',
+                value: backendOptionsMap.voxBox,
+                disabled: props.source === modelSourceMap.ollama_library_value
+              }
+            ]}
+            disabled={
+              action === PageAction.EDIT &&
+              props.source !== modelSourceMap.local_path_value
+            }
+          ></SealSelect>
+        </div>
       </Form.Item>
       <Form.Item<FormData> name="description">
         <SealInput.TextArea

src/pages/llmodels/components/deploy-modal.tsx

@@ -1,6 +1,5 @@
 import ModalFooter from '@/components/modal-footer';
 import { PageActionType } from '@/config/types';
-import useDriver from '@/hooks/use-driver';
 import { CloseOutlined } from '@ant-design/icons';
 import { useIntl } from '@umijs/max';
 import { Button, Drawer } from 'antd';
@@ -58,9 +57,10 @@ const AddModal: React.FC<AddModalProps> = (props) => {
     modelSourceMap.huggingface_value,
     modelSourceMap.modelscope_value
   ];
-  const { start } = useDriver({
-    steps
-  });
+  // const { start } = useDriver({
+  //   steps,
+  //   id: 'deploy-model'
+  // });
   const form = useRef<any>({});
   const intl = useIntl();
   const [selectedModel, setSelectedModel] = useState<any>({});
@@ -128,7 +128,9 @@ const AddModal: React.FC<AddModalProps> = (props) => {
 
   // useEffect(() => {
   //   if (open && loadfinish) {
-  //     start();
+  //     setTimeout(() => {
+  //       start();
+  //     }, 1000);
   //   }
   // }, [loadfinish, open]);
 

src/pages/llmodels/components/instance-item.tsx

@@ -8,8 +8,8 @@ import {
   ListItem as WorkerListItem
 } from '@/pages/resources/config/types';
 import {
+  DeleteOutlined,
   FieldTimeOutlined,
-  FileSyncOutlined,
   InfoCircleOutlined
 } from '@ant-design/icons';
 import { useIntl } from '@umijs/max';
@@ -71,12 +71,12 @@ const InstanceItem: React.FC<InstanceItemProps> = ({
       icon: <FieldTimeOutlined />
     },
     {
-      label: 'common.button.recreate',
+      label: 'common.button.delrecreate',
       key: 'delete',
       props: {
-        danger: false
+        danger: true
       },
-      icon: <FileSyncOutlined />
+      icon: <DeleteOutlined />
     }
   ];
 

src/pages/llmodels/components/search-model.tsx

@@ -1,4 +1,8 @@
-import { BulbOutlined, InfoCircleOutlined } from '@ant-design/icons';
+import {
+  BulbOutlined,
+  QuestionCircleOutlined,
+  WarningOutlined
+} from '@ant-design/icons';
 import { useIntl } from '@umijs/max';
 import { Checkbox, Select, Tooltip } from 'antd';
 import _ from 'lodash';
@@ -233,6 +237,12 @@ const SearchModel: React.FC<SearchInputProps> = (props) => {
           onChange={handleSearchInputChange}
           modelSource={modelSource}
         ></SearchInput>
+        <div className="gguf-tips">
+          <WarningOutlined className="font-size-14 m-r-5 warning" />
+          <span>
+            {intl.formatMessage({ id: 'models.form.search.gguftips' })}
+          </span>
+        </div>
         <div className={SearchStyle.filter}>
           <span>
             <span className="value">
@@ -268,7 +278,7 @@ const SearchModel: React.FC<SearchInputProps> = (props) => {
                   }
                 >
                   GGUF
-                  <InfoCircleOutlined className="m-l-4" />
+                  <QuestionCircleOutlined className="m-l-4" />
                 </Tooltip>
               </Checkbox>
             </span>
@@ -318,6 +328,7 @@ const SearchModel: React.FC<SearchInputProps> = (props) => {
           </div>
         )}
       </div>
+
       {
         <SearchResult
           loading={dataSource.loading}

src/pages/llmodels/components/search-result.tsx

@@ -87,7 +87,7 @@ const SearchResult: React.FC<SearchResultProps> = (props) => {
     );
   };
   return (
-    <SimpleBar style={{ height: 'calc(100vh - 194px)' }}>
+    <SimpleBar style={{ height: 'calc(100vh - 224px)' }}>
       <div style={{ ...props.style }} className="search-result-wrap">
         <Spin spinning={props.loading}>
           <div style={{ minHeight: 200 }}>

src/pages/llmodels/components/table-list.tsx

@@ -449,7 +449,7 @@ const Models: React.FC<ModelsProps> = ({
   const handleDeleteInstace = (row: any, list: ModelInstanceListItem[]) => {
     modalRef.current.show({
       content: 'models.instances',
-      okText: 'common.button.recreate',
+      okText: 'common.button.delrecreate',
       name: row.name,
       async onOk() {
         await deleteModelInstance(row.id);

src/pages/llmodels/config/index.ts

@@ -345,3 +345,11 @@ export const getVllmCliArgs = (inputString: string) => {
 
   return result;
 };
+
+export const modelLabels = [
+  { label: 'Image', value: 'image_only' },
+  { label: 'Text-to-speech', value: 'text_to_speech' },
+  { label: 'Speech-to-text', value: 'speech_to_text' },
+  { label: 'reranker', value: 'reranker' },
+  { label: 'Embedding', value: 'embedding_only' }
+];

src/pages/llmodels/config/llama-config.ts

@@ -104,6 +104,10 @@ const options = [
   {
     label: '--image-no-vae-tiling',
     value: '--image-no-vae-tiling'
+  },
+  {
+    label: '--mmproj',
+    value: '--mmproj'
   }
 ];
 

src/pages/llmodels/style/search-result.less

@@ -8,6 +8,17 @@
   // height: calc(100vh - 194px);
 }
 
+.gguf-tips {
+  display: flex;
+  align-items: center;
+  padding: 8px 0;
+  color: var(--ant-color-text-secondary);
+
+  .warning {
+    color: var(--ant-color-warning);
+  }
+}
+
 .search-bar {
   left: 0;
   right: 0;

src/pages/resources/components/add-worker.tsx

@@ -1,8 +1,8 @@
-import HighlightCode from '@/components/highlight-code';
 import { useIntl } from '@umijs/max';
-import { Modal } from 'antd';
+import { Modal, Tabs, TabsProps } from 'antd';
 import React from 'react';
-import { addWorkerGuide } from '../config';
+import ContainerInstall from './container-install';
+import ScriptInstall from './script-install';
 
 type ViewModalProps = {
   open: boolean;
@@ -12,8 +12,21 @@ type ViewModalProps = {
 const AddWorker: React.FC<ViewModalProps> = (props) => {
   const { open, onCancel } = props || {};
   const intl = useIntl();
+  const [token, setToken] = React.useState('');
+  const [activeKey, setActiveKey] = React.useState('script');
 
-  const origin = window.location.origin;
+  const items: TabsProps['items'] = [
+    {
+      key: 'script',
+      label: 'Script Installation',
+      children: <ScriptInstall token={token}></ScriptInstall>
+    },
+    {
+      key: 'container',
+      label: 'Container Installation',
+      children: <ContainerInstall token={token} />
+    }
+  ];
 
   return (
     <Modal
@@ -26,49 +39,19 @@ const AddWorker: React.FC<ViewModalProps> = (props) => {
       maskClosable={false}
       keyboard={false}
       width={600}
+      styles={{
+        body: {
+          height: 310
+        }
+      }}
       footer={null}
     >
-      <div>
-        <h3>1. {intl.formatMessage({ id: 'resources.worker.add.step1' })}</h3>
-        <h4>{intl.formatMessage({ id: 'resources.worker.linuxormaxos' })}</h4>
-        <HighlightCode
-          code={addWorkerGuide.mac.getToken}
-          theme="dark"
-        ></HighlightCode>
-        <h4>Windows </h4>
-        <HighlightCode
-          code={addWorkerGuide.win.getToken}
-          theme="dark"
-        ></HighlightCode>
-        <h3>
-          2. {intl.formatMessage({ id: 'resources.worker.add.step2' })}{' '}
-          <span
-            className="font-size-12"
-            style={{ color: 'var(--ant-color-text-tertiary)' }}
-            dangerouslySetInnerHTML={{
-              __html: `(${intl.formatMessage({
-                id: 'resources.worker.add.step2.tips'
-              })})`
-            }}
-          ></span>
-        </h3>
-        <h4>{intl.formatMessage({ id: 'resources.worker.linuxormaxos' })}</h4>
-        <HighlightCode
-          code={addWorkerGuide.mac.registerWorker(origin)}
-          theme="dark"
-        ></HighlightCode>
-        <h4>Windows </h4>
-        <HighlightCode
-          theme="dark"
-          code={addWorkerGuide.win.registerWorker(origin)}
-        ></HighlightCode>
-        <h4>Docker </h4>
-        <HighlightCode
-          theme="dark"
-          code={addWorkerGuide.docker.registerWorker(origin)}
-        ></HighlightCode>
-        <h3>3. {intl.formatMessage({ id: 'resources.worker.add.step3' })}</h3>
-      </div>
+      <Tabs
+        items={items}
+        activeKey={activeKey}
+        type="card"
+        onChange={(key) => setActiveKey(key)}
+      ></Tabs>
     </Modal>
   );
 };

src/pages/resources/components/container-install.tsx

@@ -0,0 +1,86 @@
+import { GPUStackVersionAtom } from '@/atoms/user';
+import { getAtomStorage } from '@/atoms/utils';
+import HighlightCode from '@/components/highlight-code';
+import { BulbOutlined, WarningOutlined } from '@ant-design/icons';
+import { useIntl } from '@umijs/max';
+import { Radio } from 'antd';
+import React from 'react';
+import { addWorkerGuide, containerInstallOptions } from '../config';
+import './styles/installation.less';
+
+type ViewModalProps = {
+  token: string;
+};
+
+const AddWorker: React.FC<ViewModalProps> = (props) => {
+  const intl = useIntl();
+
+  const origin = window.location.origin;
+  const [activeKey, setActiveKey] = React.useState('cuda');
+  const versionInfo = getAtomStorage(GPUStackVersionAtom);
+
+  const code = React.useMemo(() => {
+    let version = versionInfo?.version;
+    if (!version || !versionInfo.isProduction) {
+      version = 'main';
+    }
+    if (activeKey === 'cuda') {
+      return addWorkerGuide.docker.registerWorker({
+        server: origin,
+        tag: version,
+        token: props.token
+      });
+    }
+    return addWorkerGuide.docker.registerWorker({
+      server: origin,
+      tag: `${version}-${activeKey}`,
+      token: props.token
+    });
+  }, [versionInfo, activeKey, props.token, origin]);
+
+  return (
+    <div className="container-install">
+      <div className="notes">
+        <span>
+          1.{' '}
+          <span>
+            {intl.formatMessage({ id: 'resources.worker.container.supported' })}
+          </span>
+          <WarningOutlined
+            style={{ color: 'var(--ant-color-warning)' }}
+            className="font-size-14 m-l-5"
+          />
+        </span>
+        <span>
+          2.{' '}
+          {intl.formatMessage(
+            { id: 'resources.worker.current.version' },
+            { version: versionInfo.version }
+          )}
+        </span>
+        <span>
+          3. {intl.formatMessage({ id: 'resources.worker.select.command' })}
+        </span>
+      </div>
+      <div className="m-b-20">
+        <Radio.Group
+          block
+          options={containerInstallOptions}
+          defaultValue={activeKey}
+          value={activeKey}
+          optionType="button"
+          buttonStyle="solid"
+          onChange={(e) => setActiveKey(e.target.value)}
+          size="small"
+        />
+      </div>
+      <HighlightCode theme="dark" code={code}></HighlightCode>
+      <h3>
+        <BulbOutlined className="m-r-5"></BulbOutlined>
+        {intl.formatMessage({ id: 'resources.worker.add.step3' })}
+      </h3>
+    </div>
+  );
+};
+
+export default React.memo(AddWorker);

src/pages/resources/components/script-install.tsx

@@ -0,0 +1,40 @@
+import HighlightCode from '@/components/highlight-code';
+import { BulbOutlined } from '@ant-design/icons';
+import { useIntl } from '@umijs/max';
+import React from 'react';
+import { addWorkerGuide } from '../config';
+
+type ViewModalProps = { token: string };
+
+const AddWorker: React.FC<ViewModalProps> = (props) => {
+  const intl = useIntl();
+
+  const origin = window.location.origin;
+
+  return (
+    <div>
+      <h4>{intl.formatMessage({ id: 'resources.worker.linuxormaxos' })}</h4>
+      <HighlightCode
+        code={addWorkerGuide.mac.registerWorker({
+          server: origin,
+          token: props.token
+        })}
+        theme="dark"
+      ></HighlightCode>
+      <h4>Windows </h4>
+      <HighlightCode
+        theme="dark"
+        code={addWorkerGuide.win.registerWorker({
+          server: origin,
+          token: props.token
+        })}
+      ></HighlightCode>
+      <h3>
+        <BulbOutlined className="m-r-5"></BulbOutlined>
+        {intl.formatMessage({ id: 'resources.worker.add.step3' })}
+      </h3>
+    </div>
+  );
+};
+
+export default React.memo(AddWorker);

src/pages/resources/components/styles/installation.less

@@ -0,0 +1,13 @@
+.container-install {
+  .notes {
+    padding: 10px 0;
+    display: flex;
+    flex-direction: column;
+    gap: 5px;
+    border-left: 2px solid var(--ant-color-split);
+    margin-bottom: 20px;
+    padding-left: 16px;
+    background: var(--color-fill-sider);
+    border-radius: 2px;
+  }
+}

src/pages/resources/config/index.ts

@@ -18,22 +18,29 @@ export const status: any = {
 export const addWorkerGuide = {
   mac: {
     getToken: 'cat /var/lib/gpustack/token',
-    registerWorker(server: string) {
-      return `curl -sfL https://get.gpustack.ai | sh -s - --server-url ${server} --token mytoken`;
+    registerWorker(params: { server: string; token: string }) {
+      return `curl -sfL https://get.gpustack.ai | sh -s - --server-url ${params.server} --token ${params.token}`;
     }
   },
   win: {
     getToken:
       'Get-Content -Path (Join-Path -Path $env:APPDATA -ChildPath "gpustack\\token") -Raw',
-    registerWorker(server: string) {
-      return `Invoke-Expression "& { $((Invoke-WebRequest -Uri "https://get.gpustack.ai" -UseBasicParsing).Content) } --server-url ${server} --token mytoken"`;
+    registerWorker(params: { server: string; token: string }) {
+      return `Invoke-Expression "& { $((Invoke-WebRequest -Uri "https://get.gpustack.ai" -UseBasicParsing).Content) } --server-url ${params.server} --token ${params.token}"`;
     }
   },
   docker: {
     getToken:
       'Get-Content -Path (Join-Path -Path $env:APPDATA -ChildPath "gpustack\\token") -Raw',
-    registerWorker(server: string) {
-      return `docker run -d --gpus all --ipc=host --network=host gpustack/gpustack --server-url ${server} --token mytoken`;
+    registerWorker(params: { server: string; tag: string; token: string }) {
+      return `docker run -d --gpus all --ipc=host --network=host gpustack/gpustack:${params.tag} --server-url ${params.server} --token ${params.token}`;
     }
   }
 };
+
+export const containerInstallOptions = [
+  { label: 'CUDA', value: 'cuda' },
+  { label: 'CANN', value: 'npu' },
+  { label: 'MUSA', value: 'musa' },
+  { label: 'CPU', value: 'cpu' }
+];