mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-23 05:12:39 +00:00

* perf: local file create collection * rename middleware * perf: remove code * feat: next14 * feat: external file dataset * collection tags field * external file dataset doc * fix: ts
93 lines
2.4 KiB
TypeScript
93 lines
2.4 KiB
TypeScript
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@fastgpt/service/common/response';
|
|
import { getUploadModel } from '@fastgpt/service/common/file/multer';
|
|
import { removeFilesByPaths } from '@fastgpt/service/common/file/utils';
|
|
import fs from 'fs';
|
|
import { getAIApi } from '@fastgpt/service/core/ai/config';
|
|
import { pushWhisperUsage } from '@/service/support/wallet/usage/push';
|
|
import { authChatCert } from '@/service/support/permission/auth/chat';
|
|
import { MongoApp } from '@fastgpt/service/core/app/schema';
|
|
import { getGuideModule, splitGuideModule } from '@fastgpt/global/core/workflow/utils';
|
|
import { OutLinkChatAuthProps } from '@fastgpt/global/support/permission/chat';
|
|
import { NextAPI } from '@/service/middleware/entry';
|
|
|
|
const upload = getUploadModel({
|
|
maxSize: 2
|
|
});
|
|
|
|
async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
|
|
let filePaths: string[] = [];
|
|
|
|
try {
|
|
const {
|
|
file,
|
|
data: { appId, duration, shareId, outLinkUid, teamId: spaceTeamId, teamToken }
|
|
} = await upload.doUpload<
|
|
OutLinkChatAuthProps & {
|
|
appId: string;
|
|
duration: number;
|
|
}
|
|
>(req, res);
|
|
|
|
req.body.shareId = shareId;
|
|
req.body.outLinkUid = outLinkUid;
|
|
req.body.teamId = spaceTeamId;
|
|
req.body.teamToken = teamToken;
|
|
|
|
filePaths = [file.path];
|
|
|
|
if (!global.whisperModel) {
|
|
throw new Error('whisper model not found');
|
|
}
|
|
|
|
if (!file) {
|
|
throw new Error('file not found');
|
|
}
|
|
|
|
// auth role
|
|
const { teamId, tmbId } = await authChatCert({ req, authToken: true });
|
|
// auth app
|
|
const app = await MongoApp.findById(appId, 'modules').lean();
|
|
if (!app) {
|
|
throw new Error('app not found');
|
|
}
|
|
const { whisperConfig } = splitGuideModule(getGuideModule(app?.modules));
|
|
if (!whisperConfig?.open) {
|
|
throw new Error('Whisper is not open in the app');
|
|
}
|
|
|
|
const ai = getAIApi();
|
|
|
|
const result = await ai.audio.transcriptions.create({
|
|
file: fs.createReadStream(file.path),
|
|
model: global.whisperModel.model
|
|
});
|
|
|
|
pushWhisperUsage({
|
|
teamId,
|
|
tmbId,
|
|
duration
|
|
});
|
|
|
|
jsonRes(res, {
|
|
data: result.text
|
|
});
|
|
} catch (err) {
|
|
console.log(err);
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error: err
|
|
});
|
|
}
|
|
|
|
removeFilesByPaths(filePaths);
|
|
}
|
|
|
|
export default NextAPI(handler);
|
|
|
|
export const config = {
|
|
api: {
|
|
bodyParser: false
|
|
}
|
|
};
|