azure_openai_test.go RawHistoryFolder

1package provider
2
3import (
4	"context"
5	"encoding/json"
6	"io"
7	"net/http"
8	"strings"
9	"testing"
10)
11
12func TestAzureOpenAIResponsesRequestShape(t *testing.T) {
13	t.Setenv("AZURE_OPENAI_API_VERSION", "v1")
14	t.Setenv("AZURE_OPENAI_DEPLOYMENT_NAME_MAP", "gpt-5.6-sol=sol-preview")
15
16	named := NewAzureOpenAIResponses("azure-key", "https://example.openai.azure.com").(*renamedClient)
17	c := named.inner.(*codexClient)
18	transport := c.http.Transport.(*azureResponsesTransport)
19	var gotReq *http.Request
20	var gotBody codexRequest
21	transport.inner = roundTripFunc(func(r *http.Request) (*http.Response, error) {
22		gotReq = r
23		if err := json.NewDecoder(r.Body).Decode(&gotBody); err != nil {
24			t.Fatal(err)
25		}
26		return &http.Response{
27			StatusCode: http.StatusOK,
28			Header:     http.Header{"Content-Type": []string{"text/event-stream"}},
29			Body:       io.NopCloser(strings.NewReader("")),
30		}, nil
31	})
32
33	events, err := c.Stream(context.Background(), Request{
34		Model:     "gpt-5.6-sol",
35		Reasoning: "max",
36		Messages:  []Message{{Role: RoleUser, Content: []Content{TextBlock{Text: "hi"}}}},
37	})
38	if err != nil {
39		t.Fatal(err)
40	}
41	var startProvider string
42	for event := range events {
43		if start, ok := event.(EventStart); ok {
44			startProvider = start.Provider
45		}
46	}
47
48	if startProvider != "azure-openai-responses" {
49		t.Fatalf("start provider = %q", startProvider)
50	}
51	if gotReq == nil {
52		t.Fatal("request was not sent")
53	}
54	if gotReq.URL.String() != "https://example.openai.azure.com/openai/v1/responses?api-version=v1" {
55		t.Fatalf("url = %q", gotReq.URL.String())
56	}
57	if gotReq.Header.Get("api-key") != "azure-key" || gotReq.Header.Get("authorization") != "" {
58		t.Fatalf("headers = %#v", gotReq.Header)
59	}
60	if gotBody.Model != "sol-preview" {
61		t.Fatalf("model = %q", gotBody.Model)
62	}
63	if gotBody.Reasoning == nil || gotBody.Reasoning.Effort != "max" {
64		t.Fatalf("reasoning = %+v", gotBody.Reasoning)
65	}
66	if gotBody.PromptCacheKey != "" {
67		t.Fatalf("prompt_cache_key = %q", gotBody.PromptCacheKey)
68	}
69}