mirror of
https://github.com/router-for-me/CLIProxyAPIPlus.git
synced 2026-03-07 22:33:30 +00:00
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package claude
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func TestConvertClaudeRequestToCLI_ToolChoice_SpecificTool(t *testing.T) {
|
|
inputJSON := []byte(`{
|
|
"model": "gemini-3-flash-preview",
|
|
"messages": [
|
|
{
|
|
"role": "user",
|
|
"content": [
|
|
{"type": "text", "text": "hi"}
|
|
]
|
|
}
|
|
],
|
|
"tools": [
|
|
{
|
|
"name": "json",
|
|
"description": "A JSON tool",
|
|
"input_schema": {
|
|
"type": "object",
|
|
"properties": {}
|
|
}
|
|
}
|
|
],
|
|
"tool_choice": {"type": "tool", "name": "json"}
|
|
}`)
|
|
|
|
output := ConvertClaudeRequestToCLI("gemini-3-flash-preview", inputJSON, false)
|
|
|
|
if got := gjson.GetBytes(output, "request.toolConfig.functionCallingConfig.mode").String(); got != "ANY" {
|
|
t.Fatalf("Expected request.toolConfig.functionCallingConfig.mode 'ANY', got '%s'", got)
|
|
}
|
|
allowed := gjson.GetBytes(output, "request.toolConfig.functionCallingConfig.allowedFunctionNames").Array()
|
|
if len(allowed) != 1 || allowed[0].String() != "json" {
|
|
t.Fatalf("Expected allowedFunctionNames ['json'], got %s", gjson.GetBytes(output, "request.toolConfig.functionCallingConfig.allowedFunctionNames").Raw)
|
|
}
|
|
}
|