5191ea6f55
* Modify the test module, add the file upload test, and add the image editing api * fix golangci-lint * fix golangci-lint * Static file deletion, file directory name modification * fix * test-server-related logic encapsulated in a single tidy struct --------- Co-authored-by: julian_huang <julian.huang@yuansuan.com>
104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
package gogpt_test
|
|
|
|
import (
|
|
. "github.com/sashabaranov/go-gpt3"
|
|
"github.com/sashabaranov/go-gpt3/internal/test"
|
|
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestEdits Tests the edits endpoint of the API using the mocked server.
|
|
func TestEdits(t *testing.T) {
|
|
server := test.NewTestServer()
|
|
server.RegisterHandler("/v1/edits", handleEditEndpoint)
|
|
// create the test server
|
|
var err error
|
|
ts := server.OpenAITestServer()
|
|
ts.Start()
|
|
defer ts.Close()
|
|
|
|
client := NewClient(test.GetTestToken())
|
|
ctx := context.Background()
|
|
client.BaseURL = ts.URL + "/v1"
|
|
|
|
// create an edit request
|
|
model := "ada"
|
|
editReq := EditsRequest{
|
|
Model: &model,
|
|
Input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, " +
|
|
"sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim" +
|
|
" ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip" +
|
|
" ex ea commodo consequat. Duis aute irure dolor in reprehe",
|
|
Instruction: "test instruction",
|
|
N: 3,
|
|
}
|
|
response, err := client.Edits(ctx, editReq)
|
|
if err != nil {
|
|
t.Fatalf("Edits error: %v", err)
|
|
}
|
|
if len(response.Choices) != editReq.N {
|
|
t.Fatalf("edits does not properly return the correct number of choices")
|
|
}
|
|
}
|
|
|
|
// handleEditEndpoint Handles the edit endpoint by the test server.
|
|
func handleEditEndpoint(w http.ResponseWriter, r *http.Request) {
|
|
var err error
|
|
var resBytes []byte
|
|
|
|
// edits only accepts POST requests
|
|
if r.Method != "POST" {
|
|
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
|
}
|
|
var editReq EditsRequest
|
|
editReq, err = getEditBody(r)
|
|
if err != nil {
|
|
http.Error(w, "could not read request", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
// create a response
|
|
res := EditsResponse{
|
|
Object: "test-object",
|
|
Created: time.Now().Unix(),
|
|
}
|
|
// edit and calculate token usage
|
|
editString := "edited by mocked OpenAI server :)"
|
|
inputTokens := numTokens(editReq.Input+editReq.Instruction) * editReq.N
|
|
completionTokens := int(float32(len(editString))/4) * editReq.N
|
|
for i := 0; i < editReq.N; i++ {
|
|
// instruction will be hidden and only seen by OpenAI
|
|
res.Choices = append(res.Choices, EditsChoice{
|
|
Text: editReq.Input + editString,
|
|
Index: i,
|
|
})
|
|
}
|
|
res.Usage = Usage{
|
|
PromptTokens: inputTokens,
|
|
CompletionTokens: completionTokens,
|
|
TotalTokens: inputTokens + completionTokens,
|
|
}
|
|
resBytes, _ = json.Marshal(res)
|
|
fmt.Fprint(w, string(resBytes))
|
|
}
|
|
|
|
// getEditBody Returns the body of the request to create an edit.
|
|
func getEditBody(r *http.Request) (EditsRequest, error) {
|
|
edit := EditsRequest{}
|
|
// read the request body
|
|
reqBody, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
return EditsRequest{}, err
|
|
}
|
|
err = json.Unmarshal(reqBody, &edit)
|
|
if err != nil {
|
|
return EditsRequest{}, err
|
|
}
|
|
return edit, nil
|
|
}
|