From 6c70f0601dd3785a4484de7fa372b11171c98dee Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 23 Aug 2023 15:08:21 +0800
Subject: [PATCH] perf: vector unit

---
 client/src/pages/api/openapi/plugin/vector.ts | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/client/src/pages/api/openapi/plugin/vector.ts b/client/src/pages/api/openapi/plugin/vector.ts
index f4b1fbfcb..db76067f1 100644
--- a/client/src/pages/api/openapi/plugin/vector.ts
+++ b/client/src/pages/api/openapi/plugin/vector.ts
@@ -67,13 +67,13 @@ export async function getVector({
       }
     )
     .then((res) => {
-      if (!res.data?.usage?.total_tokens) {
+      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) => item.embedding)
+        vectors: res.data.data.map((item) => unityDimensional(item.embedding))
       };
     });
 
@@ -86,3 +86,12 @@ export async function getVector({
 
   return result;
 }
+
+function unityDimensional(vector: number[]) {
+  let resultVector = vector;
+  const vectorLen = vector.length;
+
+  const zeroVector = new Array(1536 - vectorLen).fill(0);
+
+  return resultVector.concat(zeroVector);
+}