Skip to content

Commit 2a288e6

Browse files
brauliopfKernelDeimos
authored andcommitted
add temp and max_tokens to Gemini Service
1 parent 1e6e70b commit 2a288e6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/backend/src/modules/puterai/GeminiService.js

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,17 @@ class GeminiService extends BaseService {
3131
return model_names;
3232
},
3333

34-
async complete ({ messages, stream, model, tools }) {
34+
async complete ({ messages, stream, model, tools, max_tokens, temperature }) {
3535
tools = FunctionCalling.make_gemini_tools(tools);
3636

3737
const genAI = new GoogleGenerativeAI(this.config.apiKey);
3838
const genModel = genAI.getGenerativeModel({
3939
model: model ?? 'gemini-2.0-flash',
4040
tools,
41+
generationConfig: {
42+
temperature: temperature, // Set temperature (0.0 to 1.0). Defaults to 0.7
43+
maxOutputTokens: max_tokens, // Note: it's maxOutputTokens, not max_tokens
44+
}
4145
});
4246

4347
messages = await GeminiSquareHole.process_input_messages(messages);

0 commit comments

Comments
 (0)