jason/ThunderAI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

new internal messaging method in the ollama api call

Browse source
This commit is contained in:
Mic 2024-08-23 13:52:48 +02:00
parent 25b84ba8b2
commit 9698eb1d01
2 changed files with 47 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api_webchat/controller.js
View file

@ -23,6 +23,7 @@
// Get the LLM to be used // Get the LLM to be used
const urlParams = new URLSearchParams(window.location.search); const urlParams = new URLSearchParams(window.location.search);
const llm = urlParams.get('llm'); const llm = urlParams.get('llm');
const call_id = urlParams.get('call_id');
//console.log(">>>>>>>>>>> llm: " + llm); //console.log(">>>>>>>>>>> llm: " + llm);
@ -35,6 +36,7 @@ switch (llm) {
worker = new Worker('model-worker-openai.js', { type: 'module' }); worker = new Worker('model-worker-openai.js', { type: 'module' });
break; break;
case "ollama_api": { case "ollama_api": {
browser.runtime.sendMessage({command: "ollama_api_ready_"+call_id});
worker = new Worker('model-worker-ollama.js', { type: 'module' }); worker = new Worker('model-worker-ollama.js', { type: 'module' });
break; break;
} }

41
mzta-background.js
View file

@ -20,7 +20,7 @@ import { mzta_script } from './js/mzta-chatgpt.js';
import { prefs_default } from './options/mzta-options-default.js'; import { prefs_default } from './options/mzta-options-default.js';
import { mzta_Menus } from './js/mzta-menus.js'; import { mzta_Menus } from './js/mzta-menus.js';
import { taLogger } from './js/mzta-logger.js'; import { taLogger } from './js/mzta-logger.js';
import { getCurrentIdentity, getOriginalBody, replaceBody, setBody, i18nConditionalGet, isThunderbird128OrGreater } from './js/mzta-utils.js'; import { getCurrentIdentity, getOriginalBody, replaceBody, setBody, i18nConditionalGet, generateCallID } from './js/mzta-utils.js';
var createdWindowID = null; var createdWindowID = null;
var original_html = ''; var original_html = '';
@ -278,36 +278,25 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
case 'ollama_api': case 'ollama_api':
// We are using the Ollama API // We are using the Ollama API
let newWindow3 = await browser.windows.create({
url: browser.runtime.getURL('api_webchat/index.html?llm='+prefs.connection_type), taLog.log("Ollama API window opening...");
let rand_call_id = '_ollama_' + generateCallID();
await browser.windows.create({
url: browser.runtime.getURL('api_webchat/index.html?llm='+prefs.connection_type+'&call_id='+rand_call_id),
type: "popup", type: "popup",
width: prefs.chatgpt_win_width, width: prefs.chatgpt_win_width,
height: prefs.chatgpt_win_height height: prefs.chatgpt_win_height
}); });
createdWindowID = newWindow3.id; browser.runtime.onMessage.addListener(async (message, sender, sendResponse) => {
let createdTab3 = newWindow3.tabs[0];
// if(await isThunderbird128OrGreater()){ if (message.command === "ollama_api_ready_"+rand_call_id) {
// Wait for tab loaded. taLog.log("Ollama API window ready.");
// await new Promise(resolve => {
// const tabIsLoaded3 = tab => { let newWindow = await browser.windows.get(message.window_id);
// return tab.status == "complete" && tab.url != "about:blank"; let createdTab3 = newWindow.tabs[0];
// };
// const listener3 = (tabId, changeInfo, updatedTab) => {
// if (tabIsLoaded3(updatedTab)) {
// browser.tabs.onUpdated.removeListener(listener3);
// resolve();
// }
// }
// // Early exit if loaded already
// if (tabIsLoaded3(createdTab3)) {
// resolve();
// } else {
// browser.tabs.onUpdated.addListener(listener3);
// }
// });
//}
let mailMessage3 = await browser.messageDisplay.getDisplayedMessage(curr_tabId); let mailMessage3 = await browser.messageDisplay.getDisplayedMessage(curr_tabId);
let mailMessageId3 = -1; let mailMessageId3 = -1;
@ -337,6 +326,8 @@ async function openChatGPT(promptText, action, curr_tabId, prompt_name = '', do_
} }
await new Promise(resolve => setTimeout(resolve, 500)); await new Promise(resolve => setTimeout(resolve, 500));
} }
}
});
break; // ollama_api break; // ollama_api
} }
} }