54 lines
1.7 KiB
TypeScript
54 lines
1.7 KiB
TypeScript
import express from "express";
|
||
import { success, error } from "@/lib/responseFormat";
|
||
import u from "@/utils";
|
||
import { validateFields } from "@/middleware/middleware";
|
||
import { z } from "zod";
|
||
const router = express.Router();
|
||
|
||
// 检查语言模型
|
||
export default router.post(
|
||
"/",
|
||
validateFields({
|
||
modelName: z.string().optional(),
|
||
apiKey: z.string(),
|
||
baseURL: z.string().optional(),
|
||
manufacturer: z.string(),
|
||
}),
|
||
async (req, res) => {
|
||
const { modelName, apiKey, baseURL, manufacturer } = req.body;
|
||
try {
|
||
const image = await u.ai.image({
|
||
prompt: "生成16：9 四宫格图片，第一宫格是一只猫，第二宫格是一只狗， 第三宫格是一只老虎，第四宫格是猪。保证四宫格图片标准四等分",
|
||
imageBase64: [],
|
||
aspectRatio: "16:9",
|
||
size: "1K",
|
||
});
|
||
res.status(200).send(success(image));
|
||
} catch (e: any) {
|
||
console.log("%c Line:28 🥒 e", "background:#fca650", e);
|
||
return res.status(500).send(error(e?.response?.data ?? e?.message ?? "生成失败"));
|
||
}
|
||
|
||
// try {
|
||
// const contentStr = await u.ai.generateImage(
|
||
// {
|
||
// prompt: "2D cat",
|
||
// imageBase64: [],
|
||
// aspectRatio: "16:9",
|
||
// size: "1K",
|
||
// },
|
||
// {
|
||
// model: modelName,
|
||
// apiKey,
|
||
// baseURL,
|
||
// manufacturer,
|
||
// },
|
||
// );
|
||
// res.status(200).send(success(contentStr));
|
||
// } catch (err: any) {
|
||
// const message = err?.response?.data?.error?.message || err?.error?.message || "模型调用失败";
|
||
// res.status(500).send(error(message));
|
||
// }
|
||
},
|
||
);
|