From ae2887e9568c08cdc447abfcd4adeaf3ec0fa92d Mon Sep 17 00:00:00 2001
From: Archer <545436317@qq.com>
Date: Mon, 11 Sep 2023 10:15:52 +0800
Subject: [PATCH] fix: file_id undefined bug (#275)

---
 client/src/pages/api/admin/initv44.ts         | 13 +++++++++++--
 client/src/pages/api/openapi/plugin/vector.ts |  1 +
 client/src/service/pg.ts                      |  2 +-
 3 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/client/src/pages/api/admin/initv44.ts b/client/src/pages/api/admin/initv44.ts
index f6db795f6..7c0a78596 100644
--- a/client/src/pages/api/admin/initv44.ts
+++ b/client/src/pages/api/admin/initv44.ts
@@ -3,7 +3,9 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { authUser } from '@/service/utils/auth';
 import { connectToDatabase, KB } from '@/service/mongo';
-import { KbTypeEnum, KbTypeMap } from '@/constants/kb';
+import { KbTypeEnum } from '@/constants/kb';
+import { PgClient } from '@/service/pg';
+import { PgTrainingTableName } from '@/constants/plugin';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
@@ -22,7 +24,14 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       }
     );
 
-    jsonRes(res, {});
+    const response = await PgClient.update(PgTrainingTableName, {
+      where: [['file_id', 'undefined']],
+      values: [{ key: 'file_id', value: '' }]
+    });
+
+    jsonRes(res, {
+      data: response.rowCount
+    });
   } catch (error) {
     jsonRes(res, {
       code: 500,
diff --git a/client/src/pages/api/openapi/plugin/vector.ts b/client/src/pages/api/openapi/plugin/vector.ts
index 72181e889..356cfa620 100644
--- a/client/src/pages/api/openapi/plugin/vector.ts
+++ b/client/src/pages/api/openapi/plugin/vector.ts
@@ -68,6 +68,7 @@ export async function getVector({
     )
     .then(async (res) => {
       if (!res.data?.data?.[0]?.embedding) {
+        console.log(res.data);
         // @ts-ignore
         return Promise.reject(res.data?.err?.message || 'Embedding API Error');
       }
diff --git a/client/src/service/pg.ts b/client/src/service/pg.ts
index edffeb443..090be6fb6 100644
--- a/client/src/service/pg.ts
+++ b/client/src/service/pg.ts
@@ -179,7 +179,7 @@ export const insertKbItem = ({
       { key: 'user_id', value: userId },
       { key: 'kb_id', value: kbId },
       { key: 'source', value: item.source?.slice(0, 30)?.trim() || '' },
-      { key: 'file_id', value: item.file_id },
+      { key: 'file_id', value: item.file_id || '' },
       { key: 'q', value: item.q.replace(/'/g, '"') },
       { key: 'a', value: item.a.replace(/'/g, '"') },
       { key: 'vector', value: `[${item.vector}]` }