fix: use any for n_epochs (#499)

* fix: use custom marshaler for n_epochs

* chore: use any for n_epochs
This commit is contained in:
Simone Vellei
2023-10-06 11:32:21 +02:00
committed by GitHub
parent 84f77a0acd
commit 533935e4fc
2 changed files with 18 additions and 3 deletions

View File

@@ -24,7 +24,7 @@ type FineTuningJob struct {
} }
type Hyperparameters struct { type Hyperparameters struct {
Epochs int `json:"n_epochs"` Epochs any `json:"n_epochs,omitempty"`
} }
type FineTuningJobRequest struct { type FineTuningJobRequest struct {

View File

@@ -21,8 +21,23 @@ func TestFineTuningJob(t *testing.T) {
server.RegisterHandler( server.RegisterHandler(
"/v1/fine_tuning/jobs", "/v1/fine_tuning/jobs",
func(w http.ResponseWriter, r *http.Request) { func(w http.ResponseWriter, r *http.Request) {
var resBytes []byte resBytes, _ := json.Marshal(FineTuningJob{
resBytes, _ = json.Marshal(FineTuningJob{}) Object: "fine_tuning.job",
ID: testFineTuninigJobID,
Model: "davinci-002",
CreatedAt: 1692661014,
FinishedAt: 1692661190,
FineTunedModel: "ft:davinci-002:my-org:custom_suffix:7q8mpxmy",
OrganizationID: "org-123",
ResultFiles: []string{"file-abc123"},
Status: "succeeded",
ValidationFile: "",
TrainingFile: "file-abc123",
Hyperparameters: Hyperparameters{
Epochs: "auto",
},
TrainedTokens: 5768,
})
fmt.Fprintln(w, string(resBytes)) fmt.Fprintln(w, string(resBytes))
}, },
) )