瀏覽代碼

向量模型通过模型管理配置,修改前端获取模型list增加模型类型参数

winkey 3 周之前
父節點
當前提交
b168b4a20c

+ 4 - 1
src/api/model.ts

@@ -4,10 +4,13 @@ import request from '@/utils/request/req';
  * 查询未隐藏模型
  * @returns 
  */
-export function modelList() {
+export function modelList(category: string) {
 	return request({
 		url: '/system/model/modelList',
 		method: 'get',
+		params: {
+            category: category
+        }
 	})
 }
 

+ 1 - 1
src/components/common/PromptStore/index.vue

@@ -213,7 +213,7 @@ const tableData = ref([]);
 const fetchData1 = async () => {
 	try {
 		// 发起一个请求
-		const [err, result] = await to(modelList());
+		const [err, result] = await to(modelList('chat'));
 
 		if (err) {
 			message.error(err.message)

+ 1 - 1
src/views/fanyi/components/textComponent.vue

@@ -149,7 +149,7 @@ onMounted(() => {
 
 async function getModelList() {
 	try {
-		const res = await modelList();
+		const res = await modelList('chat');
 		modelListData.value = res.data;
 		model.value = modelListData.value[0]?.modelDescribe || "";
 	} catch (error) {

+ 16 - 6
src/views/knowledge/index.vue

@@ -26,9 +26,10 @@ import {
 import to from "await-to-js";
 import { useRouter } from "vue-router";
 import { t } from "@/locales";
+import { modelList } from "@/api/model";
 
 onMounted(() => {
-	fetchData();
+	fetchData(),getModelList();
 });
 
 const router = useRouter();
@@ -102,13 +103,20 @@ const placement = ref<DrawerPlacement>("right");
 
 const getVector = reactive([
 	{ label: "weaviate", value: "weaviate" },
-	{ label: "milvus", value: "milvus" },
+	{ label: "milvus", value: "milvus" }, 
 ]);
 
-const getVectorModel = reactive([
-	{ label: "text-embedding-3-small", value: "text-embedding-3-small" },
-	{ label: "bge-large-zh-v1.5", value: "quentinz/bge-large-zh-v1.5" },
-]);
+const getVectorModel = ref([]);
+
+async function getModelList() {
+	try {
+		const res = await modelList('vector');
+		getVectorModel.value = res.data;
+	} catch (error) {
+		console.error("获取模型列表失败:", error);
+		message.error("获取模型列表失败");
+	}
+}
 
 const createColumns = () => {
 	return [
@@ -318,6 +326,8 @@ const columns = ref(createColumns());
 								<n-select
 									:options="getVectorModel"
 									v-model:value="formValue.vectorModel"
+									value-field="modelDescribe"
+									label-field="modelName"
 									placeholder="请选择向量模型"
 									clearable
 								></n-select>

+ 1 - 1
src/views/mj/aiModel.vue

@@ -24,7 +24,7 @@ const config = ref([])
 const fetchData = async () => {
 	try {
 		// 发起一个请求
-		const [err, result] = await to(modelList());
+		const [err, result] = await to(modelList('chat'));
 
 		if (err) {
 			message.error(err.message)