@@ -14,6 +14,7 @@ import (
14
14
"github.com/openai/openai-go"
15
15
"github.com/openai/openai-go/internal"
16
16
"github.com/openai/openai-go/option"
17
+ "github.com/openai/openai-go/shared"
17
18
)
18
19
19
20
type closureTransport struct {
@@ -43,7 +44,7 @@ func TestUserAgentHeader(t *testing.T) {
43
44
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
44
45
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
45
46
}}),
46
- Model : openai .F (openai .ChatModelO3Mini ),
47
+ Model : openai .F (shared .ChatModelO3Mini ),
47
48
})
48
49
if userAgent != fmt .Sprintf ("OpenAI/Go %s" , internal .PackageVersion ) {
49
50
t .Errorf ("Expected User-Agent to be correct, but got: %#v" , userAgent )
@@ -72,7 +73,7 @@ func TestRetryAfter(t *testing.T) {
72
73
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
73
74
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
74
75
}}),
75
- Model : openai .F (openai .ChatModelO3Mini ),
76
+ Model : openai .F (shared .ChatModelO3Mini ),
76
77
})
77
78
if err == nil {
78
79
t .Error ("Expected there to be a cancel error" )
@@ -112,7 +113,7 @@ func TestDeleteRetryCountHeader(t *testing.T) {
112
113
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
113
114
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
114
115
}}),
115
- Model : openai .F (openai .ChatModelO3Mini ),
116
+ Model : openai .F (shared .ChatModelO3Mini ),
116
117
})
117
118
if err == nil {
118
119
t .Error ("Expected there to be a cancel error" )
@@ -147,7 +148,7 @@ func TestOverwriteRetryCountHeader(t *testing.T) {
147
148
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
148
149
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
149
150
}}),
150
- Model : openai .F (openai .ChatModelO3Mini ),
151
+ Model : openai .F (shared .ChatModelO3Mini ),
151
152
})
152
153
if err == nil {
153
154
t .Error ("Expected there to be a cancel error" )
@@ -181,7 +182,7 @@ func TestRetryAfterMs(t *testing.T) {
181
182
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
182
183
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
183
184
}}),
184
- Model : openai .F (openai .ChatModelO3Mini ),
185
+ Model : openai .F (shared .ChatModelO3Mini ),
185
186
})
186
187
if err == nil {
187
188
t .Error ("Expected there to be a cancel error" )
@@ -209,7 +210,7 @@ func TestContextCancel(t *testing.T) {
209
210
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
210
211
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
211
212
}}),
212
- Model : openai .F (openai .ChatModelO3Mini ),
213
+ Model : openai .F (shared .ChatModelO3Mini ),
213
214
})
214
215
if err == nil {
215
216
t .Error ("Expected there to be a cancel error" )
@@ -234,7 +235,7 @@ func TestContextCancelDelay(t *testing.T) {
234
235
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
235
236
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
236
237
}}),
237
- Model : openai .F (openai .ChatModelO3Mini ),
238
+ Model : openai .F (shared .ChatModelO3Mini ),
238
239
})
239
240
if err == nil {
240
241
t .Error ("expected there to be a cancel error" )
@@ -265,7 +266,7 @@ func TestContextDeadline(t *testing.T) {
265
266
Role : openai .F (openai .ChatCompletionUserMessageParamRoleUser ),
266
267
Content : openai .F ([]openai.ChatCompletionContentPartUnionParam {openai.ChatCompletionContentPartTextParam {Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
267
268
}}),
268
- Model : openai .F (openai .ChatModelO3Mini ),
269
+ Model : openai .F (shared .ChatModelO3Mini ),
269
270
})
270
271
if err == nil {
271
272
t .Error ("expected there to be a deadline error" )
@@ -315,7 +316,7 @@ func TestContextDeadlineStreaming(t *testing.T) {
315
316
Content : openai .F ([]openai.ChatCompletionContentPartTextParam {{Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
316
317
Role : openai .F (openai .ChatCompletionDeveloperMessageParamRoleDeveloper ),
317
318
}}),
318
- Model : openai .F (openai .ChatModelO3Mini ),
319
+ Model : openai .F (shared .ChatModelO3Mini ),
319
320
})
320
321
for stream .Next () {
321
322
_ = stream .Current ()
@@ -367,7 +368,7 @@ func TestContextDeadlineStreamingWithRequestTimeout(t *testing.T) {
367
368
Content : openai .F ([]openai.ChatCompletionContentPartTextParam {{Text : openai .F ("text" ), Type : openai .F (openai .ChatCompletionContentPartTextTypeText )}}),
368
369
Role : openai .F (openai .ChatCompletionDeveloperMessageParamRoleDeveloper ),
369
370
}}),
370
- Model : openai .F (openai .ChatModelO3Mini ),
371
+ Model : openai .F (shared .ChatModelO3Mini ),
371
372
},
372
373
option .WithRequestTimeout ((100 * time .Millisecond )),
373
374
)
0 commit comments