Modify the test module, add the file upload test, and add the image edit api (#63)

* 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>
2023-02-12 02:51:53 +08:00
parent ae06df7d9f
commit 5191ea6f55
12 changed files with 719 additions and 426 deletions
--- a/edits_test.go
+++ b/edits_test.go
@@ -0,0 +1,103 @@
+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
+}