mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 13:03:50 +00:00
rerank api (#525)
This commit is contained in:
@@ -5,12 +5,12 @@ import { withNextCors } from '@fastgpt/service/common/middle/cors';
|
||||
import { pushReRankBill } from '@/service/support/wallet/bill/push';
|
||||
import { connectToDatabase } from '@/service/mongo';
|
||||
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';
|
||||
|
||||
export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
|
||||
let { query, inputs } = req.body as PostReRankProps;
|
||||
try {
|
||||
let { query, inputs } = req.body as PostReRankProps;
|
||||
await connectToDatabase();
|
||||
const { teamId, tmbId } = await authCert({
|
||||
req,
|
||||
@@ -29,14 +29,15 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
|
||||
source: 'api'
|
||||
});
|
||||
|
||||
jsonRes(res, {
|
||||
jsonRes<PostReRankResponse>(res, {
|
||||
data: result
|
||||
});
|
||||
} catch (err) {
|
||||
console.log(err);
|
||||
jsonRes(res, {
|
||||
code: 500,
|
||||
error: err
|
||||
jsonRes<PostReRankResponse>(res, {
|
||||
data: inputs.map((input) => ({
|
||||
id: input.id
|
||||
}))
|
||||
});
|
||||
}
|
||||
});
|
||||
|
@@ -398,7 +398,7 @@ export async function reRankSearchResult({
|
||||
if (!target) return null;
|
||||
return {
|
||||
...target,
|
||||
score: item.score || 0
|
||||
score: item.score || target.score
|
||||
};
|
||||
})
|
||||
.filter(Boolean) as SearchDataResponseItemType[];
|
||||
|
Reference in New Issue
Block a user