feat: align provider runtimes with cliproxyapi

2026-05-22 07:27:35 +08:00 · 2026-03-12 17:57:00 +08:00
parent 1775b0ec86
commit 92fba9eb74
13 changed files with 5101 additions and 45 deletions
--- a/pkg/providers/antigravity_provider_test.go
+++ b/pkg/providers/antigravity_provider_test.go
@@ -0,0 +1,101 @@
+package providers
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestAntigravityBuildRequestBody(t *testing.T) {
+	p := NewAntigravityProvider("openai", "", "", "gemini-2.5-pro", false, "oauth", 0, nil)
+	body := p.buildRequestBody([]Message{
+		{Role: "system", Content: "You are helpful."},
+		{Role: "user", Content: "hello"},
+		{
+			Role:    "assistant",
+			Content: "calling tool",
+			ToolCalls: []ToolCall{{
+				ID:   "call_1",
+				Name: "lookup",
+				Function: &FunctionCall{
+					Name:      "lookup",
+					Arguments: `{"q":"weather"}`,
+				},
+			}},
+		},
+		{Role: "tool", ToolCallID: "call_1", Content: `{"ok":true}`},
+	}, []ToolDefinition{{
+		Type: "function",
+		Function: ToolFunctionDefinition{
+			Name:        "lookup",
+			Description: "Lookup data",
+			Parameters: map[string]interface{}{
+				"type": "object",
+			},
+		},
+	}}, "gemini-2.5-pro", map[string]interface{}{
+		"max_tokens":  256,
+		"temperature": 0.2,
+	}, &oauthSession{ProjectID: "demo-project"}, false)
+
+	if got := body["project"]; got != "demo-project" {
+		t.Fatalf("expected project id to be preserved, got %#v", got)
+	}
+	request := mapFromAny(body["request"])
+	if system := asString(mapFromAny(request["systemInstruction"])["parts"].([]map[string]any)[0]["text"]); system != "You are helpful." {
+		t.Fatalf("expected system instruction, got %q", system)
+	}
+	if got := len(request["contents"].([]map[string]any)); got != 3 {
+		t.Fatalf("expected 3 content entries, got %d", got)
+	}
+	gen := mapFromAny(request["generationConfig"])
+	if got := intValue(gen["maxOutputTokens"]); got != 256 {
+		t.Fatalf("expected maxOutputTokens, got %#v", gen["maxOutputTokens"])
+	}
+	if got := gen["temperature"]; got != 0.2 {
+		t.Fatalf("expected temperature, got %#v", got)
+	}
+}
+
+func TestParseAntigravityResponse(t *testing.T) {
+	raw := []byte(`{
+		"response": {
+			"candidates": [{
+				"finishReason": "STOP",
+				"content": {
+					"parts": [
+						{"text": "hello"},
+						{"functionCall": {"id": "call_1", "name": "lookup", "args": {"q":"weather"}}}
+					]
+				}
+			}],
+			"usageMetadata": {
+				"promptTokenCount": 11,
+				"candidatesTokenCount": 7,
+				"totalTokenCount": 18
+			}
+		}
+	}`)
+	resp, err := parseAntigravityResponse(raw)
+	if err != nil {
+		t.Fatalf("parse response: %v", err)
+	}
+	if resp.Content != "hello" {
+		t.Fatalf("expected content, got %q", resp.Content)
+	}
+	if resp.FinishReason != "STOP" {
+		t.Fatalf("expected finish reason passthrough, got %q", resp.FinishReason)
+	}
+	if len(resp.ToolCalls) != 1 || resp.ToolCalls[0].Name != "lookup" {
+		t.Fatalf("expected tool call, got %#v", resp.ToolCalls)
+	}
+	if resp.Usage == nil || resp.Usage.TotalTokens != 18 {
+		t.Fatalf("expected usage, got %#v", resp.Usage)
+	}
+	var args map[string]any
+	if err := json.Unmarshal([]byte(resp.ToolCalls[0].Function.Arguments), &args); err != nil {
+		t.Fatalf("decode args: %v", err)
+	}
+	if got := asString(args["q"]); got != "weather" {
+		t.Fatalf("expected tool args, got %#v", args)
+	}
+}