394 lines
11 KiB
Vue
394 lines
11 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" v-for="(item, key) in modelKeys" :key="key" :style="{backgroundColor: modelProvider == item ? colorMap[item] : white }">
|
|
<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 } from 'vue'
|
|
import { useConfigStore } from '@/stores/config';
|
|
import {
|
|
ReloadOutlined,
|
|
SettingOutlined,
|
|
CodeOutlined,
|
|
ExceptionOutlined,
|
|
FolderOutlined,
|
|
} from '@ant-design/icons-vue';
|
|
import HeaderComponent from '@/components/HeaderComponent.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 state = reactive({
|
|
loading: false,
|
|
section: 'base'
|
|
})
|
|
|
|
const colorMap = reactive({
|
|
siliconflow: '#FFECFF',
|
|
zhipu: '#EFF1FE',
|
|
qianfan: '#E8F5FE',
|
|
deepseek: '#D3DCFF',
|
|
openai: '#E5E7EB',
|
|
vllm: '#E5E7EB',
|
|
bailian: '#EFF1FE',
|
|
})
|
|
|
|
// 筛选 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
|
|
message.info('修改配置后需要重启服务才能生效')
|
|
}
|
|
}
|
|
|
|
configStore.setConfigValue(key, e)
|
|
}
|
|
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;
|
|
background: inherit;
|
|
position: relative;
|
|
}
|
|
|
|
.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 {
|
|
background-color: var(--gray-10);
|
|
border: 1px solid var(--gray-300);
|
|
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: green;
|
|
border-radius: 50%;
|
|
box-shadow: 0 0 10px 1px rgba( 0,128, 0, 0.5);
|
|
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);
|
|
background-color: var(--gray-50);
|
|
padding: 10px 16px;
|
|
display: flex;
|
|
gap: 6px;
|
|
justify-content: space-between;
|
|
align-items: center;
|
|
cursor: pointer;
|
|
box-sizing: border-box;
|
|
background-color: rgba(255, 255, 255, 0.6);
|
|
&:hover {
|
|
border-color: var(--gray-400);
|
|
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);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
</style> |