udpate some type to optimize usage (#139)

This commit is contained in:
jooyyy.eth
2023-03-09 22:53:50 +08:00
committed by GitHub
parent 34e1e10b5f
commit 11052c6106

View File

@@ -24,23 +24,24 @@ type FineTuneRequest struct {
} }
type FineTune struct { type FineTune struct {
ID string `json:"id"` ID string `json:"id"`
Object string `json:"object"` Object string `json:"object"`
Model string `json:"model"` Model string `json:"model"`
CreatedAt int `json:"created_at"` CreatedAt int64 `json:"created_at"`
FineTunedModel string `json:"fine_tuned_model"` FineTuneEventList []FineTuneEvent `json:"events,omitempty"`
Hyperparams FineTuneHyperParams `json:"hyperparams"` FineTunedModel string `json:"fine_tuned_model"`
OrganizationID string `json:"organization_id"` HyperParams FineTuneHyperParams `json:"hyperparams"`
ResultFiles []File `json:"result_files"` OrganizationID string `json:"organization_id"`
Status string `json:"status"` ResultFiles []File `json:"result_files"`
ValidationFiles []File `json:"validation_files"` Status string `json:"status"`
TrainingFiles []File `json:"training_files"` ValidationFiles []File `json:"validation_files"`
UpdatedAt int `json:"updated_at"` TrainingFiles []File `json:"training_files"`
UpdatedAt int64 `json:"updated_at"`
} }
type FineTuneEvent struct { type FineTuneEvent struct {
Object string `json:"object"` Object string `json:"object"`
CreatedAt int `json:"created_at"` CreatedAt int64 `json:"created_at"`
Level string `json:"level"` Level string `json:"level"`
Message string `json:"message"` Message string `json:"message"`
} }
@@ -84,7 +85,7 @@ func (c *Client) CreateFineTune(ctx context.Context, request FineTuneRequest) (r
return return
} }
// Cancel a fine-tune job. // CancelFineTune cancel a fine-tune job.
func (c *Client) CancelFineTune(ctx context.Context, fineTuneID string) (response FineTune, err error) { func (c *Client) CancelFineTune(ctx context.Context, fineTuneID string) (response FineTune, err error) {
req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.fullURL("/fine-tunes/"+fineTuneID+"/cancel"), nil) req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.fullURL("/fine-tunes/"+fineTuneID+"/cancel"), nil)
if err != nil { if err != nil {