32 lines
752 B
TypeScript
32 lines
752 B
TypeScript
import express from "express";
|
|
import u from "@/utils";
|
|
import { z } from "zod";
|
|
import { success } from "@/lib/responseFormat";
|
|
import { validateFields } from "@/middleware/middleware";
|
|
const router = express.Router();
|
|
|
|
// 获取视频模型
|
|
export default router.post(
|
|
"/",
|
|
validateFields({
|
|
userId: z.number(),
|
|
}),
|
|
async (req, res) => {
|
|
const { userId } = req.body;
|
|
|
|
const data = await u.db("t_config").where("userId", userId).select("model");
|
|
const modelData = [];
|
|
|
|
for (const item of data) {
|
|
if (item.model?.includes("sora")) {
|
|
modelData.push("sora");
|
|
}
|
|
if (item.model?.includes("doubao")) {
|
|
modelData.push("doubao");
|
|
}
|
|
}
|
|
|
|
res.status(200).send(success(modelData));
|
|
}
|
|
);
|