4.6.7-alpha commit (#743)

Co-authored-by: Archer <545436317@qq.com>
Co-authored-by: heheer <71265218+newfish-cmyk@users.noreply.github.com>

packages/service/common/vectorStore/controller.ts

@@ -10,6 +10,7 @@ const getVectorObj = () => {
 export const initVectorStore = getVectorObj().init;
 export const deleteDatasetDataVector = getVectorObj().delete;
 export const recallFromVectorStore = getVectorObj().recall;
+export const checkVectorDataExist = getVectorObj().checkDataExist;
 export const getVectorDataByTime = getVectorObj().getVectorDataByTime;
 export const getVectorCountByTeamId = getVectorObj().getVectorCountByTeamId;
 
@@ -21,7 +22,7 @@ export const insertDatasetDataVector = async ({
   query: string;
   model: string;
 }) => {
-  const { vectors, tokens } = await getVectorsByText({
+  const { vectors, charsLength } = await getVectorsByText({
     model,
     input: query
   });
@@ -31,32 +32,27 @@ export const insertDatasetDataVector = async ({
   });
 
   return {
-    tokens,
+    charsLength,
     insertId
   };
 };
 
 export const updateDatasetDataVector = async ({
   id,
-  query,
-  model
-}: {
+  ...props
+}: InsertVectorProps & {
   id: string;
   query: string;
   model: string;
 }) => {
-  // get vector
-  const { vectors, tokens } = await getVectorsByText({
-    model,
-    input: query
+  // insert new vector
+  const { charsLength, insertId } = await insertDatasetDataVector(props);
+
+  // delete old vector
+  await deleteDatasetDataVector({
+    teamId: props.teamId,
+    id
   });
 
-  await getVectorObj().update({
-    id,
-    vectors
-  });
-
-  return {
-    tokens
-  };
+  return { charsLength, insertId };
 };