adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(web): custom model add thinking capability

Browse Source
This commit is contained in:
zhaoying
2026-03-31 15:48:19 +08:00
parent 52ae914e17
commit b40f4829cb
2 changed files with 39 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
web/src/views/ModelManagement/components/CustomModelModal.tsx
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing
* @Date: 2026-02-03 16:49:28
* @Last Modified by: ZhaoYing
* @Last Modified time: 2026-03-25 14:07:10
* @Last Modified time: 2026-03-31 13:56:18
*/
/**
* Custom Model Modal
@@ -11,7 +11,7 @@
*/
import { forwardRef, useEffect, useImperativeHandle, useState } from 'react';
import { Form, Input, App, Checkbox, Button } from 'antd';
import { Form, Input, App, Checkbox, Button, Row, Col } from 'antd';
import { useTranslation } from 'react-i18next';
import type { CustomModelForm, ModelListItem, CustomModelModalRef, CustomModelModalProps } from '../types';
@@ -72,6 +72,7 @@ const CustomModelModal = forwardRef<CustomModelModalRef, CustomModelModalProps>(
is_vision: capability?.includes('vision') || false,
is_video: capability?.includes('video') || false,
is_audio: capability?.includes('audio') || false,
is_thinking: capability?.includes('thinking') || false,
});
} else {
setIsEdit(false);
@@ -101,7 +102,7 @@ const CustomModelModal = forwardRef<CustomModelModalRef, CustomModelModalProps>(
form
.validateFields()
.then((values) => {
const { logo, type, is_vision, is_video, is_audio, is_omni, ...rest } = values;
const { logo, type, is_vision, is_video, is_audio, is_omni, is_thinking, ...rest } = values;
const formData: CustomModelForm = {
...rest,
type,
@@ -120,6 +121,9 @@ const CustomModelModal = forwardRef<CustomModelModalRef, CustomModelModalProps>(
capability.push('video')
}
}
if (is_thinking) {
capability.push('thinking')
}
formData.capability = capability
formData.is_omni = is_omni
@@ -238,21 +242,34 @@ const CustomModelModal = forwardRef<CustomModelModalRef, CustomModelModalProps>(
<Input placeholder="https://api.example.com/v1" />
</Form.Item>
{!['embedding', 'rerank'].includes(modelType as string) &&
<>
<Form.Item name="is_omni" valuePropName="checked" className="rb:mb-2!">
<Checkbox>{t('modelNew.is_omni')}</Checkbox>
</Form.Item>
<Form.Item name="is_vision" valuePropName="checked" className="rb:mb-2!">
<Checkbox disabled={isOmni}>{t('modelNew.is_vision')}</Checkbox>
</Form.Item>
<Form.Item name="is_video" valuePropName="checked" className="rb:mb-2!">
<Checkbox disabled={isOmni}>{t('modelNew.is_video')}</Checkbox>
</Form.Item>
<Form.Item name="is_audio" valuePropName="checked" className="rb:mb-0!">
<Checkbox disabled={isOmni}>{t('modelNew.is_audio')}</Checkbox>
</Form.Item>
</>
{['llm', 'chat'].includes(modelType as string) &&
<Row gutter={16}>
<Col span={24}>
<Form.Item name="is_omni" valuePropName="checked" className="rb:mb-2!">
<Checkbox>{t('modelNew.is_omni')}</Checkbox>
</Form.Item>
</Col>
<Col span={8}>
<Form.Item name="is_vision" valuePropName="checked" className="rb:mb-2!">
<Checkbox disabled={isOmni}>{t('modelNew.is_vision')}</Checkbox>
</Form.Item>
</Col>
<Col span={8}>
<Form.Item name="is_video" valuePropName="checked" className="rb:mb-2!">
<Checkbox disabled={isOmni}>{t('modelNew.is_video')}</Checkbox>
</Form.Item>
</Col>
<Col span={8}>
<Form.Item name="is_audio" valuePropName="checked" className="rb:mb-2!">
<Checkbox disabled={isOmni}>{t('modelNew.is_audio')}</Checkbox>
</Form.Item>
</Col>
<Col span={24}>
<Form.Item name="is_thinking" valuePropName="checked" className="rb:mb-0!">
<Checkbox>{t('modelNew.is_thinking')}</Checkbox>
</Form.Item>
</Col>
</Row>
}
</Form>
</RbModal>

7
web/src/views/ModelManagement/types.ts
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing
* @Date: 2026-02-03 16:50:18
* @Last Modified by: ZhaoYing
* @Last Modified time: 2026-03-25 12:28:10
* @Last Modified time: 2026-03-31 15:48:02
*/
/**
* Type definitions for Model Management
@@ -295,7 +295,8 @@ export interface CustomModelForm {
is_video?: boolean;
is_audio?: boolean;
is_omni?: boolean;
capability?: string[];
is_thinking?: boolean;
capability?: Capability[];
}
/**
@@ -324,7 +325,7 @@ export interface BaseRef {
modelListDetailRefresh?: () => void;
}
export type Capability = 'vision' | 'audio' | 'video';
export type Capability = 'vision' | 'audio' | 'video' | 'thinking';
export interface Model {
name: string;
type: string;