570 lines
17 KiB
Vue
570 lines
17 KiB
Vue
<template>
|
||
<div class="">
|
||
<HeaderComponent title="设置" class="setting-header">
|
||
<template #description>
|
||
<p>配置文件也可以在 <code>saves/config/config.yaml</code> 中修改</p>
|
||
</template>
|
||
<template #actions>
|
||
<a-button type="primary" v-if="isNeedRestart" @click="sendRestart">
|
||
<ReloadOutlined />需要重启
|
||
</a-button>
|
||
<a-button v-else @click="sendRestart">
|
||
<ReloadOutlined />重启服务
|
||
</a-button>
|
||
</template>
|
||
</HeaderComponent>
|
||
<div class="setting-container layout-container">
|
||
<div class="sider">
|
||
<a-button type="text" :class="{ activesec: state.section === 'base'}" @click="state.section='base'" :icon="h(SettingOutlined)"> 基本设置 </a-button>
|
||
<a-button type="text" :class="{ activesec: state.section === 'model'}" @click="state.section='model'" :icon="h(CodeOutlined)"> 模型配置 </a-button>
|
||
<a-button type="text" :class="{ activesec: state.section === 'path'}" @click="state.section='path'" :icon="h(FolderOutlined)"> 路径配置 </a-button>
|
||
</div>
|
||
<div class="setting" v-if="state.section === 'base'">
|
||
<h3>基础模型配置</h3>
|
||
<div class="section">
|
||
<div class="card">
|
||
<span class="label">{{ items?.embed_model.des }}</span>
|
||
<a-select style="width: 200px"
|
||
:value="configStore.config?.embed_model"
|
||
@change="handleChange('embed_model', $event)"
|
||
>
|
||
<a-select-option
|
||
v-for="(name, idx) in items?.embed_model.choices" :key="idx"
|
||
:value="name">{{ name }}
|
||
</a-select-option>
|
||
</a-select>
|
||
</div>
|
||
<div class="card">
|
||
<span class="label">{{ items?.reranker.des }}</span>
|
||
<a-select style="width: 200px"
|
||
:value="configStore.config?.reranker"
|
||
@change="handleChange('reranker', $event)"
|
||
:disabled="!configStore.config.enable_reranker"
|
||
>
|
||
<a-select-option
|
||
v-for="(name, idx) in items?.reranker.choices" :key="idx"
|
||
:value="name">{{ name }}
|
||
</a-select-option>
|
||
</a-select>
|
||
</div>
|
||
</div>
|
||
<h3>功能配置</h3>
|
||
<div class="section">
|
||
<div class="card">
|
||
<span class="label">{{ items?.enable_knowledge_base.des }}</span>
|
||
<a-switch
|
||
:checked="configStore.config.enable_knowledge_base"
|
||
@change="handleChange('enable_knowledge_base', !configStore.config.enable_knowledge_base)"
|
||
/>
|
||
</div>
|
||
<div class="card">
|
||
<span class="label">{{ items?.enable_knowledge_graph.des }}</span>
|
||
<a-switch
|
||
:checked="configStore.config.enable_knowledge_graph"
|
||
@change="handleChange('enable_knowledge_graph', !configStore.config.enable_knowledge_graph)"
|
||
/>
|
||
</div>
|
||
<div class="card">
|
||
<span class="label">{{ items?.enable_search_engine.des }}</span>
|
||
<a-switch
|
||
:checked="configStore.config.enable_search_engine"
|
||
@change="handleChange('enable_search_engine', !configStore.config.enable_search_engine)"
|
||
/>
|
||
</div>
|
||
<div class="card">
|
||
<span class="label">{{ items?.enable_reranker.des }}</span>
|
||
<a-switch
|
||
:checked="configStore.config.enable_reranker"
|
||
@change="handleChange('enable_reranker', !configStore.config.enable_reranker)"
|
||
/>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<div class="setting" v-if="state.section == 'model'">
|
||
<h3>模型配置</h3>
|
||
<p>请在 <code>src/.env</code> 文件中配置对应的 APIKEY</p>
|
||
<div class="model-provider-card">
|
||
<div class="card-header">
|
||
<h3>自定义模型</h3>
|
||
</div>
|
||
<div class="card-body">
|
||
<div
|
||
:class="{'model_selected': modelProvider == 'custom' && configStore.config.model_name == item.name, 'card-models': true, 'custom-model': true}"
|
||
v-for="(item, key) in configStore.config.custom_models" :key="key"
|
||
@click="handleChange('model_provider', 'custom'); handleChange('model_name', item.name)"
|
||
>
|
||
<div class="card-models__header">
|
||
<div class="name">{{ item.name }}</div>
|
||
<div class="action">
|
||
<!-- 添加确认删除 -->
|
||
<a-popconfirm
|
||
title="确认删除该模型?"
|
||
@confirm="handleDeleteCustomModel(item.name)"
|
||
okText="确认删除"
|
||
cancelText="取消"
|
||
ok-type="danger"
|
||
:disabled="configStore.config.model_name == item.name"
|
||
>
|
||
<a-button type="text" :disabled="configStore.config.model_name == item.name" @click.stop><DeleteOutlined /></a-button>
|
||
</a-popconfirm>
|
||
<a-button type="text" @click.stop="handleEditCustomModel(item)"><EditOutlined /></a-button>
|
||
</div>
|
||
</div>
|
||
<div class="api_base">{{ item.api_base }}</div>
|
||
<!-- <div class="select-btn"></div> -->
|
||
</div>
|
||
<div class="card-models custom-model" @click="customModel.visible=true">
|
||
<div class="card-models__header">
|
||
<div class="name"> + 添加模型</div>
|
||
</div>
|
||
<div class="api_base">添加兼容 OpenAI 的模型</div>
|
||
<a-modal
|
||
class="custom-model-modal"
|
||
v-model:open="customModel.visible"
|
||
:title="customModel.modelTitle"
|
||
@ok="handleAddCustomModel"
|
||
@cancel="handleCancelCustomModel"
|
||
:okText="'确认'"
|
||
:cancelText="'取消'"
|
||
:okButtonProps="{disabled: !customModel.name || !customModel.api_base}"
|
||
:ok-type="'primary'"
|
||
>
|
||
<p>添加的模型是兼容 OpenAI 的模型，比如 vllm，Ollama。</p>
|
||
<a-form :model="customModel" layout="vertical" >
|
||
<a-form-item label="模型名称" name="name" :rules="[{ required: true, message: '请输入模型名称' }]">
|
||
<a-input v-model:value="customModel.name" />
|
||
</a-form-item>
|
||
<a-form-item label="API Base" name="api_base" :rules="[{ required: true, message: '请输入API Base' }]">
|
||
<a-input v-model:value="customModel.api_base"/>
|
||
</a-form-item>
|
||
<a-form-item label="API KEY" name="api_key">
|
||
<a-input v-model:value="customModel.api_key" autocomplete="off"/>
|
||
</a-form-item>
|
||
</a-form>
|
||
</a-modal>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<div class="model-provider-card" v-for="(item, key) in modelKeys" :key="key">
|
||
<div class="card-header">
|
||
<h3>{{ modelNames[item].name }}</h3>
|
||
<a :href="modelNames[item].url" target="_blank">详情</a>
|
||
<div v-if="modelStatus[item]" class="success"></div>
|
||
</div>
|
||
<div class="card-body" v-if="modelStatus[item]">
|
||
<div
|
||
:class="{'model_selected': modelProvider == item && configStore.config.model_name == model, 'card-models': true}"
|
||
v-for="(model, idx) in modelNames[item].models" :key="idx"
|
||
@click="handleChange('model_provider', item); handleChange('model_name', model)"
|
||
>
|
||
<div class="model_name">{{ model }}</div>
|
||
<!-- <div class="select-btn"></div> -->
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<div class="model-provider-card" v-for="(item, key) in notModelKeys" :key="key">
|
||
<div class="card-header">
|
||
<h3>{{ modelNames[item].name }}</h3>
|
||
<a :href="modelNames[item].url" target="_blank">详情</a>
|
||
<div class="missing-keys">
|
||
需配置 <span v-for="(key, idx) in modelNames[item].env" :key="idx">{{ key }}</span>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
|
||
<script setup>
|
||
import { message } from 'ant-design-vue';
|
||
import { computed, reactive, ref, h, watch } from 'vue'
|
||
import { useConfigStore } from '@/stores/config';
|
||
import {
|
||
ReloadOutlined,
|
||
SettingOutlined,
|
||
CodeOutlined,
|
||
ExceptionOutlined,
|
||
FolderOutlined,
|
||
DeleteOutlined,
|
||
EditOutlined,
|
||
} from '@ant-design/icons-vue';
|
||
import HeaderComponent from '@/components/HeaderComponent.vue';
|
||
import { notification, Button } from 'ant-design-vue';
|
||
|
||
const configStore = useConfigStore()
|
||
const items = computed(() => configStore.config._config_items)
|
||
const modelNames = computed(() => configStore.config?.model_names)
|
||
const modelStatus = computed(() => configStore.config?.model_provider_status)
|
||
const modelProvider = computed(() => configStore.config?.model_provider)
|
||
const isNeedRestart = ref(false)
|
||
const customModel = reactive({
|
||
modelTitle: '添加自定义模型',
|
||
visible: false,
|
||
name: '',
|
||
api_key: '',
|
||
api_base: '',
|
||
edit_type: 'add',
|
||
})
|
||
const state = reactive({
|
||
loading: false,
|
||
section: 'base'
|
||
})
|
||
|
||
// 筛选 modelStatus 中为真的key
|
||
const modelKeys = computed(() => {
|
||
return Object.keys(modelStatus.value).filter(key => modelStatus.value[key])
|
||
})
|
||
|
||
const notModelKeys = computed(() => {
|
||
return Object.keys(modelStatus.value).filter(key => !modelStatus.value[key])
|
||
})
|
||
|
||
const handleChange = (key, e) => {
|
||
if (key == 'enable_knowledge_graph' && e && !configStore.config.enable_knowledge_base) {
|
||
message.error('启动知识图谱必须请先启用知识库功能')
|
||
return
|
||
}
|
||
|
||
if (key == 'enable_knowledge_base' && !e && configStore.config.enable_knowledge_graph) {
|
||
message.error('关闭知识库功能必须请先关闭知识图谱功能')
|
||
return
|
||
}
|
||
|
||
console.log('Change', key, e)
|
||
|
||
if (key == 'enable_reranker'
|
||
|| key == 'enable_knowledge_graph'
|
||
|| key == 'enable_knowledge_base'
|
||
|| key == 'model_provider'
|
||
|| key == 'model_name'
|
||
|| key == 'embed_model'
|
||
|| key == 'reranker') {
|
||
if (!isNeedRestart.value) {
|
||
isNeedRestart.value = true
|
||
notification.info({
|
||
message: '需要重启服务',
|
||
description: '请点击右下角按钮重启服务',
|
||
placement: 'topLeft',
|
||
duration: 0,
|
||
btn: h(Button, { type: 'primary', onClick: sendRestart }, '立即重启')
|
||
})
|
||
}
|
||
}
|
||
|
||
configStore.setConfigValue(key, e)
|
||
}
|
||
|
||
const handleAddCustomModel = async () => {
|
||
if (!customModel.name || !customModel.api_base) {
|
||
message.error('请填写完整模型信息')
|
||
return
|
||
}
|
||
|
||
if (!configStore.config.custom_models) {
|
||
configStore.config.custom_models = []
|
||
}
|
||
|
||
if (configStore.config.custom_models.find(item => item.name == customModel.name)) {
|
||
message.error('模型名称已存在')
|
||
return
|
||
}
|
||
|
||
if (customModel.edit_type == 'add') {
|
||
configStore.config.custom_models.push(customModel)
|
||
} else {
|
||
configStore.config.custom_models = configStore.config.custom_models.map(item => {
|
||
if (item.name == customModel.name) {
|
||
return customModel
|
||
}
|
||
return item
|
||
})
|
||
}
|
||
|
||
customModel.visible = false
|
||
await configStore.setConfigValue('custom_models', configStore.config.custom_models)
|
||
configStore.refreshConfig()
|
||
message.success('添加自定义模型成功')
|
||
}
|
||
|
||
const handleDeleteCustomModel = (name) => {
|
||
configStore.config.custom_models = configStore.config.custom_models.filter(item => item.name != name)
|
||
configStore.setConfigValue('custom_models', configStore.config.custom_models)
|
||
configStore.refreshConfig()
|
||
}
|
||
|
||
const handleEditCustomModel = (item) => {
|
||
customModel.modelTitle = '编辑自定义模型'
|
||
customModel.name = item.name
|
||
customModel.api_key = item.api_key
|
||
customModel.api_base = item.api_base
|
||
customModel.visible = true
|
||
customModel.edit_type = 'edit'
|
||
}
|
||
|
||
const handleCancelCustomModel = () => {
|
||
customModel.name = ''
|
||
customModel.api_key = ''
|
||
customModel.api_base = ''
|
||
customModel.visible = false
|
||
}
|
||
|
||
const sendRestart = () => {
|
||
console.log('Restarting...')
|
||
message.loading({ content: '重新加载模型中', key: "restart", duration: 0 });
|
||
fetch('/api/restart', {
|
||
method: 'POST',
|
||
}).then(() => {
|
||
console.log('Restarted')
|
||
message.success({ content: '重新加载完成!', key: "restart", duration: 2 });
|
||
setTimeout(() => {
|
||
window.location.reload()
|
||
}, 200)
|
||
})
|
||
}
|
||
</script>
|
||
|
||
<style lang="less" scoped>
|
||
.setting-header {
|
||
height: 100px;
|
||
}
|
||
|
||
.setting-header p {
|
||
margin: 8px 0 0;
|
||
}
|
||
|
||
.setting-container {
|
||
padding: 0;
|
||
box-sizing: border-box;
|
||
display: flex;
|
||
position: relative;
|
||
min-height: 100%;
|
||
}
|
||
|
||
.sider {
|
||
width: 200px;
|
||
height: 100%;
|
||
padding: 0 20px;
|
||
position: sticky;
|
||
top: 100px;
|
||
display: flex;
|
||
flex-direction: column;
|
||
align-items: center;
|
||
border-right: 1px solid var(--main-light-3);
|
||
gap: 8px;
|
||
padding-top: 20px;
|
||
|
||
|
||
& > * {
|
||
width: 100%;
|
||
height: auto;
|
||
padding: 6px 16px;
|
||
cursor: pointer;
|
||
transition: all 0.1s;
|
||
text-align: left;
|
||
font-size: 15px;
|
||
border-radius: 8px;
|
||
color: var(--gray-700);
|
||
|
||
&:hover {
|
||
background: var(--gray-100);
|
||
}
|
||
|
||
&.activesec {
|
||
background: var(--gray-200);
|
||
color: var(--gray-900);
|
||
}
|
||
}
|
||
}
|
||
|
||
.setting {
|
||
width: 100%;
|
||
flex: 1;
|
||
margin: 0 auto;
|
||
height: 100%;
|
||
padding: 0 20px;
|
||
margin-bottom: 40px;
|
||
|
||
h3 {
|
||
margin-top: 20px;
|
||
}
|
||
|
||
.section {
|
||
margin-top: 20px;
|
||
background-color: var(--gray-10);
|
||
padding: 20px;
|
||
border-radius: 8px;
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 16px;
|
||
// box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
|
||
border: 1px solid var(--gray-300);
|
||
}
|
||
|
||
.card {
|
||
display: flex;
|
||
align-items: center;
|
||
justify-content: space-between;
|
||
|
||
.label {
|
||
margin-right: 20px;
|
||
|
||
button {
|
||
margin-left: 10px;
|
||
height: 24px;
|
||
padding: 0 8px;
|
||
font-size: smaller;
|
||
}
|
||
}
|
||
}
|
||
|
||
.model-provider-card {
|
||
border: 1px solid var(--gray-300);
|
||
background-color: white;
|
||
border-radius: 8px;
|
||
margin-bottom: 16px;
|
||
padding: 16px;
|
||
.card-header {
|
||
display: flex;
|
||
align-items: baseline;
|
||
gap: 10px;
|
||
|
||
h3 {
|
||
margin: 0;
|
||
font-size: 1rem;
|
||
}
|
||
|
||
a {
|
||
text-decoration: none;
|
||
color: var(--gray-700);
|
||
font-size: 14px;
|
||
transition: all 0.1s;
|
||
|
||
&:hover {
|
||
color: var(--gray-900);
|
||
}
|
||
}
|
||
|
||
.success, .missing-keys {
|
||
margin-left: auto;
|
||
}
|
||
|
||
.success {
|
||
width: 1rem;
|
||
height: 1rem;
|
||
background-color: rgb(91, 186, 91);
|
||
border-radius: 50%;
|
||
box-shadow: 0 0 10px 1px rgba( 0,128, 0, 0.2);
|
||
border: 2px solid white;
|
||
}
|
||
|
||
.missing-keys {
|
||
color: var(--gray-600);
|
||
& > span {
|
||
margin-left: 10px;
|
||
}
|
||
}
|
||
}
|
||
|
||
.card-body {
|
||
display: grid;
|
||
grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
|
||
gap: 10px;
|
||
margin-top: 10px;
|
||
.card-models {
|
||
width: 100%;
|
||
border-radius: 8px;
|
||
border: 1px solid var(--gray-300);
|
||
padding: 10px 16px;
|
||
display: flex;
|
||
gap: 6px;
|
||
justify-content: space-between;
|
||
align-items: center;
|
||
cursor: pointer;
|
||
box-sizing: border-box;
|
||
background-color: var(--gray-10);
|
||
transition: box-shadow 0.1s;
|
||
&:hover {
|
||
box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
|
||
}
|
||
.model_name {
|
||
font-size: 14px;
|
||
color: var(--gray-800);
|
||
}
|
||
.select-btn {
|
||
width: 16px;
|
||
height: 16px;
|
||
flex: 0 0 16px;
|
||
border-radius: 50%;
|
||
}
|
||
|
||
&.model_selected {
|
||
border: 2px solid var(--main-color);
|
||
padding: 9px 15px;
|
||
.model_name {
|
||
color: var(--gray-1000)
|
||
}
|
||
.select-btn {
|
||
border-color: var(--main-color);
|
||
border: 2px solid var(--main-color);
|
||
}
|
||
}
|
||
|
||
&.custom-model {
|
||
display: flex;
|
||
flex-direction: column;
|
||
align-items: flex-start;
|
||
padding-right: 8px;
|
||
gap: 10px;
|
||
.card-models__header {
|
||
width: 100%;
|
||
height: 32px;
|
||
display: flex;
|
||
justify-content: flex-start;
|
||
align-items: center;
|
||
.name {
|
||
color: var(--gray-1000);
|
||
font-weight: bold;
|
||
}
|
||
.action {
|
||
opacity: 0;
|
||
user-select: none;
|
||
margin-left: auto;
|
||
button {
|
||
padding: 4px 8px;
|
||
}
|
||
}
|
||
.custom-model-modal {
|
||
.ant-form-item {
|
||
margin-bottom: 10px;
|
||
}
|
||
}
|
||
}
|
||
.api_base {
|
||
font-size: 12px;
|
||
color: var(--gray-600);
|
||
}
|
||
|
||
&:hover {
|
||
.card-models__header {
|
||
.action {
|
||
opacity: 1;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
&.model_selected.custom-model {
|
||
padding: 9px 7px 9px 15px;
|
||
.card-models__header {
|
||
.action {
|
||
opacity: 1;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
}
|
||
</style> |