Luis Pater
e50cabac4b
- Updated all references from v6 to v7 for `github.com/router-for-me/CLIProxyAPI`. - Ensured consistency in imports within core libraries, tests, and integration tests. - Added missing tests for new features in Redis Protocol integration.
36 lines
1.3 KiB
Go
36 lines
1.3 KiB
Go
package responses
|
|
|
|
import (
|
|
"context"
|
|
|
|
. "github.com/router-for-me/CLIProxyAPI/v7/internal/translator/gemini/openai/responses"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func ConvertGeminiCLIResponseToOpenAIResponses(ctx context.Context, modelName string, originalRequestRawJSON, requestRawJSON, rawJSON []byte, param *any) [][]byte {
|
|
responseResult := gjson.GetBytes(rawJSON, "response")
|
|
if responseResult.Exists() {
|
|
rawJSON = []byte(responseResult.Raw)
|
|
}
|
|
return ConvertGeminiResponseToOpenAIResponses(ctx, modelName, originalRequestRawJSON, requestRawJSON, rawJSON, param)
|
|
}
|
|
|
|
func ConvertGeminiCLIResponseToOpenAIResponsesNonStream(ctx context.Context, modelName string, originalRequestRawJSON, requestRawJSON, rawJSON []byte, param *any) []byte {
|
|
responseResult := gjson.GetBytes(rawJSON, "response")
|
|
if responseResult.Exists() {
|
|
rawJSON = []byte(responseResult.Raw)
|
|
}
|
|
|
|
requestResult := gjson.GetBytes(originalRequestRawJSON, "request")
|
|
if responseResult.Exists() {
|
|
originalRequestRawJSON = []byte(requestResult.Raw)
|
|
}
|
|
|
|
requestResult = gjson.GetBytes(requestRawJSON, "request")
|
|
if responseResult.Exists() {
|
|
requestRawJSON = []byte(requestResult.Raw)
|
|
}
|
|
|
|
return ConvertGeminiResponseToOpenAIResponsesNonStream(ctx, modelName, originalRequestRawJSON, requestRawJSON, rawJSON, param)
|
|
}
|