From 867d69659f6a155521a255dde0dc4034e01e83fd Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 19 Apr 2023 10:03:11 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20sql=20=E5=B0=81=E8=A3=85?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/service/events/generateVector.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/service/events/generateVector.ts b/src/service/events/generateVector.ts
index e137ffbda..76f462572 100644
--- a/src/service/events/generateVector.ts
+++ b/src/service/events/generateVector.ts
@@ -1,7 +1,6 @@
-import { connectRedis } from '../redis';
 import { openaiCreateEmbedding, getOpenApiKey } from '../utils/openai';
 import { openaiError2 } from '../errorCode';
-import { connectPg, PgClient } from '@/service/pg';
+import { PgClient } from '@/service/pg';
 
 export async function generateVector(next = false): Promise<any> {
   if (process.env.queueTask !== '1') {
@@ -86,8 +85,9 @@ export async function generateVector(next = false): Promise<any> {
     // 没有余额或者凭证错误时，拒绝任务
     if (dataId && openaiError2[error?.response?.data?.error?.type]) {
       console.log('删除向量生成任务记录');
-      const redis = await connectRedis();
-      redis.del(dataId);
+      await PgClient.delete('modelData', {
+        where: [['id', dataId]]
+      });
       generateVector(true);
       return;
     }