<template>
|
<a-button
|
v-if="typeAngint == 'add'"
|
type="primary"
|
@click="handleClick"
|
style="margin-left: 10px">
|
<template #icon>
|
<icon-plus />
|
</template>
|
新建智能体
|
</a-button>
|
<a-button
|
v-if="typeAngint == 'edit'"
|
type="text"
|
size="small"
|
@click="editClick"
|
>
|
<template #icon>
|
<icon-tool />
|
</template>
|
</a-button>
|
<a-modal
|
v-model:visible="visible"
|
title="智能体配置"
|
@before-open="handleOpened"
|
@cancel="handleCancel"
|
:ok-loading="loading"
|
@before-ok="handleBeforeOk"
|
:footer="true"
|
title-align="start"
|
width="600px"
|
>
|
<a-row class="">
|
<a-col :span="24">
|
<a-tabs type="capsule" size="large" :active-key="activeKey" @change="handleTabChange">
|
<a-tab-pane key="1" title="助理设置">
|
<a-divider style="margin-bottom: 20px;margin-top: 0" />
|
<div style="width: 100%;" :style="{height:height}">
|
<a-form
|
ref="formRef"
|
:rules="rules"
|
:model="form"
|
@submit="handleSubmit"
|
:style="{ width: '90%', margin: '0 auto' }"
|
>
|
<a-form-item field="name" label="智能体名称">
|
<a-input v-model="form.name" placeholder="请输入名称" />
|
</a-form-item>
|
<a-form-item label="智能体图标">
|
<a-space direction="vertical" :style="{ width: '100%' }">
|
<Upload
|
:action="uploadAction"
|
:limit="1"
|
:url="form.icon"
|
@update:fileList="updateFileList"
|
@success="handleSuccess"
|
></Upload>
|
</a-space>
|
</a-form-item>
|
<a-form-item label="空回复">
|
<a-input v-model="form.prompt_config.empty_response" placeholder="" />
|
</a-form-item>
|
<a-form-item label="设置开场白">
|
<a-textarea
|
v-model="form.prompt_config.prologue"
|
placeholder=""
|
style="height: 100px"
|
/>
|
</a-form-item>
|
<a-form-item label="显示引文">
|
<a-space direction="vertical" >
|
<a-switch v-model="form.prompt_config.quote" size="small" />
|
</a-space>
|
</a-form-item>
|
<a-form-item label="Self-RAG">
|
<a-space direction="vertical">
|
<a-switch v-model="form.prompt_config.self_rag" size="small" />
|
</a-space>
|
</a-form-item>
|
<a-form-item field="kb_ids" label="知识库">
|
<a-select
|
v-model="form.kb_ids"
|
placeholder="请选择 ..."
|
multiple
|
>
|
<a-option
|
v-for="item in tabs"
|
:key="item.id"
|
:value="item.id"
|
>
|
{{ item.name }}
|
</a-option>
|
</a-select>
|
</a-form-item>
|
<!-- <a-form-item>-->
|
<!-- <div style="width: 100%; text-align: right">-->
|
<!-- <a-button @click="visible = false">取消</a-button>-->
|
<!-- <a-button style="margin-left: 10px" type="primary" html-type="submit"-->
|
<!-- >确定</a-button-->
|
<!-- >-->
|
<!-- </div>-->
|
<!-- </a-form-item>-->
|
</a-form>
|
</div>
|
</a-tab-pane>
|
<a-tab-pane key="2" title="提示引擎">
|
<a-divider style="margin-bottom: 20px;margin-top: 0" />
|
<a-scrollbar style="width: 100%;overflow: auto" :style="{height:height}">
|
<a-form
|
ref="formRef1"
|
:rules="rules"
|
:model="form"
|
@submit="handleSubmit"
|
:style="{ width: '90%', margin: '0 auto' }"
|
>
|
<a-form-item field="prompt_config.system" label="系统">
|
<a-textarea
|
v-model="form.prompt_config.system"
|
placeholder=""
|
style="height: 180px"
|
/>
|
</a-form-item>
|
<a-divider style="margin: 0;margin-bottom: 10px" />
|
<a-form-item label="相似度阈值">
|
<a-slider v-model="form.similarity_threshold" :step="0.01" :min="0" :max="1" />
|
</a-form-item>
|
<a-form-item label="关键字相似度权重">
|
<a-slider v-model="form.vector_similarity_weight" :step="0.01" :min="0" :max="1" />
|
</a-form-item>
|
<a-form-item label="Top N">
|
<a-slider v-model="form.top_n" :min="0" :max="30" />
|
</a-form-item>
|
<a-form-item label="Rerank模型" >
|
<a-space direction="vertical" size="large">
|
<a-select :size="'large'" v-model="form.rerank_id" placeholder="请选择 ..." allow-clear>
|
<a-optgroup :label="index" v-for="(item,index) in rankModelList" :key=index>
|
<a-option
|
v-for="(obj) in item"
|
:key="obj.fid"
|
:disabled="!obj.available"
|
:value="obj.llm_name"
|
>
|
{{obj.llm_name}}
|
</a-option>
|
</a-optgroup>
|
</a-select>
|
</a-space>
|
</a-form-item>
|
<a-form-item v-if="form.rerank_id" label="Top-K">
|
<a-slider v-model="form.top_k" :min="1" :max="2048" />
|
</a-form-item>
|
<a-form-item label="变量">
|
<div style="width: 100%;text-align: right" >
|
<a-button type="outline" @click="addVariable" style="" size="mini" shape="round">
|
新建
|
</a-button>
|
</div>
|
</a-form-item>
|
<a-form-item label="">
|
<a-table
|
:columns="columns"
|
:data="form.prompt_config.parameters"
|
style="width: 100%"
|
>
|
<template #key="{ record, rowIndex }">
|
<a-input v-model="record.key" placeholder="" />
|
<!-- <a-input v-model="form.prompt_config.parameters[rowIndex].key" />-->
|
</template>
|
<template #optional="{ record }">
|
<a-switch v-model="record.optional" size="small" />
|
</template>
|
<template #action="{ record }">
|
<a-button type="text" @click="deleteParameters(record)" style="margin-left: 0" size="small">
|
<template #icon>
|
<icon-delete />
|
</template>
|
</a-button>
|
</template>
|
</a-table>
|
</a-form-item>
|
</a-form>
|
</a-scrollbar>
|
</a-tab-pane>
|
<a-tab-pane key="3" title="模型设置">
|
<a-divider style="margin-bottom: 20px;margin-top: 0" />
|
<div style="width: 100%;" :style="{height:height}">
|
<a-form
|
ref="formRef2"
|
:rules="rules"
|
:model="form"
|
@submit="handleSubmit"
|
:style="{ width: '90%', margin: '0 auto' }"
|
>
|
<a-form-item field="llm_id" label="模型">
|
<a-space direction="vertical" size="large">
|
<a-select :size="'large'" field="llm_id" v-model="form.llm_id" style="width: 400px" placeholder="请选择">
|
<a-optgroup
|
:label="index"
|
v-for="(item, index) in modelList"
|
:key="index"
|
>
|
<a-option
|
v-for="obj in item"
|
:key="obj.fid"
|
:disabled="!obj.available"
|
:value="obj.llm_id"
|
>
|
{{ obj.llm_name }}
|
</a-option>
|
</a-optgroup>
|
</a-select>
|
</a-space>
|
</a-form-item>
|
<a-divider style="margin-bottom: 20px;margin-top: 0" />
|
<a-form-item label="自由">
|
<a-select default-value="2" :style="{width:'400px'}" placeholder="">
|
<a-option value="1">即兴创作</a-option>
|
<a-option value="2">精确</a-option>
|
<a-option value="3">平衡</a-option>
|
</a-select>
|
</a-form-item>
|
<a-form-item label="温度">
|
<a-switch size="small" v-model="temperature" />
|
<a-space direction="vertical" size="large">
|
<a-slider :disabled="!temperature" v-model="form.llm_setting.temperature" :step="0.01" :min="0" :max="1" :style="{ width: '350px', marginLeft: '20px' }" show-input />
|
</a-space>
|
</a-form-item>
|
<a-form-item label="top P">
|
<a-switch size="small" v-model="top_p" />
|
<a-space direction="vertical" size="large">
|
<a-slider :disabled="!top_p" v-model="form.llm_setting.top_p" :step="0.01" :min="0" :max="1" :style="{ width: '350px', marginLeft: '20px' }" show-input />
|
</a-space>
|
</a-form-item>
|
<a-form-item label="出席处罚">
|
<a-switch size="small" v-model="presence_penalty" />
|
<a-space direction="vertical" size="large">
|
<a-slider :disabled="!presence_penalty" v-model="form.llm_setting.presence_penalty" :step="0.01" :min="0" :max="1" :style="{ width: '350px', marginLeft: '20px' }" show-input />
|
</a-space>
|
</a-form-item>
|
<a-form-item label="频率处罚">
|
<a-switch size="small" v-model="frequency_penalty" />
|
<a-space direction="vertical" size="large">
|
<a-slider :disabled="!frequency_penalty" v-model="form.llm_setting.frequency_penalty" :step="0.01" :min="0" :max="1" :style="{ width: '350px', marginLeft: '20px' }" show-input />
|
</a-space>
|
</a-form-item>
|
<a-form-item label="最大token数">
|
<a-switch size="small" v-model="max_tokens" />
|
<a-space direction="vertical" size="large">
|
<a-slider :disabled="!max_tokens" v-model="form.llm_setting.max_tokens" :min="0" :max="2048" :style="{ width: '350px', marginLeft: '20px' }" show-input />
|
</a-space>
|
</a-form-item>
|
</a-form>
|
</div>
|
</a-tab-pane>
|
</a-tabs>
|
</a-col>
|
</a-row>
|
</a-modal>
|
</template>
|
|
<script lang="ts" setup>
|
import { onMounted, onBeforeMount, reactive, ref, nextTick } from "vue";
|
import { kbdocumentupload, queryKbList, queryModelList } from "@/api/kbList";
|
import useLoading from "@/hooks/loading";
|
import { Message } from "@arco-design/web-vue";
|
import { dialogSet } from "@/api/Agent";
|
import EventBus from "@/utils/EventBus";
|
const { loading,setLoading } = useLoading(true);
|
const visible = ref(false);
|
const modelList = ref({});
|
const rankModelList = ref({});
|
let tabs = ref([]);
|
const file = ref();
|
const formRef = ref();
|
const formRef1 = ref();
|
const formRef2 = ref();
|
const activeKey = ref('1');
|
const form = reactive({
|
name: "",
|
icon: "",
|
language: "English",
|
prompt_config: {
|
empty_response: "",
|
prologue: "你好！ 我是你的助理，有什么可以帮到你的吗？",
|
quote: true,
|
self_rag: false,
|
system: "你是一个智能助手，请总结知识库的内容来回答问题，请列举知识库中的数据详细回答。当所有知识库内容都与问题无关时，你的回答必须包括“知识库中未找到您要的答案！”这句话。" +
|
"回答需要考虑聊天历史。\n 以下是知识库：\n {knowledge}\n 以上是知识库。",
|
parameters: [
|
{
|
index: 0,
|
key: "knowledge",
|
optional: false
|
}
|
]
|
},
|
kb_ids: [],
|
llm_id: "qwen-plus",
|
llm_setting: {
|
temperature: 0.1,
|
top_p: 0.3,
|
presence_penalty: 0.4,
|
frequency_penalty: 0.7,
|
max_tokens: 512
|
},
|
similarity_threshold: 0.2,
|
vector_similarity_weight: 0.30000000000000004,
|
top_n: 8,
|
rerank_id:'',
|
top_k:1024,
|
});
|
let temperature = ref(true);
|
let top_p = ref(true);
|
let presence_penalty = ref(true);
|
let frequency_penalty = ref(true);
|
let max_tokens = ref(true);
|
|
|
const height = ref('calc(500px)');
|
const props = defineProps(['typeAngint', 'formData']);
|
const emit = defineEmits(['queryList']);
|
const columns = [
|
{
|
title: '关键字',
|
dataIndex: 'key',
|
slotName: 'key',
|
},
|
{
|
title: '可选',
|
dataIndex: 'optional',
|
slotName: 'optional',
|
},
|
{
|
title: '操作',
|
dataIndex: 'action',
|
slotName: 'action',
|
},
|
];
|
|
const formatter = (value) => {
|
return `${(value / 100).toFixed(2)}`
|
};
|
|
const uploadAction = '/api/v1/llm/upload'; // 替换为你的上传API
|
const fileList = ref([]);
|
const imageUrls = ref([]);
|
|
const updateFileList = (newFileList) => {
|
fileList.value = newFileList;
|
console.log(newFileList, 88);
|
};
|
|
const handleSuccess = (urls) => {
|
imageUrls.value = urls; // 拿到上传的图片地址
|
form.icon = urls[0];
|
};
|
|
const rules = {
|
name: [
|
{
|
required: true,
|
message: "名称不允许为空"
|
}
|
],
|
"prompt_config.system": [
|
{
|
required: true,
|
message: "系统不允许为空"
|
}
|
],
|
llm_id: [
|
{
|
required: true,
|
message: '模型不能为空',
|
},
|
],
|
kb_ids: [
|
{
|
required: true,
|
message: '知识库不能为空',
|
},
|
],
|
};
|
|
const handleSubmit = ({ values, errors }) => {
|
console.log("values:", values, "\nerrors:", errors);
|
if (!errors) {
|
|
}
|
};
|
|
const handleClick = () => {
|
visible.value = true;
|
};
|
defineExpose({
|
handleClick
|
});
|
|
const handleCancel = () => {
|
visible.value = false;
|
formRef.value.resetFields();
|
formRef1.value.resetFields();
|
formRef2.value.resetFields();
|
form.name = "";
|
emit('queryList');
|
};
|
|
const handleOpened = (el) => {
|
// Object.assign(form,{
|
// name: '',// 用户名
|
// nameJoin: '',// 昵称
|
// post: '',// 岗位
|
// txt: '',// 备注
|
// });
|
activeKey.value= '1'
|
formRef.value.resetFields();
|
formRef1.value.resetFields();
|
formRef2.value.resetFields();
|
|
if(props.typeAngint=='add'){
|
form.name = "";
|
form.prompt_config.system = "你是一个智能助手，请总结知识库的内容来回答问题，请列举知识库中的数据详细回答。当所有知识库内容都与问题无关时，你的回答必须包括“知识库中未找到您要的答案！”这句话。" +
|
"回答需要考虑聊天历史。\n 以下是知识库：\n {knowledge}\n 以上是知识库。";
|
}
|
|
};
|
|
|
const handleBeforeOk = async (done) => {
|
formRef.value.validate().then(res => {
|
// console.log('res:', res)
|
})
|
formRef1.value.validate().then(res => {
|
// console.log('res:', res)
|
})
|
formRef2.value.validate().then(res => {
|
// console.log('res:', res)
|
})
|
if (form.name && form.kb_ids.length>0 && form.prompt_config.system && form.llm_id) {
|
let title = '创建成功';
|
let formNew = { ...form };
|
if(formNew.rerank_id){
|
delete formNew.top_k;
|
}
|
|
if(formNew.rerank_id){
|
delete formNew.top_k;
|
}
|
|
if (props.typeAngint == 'edit') {
|
formNew.dialog_id = form.id;
|
delete formNew.id;
|
delete formNew.off;
|
title = '修改成功';
|
}
|
setLoading(true)
|
try {
|
const data = await dialogSet(formNew);
|
// console.log(data, 'data');
|
if (data.code == 0){
|
Message.success(title);
|
handleCancel();
|
EventBus.emit('queryList');
|
}else {
|
Message.error(data.msg);
|
}
|
done(true);
|
setLoading(false)
|
} catch (err) {
|
// you can report use errorHandler or other
|
setLoading(false)
|
}
|
}else {
|
Message.warning('请填写必填项');
|
done(false)
|
}
|
};
|
|
const editClick = (data) => {
|
visible.value = true;
|
nextTick(()=>{
|
// console.log(props.formData,'传入数据');
|
Object.assign(form, props.formData);
|
console.log(form,'表单数据');
|
})
|
};
|
|
const addVariable = () => {
|
form.prompt_config.parameters.push({
|
index: form.prompt_config.parameters.length,
|
key: "",
|
optional: true
|
});
|
nextTick(() => {
|
formRef.value.validate();
|
});
|
}
|
|
const deleteParameters = (record) => {
|
// console.log(record, 'record');
|
// console.log(form.prompt_config.parameters);
|
form.prompt_config.parameters = form.prompt_config.parameters.filter(item => item.index !== record.index);
|
|
}
|
|
function handleTabChange(key) {
|
// 处理标签改变事件
|
console.log('Changed to tab:', key);
|
activeKey.value=key;
|
}
|
|
const queryModel = async (params) => {
|
try {
|
const data = await queryModelList(params);
|
// console.log(data.data, '大模型列表');
|
modelList.value = data.data;
|
rankModelList.value = {
|
BAAI: [data.data.BAAI[1]],
|
Jina: data.data.Jina,
|
youdao: data.data.youdao
|
};
|
} catch (err) {
|
// you can report use errorHandler or other
|
} finally {
|
}
|
};
|
|
|
const knowledgeData = async (params = { page: 1, page_size: 20 }) => {
|
setLoading(true);
|
try {
|
const { data } = await queryKbList(params);
|
// console.log(data, 'data');
|
nextTick(() => {
|
tabs.value = data;
|
// console.log(tabs.value, 'tabs');
|
});
|
} catch (err) {
|
// you can report use errorHandler or other
|
} finally {
|
setLoading(false);
|
}
|
};
|
|
onBeforeMount(() => {
|
queryModel({});
|
knowledgeData();
|
});
|
onMounted(() => {});
|
|
|
|
|
</script>
|
<style lang="less" scoped>
|
:deep(.arco-tabs-nav-tab-list) {
|
width: 100%;
|
|
.arco-tabs-tab {
|
width: 33%;
|
|
.arco-tabs-tab-title {
|
width: 100%;
|
display: inline-block;
|
text-align: center;
|
}
|
}
|
}
|
</style>
|
