spacellama/background.js
2024-10-13 17:54:38 +02:00

64 lines
1.8 KiB
JavaScript

console.log("Background script loaded");
browser.browserAction.onClicked.addListener(() => {
browser.sidebarAction.toggle();
});
browser.runtime.onMessage.addListener((request, sender, sendResponse) => {
if (request.action === "summarize") {
summarizeContent(request.content, request.systemPrompt)
.then((summary) => {
sendResponse({ summary });
})
.catch((error) => {
console.error("Error in summarizeContent:", error);
sendResponse({ error: error.toString(), details: error.details });
});
return true; // Indicates that we will send a response asynchronously
}
});
async function summarizeContent(content, systemPrompt) {
const settings = await browser.storage.local.get([
"ollamaEndpoint",
"ollamaModel",
]);
const endpoint = `${
settings.ollamaEndpoint || "http://localhost:11434"
}/api/generate`;
const model = settings.ollamaModel || "llama3.1:8b";
try {
console.log(`Using system prompt: ${systemPrompt}`);
const response = await fetch(endpoint, {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({
prompt: `${systemPrompt}\n\nFollow the above instructions and summarize the following text:\n\n${content}`,
model: model,
stream: false,
}),
});
if (!response.ok) {
const errorText = await response.text();
throw new Error(
`HTTP error! status: ${response.status}, message: ${errorText}`
);
}
const data = await response.json();
return data.response;
} catch (error) {
console.error("Error details:", error);
error.details = {
endpoint: endpoint,
model: model,
message: error.message,
};
throw error;
}
}