4.6.3-website dataset (#532)

projects/app/src/pages/api/v1/embeddings.ts

@@ -6,6 +6,8 @@ import { pushGenerateVectorBill } from '@/service/support/wallet/bill/push';
 import { connectToDatabase } from '@/service/mongo';
 import { authTeamBalance } from '@/service/support/permission/auth/bill';
 import { getVectorsByText, GetVectorProps } from '@/service/core/ai/vector';
+import { updateApiKeyUsage } from '@fastgpt/service/support/openapi/tools';
+import { getBillSourceByAuthType } from '@fastgpt/global/support/wallet/bill/tools';
 
 type Props = GetVectorProps & {
   billId?: string;
@@ -15,24 +17,21 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
   try {
     let { input, model, billId } = req.body as Props;
     await connectToDatabase();
-    const { teamId, tmbId } = await authCert({ req, authToken: true, authApiKey: true });
 
-    if (!Array.isArray(input) || typeof input !== 'string') {
+    if (!Array.isArray(input) && typeof input !== 'string') {
       throw new Error('input is nor array or string');
     }
 
+    const { teamId, tmbId, apikey, authType } = await authCert({
+      req,
+      authToken: true,
+      authApiKey: true
+    });
+
     await authTeamBalance(teamId);
 
     const { tokenLen, vectors } = await getVectorsByText({ input, model });
 
-    pushGenerateVectorBill({
-      teamId,
-      tmbId,
-      tokenLen: tokenLen,
-      model,
-      billId
-    });
-
     jsonRes(res, {
       data: {
         object: 'list',
@@ -48,6 +47,22 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
         }
       }
     });
+
+    const { total } = pushGenerateVectorBill({
+      teamId,
+      tmbId,
+      tokenLen,
+      model,
+      billId,
+      source: getBillSourceByAuthType({ authType })
+    });
+
+    if (apikey) {
+      updateApiKeyUsage({
+        apikey,
+        usage: total
+      });
+    }
   } catch (err) {
     console.log(err);
     jsonRes(res, {