chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-02 20:40:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle 'n' parameter for candidate count in requests

Browse Source 
Added handling for the 'n' parameter to set candidate count in generationConfig.
This commit is contained in:
extremk
2026-01-10 18:48:33 +08:00
committed by GitHub
parent 6448d0ee7c
commit 14c74e5e84
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/translator/gemini/openai/chat-completions/gemini_openai_request.go
View File

@@ -99,6 +99,13 @@ func ConvertOpenAIRequestToGemini(modelName string, inputRawJSON []byte, _ bool)
out, _ = sjson.SetBytes(out, "generationConfig.topK", tkr.Num) out, _ = sjson.SetBytes(out, "generationConfig.topK", tkr.Num)
} }
// Candidate count (OpenAI 'n' parameter)
if n := gjson.GetBytes(rawJSON, "n"); n.Exists() && n.Type == gjson.Number {
if val := n.Int(); val > 1 {
out, _ = sjson.SetBytes(out, "generationConfig.candidateCount", val)
}
}
// Map OpenAI modalities -> Gemini generationConfig.responseModalities // Map OpenAI modalities -> Gemini generationConfig.responseModalities
// e.g. "modalities": ["image", "text"] -> ["IMAGE", "TEXT"] // e.g. "modalities": ["image", "text"] -> ["IMAGE", "TEXT"]
if mods := gjson.GetBytes(rawJSON, "modalities"); mods.Exists() && mods.IsArray() { if mods := gjson.GetBytes(rawJSON, "modalities"); mods.Exists() && mods.IsArray() {