58 lines
1.7 KiB
JavaScript
58 lines
1.7 KiB
JavaScript
/**
|
|
* Local API smoke test. Usage:
|
|
* node scripts/test-api.mjs [baseUrl]
|
|
* Default baseUrl: http://127.0.0.1:8788
|
|
*/
|
|
const base = process.argv[2] ?? "http://127.0.0.1:8788";
|
|
const apiKey = process.env.API_KEY ?? "sk-shumengya666";
|
|
const headers = {
|
|
authorization: `Bearer ${apiKey}`,
|
|
"content-type": "application/json",
|
|
};
|
|
|
|
async function main() {
|
|
console.log("=== GET /v1/models ===");
|
|
const modelsRes = await fetch(`${base}/v1/models`, { headers });
|
|
console.log("status:", modelsRes.status);
|
|
const modelsText = await modelsRes.text();
|
|
if (!modelsRes.ok) {
|
|
console.log(modelsText.slice(0, 500));
|
|
process.exit(1);
|
|
}
|
|
const models = JSON.parse(modelsText);
|
|
const ids = (models.data ?? []).map((m) => m.id);
|
|
console.log("count:", ids.length);
|
|
console.log("sample:", ids.slice(0, 15).join(", "));
|
|
|
|
const testModels = [
|
|
ids.find((id) => id.includes("gpt-4o")) ?? ids[0],
|
|
ids.find((id) => id.includes("claude")) ?? ids[1],
|
|
].filter(Boolean);
|
|
|
|
for (const model of [...new Set(testModels)].slice(0, 2)) {
|
|
console.log(`\n=== POST /v1/chat/completions (${model}) ===`);
|
|
const chatRes = await fetch(`${base}/v1/chat/completions`, {
|
|
method: "POST",
|
|
headers,
|
|
body: JSON.stringify({
|
|
model,
|
|
max_tokens: 32,
|
|
messages: [{ role: "user", content: "Reply with exactly: ok" }],
|
|
}),
|
|
});
|
|
console.log("status:", chatRes.status);
|
|
const chatBody = await chatRes.text();
|
|
if (!chatRes.ok) {
|
|
console.log(chatBody.slice(0, 400));
|
|
continue;
|
|
}
|
|
const chat = JSON.parse(chatBody);
|
|
console.log("reply:", chat.choices?.[0]?.message?.content?.slice(0, 120));
|
|
}
|
|
}
|
|
|
|
main().catch((e) => {
|
|
console.error(e);
|
|
process.exit(1);
|
|
});
|