669 lines
20 KiB
Vue
669 lines
20 KiB
Vue
<template>
|
||
<div class="">
|
||
<HeaderComponent title="设置" class="setting-header">
|
||
<template #description>
|
||
<p>配置文件也可以在 <code>saves/config/base.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" v-if="state.windowWidth > 520">
|
||
<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.windowWidth <= 520 || state.section === 'base'">
|
||
<h3>基础模型配置</h3>
|
||
<div class="section">
|
||
<div class="card card-select">
|
||
<span class="label">{{ items?.embed_model.des }}</span>
|
||
<a-select style="width: 300px"
|
||
: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 card-select">
|
||
<span class="label">{{ items?.reranker.des }}</span>
|
||
<a-select style="width: 300px"
|
||
: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_web_search.des }}</span>
|
||
<a-switch
|
||
:checked="configStore.config.enable_web_search"
|
||
@change="handleChange('enable_web_search', !configStore.config.enable_web_search)"
|
||
/>
|
||
</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>
|
||
<h3>检索配置</h3>
|
||
<div class="section">
|
||
<div class="card card-select">
|
||
<span class="label">{{ items?.use_rewrite_query.des }}</span>
|
||
<a-select style="width: 200px"
|
||
:value="configStore.config?.use_rewrite_query"
|
||
@change="handleChange('use_rewrite_query', $event)"
|
||
>
|
||
<a-select-option
|
||
v-for="(name, idx) in items?.use_rewrite_query.choices" :key="idx"
|
||
:value="name">{{ name }}
|
||
</a-select-option>
|
||
</a-select>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<div class="setting" v-if="state.windowWidth <= 520 || 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.custom_id, 'card-models': true, 'custom-model': true}"
|
||
v-for="(item, key) in configStore.config.custom_models" :key="item.custom_id"
|
||
@click="handleChange('model_provider', 'custom'); handleChange('model_name', item.custom_id)"
|
||
>
|
||
<div class="card-models__header">
|
||
<div class="name">{{ item.name }}</div>
|
||
<div class="action">
|
||
<a-popconfirm
|
||
title="确认删除该模型?"
|
||
@confirm="handleDeleteCustomModel(item.custom_id)"
|
||
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>
|
||
<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">
|
||
<!-- <div v-if="modelStatus[item]" class="success"></div> -->
|
||
<div :class="{'model-icon': true, 'available': modelStatus[item]}">
|
||
<img :src="modelIcons[item]" alt="模型图标">
|
||
</div>
|
||
<h3>{{ modelNames[item].name }}</h3>
|
||
<a :href="modelNames[item].url" target="_blank"><InfoCircleOutlined /></a>
|
||
</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>
|
||
</div>
|
||
</div>
|
||
<div class="model-provider-card" v-for="(item, key) in notModelKeys" :key="key">
|
||
<div class="card-header">
|
||
<h3 style="font-weight: 400">{{ modelNames[item].name }}</h3>
|
||
<a :href="modelNames[item].url" target="_blank"><InfoCircleOutlined /></a>
|
||
<div class="missing-keys">
|
||
<small>需配置</small> <span v-for="(key, idx) in modelNames[item].env" :key="idx">{{ key }}</span>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
<div class="setting" v-if="state.windowWidth <= 520 || state.section ==='path'">
|
||
<h3>本地模型配置</h3>
|
||
<p>如果是 Docker 启动，务必确保在 docker-compose.dev.yaml 中添加了 volumes 映射。</p>
|
||
<TableConfigComponent
|
||
:config="configStore.config?.model_local_paths"
|
||
@update:config="handleModelLocalPathsUpdate"
|
||
/>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
|
||
<script setup>
|
||
import { message } from 'ant-design-vue';
|
||
import { computed, reactive, ref, h, watch, onMounted, onUnmounted } from 'vue'
|
||
import { useConfigStore } from '@/stores/config';
|
||
import {
|
||
ReloadOutlined,
|
||
SettingOutlined,
|
||
CodeOutlined,
|
||
ExceptionOutlined,
|
||
FolderOutlined,
|
||
DeleteOutlined,
|
||
EditOutlined,
|
||
InfoCircleOutlined,
|
||
} from '@ant-design/icons-vue';
|
||
import HeaderComponent from '@/components/HeaderComponent.vue';
|
||
import TableConfigComponent from '@/components/TableConfigComponent.vue';
|
||
import { notification, Button } from 'ant-design-vue';
|
||
import { modelIcons } from '@/utils/modelIcon'
|
||
|
||
|
||
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,
|
||
custom_id: '', // Add this line
|
||
name: '',
|
||
api_key: '',
|
||
api_base: '',
|
||
edit_type: 'add',
|
||
})
|
||
const state = reactive({
|
||
loading: false,
|
||
section: 'base',
|
||
windowWidth: window?.innerWidth || 0
|
||
})
|
||
|
||
// 筛选 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 generateRandomHash = (length) => {
|
||
let chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
|
||
let hash = '';
|
||
for (let i = 0; i < length; i++) {
|
||
hash += chars.charAt(Math.floor(Math.random() * chars.length));
|
||
}
|
||
return hash;
|
||
}
|
||
|
||
const handleModelLocalPathsUpdate = (config) => {
|
||
handleChange('model_local_paths', config)
|
||
}
|
||
|
||
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
|
||
}
|
||
|
||
// 这些都是需要重启的配置
|
||
if (key == 'enable_reranker'
|
||
|| key == 'enable_knowledge_graph'
|
||
|| key == 'enable_knowledge_base'
|
||
|| key == 'enable_web_search'
|
||
|| key == 'embed_model'
|
||
|| key == 'reranker'
|
||
|| key == 'model_local_paths') {
|
||
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 = []
|
||
}
|
||
|
||
|
||
const model_info = {
|
||
custom_id: customModel.custom_id,
|
||
name: customModel.name,
|
||
api_key: customModel.api_key,
|
||
api_base: customModel.api_base,
|
||
}
|
||
|
||
if (customModel.edit_type == 'add') {
|
||
if (configStore.config.custom_models.find(item => item.custom_id == customModel.custom_id)) {
|
||
message.error('模型ID已存在')
|
||
return
|
||
}
|
||
const modelHash = generateRandomHash(4)
|
||
model_info.custom_id = `${customModel.name}-${modelHash}`
|
||
configStore.config.custom_models.push(model_info)
|
||
} else {
|
||
configStore.config.custom_models = configStore.config.custom_models.map(item => {
|
||
if (item.custom_id == customModel.custom_id) {
|
||
if (item.name != customModel.name) {
|
||
const modelHash = generateRandomHash(4)
|
||
model_info.custom_id = `${customModel.name}-${modelHash}`
|
||
configStore.setConfigValue('model_name', model_info.custom_id)
|
||
}
|
||
return model_info
|
||
}
|
||
return item
|
||
})
|
||
}
|
||
|
||
customModel.visible = false
|
||
await configStore.setConfigValue('custom_models', configStore.config.custom_models)
|
||
message.success('添加自定义模型成功')
|
||
}
|
||
|
||
const handleDeleteCustomModel = (custom_id) => {
|
||
const updatedModels = configStore.config.custom_models.filter(item => item.custom_id !== custom_id);
|
||
configStore.setConfigValue('custom_models', updatedModels);
|
||
}
|
||
|
||
const handleEditCustomModel = (item) => {
|
||
customModel.modelTitle = '编辑自定义模型'
|
||
customModel.custom_id = item.custom_id
|
||
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.custom_id = ''
|
||
customModel.name = ''
|
||
customModel.api_key = ''
|
||
customModel.api_base = ''
|
||
customModel.visible = false
|
||
}
|
||
|
||
const updateWindowWidth = () => {
|
||
state.windowWidth = window?.innerWidth || 0
|
||
}
|
||
|
||
onMounted(() => {
|
||
updateWindowWidth()
|
||
window.addEventListener('resize', updateWindowWidth)
|
||
})
|
||
|
||
onUnmounted(() => {
|
||
window.removeEventListener('resize', updateWindowWidth)
|
||
})
|
||
|
||
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: 12px;
|
||
padding: 12px;
|
||
.card-header {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 10px;
|
||
|
||
.model-icon {
|
||
width: 24px;
|
||
height: 24px;
|
||
// 灰度
|
||
filter: grayscale(100%);
|
||
img {
|
||
width: 100%;
|
||
height: 100%;
|
||
}
|
||
|
||
&.available {
|
||
filter: grayscale(0%);
|
||
}
|
||
}
|
||
|
||
|
||
h3 {
|
||
margin: 0;
|
||
font-size: 1rem;
|
||
font-weight: bold;
|
||
}
|
||
|
||
a {
|
||
text-decoration: none;
|
||
color: var(--gray-500);
|
||
font-size: 12px;
|
||
transition: all 0.1s;
|
||
|
||
&:hover {
|
||
color: var(--gray-900);
|
||
}
|
||
}
|
||
|
||
.details, .missing-keys {
|
||
margin-left: auto;
|
||
}
|
||
|
||
.success {
|
||
width: 0.75rem;
|
||
height: 0.75rem;
|
||
background-color: rgb(91, 186, 91);
|
||
border-radius: 50%;
|
||
box-shadow: 0 0 10px 1px rgba( 0,128, 0, 0.1);
|
||
border: 2px solid white;
|
||
}
|
||
|
||
.missing-keys {
|
||
color: var(--gray-600);
|
||
& > span {
|
||
margin-left: 10px;
|
||
user-select: all;
|
||
}
|
||
}
|
||
}
|
||
|
||
.card-body {
|
||
display: grid;
|
||
grid-template-columns: repeat(auto-fill, minmax(200px, 1fr));
|
||
gap: 12px;
|
||
margin-top: 10px;
|
||
.card-models {
|
||
width: 100%;
|
||
border-radius: 8px;
|
||
border: 1px solid var(--gray-300);
|
||
padding: 12px 16px;
|
||
display: flex;
|
||
gap: 6px;
|
||
justify-content: space-between;
|
||
align-items: center;
|
||
cursor: pointer;
|
||
box-sizing: border-box;
|
||
background-color: var(--gray-50);
|
||
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-900);
|
||
}
|
||
.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;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
|
||
@media (max-width: 520px) {
|
||
.setting-container {
|
||
flex-direction: column;
|
||
}
|
||
|
||
.card.card-select {
|
||
gap: 0.75rem;
|
||
align-items: flex-start;
|
||
flex-direction: column;
|
||
}
|
||
}
|
||
</style>
|
||
|