255 lines
8.4 KiB
JavaScript
255 lines
8.4 KiB
JavaScript
import React, { useEffect, useState } from 'react';
|
||
import { Button, Form, Header, Message, Segment } from 'semantic-ui-react';
|
||
import { useParams } from 'react-router-dom';
|
||
import { API, showError, showInfo, showSuccess } from '../../helpers';
|
||
import { CHANNEL_OPTIONS } from '../../constants';
|
||
|
||
const EditChannel = () => {
|
||
const params = useParams();
|
||
const channelId = params.id;
|
||
const isEdit = channelId !== undefined;
|
||
const [loading, setLoading] = useState(isEdit);
|
||
const originInputs = {
|
||
name: '',
|
||
type: 1,
|
||
key: '',
|
||
base_url: '',
|
||
other: '',
|
||
group: 'default',
|
||
models: [],
|
||
};
|
||
const [batch, setBatch] = useState(false);
|
||
const [inputs, setInputs] = useState(originInputs);
|
||
const [modelOptions, setModelOptions] = useState([]);
|
||
const [basicModels, setBasicModels] = useState([]);
|
||
const [fullModels, setFullModels] = useState([]);
|
||
const handleInputChange = (e, { name, value }) => {
|
||
setInputs((inputs) => ({ ...inputs, [name]: value }));
|
||
};
|
||
|
||
const loadChannel = async () => {
|
||
let res = await API.get(`/api/channel/${channelId}`);
|
||
const { success, message, data } = res.data;
|
||
if (success) {
|
||
if (data.models === "") {
|
||
data.models = []
|
||
} else {
|
||
data.models = data.models.split(",")
|
||
}
|
||
setInputs(data);
|
||
} else {
|
||
showError(message);
|
||
}
|
||
setLoading(false);
|
||
};
|
||
|
||
const fetchModels = async () => {
|
||
try {
|
||
let res = await API.get(`/api/channel/models`);
|
||
setModelOptions(res.data.data.map((model) => ({
|
||
key: model.id,
|
||
text: model.id,
|
||
value: model.id,
|
||
})));
|
||
setFullModels(res.data.data.map((model) => model.id));
|
||
setBasicModels(res.data.data.filter((model) => !model.id.startsWith("gpt-4")).map((model) => model.id));
|
||
} catch (error) {
|
||
showError(error.message);
|
||
}
|
||
};
|
||
|
||
useEffect(() => {
|
||
if (isEdit) {
|
||
loadChannel().then();
|
||
}
|
||
fetchModels().then();
|
||
}, []);
|
||
|
||
const submit = async () => {
|
||
if (!isEdit && (inputs.name === '' || inputs.key === '')) {
|
||
showInfo('请填写渠道名称和渠道密钥！');
|
||
return;
|
||
}
|
||
let localInputs = inputs;
|
||
if (localInputs.base_url.endsWith('/')) {
|
||
localInputs.base_url = localInputs.base_url.slice(0, localInputs.base_url.length - 1);
|
||
}
|
||
if (localInputs.type === 3 && localInputs.other === '') {
|
||
localInputs.other = '2023-03-15-preview';
|
||
}
|
||
let res;
|
||
localInputs.models = localInputs.models.join(",")
|
||
if (isEdit) {
|
||
res = await API.put(`/api/channel/`, { ...localInputs, id: parseInt(channelId) });
|
||
} else {
|
||
res = await API.post(`/api/channel/`, localInputs);
|
||
}
|
||
const { success, message } = res.data;
|
||
if (success) {
|
||
if (isEdit) {
|
||
showSuccess('渠道更新成功！');
|
||
} else {
|
||
showSuccess('渠道创建成功！');
|
||
setInputs(originInputs);
|
||
}
|
||
} else {
|
||
showError(message);
|
||
}
|
||
};
|
||
|
||
return (
|
||
<>
|
||
<Segment loading={loading}>
|
||
<Header as='h3'>{isEdit ? '更新渠道信息' : '创建新的渠道'}</Header>
|
||
<Form autoComplete='new-password'>
|
||
<Form.Field>
|
||
<Form.Select
|
||
label='类型'
|
||
name='type'
|
||
options={CHANNEL_OPTIONS}
|
||
value={inputs.type}
|
||
onChange={handleInputChange}
|
||
/>
|
||
</Form.Field>
|
||
{
|
||
inputs.type === 3 && (
|
||
<>
|
||
<Message>
|
||
注意，<strong>模型部署名称必须和模型名称保持一致</strong>，因为 One API 会把请求体中的 model
|
||
参数替换为你的部署名称（模型名称中的点会被剔除），<a target='_blank'
|
||
href='https://github.com/songquanpeng/one-api/issues/133?notification_referrer_id=NT_kwDOAmJSYrM2NjIwMzI3NDgyOjM5OTk4MDUw#issuecomment-1571602271'>图片演示</a>。
|
||
</Message>
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='AZURE_OPENAI_ENDPOINT'
|
||
name='base_url'
|
||
placeholder={'请输入 AZURE_OPENAI_ENDPOINT，例如：https://docs-test-001.openai.azure.com'}
|
||
onChange={handleInputChange}
|
||
value={inputs.base_url}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='默认 API 版本'
|
||
name='other'
|
||
placeholder={'请输入默认 API 版本，例如：2023-03-15-preview，该配置可以被实际的请求查询参数所覆盖'}
|
||
onChange={handleInputChange}
|
||
value={inputs.other}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
</>
|
||
)
|
||
}
|
||
{
|
||
inputs.type === 8 && (
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='Base URL'
|
||
name='base_url'
|
||
placeholder={'请输入自定义渠道的 Base URL，例如：https://openai.justsong.cn'}
|
||
onChange={handleInputChange}
|
||
value={inputs.base_url}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
)
|
||
}
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='名称'
|
||
name='name'
|
||
placeholder={'请输入名称'}
|
||
onChange={handleInputChange}
|
||
value={inputs.name}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='分组'
|
||
name='group'
|
||
placeholder={'请输入分组'}
|
||
onChange={handleInputChange}
|
||
value={inputs.group}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
<Form.Field>
|
||
<Form.Dropdown
|
||
label='模型'
|
||
placeholder={'请选择该通道所支持的模型'}
|
||
name='models'
|
||
fluid
|
||
multiple
|
||
selection
|
||
onChange={handleInputChange}
|
||
value={inputs.models}
|
||
autoComplete='new-password'
|
||
options={modelOptions}
|
||
/>
|
||
</Form.Field>
|
||
<div style={{ lineHeight: '40px', marginBottom: '12px'}}>
|
||
<Button type={'button'} onClick={() => {
|
||
handleInputChange(null, { name: 'models', value: basicModels });
|
||
}}>填入基础模型</Button>
|
||
<Button type={'button'} onClick={() => {
|
||
handleInputChange(null, { name: 'models', value: fullModels });
|
||
}}>填入所有模型</Button>
|
||
</div>
|
||
{
|
||
inputs.type === 1 && (
|
||
<Form.Field>
|
||
<Form.Input
|
||
label='代理'
|
||
name='base_url'
|
||
placeholder={'请输入 OpenAI API 代理地址，如果不需要请留空，格式为：https://api.openai.com'}
|
||
onChange={handleInputChange}
|
||
value={inputs.base_url}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
)
|
||
}
|
||
{
|
||
batch ? <Form.Field>
|
||
<Form.TextArea
|
||
label='密钥'
|
||
name='key'
|
||
placeholder={'请输入密钥，一行一个'}
|
||
onChange={handleInputChange}
|
||
value={inputs.key}
|
||
style={{ minHeight: 150, fontFamily: 'JetBrains Mono, Consolas' }}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field> : <Form.Field>
|
||
<Form.Input
|
||
label='密钥'
|
||
name='key'
|
||
placeholder={'请输入密钥'}
|
||
onChange={handleInputChange}
|
||
value={inputs.key}
|
||
autoComplete='new-password'
|
||
/>
|
||
</Form.Field>
|
||
}
|
||
{
|
||
!isEdit && (
|
||
<Form.Checkbox
|
||
checked={batch}
|
||
label='批量创建'
|
||
name='batch'
|
||
onChange={() => setBatch(!batch)}
|
||
/>
|
||
)
|
||
}
|
||
<Button positive onClick={submit}>提交</Button>
|
||
</Form>
|
||
</Segment>
|
||
</>
|
||
);
|
||
};
|
||
|
||
export default EditChannel;
|