feat: add api access info modal

This commit is contained in:
jialin
2025-04-10 17:30:26 +08:00
parent 7c79accda7
commit 48580cb5b0
10 changed files with 313 additions and 24 deletions
@@ -0,0 +1,197 @@
import AutoTooltip from '@/components/auto-tooltip';
import CopyButton from '@/components/copy-button';
import IconFont from '@/components/icon-font';
import ScrollerModal from '@/components/scroller-modal';
import { BulbOutlined } from '@ant-design/icons';
import { useIntl, useNavigate } from '@umijs/max';
import { Button, Tag } from 'antd';
import _ from 'lodash';
import { useMemo } from 'react';
import styled from 'styled-components';
import { modelCategoriesMap } from '../config';
const ApiAccessInfoWrapper = styled.div`
display: flex;
flex-direction: column;
gap: 12px;
padding-left: 20px;
.item {
display: flex;
justify-content: space-between;
align-items: center;
.label-wrapper {
display: flex;
justify-content: flex-start;
align-items: center;
gap: 16px;
.label {
width: 86px;
font-weight: 600;
}
.value {
display: flex;
align-items: center;
color: var(--ant-color-text-secondary);
}
}
.copy-btn {
display: flex;
align-items: center;
gap: 4px;
}
}
`;
const Tips = styled.div`
color: var(--ant-color-text-secondary);
font-size: var(--font-size-small);
.tips {
display: flex;
align-items: flex-start;
gap: 8px;
}
dd {
margin-bottom: 16px;
}
`;
const APITAG = styled(Tag)`
border-radius: 12px;
margin: 0;
margin-left: 8px;
`;
const CreateButton = styled(Button)`
padding-inline: 0;
`;
interface ApiAccessInfoProps {
open: boolean;
data: any;
onClose: () => void;
}
const ApiAccessInfo = ({ open, data, onClose }: ApiAccessInfoProps) => {
const intl = useIntl();
const navigate = useNavigate();
const endPoint = `${window.location.origin}/v1`;
const isRanker = useMemo(() => {
return _.includes(data.categories, modelCategoriesMap.reranker);
}, [data]);
const handleClose = () => {
onClose();
};
return (
<ScrollerModal
open={open}
style={{
top: '20%'
}}
title={intl.formatMessage({ id: 'models.table.button.apiAccessInfo' })}
width={510}
destroyOnClose
closable={true}
maskClosable={false}
onOk={handleClose}
onCancel={handleClose}
styles={{
content: {
padding: '0 0 16px 0'
},
header: {
padding: 'var(--ant-modal-content-padding)',
paddingBottom: '0'
},
body: {
padding: '16px 24px 32px'
},
footer: {
padding: '16px 24px',
margin: '0'
}
}}
footer={null}
>
<Tips>
<dl className="tips">
<dt>
<BulbOutlined />
</dt>
<dd>
{intl.formatMessage({
id: 'models.table.button.apiAccessInfo.tips'
})}
</dd>
</dl>
</Tips>
<ApiAccessInfoWrapper>
<div className="item">
<span className="label-wrapper">
<span className="label">
{intl.formatMessage({ id: 'models.table.apiAccessInfo.enpoint' })}
</span>
<span className="value">
<AutoTooltip ghost maxWidth={200}>
{endPoint}
</AutoTooltip>
<APITAG color="geekblue">
{intl.formatMessage({
id: isRanker
? 'models.table.apiAccessInfo.jinaCompatible'
: 'models.table.apiAccessInfo.openaiCompatible'
})}
</APITAG>
</span>
</span>
<span className="copy-btn">
<CopyButton text={endPoint} type="link" size="small"></CopyButton>
</span>
</div>
<div className="item">
<span className="label-wrapper">
<span className="label">
{intl.formatMessage({
id: 'models.table.apiAccessInfo.modelName'
})}
</span>
<span className="value">
<AutoTooltip ghost maxWidth={300}>
{data.name}
</AutoTooltip>
</span>
</span>
<span className="copy-btn">
<CopyButton text={data.name} type="link" size="small"></CopyButton>
</span>
</div>
<div className="item">
<span className="label-wrapper">
<span className="label">
{intl.formatMessage({ id: 'models.table.apiAccessInfo.apikey' })}
</span>
<span className="value">
<CreateButton
type="link"
size="small"
onClick={() => navigate('/api-keys')}
>
{intl.formatMessage({
id: 'models.table.apiAccessInfo.gotoCreate'
})}
<IconFont
type="icon-external-link"
className="font-size-14"
></IconFont>
</CreateButton>
</span>
</span>
</div>
</ApiAccessInfoWrapper>
</ScrollerModal>
);
};
export default ApiAccessInfo;
+30 -1
View File
@@ -78,12 +78,12 @@ import {
SourceType
} from '../config/types';
import { useGenerateFormEditInitialValues } from '../hooks';
import APIAccessInfoModal from './api-access-info';
import DeployModal from './deploy-modal';
import Instances from './instances';
import ModelTag from './model-tag';
import UpdateModel from './update-modal';
import ViewLogsModal from './view-logs-modal';
interface ModelsProps {
handleSearch: () => void;
handleNameChange: (e: any) => void;
@@ -174,6 +174,10 @@ const Models: React.FC<ModelsProps> = ({
defaultSortOrder: 'descend'
});
const [apiAccessInfo, setAPIAccessInfo] = useState<any>({
show: false,
data: {}
});
const [openLogModal, setOpenLogModal] = useState(false);
const [openAddModal, setOpenAddModal] = useState(false);
const [openDeployModal, setOpenDeployModal] = useState<{
@@ -438,6 +442,17 @@ const Models: React.FC<ModelsProps> = ({
saveScrollHeight();
};
const handleViewAPIInfo = useCallback((row: ListItem) => {
setAPIAccessInfo({
show: true,
data: {
id: row.id,
name: row.name,
categories: row.categories,
url: `${MODELS_API}/${row.id}/instances`
}
});
}, []);
const handleSelect = useCallback(
async (val: any, row: ListItem) => {
try {
@@ -456,6 +471,10 @@ const Models: React.FC<ModelsProps> = ({
updateExpandedRowKeys([row.id, ...expandedRowKeys]);
}
if (val === 'api') {
handleViewAPIInfo(row);
}
if (val === 'stop') {
modalRef.current.show({
content: 'models.instances',
@@ -848,6 +867,16 @@ const Models: React.FC<ModelsProps> = ({
onCancel={handleLogModalCancel}
></ViewLogsModal>
<DeleteModal ref={modalRef}></DeleteModal>
<APIAccessInfoModal
open={apiAccessInfo.show}
data={apiAccessInfo.data}
onClose={() => {
setAPIAccessInfo({
...apiAccessInfo,
show: false
});
}}
></APIAccessInfoModal>
</>
);
};
+8 -1
View File
@@ -1,6 +1,7 @@
import IconFont from '@/components/icon-font';
import HotKeys from '@/config/hotkeys';
import {
ApiOutlined,
DeleteOutlined,
DownloadOutlined,
EditOutlined,
@@ -19,6 +20,7 @@ const icons = {
ThunderboltOutlined: React.createElement(ThunderboltOutlined),
RetweetOutlined: React.createElement(RetweetOutlined),
DownloadOutlined: React.createElement(DownloadOutlined),
ApiOutlined: React.createElement(ApiOutlined),
Stop: React.createElement(IconFont, { type: 'icon-stop1' }),
Play: React.createElement(IconFont, { type: 'icon-outline-play' }),
Catalog: React.createElement(IconFont, { type: 'icon-catalog' }),
@@ -76,6 +78,11 @@ export const ActionList: ActionItem[] = [
key: 'chat',
icon: icons.ExperimentOutlined
},
{
label: 'models.table.button.apiAccessInfo',
key: 'api',
icon: icons.ApiOutlined
},
{
label: 'common.button.stop',
key: 'stop',
@@ -177,7 +184,7 @@ export const generateSource = (record: any) => {
export const setModelActionList = (record: any) => {
return _.filter(ActionList, (action: any) => {
if (action.key === 'chat') {
if (action.key === 'chat' || action.key === 'api') {
return record.ready_replicas > 0;
}
if (action.key === 'start') {
+8 -6
View File
@@ -306,14 +306,16 @@ export const useCheckCompatibility = () => {
if (hasClaim) {
const ram = convertFileSize(resource_claim.ram, 2);
const vram = convertFileSize(resource_claim.vram, 2);
let messageId = 'models.form.check.claims';
if (!ram) {
messageId = 'models.form.check.claims2';
}
if (!vram) {
messageId = 'models.form.check.claims3';
}
msgData = {
title: intl.formatMessage({ id: 'models.form.check.passed' }),
message: ram
? intl.formatMessage(
{ id: 'models.form.check.claims' },
{ ram, vram }
)
: intl.formatMessage({ id: 'models.form.check.claims2' }, { vram })
message: intl.formatMessage({ id: messageId }, { ram, vram })
};
}