mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 21:10:51 +08:00
9fce13fe03
- Updated all `github.com/luispater/CLIProxyAPI/internal/...` imports to point to `github.com/luispater/CLIProxyAPI/v5/internal/...`. - Adjusted `go.mod` to specify `module github.com/luispater/CLIProxyAPI/v5`.
44 lines
2.1 KiB
Go
44 lines
2.1 KiB
Go
// Package geminiCLI provides request translation functionality for Gemini CLI to Codex API compatibility.
|
|
// It handles parsing and transforming Gemini CLI API requests into Codex API format,
|
|
// extracting model information, system instructions, message contents, and tool declarations.
|
|
// The package performs JSON data transformation to ensure compatibility
|
|
// between Gemini CLI API format and Codex API's expected format.
|
|
package geminiCLI
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
. "github.com/luispater/CLIProxyAPI/v5/internal/translator/codex/gemini"
|
|
"github.com/tidwall/gjson"
|
|
"github.com/tidwall/sjson"
|
|
)
|
|
|
|
// ConvertGeminiCLIRequestToCodex parses and transforms a Gemini CLI API request into Codex API format.
|
|
// It extracts the model name, system instruction, message contents, and tool declarations
|
|
// from the raw JSON request and returns them in the format expected by the Codex API.
|
|
// The function performs the following transformations:
|
|
// 1. Extracts the inner request object and promotes it to the top level
|
|
// 2. Restores the model information at the top level
|
|
// 3. Converts systemInstruction field to system_instruction for Codex compatibility
|
|
// 4. Delegates to the Gemini-to-Codex conversion function for further processing
|
|
//
|
|
// Parameters:
|
|
// - modelName: The name of the model to use for the request
|
|
// - rawJSON: The raw JSON request data from the Gemini CLI API
|
|
// - stream: A boolean indicating if the request is for a streaming response
|
|
//
|
|
// Returns:
|
|
// - []byte: The transformed request data in Codex API format
|
|
func ConvertGeminiCLIRequestToCodex(modelName string, inputRawJSON []byte, stream bool) []byte {
|
|
rawJSON := bytes.Clone(inputRawJSON)
|
|
|
|
rawJSON = []byte(gjson.GetBytes(rawJSON, "request").Raw)
|
|
rawJSON, _ = sjson.SetBytes(rawJSON, "model", modelName)
|
|
if gjson.GetBytes(rawJSON, "systemInstruction").Exists() {
|
|
rawJSON, _ = sjson.SetRawBytes(rawJSON, "system_instruction", []byte(gjson.GetBytes(rawJSON, "systemInstruction").Raw))
|
|
rawJSON, _ = sjson.DeleteBytes(rawJSON, "systemInstruction")
|
|
}
|
|
|
|
return ConvertGeminiRequestToCodex(modelName, rawJSON, stream)
|
|
}
|