mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 04:50:52 +08:00
50 lines
2.1 KiB
Go
50 lines
2.1 KiB
Go
// Package gemini_cli provides response translation functionality for Gemini API to Gemini CLI API.
|
|
// This package handles the conversion of Gemini API responses into Gemini CLI-compatible
|
|
// JSON format, transforming streaming events and non-streaming responses into the format
|
|
// expected by Gemini CLI API clients.
|
|
package geminiCLI
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"github.com/tidwall/sjson"
|
|
)
|
|
|
|
// ConvertGeminiResponseToGeminiCLI converts Gemini streaming response format to Gemini CLI single-line JSON format.
|
|
// This function processes various Gemini event types and transforms them into Gemini CLI-compatible JSON responses.
|
|
// It handles thinking content, regular text content, and function calls, outputting single-line JSON
|
|
// that matches the Gemini CLI API response format.
|
|
//
|
|
// Parameters:
|
|
// - ctx: The context for the request.
|
|
// - modelName: The name of the model.
|
|
// - rawJSON: The raw JSON response from the Gemini API.
|
|
// - param: A pointer to a parameter object for the conversion (unused).
|
|
//
|
|
// Returns:
|
|
// - []string: A slice of strings, each containing a Gemini CLI-compatible JSON response.
|
|
func ConvertGeminiResponseToGeminiCLI(_ context.Context, _ string, originalRequestRawJSON, requestRawJSON, rawJSON []byte, _ *any) []string {
|
|
if bytes.Equal(rawJSON, []byte("[DONE]")) {
|
|
return []string{}
|
|
}
|
|
json := `{"response": {}}`
|
|
rawJSON, _ = sjson.SetRawBytes([]byte(json), "response", rawJSON)
|
|
return []string{string(rawJSON)}
|
|
}
|
|
|
|
// ConvertGeminiResponseToGeminiCLINonStream converts a non-streaming Gemini response to a non-streaming Gemini CLI response.
|
|
//
|
|
// Parameters:
|
|
// - ctx: The context for the request.
|
|
// - modelName: The name of the model.
|
|
// - rawJSON: The raw JSON response from the Gemini API.
|
|
// - param: A pointer to a parameter object for the conversion (unused).
|
|
//
|
|
// Returns:
|
|
// - string: A Gemini CLI-compatible JSON response.
|
|
func ConvertGeminiResponseToGeminiCLINonStream(_ context.Context, _ string, originalRequestRawJSON, requestRawJSON, rawJSON []byte, _ *any) string {
|
|
json := `{"response": {}}`
|
|
rawJSON, _ = sjson.SetRawBytes([]byte(json), "response", rawJSON)
|
|
return string(rawJSON)
|
|
}
|