v4.6 -1 (#459)

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

@@ -0,0 +1,58 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@fastgpt/service/common/response';
+import { authCert } from '@fastgpt/service/support/permission/auth/common';
+import { withNextCors } from '@fastgpt/service/common/middle/cors';
+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';
+
+type Props = GetVectorProps & {
+  billId?: string;
+};
+
+export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
+  try {
+    let { input, model, billId } = req.body as Props;
+    await connectToDatabase();
+    const { teamId, tmbId } = await authCert({ req, authToken: true });
+
+    if (!Array.isArray(input) || typeof input !== 'string') {
+      throw new Error('input is nor array or string');
+    }
+
+    await authTeamBalance(teamId);
+
+    const { tokenLen, vectors } = await getVectorsByText({ input, model });
+
+    pushGenerateVectorBill({
+      teamId,
+      tmbId,
+      tokenLen: tokenLen,
+      model,
+      billId
+    });
+
+    jsonRes(res, {
+      data: {
+        object: 'list',
+        data: vectors.map((item, index) => ({
+          object: 'embedding',
+          index: index,
+          embedding: item
+        })),
+        model,
+        usage: {
+          prompt_tokens: tokenLen,
+          total_tokens: tokenLen
+        }
+      }
+    });
+  } catch (err) {
+    console.log(err);
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+});