perf: chat framwork

src/service/events/generateVector.ts

@@ -1,6 +1,8 @@
-import { openaiCreateEmbedding, getOpenApiKey } from '../utils/openai';
+import { openaiCreateEmbedding } from '../utils/chat/openai';
+import { getApiKey } from '../utils/auth';
 import { openaiError2 } from '../errorCode';
 import { PgClient } from '@/service/pg';
+import { embeddingModel } from '@/constants/model';
 
 export async function generateVector(next = false): Promise<any> {
   if (process.env.queueTask !== '1') {
@@ -40,11 +42,11 @@ export async function generateVector(next = false): Promise<any> {
     dataId = dataItem.id;
 
     // 获取 openapi Key
-    let userApiKey, systemKey;
+    let userApiKey, systemApiKey;
     try {
-      const res = await getOpenApiKey(dataItem.userId);
+      const res = await getApiKey({ model: embeddingModel, userId: dataItem.userId });
       userApiKey = res.userApiKey;
-      systemKey = res.systemKey;
+      systemApiKey = res.systemApiKey;
     } catch (error: any) {
       if (error?.code === 501) {
         await PgClient.delete('modelData', {
@@ -61,8 +63,8 @@ export async function generateVector(next = false): Promise<any> {
     const { vector } = await openaiCreateEmbedding({
       text: dataItem.q,
       userId: dataItem.userId,
-      isPay: !userApiKey,
-      apiKey: userApiKey || systemKey
+      userApiKey,
+      systemApiKey
     });
 
     // 更新 pg 向量和状态数据