54 lines
1.6 KiB
TypeScript
54 lines
1.6 KiB
TypeScript
|
import { PrismaClient } from "@prisma/client";
|
|||
|
|
import { generateText } from "ai";
|
|||
|
|
import { z } from "zod";
|
|||
|
|
import { generateStructuredObject } from "../lib/ai/generate-structured";
|
|||
|
|
import { parseProviderApiKeys } from "../lib/ai/keys";
|
|||
|
|
import { getModel } from "../lib/ai/provider";
|
|||
|
|
|
|||
|
|
async function main() {
|
|||
|
|
const prisma = new PrismaClient();
|
|||
|
|
const settings = await prisma.setting.findFirst();
|
|||
|
|
const apiKeys = parseProviderApiKeys(settings?.providerApiKeys);
|
|||
|
|
const schema = z.object({ answer: z.string(), score: z.number() });
|
|||
|
|
|
|||
|
|
const models = [
|
|||
|
|
"deepseek-v4-flash",
|
|||
|
|
"deepseek-v4-pro",
|
|||
|
|
"minimax-m3",
|
|||
|
|
"minimax-m2.7",
|
|||
|
|
"kimi-k2.7-code",
|
|||
|
|
"glm-5.1",
|
|||
|
|
"qwen3.7-max",
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
for (const modelId of models) {
|
|||
|
|
const model = getModel("opencode-go", modelId, apiKeys);
|
|||
|
|
process.stdout.write(`\n=== ${modelId} ===\n`);
|
|||
|
|
try {
|
|||
|
|
const { text } = await generateText({ model, prompt: "只回覆 OK" });
|
|||
|
|
console.log("generateText OK:", text.trim().slice(0, 60));
|
|||
|
|
} catch (error) {
|
|||
|
|
console.log("generateText FAIL:", error instanceof Error ? error.message : error);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const object = await generateStructuredObject({
|
|||
|
|
model,
|
|||
|
|
provider: "opencode-go",
|
|||
|
|
modelId,
|
|||
|
|
schema,
|
|||
|
|
prompt: "回傳 JSON：answer 一句話，score 為 0-100",
|
|||
|
|
});
|
|||
|
|
console.log("generateStructuredObject OK:", object);
|
|||
|
|
} catch (error) {
|
|||
|
|
console.log("generateStructuredObject FAIL:", error instanceof Error ? error.message : error);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
await prisma.$disconnect();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
main().catch((error) => {
|
|||
|
|
console.error(error);
|
|||
|
|
process.exit(1);
|
|||
|
|
});
|