centersl/chrome-extensions-samples
1
0
Fork 0
mirror of https://github.com/GoogleChrome/chrome-extensions-samples.git synced 2026-03-27 13:29:34 +07:00
Code Issues Packages Projects Releases Wiki Activity

Migrate to new prompt API (#1264)

Browse Source 
...also add a check whether the browser supports the prompt API.
This commit is contained in:
Sebastian Benz
2024-08-28 12:14:45 +02:00
committed by GitHub
parent e8d05b6830
commit cbfd8823fa
1 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
functional-samples/ai.gemini-on-device/sidepanel/index.js
View File

@@ -14,21 +14,7 @@ let session;
async function runPrompt(prompt, params) {
try {
if (!session) {
// Start by checking if it's possible to create a session based on the availability of the model, and the characteristics of the device.
const canCreate = await self.ai.canCreateTextSession();
// canCreate will be one of the following:
// * "readily": the model is available on-device and so creating will happen quickly
// * "after-download": the model is not available on-device, but the device is capable,
// so creating the session will start the download process (which can take a while).
// * "no": the model is not available for this device.
if (canCreate === 'no') {
console.warn('Built-in prompt API not available.');
throw new Error(
'Built-in prompt API not available. Join the preview program to learn how to enable it.'
);
}
console.log('Creating new text session');
session = await self.ai.createTextSession(params);
session = await window.ai.assistant.create(params);
}
return session.prompt(prompt);
} catch (e) {
@@ -49,7 +35,11 @@ async function reset() {
}
async function initDefaults() {
const defaults = await window.ai.defaultTextSessionOptions();
if (!window.ai) {
showResponse('Error: window.ai not supported in this browser');
return;
}
const defaults = await window.ai.assistant.capabilities();
console.log('Model default:', defaults);
sliderTemperature.value = defaults.temperature;
sliderTopK.value = defaults.topK;