feat: support custom gemini pro params

This commit is contained in:
Fred Liang
2023-12-24 17:24:04 +08:00
parent 75057f9a91
commit 7c3dfb7bae
2 changed files with 83 additions and 69 deletions

View File

@@ -34,6 +34,15 @@ export class GeminiProApi implements LLMApi {
};
const requestPayload = {
contents: messages,
generationConfig: {
// stopSequences: [
// "Title"
// ],
temperature: modelConfig.temperature,
maxOutputTokens: modelConfig.max_tokens,
topP: modelConfig.top_p,
// "topK": modelConfig.top_k,
},
// stream: options.config.stream,
// model: modelConfig.model,
// temperature: modelConfig.temperature,