buzzcode2007/ShopAI-Extension
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add the AI prompt and parsing improvements

Browse source 
Merge branch "development-ai" into development
This commit is contained in:
buzz-lightsnack-2007 2024-04-15 13:24:56 +08:00
commit 4d20cdec8d
3 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

1
scripts/AI/gemini.js
View file

@ -133,7 +133,6 @@ export default class gemini {
if (Object.keys(RESPONSE).includes(`error`)) {
throw new Error(RESPONSE[`error`]);
} else {
// this[`history`] = REQUEST[`contents`];
this.response = RESPONSE;
return RESPONSE;
}

18
scripts/product.js
View file

@ -93,19 +93,21 @@ export default class product {
if ((this.analysis && this.analysis != null && this.analysis != undefined) ? !((typeof this.analysis).contains(`obj`) && !Array.isArray(this.analysis)) : true) {
// Analyze the data.
const gemini = (await import(chrome.runtime.getURL("scripts/AI/gemini.js"))).default;
let analyzer = new gemini (await secretariat.read([`settings`,`analysis`,`api`,`key`]), `gemini-pro`);
let analyzer = new gemini (await secretariat.read([`settings`,`analysis`,`api`,`key`]), `gemini-pro-vision`);
// Analyze the data.
let PROMPT = [];
// Add the "system" prompt.
PROMPT.push({"text": texts.localized(`AI_message_prompt`)});
// Add the prompt.
PROMPT.push({"text": (texts.localized(`AI_message_prompt`)).concat(JSON.stringify(this.details))});
// This is the user prompt.
PROMPT.push({"text": JSON.stringify(this.details)});
// Return the analysis
this.analysis = JSON.stringify(await analyzer.generate(PROMPT));
// Run the analysis.
await analyzer.generate(PROMPT);
if (analyzer.candidate) {
// Remove all markdown formatting.
this.analysis = JSON.parse(analyzer.candidate.replace(/(```json|```|`)/g, ''));
};
};
return(this.analysis);