67 lines
1.9 KiB
TypeScript
67 lines
1.9 KiB
TypeScript
import { callZhipuAPI } from "@/lib/utils";
|
||
import { NextRequest, NextResponse } from "next/server";
|
||
|
||
async function translate(text: string, target_lang: string) {
|
||
console.log(`translate "${text}" into ${target_lang}`);
|
||
const messages = [
|
||
{
|
||
role: "user",
|
||
content: `
|
||
请推断以下文本的locale，并翻译到目标语言${target_lang}，同样需要locale信息，以JSON格式返回
|
||
[${text}]
|
||
结果如：
|
||
{
|
||
"source_locale": "zh-CN",
|
||
"target_locale": "de-DE",
|
||
"target_text": "Halo"
|
||
}
|
||
注意：
|
||
直接返回json文本，
|
||
locale如果可能有多个，选取最可能的一个，其中使用符号"-"，
|
||
locale如果推断失败，就当作是en-US
|
||
`,
|
||
},
|
||
];
|
||
try {
|
||
const response = await callZhipuAPI(messages);
|
||
let to_parse = response.choices[0].message.content.trim() as string;
|
||
if (to_parse.startsWith("`"))
|
||
to_parse = to_parse.slice(7, to_parse.length - 3);
|
||
if (to_parse.length === 0) throw Error("ai啥也每说");
|
||
return JSON.parse(to_parse);
|
||
} catch (error) {
|
||
console.error(error);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
export async function GET(request: NextRequest) {
|
||
try {
|
||
const searchParams = request.nextUrl.searchParams;
|
||
const text = searchParams.get("text");
|
||
const target_lang = searchParams.get("target");
|
||
|
||
if (!text || !target_lang) {
|
||
return NextResponse.json(
|
||
{ error: "查询参数错误", message: "text参数, target参数是必需的" },
|
||
{ status: 400 },
|
||
);
|
||
}
|
||
|
||
const textInfo = await translate(text, target_lang);
|
||
if (!textInfo) {
|
||
return NextResponse.json(
|
||
{ error: "服务暂时不可用", message: "LLM API 请求失败" },
|
||
{ status: 503 },
|
||
);
|
||
}
|
||
return NextResponse.json(textInfo, { status: 200 });
|
||
} catch (error) {
|
||
console.error("API 错误:", error);
|
||
return NextResponse.json(
|
||
{ error: "服务器内部错误", message: "请稍后重试" },
|
||
{ status: 500 },
|
||
);
|
||
}
|
||
}
|