2api
feat: configurable stream stall timeout + per-provider UI
88c4c60
Raw
History Blame Contribute Delete
3.89 kB
import { NextResponse } from "next/server";
import { detectFormat, getTargetFormat } from "open-sse/services/provider.js";
import { translateRequest } from "open-sse/translator/index.js";
import { FORMATS } from "open-sse/translator/formats.js";
import { parseModel } from "open-sse/services/model.js";
import { getProviderConnections } from "@/lib/localDb.js";
import { getExecutor } from "open-sse/executors/index.js";
export async function POST(request) {
try {
const { step, body } = await request.json();
if (!step || !body) {
return NextResponse.json({ success: false, error: "Step and body required" }, { status: 400 });
}
switch (step) {
case 1: {
// Detect provider + formats from 1_req_client.json
const clientBody = body.body || body;
const { provider, model } = parseModel(clientBody.model);
const sourceFormat = detectFormat(clientBody);
const targetFormat = getTargetFormat(provider);
return NextResponse.json({ success: true, result: { provider, model, sourceFormat, targetFormat } });
}
case 2: {
// source → OpenAI intermediate (mirrors 3_req_openai.json)
// Translate source→openai only (half of the pipeline)
const clientBody = body.body || body;
const { provider, model } = parseModel(clientBody.model);
const sourceFormat = detectFormat(clientBody);
const stream = clientBody.stream !== false;
// translateRequest(source, OPENAI) = only the first half
const result = translateRequest(sourceFormat, FORMATS.OPENAI, model, clientBody, stream, null, provider);
delete result._toolNameMap;
return NextResponse.json({ success: true, result: { body: result } });
}
case 3: {
// OpenAI intermediate → target + build URL/headers (mirrors 4_req_target.json)
const openaiBody = body.body || body;
const provider = body.provider;
const model = body.model;
if (!provider || !model) {
return NextResponse.json({ success: false, error: "provider and model required" }, { status: 400 });
}
const targetFormat = getTargetFormat(provider);
const stream = openaiBody.stream !== false;
// translateRequest(OPENAI, target) = second half of pipeline
const translated = translateRequest(FORMATS.OPENAI, targetFormat, model, openaiBody, stream, null, provider);
delete translated._toolNameMap;
// Build URL + headers via executor (same as chatCore → executor.execute)
const connections = await getProviderConnections({ provider });
const connection = connections.find(c => c.isActive !== false);
if (!connection) {
return NextResponse.json({ success: false, error: `No active connection for provider: ${provider}` }, { status: 400 });
}
const credentials = {
apiKey: connection.apiKey,
accessToken: connection.accessToken,
refreshToken: connection.refreshToken,
copilotToken: connection.copilotToken,
projectId: connection.projectId,
providerSpecificData: connection.providerSpecificData
};
const executor = getExecutor(provider);
const url = executor.buildUrl(model, stream, 0, credentials);
const headers = executor.buildHeaders(credentials, stream);
const finalBody = executor.transformRequest(model, translated, stream, credentials);
return NextResponse.json({ success: true, result: { url, headers, body: finalBody } });
}
default:
return NextResponse.json({ success: false, error: "Invalid step (1-3)" }, { status: 400 });
}
} catch (error) {
console.error("Error in translator:", error);
return NextResponse.json({ success: false, error: error.message }, { status: 500 });
}
}