chore: sync local changes to Gitea
This commit is contained in:
57
scripts/test-api.mjs
Normal file
57
scripts/test-api.mjs
Normal file
@@ -0,0 +1,57 @@
|
||||
/**
|
||||
* 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);
|
||||
});
|
||||
Reference in New Issue
Block a user