@@ -1,14 +1,14 @@
|
||||
package openai_test
|
||||
|
||||
import (
|
||||
. "github.com/sashabaranov/go-openai"
|
||||
"github.com/sashabaranov/go-openai/internal/test/checks"
|
||||
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"testing"
|
||||
|
||||
"github.com/sashabaranov/go-openai"
|
||||
"github.com/sashabaranov/go-openai/internal/test/checks"
|
||||
)
|
||||
|
||||
const testFineTuneID = "fine-tune-id"
|
||||
@@ -22,9 +22,9 @@ func TestFineTunes(t *testing.T) {
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
var resBytes []byte
|
||||
if r.Method == http.MethodGet {
|
||||
resBytes, _ = json.Marshal(FineTuneList{})
|
||||
resBytes, _ = json.Marshal(openai.FineTuneList{})
|
||||
} else {
|
||||
resBytes, _ = json.Marshal(FineTune{})
|
||||
resBytes, _ = json.Marshal(openai.FineTune{})
|
||||
}
|
||||
fmt.Fprintln(w, string(resBytes))
|
||||
},
|
||||
@@ -32,8 +32,8 @@ func TestFineTunes(t *testing.T) {
|
||||
|
||||
server.RegisterHandler(
|
||||
"/v1/fine-tunes/"+testFineTuneID+"/cancel",
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
resBytes, _ := json.Marshal(FineTune{})
|
||||
func(w http.ResponseWriter, _ *http.Request) {
|
||||
resBytes, _ := json.Marshal(openai.FineTune{})
|
||||
fmt.Fprintln(w, string(resBytes))
|
||||
},
|
||||
)
|
||||
@@ -43,9 +43,9 @@ func TestFineTunes(t *testing.T) {
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
var resBytes []byte
|
||||
if r.Method == http.MethodDelete {
|
||||
resBytes, _ = json.Marshal(FineTuneDeleteResponse{})
|
||||
resBytes, _ = json.Marshal(openai.FineTuneDeleteResponse{})
|
||||
} else {
|
||||
resBytes, _ = json.Marshal(FineTune{})
|
||||
resBytes, _ = json.Marshal(openai.FineTune{})
|
||||
}
|
||||
fmt.Fprintln(w, string(resBytes))
|
||||
},
|
||||
@@ -53,8 +53,8 @@ func TestFineTunes(t *testing.T) {
|
||||
|
||||
server.RegisterHandler(
|
||||
"/v1/fine-tunes/"+testFineTuneID+"/events",
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
resBytes, _ := json.Marshal(FineTuneEventList{})
|
||||
func(w http.ResponseWriter, _ *http.Request) {
|
||||
resBytes, _ := json.Marshal(openai.FineTuneEventList{})
|
||||
fmt.Fprintln(w, string(resBytes))
|
||||
},
|
||||
)
|
||||
@@ -64,7 +64,7 @@ func TestFineTunes(t *testing.T) {
|
||||
_, err := client.ListFineTunes(ctx)
|
||||
checks.NoError(t, err, "ListFineTunes error")
|
||||
|
||||
_, err = client.CreateFineTune(ctx, FineTuneRequest{})
|
||||
_, err = client.CreateFineTune(ctx, openai.FineTuneRequest{})
|
||||
checks.NoError(t, err, "CreateFineTune error")
|
||||
|
||||
_, err = client.CancelFineTune(ctx, testFineTuneID)
|
||||
|
||||
Reference in New Issue
Block a user