mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-19 04:40:52 +08:00
a5a25dec57
- Replaced all instances of `bytes.Clone` with direct references to enhance efficiency. - Simplified payload handling across executors and translators by eliminating unnecessary data duplication.
31 lines
1.4 KiB
Go
31 lines
1.4 KiB
Go
// Package openai provides request translation functionality for OpenAI to Gemini CLI API compatibility.
|
|
// It converts OpenAI Chat Completions requests into Gemini CLI compatible JSON using gjson/sjson only.
|
|
package chat_completions
|
|
|
|
import (
|
|
"github.com/tidwall/sjson"
|
|
)
|
|
|
|
// ConvertOpenAIRequestToOpenAI converts an OpenAI Chat Completions request (raw JSON)
|
|
// into a complete Gemini CLI request JSON. All JSON construction uses sjson and lookups use gjson.
|
|
//
|
|
// Parameters:
|
|
// - modelName: The name of the model to use for the request
|
|
// - rawJSON: The raw JSON request data from the OpenAI API
|
|
// - stream: A boolean indicating if the request is for a streaming response (unused in current implementation)
|
|
//
|
|
// Returns:
|
|
// - []byte: The transformed request data in Gemini CLI API format
|
|
func ConvertOpenAIRequestToOpenAI(modelName string, inputRawJSON []byte, _ bool) []byte {
|
|
// Update the "model" field in the JSON payload with the provided modelName
|
|
// The sjson.SetBytes function returns a new byte slice with the updated JSON.
|
|
updatedJSON, err := sjson.SetBytes(inputRawJSON, "model", modelName)
|
|
if err != nil {
|
|
// If there's an error, return the original JSON or handle the error appropriately.
|
|
// For now, we'll return the original, but in a real scenario, logging or a more robust error
|
|
// handling mechanism would be needed.
|
|
return inputRawJSON
|
|
}
|
|
return updatedJSON
|
|
}
|