use zhipuai to generate api
This commit is contained in:
@@ -1,10 +1,46 @@
|
||||
import { GoogleGenAI } from "@google/genai";
|
||||
import { NextRequest, NextResponse } from "next/server";
|
||||
import { env } from "process";
|
||||
|
||||
const api_key = env.GEMINI_API_KEY;
|
||||
const ai = new GoogleGenAI(api_key ? { apiKey: api_key } : {});
|
||||
const prompt = `[TEXT]
|
||||
const API_KEY = env.ZHIPU_API_KEY;
|
||||
|
||||
async function callZhipuAPI(messages: { role: string, content: string }[], model = 'glm-4.5-flash') {
|
||||
const url = 'https://open.bigmodel.cn/api/paas/v4/chat/completions';
|
||||
|
||||
const response = await fetch(url, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Authorization': 'Bearer ' + API_KEY,
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: model,
|
||||
messages: messages,
|
||||
temperature: 0.6
|
||||
})
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`API 调用失败: ${response.status}`);
|
||||
}
|
||||
|
||||
return await response.json();
|
||||
}
|
||||
|
||||
|
||||
|
||||
// callZhipuAPI(messages)
|
||||
// .then(result => {
|
||||
// console.log(result.choices[0].message.content);
|
||||
// })
|
||||
// .catch(error => {
|
||||
// console.error('错误:', error);
|
||||
// });
|
||||
|
||||
async function getIPAFromLLM(text: string) {
|
||||
const messages = [
|
||||
{
|
||||
role: 'user', content: `
|
||||
[TEXT]
|
||||
请推断以上文本的语言,并返回其宽式国际音标(IPA),以JSON格式
|
||||
如:
|
||||
{
|
||||
@@ -14,32 +50,48 @@ const prompt = `[TEXT]
|
||||
注意:直接返回json文本,
|
||||
不要带markdown记号,
|
||||
ipa一定要加[],
|
||||
lang的值是小写英语的语言名称`;
|
||||
|
||||
async function getIPAFromGemini(text: string) {
|
||||
const response = await ai.models.generateContent({
|
||||
model: "gemini-2.5-flash",
|
||||
contents: prompt.replace("[TEXT]", text),
|
||||
});
|
||||
if (response.text === undefined) return null;
|
||||
return JSON.parse(response.text);
|
||||
}
|
||||
|
||||
export async function ggetIPA(text: string): Promise<{ lang: string, ipa: string } | null> {
|
||||
return {
|
||||
lang: `(这是的${text}的lang)`,
|
||||
ipa: `(这是的${text}的ipa)`
|
||||
};
|
||||
lang的值是小写英语的语言名称
|
||||
`.replace('[TEXT]', text)
|
||||
}
|
||||
];
|
||||
try {
|
||||
const response = await callZhipuAPI(messages);
|
||||
return JSON.parse(response.choices[0].message.content);
|
||||
} catch (error) {
|
||||
console.error(error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
export async function GET(request: NextRequest) {
|
||||
const searchParams = request.nextUrl.searchParams;
|
||||
try {
|
||||
const searchParams = request.nextUrl.searchParams;
|
||||
const text = searchParams.get('text');
|
||||
|
||||
const text = searchParams.get('text');
|
||||
if (!text) return NextResponse.json("查询参数错误", { status: 400 });
|
||||
if (!text) {
|
||||
return NextResponse.json(
|
||||
{ error: "查询参数错误", message: "text 参数是必需的" },
|
||||
{ status: 400 }
|
||||
);
|
||||
}
|
||||
|
||||
const r = await getIPAFromGemini(text);
|
||||
if (r === null) return NextResponse.json("Gemini Api请求失败", { status: 424 });
|
||||
const ipaData = await getIPAFromLLM(text);
|
||||
|
||||
return NextResponse.json({ r }, { status: 200 });
|
||||
}
|
||||
if (!ipaData) {
|
||||
return NextResponse.json(
|
||||
{ error: "服务暂时不可用", message: "LLM API 请求失败" },
|
||||
{ status: 503 }
|
||||
);
|
||||
}
|
||||
|
||||
return NextResponse.json(ipaData, { status: 200 });
|
||||
|
||||
} catch (error) {
|
||||
console.error('API 错误:', error);
|
||||
|
||||
return NextResponse.json(
|
||||
{ error: "服务器内部错误", message: "请稍后重试" },
|
||||
{ status: 500 }
|
||||
);
|
||||
}
|
||||
}
|
||||
@@ -6,12 +6,13 @@ import { useRef, useState } from "react";
|
||||
export default function Home() {
|
||||
const respref = useRef<HTMLParagraphElement>(null);
|
||||
const inputref = useRef<HTMLTextAreaElement>(null);
|
||||
const [ipa_result, set_ipa_result] = useState<{ lang: string, ipa: string } | null>(null);
|
||||
const [genaiEnabled, setGenaiEnabled] = useState<boolean>(true);
|
||||
const [reqEnabled, setReqEnabled] = useState<boolean>(true);
|
||||
|
||||
const generateIPA = () => {
|
||||
if (!genaiEnabled) return;
|
||||
setGenaiEnabled(false);
|
||||
if (!reqEnabled) return;
|
||||
setReqEnabled(false);
|
||||
|
||||
respref.current!.innerText = '正在生成国际音标(IPA),请稍等~'
|
||||
|
||||
const text = inputref.current!.value.trim();
|
||||
if (text.length === 0) return;
|
||||
@@ -19,19 +20,21 @@ export default function Home() {
|
||||
const params = new URLSearchParams({ text: text });
|
||||
fetch(`/api/ipa?${params}`)
|
||||
.then(response => {
|
||||
if (response.ok) {
|
||||
return response.json().then((data) => {
|
||||
set_ipa_result(data);
|
||||
});
|
||||
} else {
|
||||
return response.text()
|
||||
.then(errorText => {
|
||||
console.error(errorText);
|
||||
});
|
||||
if (!response.ok) {
|
||||
return response.json().then(resj => {
|
||||
throw new Error(`HTTP ${response.status}: ${resj.error} ${resj.message}`);
|
||||
})
|
||||
}
|
||||
return response.json();
|
||||
})
|
||||
.then(data => {
|
||||
respref.current!.innerText = `LANG: ${data.lang}\nIPA:${data.ipa}`;
|
||||
})
|
||||
.catch(error => {
|
||||
respref.current!.innerText = `错误: ${error.message}`;
|
||||
})
|
||||
.finally(() => {
|
||||
setGenaiEnabled(true);
|
||||
setReqEnabled(true);
|
||||
});
|
||||
}
|
||||
const readIPA = () => {
|
||||
@@ -51,14 +54,10 @@ export default function Home() {
|
||||
</textarea>
|
||||
</div>
|
||||
<div className="m-2 flex-row flex gap-2">
|
||||
<Button onClick={generateIPA} label="生成IPA" disabled={!genaiEnabled}></Button>
|
||||
<Button onClick={generateIPA} label="生成IPA"></Button>
|
||||
<Button onClick={readIPA} label="朗读"></Button>
|
||||
{/* <Button onClick={() => { setGenaiEnabled(!genaiEnabled) }} label="test"></Button> */}
|
||||
</div>
|
||||
<div ref={respref} className="whitespace-pre-line">
|
||||
语言:{ipa_result?.lang}{'\n'}
|
||||
IPA:{ipa_result?.ipa}
|
||||
</div>
|
||||
<div ref={respref} className="whitespace-pre-line"></div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user