45 lines
1.4 KiB
JavaScript
45 lines
1.4 KiB
JavaScript
// background.js
|
||
|
||
// 监听来自 main.js 的消息
|
||
chrome.runtime.onMessageExternal.addListener((message, sender, sendResponse) => {
|
||
// 这种方式需要知道 Extension ID，更简单的方法是统一由 content.js 转发
|
||
});
|
||
|
||
// 通用监听（推荐）
|
||
chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
|
||
if (request.type === "AI_TRANSLATE") {
|
||
// 1. 先从 storage 获取配置
|
||
chrome.storage.sync.get(['aiConfig'], async (result) => {
|
||
const config = result.aiConfig;
|
||
if (!config || !config.apiKey) {
|
||
sendResponse({ success: false, error: "未配置 API Key" });
|
||
return;
|
||
}
|
||
|
||
try {
|
||
// 2. 发起请求
|
||
const response = await fetch(`${config.apiUrl}/chat/completions`, {
|
||
method: 'POST',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
'Authorization': `Bearer ${config.apiKey}`
|
||
},
|
||
body: JSON.stringify({
|
||
model: config.modelName,
|
||
messages: [
|
||
{ role: "system", content: request.systemPrompt },
|
||
{ role: "user", content: request.userInput }
|
||
],
|
||
temperature: 0.1
|
||
})
|
||
});
|
||
|
||
const data = await response.json();
|
||
sendResponse({ success: true, data: data });
|
||
} catch (err) {
|
||
sendResponse({ success: false, error: err.message });
|
||
}
|
||
});
|
||
return true; // 保持异步
|
||
}
|
||
}); |