96 lines
3 KiB
JavaScript
96 lines
3 KiB
JavaScript
|
import { NextResponse } from "next/server";
|
||
|
|
import { getProviderConnectionById, updateProviderConnection } from "@/lib/localDb";
|
||
|
|
|
||
|
|
// POST /api/providers/[id]/test - Test connection
|
||
|
|
export async function POST(request, { params }) {
|
||
|
|
try {
|
||
|
|
const { id } = await params;
|
||
|
|
const connection = await getProviderConnectionById(id);
|
||
|
|
|
||
|
|
if (!connection) {
|
||
|
|
return NextResponse.json({ error: "Connection not found" }, { status: 404 });
|
||
|
|
}
|
||
|
|
|
||
|
|
let isValid = false;
|
||
|
|
let error = null;
|
||
|
|
|
||
|
|
try {
|
||
|
|
if (connection.authType === "apikey") {
|
||
|
|
// Test API key
|
||
|
|
switch (connection.provider) {
|
||
|
|
case "openai":
|
||
|
|
const openaiRes = await fetch("https://api.openai.com/v1/models", {
|
||
|
|
headers: { "Authorization": `Bearer ${connection.apiKey}` },
|
||
|
|
});
|
||
|
|
isValid = openaiRes.ok;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case "anthropic":
|
||
|
|
const anthropicRes = await fetch("https://api.anthropic.com/v1/messages", {
|
||
|
|
method: "POST",
|
||
|
|
headers: {
|
||
|
|
"x-api-key": connection.apiKey,
|
||
|
|
"anthropic-version": "2023-06-01",
|
||
|
|
"content-type": "application/json",
|
||
|
|
},
|
||
|
|
body: JSON.stringify({
|
||
|
|
model: "claude-3-haiku-20240307",
|
||
|
|
max_tokens: 1,
|
||
|
|
messages: [{ role: "user", content: "test" }],
|
||
|
|
}),
|
||
|
|
});
|
||
|
|
isValid = anthropicRes.status !== 401;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case "gemini":
|
||
|
|
const geminiRes = await fetch(`https://generativelanguage.googleapis.com/v1/models?key=${connection.apiKey}`);
|
||
|
|
isValid = geminiRes.ok;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case "openrouter":
|
||
|
|
const openrouterRes = await fetch("https://openrouter.ai/api/v1/models", {
|
||
|
|
headers: { "Authorization": `Bearer ${connection.apiKey}` },
|
||
|
|
});
|
||
|
|
isValid = openrouterRes.ok;
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
error = "Provider test not supported";
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// OAuth - check if token exists and not expired
|
||
|
|
if (connection.accessToken) {
|
||
|
|
if (connection.expiresAt) {
|
||
|
|
const expiresAt = new Date(connection.expiresAt).getTime();
|
||
|
|
isValid = expiresAt > Date.now();
|
||
|
|
if (!isValid) error = "Token expired";
|
||
|
|
} else {
|
||
|
|
isValid = true;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
error = "No access token";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} catch (err) {
|
||
|
|
error = err.message;
|
||
|
|
isValid = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update status in db
|
||
|
|
await updateProviderConnection(id, {
|
||
|
|
testStatus: isValid ? "active" : "error",
|
||
|
|
lastError: isValid ? null : error,
|
||
|
|
lastErrorAt: isValid ? null : new Date().toISOString(),
|
||
|
|
});
|
||
|
|
|
||
|
|
return NextResponse.json({
|
||
|
|
valid: isValid,
|
||
|
|
error: isValid ? null : error,
|
||
|
|
});
|
||
|
|
} catch (error) {
|
||
|
|
console.log("Error testing connection:", error);
|
||
|
|
return NextResponse.json({ error: "Test failed" }, { status: 500 });
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|