diff --git a/client/src/pages/api/openapi/plugin/vector.ts b/client/src/pages/api/openapi/plugin/vector.ts index db76067f1..0a1a1a0a1 100644 --- a/client/src/pages/api/openapi/plugin/vector.ts +++ b/client/src/pages/api/openapi/plugin/vector.ts @@ -66,14 +66,14 @@ export async function getVector({ ...axiosConfig() } ) - .then((res) => { + .then(async (res) => { if (!res.data?.data?.[0]?.embedding) { // @ts-ignore return Promise.reject(res.data?.error?.message || 'Embedding Error'); } return { tokenLen: res.data.usage.total_tokens || 0, - vectors: res.data.data.map((item) => unityDimensional(item.embedding)) + vectors: await Promise.all(res.data.data.map((item) => unityDimensional(item.embedding))) }; }); @@ -88,6 +88,7 @@ export async function getVector({ } function unityDimensional(vector: number[]) { + if (vector.length > 1536) return Promise.reject('向量维度不能超过 1536'); let resultVector = vector; const vectorLen = vector.length;