perf: vector over range error

labring · Aug 28, 2023 · c5c3826 · c5c3826
1 parent 42fec3a
commit c5c3826
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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;