openai-go/batch_test.go at main · openai/openai-go · GitHub

Name: openai-go/batch_test.go at main · openai/openai-go · GitHub
Rating: 4.9 (1878 reviews)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

package openai_test

import (
"context"
"errors"
"os"
"testing"

"github.com/openai/openai-go"
"github.com/openai/openai-go/internal/testutil"
"github.com/openai/openai-go/option"
"github.com/openai/openai-go/shared"
)

funcTestBatchNewWithOptionalParams(t*testing.T) {
baseURL:="http://localhost:4010"
ifenvURL, ok:=os.LookupEnv("TEST_API_BASE_URL"); ok {
baseURL=envURL
 }
if!testutil.CheckTestServer(t, baseURL) {
return
 }
client:=openai.NewClient(
option.WithBaseURL(baseURL),
option.WithAPIKey("My API Key"),
 )
_, err:=client.Batches.New(context.TODO(), openai.BatchNewParams{
CompletionWindow: openai.BatchNewParamsCompletionWindow24h,
Endpoint: openai.BatchNewParamsEndpointV1Responses,
InputFileID: "input_file_id",
Metadata: shared.MetadataParam{
"foo": "string",
 },
 })
iferr!=nil {
varapierr*openai.Error
iferrors.As(err, &apierr) {
t.Log(string(apierr.DumpRequest(true)))
 }
t.Fatalf("err should be nil: %s", err.Error())
 }
}

funcTestBatchGet(t*testing.T) {
baseURL:="http://localhost:4010"
ifenvURL, ok:=os.LookupEnv("TEST_API_BASE_URL"); ok {
baseURL=envURL
 }
if!testutil.CheckTestServer(t, baseURL) {
return
 }
client:=openai.NewClient(
option.WithBaseURL(baseURL),
option.WithAPIKey("My API Key"),
 )
_, err:=client.Batches.Get(context.TODO(), "batch_id")
iferr!=nil {
varapierr*openai.Error
iferrors.As(err, &apierr) {
t.Log(string(apierr.DumpRequest(true)))
 }
t.Fatalf("err should be nil: %s", err.Error())
 }
}

funcTestBatchListWithOptionalParams(t*testing.T) {
baseURL:="http://localhost:4010"
ifenvURL, ok:=os.LookupEnv("TEST_API_BASE_URL"); ok {
baseURL=envURL
 }
if!testutil.CheckTestServer(t, baseURL) {
return
 }
client:=openai.NewClient(
option.WithBaseURL(baseURL),
option.WithAPIKey("My API Key"),
 )
_, err:=client.Batches.List(context.TODO(), openai.BatchListParams{
After: openai.String("after"),
Limit: openai.Int(0),
 })
iferr!=nil {
varapierr*openai.Error
iferrors.As(err, &apierr) {
t.Log(string(apierr.DumpRequest(true)))
 }
t.Fatalf("err should be nil: %s", err.Error())
 }
}

funcTestBatchCancel(t*testing.T) {
baseURL:="http://localhost:4010"
ifenvURL, ok:=os.LookupEnv("TEST_API_BASE_URL"); ok {
baseURL=envURL
 }
if!testutil.CheckTestServer(t, baseURL) {
return
 }
client:=openai.NewClient(
option.WithBaseURL(baseURL),
option.WithAPIKey("My API Key"),
 )
_, err:=client.Batches.Cancel(context.TODO(), "batch_id")
iferr!=nil {
varapierr*openai.Error
iferrors.As(err, &apierr) {
t.Log(string(apierr.DumpRequest(true)))
 }
t.Fatalf("err should be nil: %s", err.Error())
 }
}