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

Include AI system prompt and analyzer function

Browse source
This commit is contained in:
buzz-lightsnack-2007 2024-04-11 18:39:52 +08:00
parent bad5ea8c1a
commit 586578fe45
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
scripts/product.js
View file

@ -90,17 +90,23 @@ export default class product {
// Stop when the data is already analyzed. // Stop when the data is already analyzed.
if (this[`analysis`]) {return(this.analysis)} if (this[`analysis`]) {return(this.analysis)}
else if (this.status ? (!this.status.update) : false) {this.analysis = await secretariat.read([`sites`, this.URL, `analysis`]);} else if (this.status ? (!this.status.update) : false) {this.analysis = await secretariat.read([`sites`, this.URL, `analysis`]);}
else { if ((this.analysis && this.analysis != null && this.analysis != undefined) ? !((typeof this.analysis).contains(`obj`) && !Array.isArray(this.analysis)) : true) {
// Analyze the data. // Analyze the data.
const gemini = (await import(chrome.runtime.getURL("scripts/AI/gemini.js"))).default; 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`);
// Analyze the data. // Analyze the data.
await analyzer.generate(this.details); let PROMPT = [];
// Add the "system" prompt.
PROMPT.push({"text": texts.localized(`AI_message_prompt`)});
// This is the user prompt.
PROMPT.push({"text": JSON.stringify(this.details)});
// Return the analysis // Return the analysis
this.analysis = analyzer.response; this.analysis = JSON.stringify(await analyzer.generate(PROMPT));
} };
return(this.analysis); return(this.analysis);
}; };