-
Notifications
You must be signed in to change notification settings - Fork 0
/
background.js
64 lines (57 loc) · 2.44 KB
/
background.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
browser.runtime.onMessage.addListener((message, sender, sendResponse) => {
if (message.action === 'summarize') {
const articleText = message.data;
const summaryType = message.summaryType;
console.log('Received article text:', articleText);
console.log('Summary type:', summaryType);
if (!articleText) {
console.error('No article text provided.');
sendResponse({ success: false, error: 'No article text provided.' });
return true;
}
browser.storage.sync.get(['model', 'apiKey', 'shortPrompt', 'longPrompt']).then((result) => {
const model = result.model || 'anthropic/claude-3-haiku';
const apiKey = result.apiKey;
const prompt = summaryType === 'short' ?
(result.shortPrompt || 'Summarize the following article for personal research purposes, use bullet points:') :
(result.longPrompt || 'Summarise comprehensively');
if (!apiKey) {
sendResponse({ success: false, error: 'API key not provided. Go to the extension options to set it.'});
return;
}
const requestBody = JSON.stringify({
model: model,
messages: [
{ role: 'user', content: `${prompt}\n\n${articleText}` }
]
});
console.log('Request body:', requestBody);
return fetch('https://openrouter.ai/api/v1/chat/completions', {
method: 'POST',
headers: {
'Authorization': `Bearer ${apiKey}`,
'Content-Type': 'application/json'
},
body: requestBody
});
})
.then(response => {
if (!response.ok) {
throw new Error(`Network response was not ok: ${response.statusText}`);
}
return response.json();
})
.then(data => {
console.log('Received response from API:', data);
const summary = data.choices[0].message.content;
console.log('Summary:', summary);
sendResponse({ success: true, summary: summary });
})
.catch(error => {
console.error('Error:', error);
sendResponse({ success: false, error: error.message });
});
// Indicate that we will send a response asynchronously
return true;
}
});