95 lines
3.5 KiB
JavaScript
95 lines
3.5 KiB
JavaScript
// Changing defaultConfigurations requires changing popup/popup.js
|
|
const defaultConfigurations = {
|
|
openai: {
|
|
url: 'https://api.openai.com/v1/chat/completions',
|
|
base: {
|
|
n: 1,
|
|
temperature: 0.5,
|
|
model: 'gpt-3.5-turbo'
|
|
},
|
|
Complete: {
|
|
max_tokens: 512,
|
|
messages: [{
|
|
role: 'system',
|
|
content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Continues the given text. No need to rewrite the given text.'
|
|
}]
|
|
},
|
|
Improve: {
|
|
messages: [{
|
|
role: 'system',
|
|
content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Style and improves the given text.'
|
|
}]
|
|
},
|
|
Ask: {
|
|
messages: [{
|
|
role: 'system',
|
|
content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Address the given questions/request without introduction/explanations.'
|
|
}]
|
|
}
|
|
}
|
|
}
|
|
|
|
const settings = [
|
|
{ key: 'Complete', shortcut: 'Alt+C', status: 'enabled', type: 'Command' },
|
|
{ key: 'Improve', shortcut: 'Alt+I', status: 'enabled', type: 'Command' },
|
|
{ key: 'Ask', shortcut: 'Alt+A', status: 'enabled', type: 'Command' },
|
|
{ key: 'RequestConfiguration', value: defaultConfigurations, type: 'Configuration' }
|
|
]
|
|
|
|
async function sendMessage(message) {
|
|
const [tab] = await chrome.tabs.query({ active: true, lastFocusedWindow: true })
|
|
if (tab == null || tab.url?.startsWith('chrome://')) return undefined
|
|
chrome.tabs.sendMessage(tab.id, message)
|
|
// do something with response here, not outside the function
|
|
// console.log(response)
|
|
}
|
|
|
|
function addListener(commandName) {
|
|
chrome.commands.onCommand.addListener((command) => {
|
|
if (command !== commandName) return
|
|
console.log(`Command ${command} triggered`)
|
|
sendMessage({ command: command })
|
|
})
|
|
}
|
|
|
|
// Only use this function during the initial installation phase. After
|
|
// installation the user may have intentionally unassigned commands.
|
|
// Example for install commands: [{"description":"","name":"_execute_action","shortcut":""},{"description":"Use the selected text to ask GPT. It adds to the beginning of the selected text: 'In Latex, '","name":"Ask","shortcut":""},{"description":"Complete selected text","name":"Complete","shortcut":""},{"description":"Improve selected text","name":"Improve","shortcut":""}]
|
|
async function checkCommandShortcuts() {
|
|
chrome.commands.getAll((commands) => {
|
|
for (let { name, shortcut } of commands) {
|
|
let command =
|
|
settings.filter(({ type, key }) => 'Command' === type && name === key)
|
|
if (command.length > 0) {
|
|
command = command[0]
|
|
if (shortcut === '') {
|
|
command.status = 'error'
|
|
}
|
|
chrome.storage.local.set({ [command.key]: command })
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
chrome.runtime.onInstalled.addListener((reason) => {
|
|
if (reason.reason === chrome.runtime.OnInstalledReason.INSTALL) {
|
|
checkCommandShortcuts()
|
|
chrome.storage.local.set({ RequestConfiguration: defaultConfigurations })
|
|
} else if (reason.reason === chrome.runtime.OnInstalledReason.UPDATE) {
|
|
let newVersion = chrome.runtime.getManifest().version
|
|
let oldVersion = reason.previousVersion
|
|
let oldVersionArray = oldVersion.split('.')
|
|
if (parseInt(oldVersionArray[0]) === 1 && parseInt(oldVersionArray[1]) < 4) {
|
|
chrome.storage.local.set({ RequestConfiguration: defaultConfigurations })
|
|
}
|
|
}
|
|
})
|
|
|
|
async function setup() {
|
|
addListener('Improve')
|
|
addListener('Complete')
|
|
addListener('Ask')
|
|
}
|
|
|
|
setup()
|