vertex_replay_test.go RawHistoryFolder

1package provider
2
3import "testing"
4
5func TestVertexReplaysOwnReasoning(t *testing.T) {
6	t.Setenv("GOOGLE_CLOUD_API_KEY", "synthetic-key")
7	t.Setenv("GOOGLE_CLOUD_PROJECT", "test-project")
8	client := NewVertex("", "").(*renamedClient)
9	inner := client.inner.(*geminiClient)
10	for _, origin := range []string{"google-vertex", "google"} {
11		t.Run(origin, func(t *testing.T) {
12			wire, _, err := inner.buildRequest(Request{Model: "gemini-2.5-pro", Messages: []Message{WithMessageOrigin(Message{Role: RoleAssistant, Content: []Content{ReasoningBlock{Summary: "summary", Encrypted: "opaque"}, TextBlock{Text: "answer", ThoughtSignature: "signature"}}}, origin, "gemini-2.5-pro")}})
13			if err != nil {
14				t.Fatal(err)
15			}
16			parts := wire.Contents[0].Parts
17			if origin == "google-vertex" {
18				if len(parts) != 2 || parts[0].ThoughtSignature != "opaque" || parts[1].ThoughtSignature != "signature" {
19					t.Fatalf("own metadata lost: %+v", parts)
20				}
21			} else if len(parts) != 1 || parts[0].ThoughtSignature != "" {
22				t.Fatalf("foreign metadata replayed: %+v", parts)
23			}
24		})
25	}
26}