mirror of
https://github.com/HeyPuter/puter.git
synced 2026-05-06 09:30:49 +00:00
add temp and max_tokens to Gemini Service
This commit is contained in:
committed by
Eric Dubé
parent
1e6e70b553
commit
2a288e6cf0
@@ -31,13 +31,17 @@ class GeminiService extends BaseService {
|
||||
return model_names;
|
||||
},
|
||||
|
||||
async complete ({ messages, stream, model, tools }) {
|
||||
async complete ({ messages, stream, model, tools, max_tokens, temperature }) {
|
||||
tools = FunctionCalling.make_gemini_tools(tools);
|
||||
|
||||
const genAI = new GoogleGenerativeAI(this.config.apiKey);
|
||||
const genModel = genAI.getGenerativeModel({
|
||||
model: model ?? 'gemini-2.0-flash',
|
||||
tools,
|
||||
generationConfig: {
|
||||
temperature: temperature, // Set temperature (0.0 to 1.0). Defaults to 0.7
|
||||
maxOutputTokens: max_tokens, // Note: it's maxOutputTokens, not max_tokens
|
||||
}
|
||||
});
|
||||
|
||||
messages = await GeminiSquareHole.process_input_messages(messages);
|
||||
|
||||
Reference in New Issue
Block a user