mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-02 20:40:52 +08:00
e5ed2cba4a
Implements functionality to parse model names with provider information in the format "provider://model" This allows dynamic provider selection rather than relying only on predefined mappings. The change affects all execution methods to properly handle these dynamic model specifications while maintaining compatibility with the existing approach for standard model names.
32 lines
1.4 KiB
Go
32 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 (
|
|
"bytes"
|
|
"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 bytes.Clone(inputRawJSON)
|
|
}
|
|
return updatedJSON
|
|
}
|