rerank api (#525)

packages/global/core/ai/api.d.ts

@@ -2,4 +2,4 @@ export type PostReRankProps = {
   query: string;
   query: string;
   inputs: { id: string; text: string }[];
   inputs: { id: string; text: string }[];
 };
 };
-export type PostReRankResponse = { id: string; score: number }[];
+export type PostReRankResponse = { id: string; score?: number }[];

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

@@ -5,12 +5,12 @@ import { withNextCors } from '@fastgpt/service/common/middle/cors';
 import { pushReRankBill } from '@/service/support/wallet/bill/push';
 import { pushReRankBill } from '@/service/support/wallet/bill/push';
 import { connectToDatabase } from '@/service/mongo';
 import { connectToDatabase } from '@/service/mongo';
 import { authTeamBalance } from '@/service/support/permission/auth/bill';
 import { authTeamBalance } from '@/service/support/permission/auth/bill';
-import { PostReRankProps } from '@fastgpt/global/core/ai/api';
+import { PostReRankProps, PostReRankResponse } from '@fastgpt/global/core/ai/api';
 import { reRankRecall } from '@/service/core/ai/rerank';
 import { reRankRecall } from '@/service/core/ai/rerank';
 
 
 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
-  try {
   let { query, inputs } = req.body as PostReRankProps;
   let { query, inputs } = req.body as PostReRankProps;
+  try {
     await connectToDatabase();
     await connectToDatabase();
     const { teamId, tmbId } = await authCert({
     const { teamId, tmbId } = await authCert({
       req,
       req,
@@ -29,14 +29,15 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
       source: 'api'
       source: 'api'
     });
     });
 
 
-    jsonRes(res, {
+    jsonRes<PostReRankResponse>(res, {
       data: result
       data: result
     });
     });
   } catch (err) {
   } catch (err) {
     console.log(err);
     console.log(err);
-    jsonRes(res, {
-      code: 500,
-      error: err
+    jsonRes<PostReRankResponse>(res, {
+      data: inputs.map((input) => ({
+        id: input.id
+      }))
     });
     });
   }
   }
 });
 });

projects/app/src/service/core/dataset/data/pg.ts

@@ -398,7 +398,7 @@ export async function reRankSearchResult({
         if (!target) return null;
         if (!target) return null;
         return {
         return {
           ...target,
           ...target,
-          score: item.score || 0
+          score: item.score || target.score
         };
         };
       })
       })
       .filter(Boolean) as SearchDataResponseItemType[];
       .filter(Boolean) as SearchDataResponseItemType[];