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

temperature added to the google gemini class. see #572

Browse source
This commit is contained in:
mic 2025-12-26 23:16:56 +01:00
parent de7a2ca831
commit 3c089fc04e
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
js/api/google_gemini.js
View file

@ -25,6 +25,7 @@ export class GoogleGemini {
system_instruction = ''; system_instruction = '';
stream = false; stream = false;
thinking_budget = ''; // Model default thinking_budget = ''; // Model default
temperature = ''; // no temperature defined
constructor({ constructor({
apiKey = '', apiKey = '',
@ -32,12 +33,14 @@ export class GoogleGemini {
system_instruction = '', system_instruction = '',
stream = false, stream = false,
thinking_budget = '', thinking_budget = '',
temperature = '',
} = {}) { } = {}) {
this.apiKey = apiKey; this.apiKey = apiKey;
this.model = model; this.model = model;
this.system_instruction = system_instruction; this.system_instruction = system_instruction;
this.stream = stream; this.stream = stream;
this.thinking_budget = String(thinking_budget ?? '').trim(); this.thinking_budget = String(thinking_budget ?? '').trim();
this.temperature = String(temperature ?? '').trim();
/* Info from: https://ai.google.dev/gemini-api/docs/thinking?#set-budget /* Info from: https://ai.google.dev/gemini-api/docs/thinking?#set-budget
# Turn on thinking with a specific token limit: "thinking_budget": 1024 # Turn on thinking with a specific token limit: "thinking_budget": 1024
# Thinking off: "thinking_budget": 0 # Thinking off: "thinking_budget": 0
@ -87,7 +90,8 @@ export class GoogleGemini {
try { try {
let google_gemini_body = { let google_gemini_body = {
contents:messages contents: messages,
generationConfig: {},
}; };
// console.log("[ThunderAI] Google Gemini API system_instruction: " + JSON.stringify(this.system_instruction)); // console.log("[ThunderAI] Google Gemini API system_instruction: " + JSON.stringify(this.system_instruction));
@ -101,13 +105,15 @@ export class GoogleGemini {
} }
if(this.thinking_budget !== '') { if(this.thinking_budget !== '') {
google_gemini_body.generationConfig = { google_gemini_body.generationConfig.thinkingConfig = {
thinkingConfig: {
thinking_budget: this.thinking_budget, thinking_budget: this.thinking_budget,
}
}; };
} }
if(this.temperature !== ''){
google_gemini_body.generationConfig.temperature = this.temperature;
}
// console.log("[ThunderAI] Google Gemini API request: " + JSON.stringify(google_gemini_body)); // console.log("[ThunderAI] Google Gemini API request: " + JSON.stringify(google_gemini_body));
const response = await fetch("https://generativelanguage.googleapis.com/v1beta/models/" + this.model + ":" + (this.stream ? 'streamGenerateContent?alt=sse&' : 'generateContent?') + "key=" + this.apiKey, { const response = await fetch("https://generativelanguage.googleapis.com/v1beta/models/" + this.model + ":" + (this.stream ? 'streamGenerateContent?alt=sse&' : 'generateContent?') + "key=" + this.apiKey, {