sammyjava
diff --git a/‎api/src/main/java/com/theokanning/openai/edit/EditRequest.java
Copy file name to clipboardExpand all lines: api/src/main/java/com/theokanning/openai/edit/EditRequest.java
+9-4Lines changed: 9 additions & 4 deletions b/‎api/src/main/java/com/theokanning/openai/edit/EditRequest.java
Copy file name to clipboardExpand all lines: api/src/main/java/com/theokanning/openai/edit/EditRequest.java
+9-4Lines changed: 9 additions & 4 deletions
diff --git a/‎api/src/main/java/com/theokanning/openai/edit/EditResult.java
Copy file name to clipboardExpand all lines: api/src/main/java/com/theokanning/openai/edit/EditResult.java
+5Lines changed: 5 additions & 0 deletions b/‎api/src/main/java/com/theokanning/openai/edit/EditResult.java
Copy file name to clipboardExpand all lines: api/src/main/java/com/theokanning/openai/edit/EditResult.java
+5Lines changed: 5 additions & 0 deletions
diff --git a/‎api/src/main/java/com/theokanning/openai/edit/EditUsage.java
Copy file name to clipboard
+27Lines changed: 27 additions & 0 deletions b/‎api/src/main/java/com/theokanning/openai/edit/EditUsage.java
Copy file name to clipboard
+27Lines changed: 27 additions & 0 deletions
diff --git a/‎client/src/test/java/com/theokanning/openai/EditTest.java
Copy file name to clipboardExpand all lines: client/src/test/java/com/theokanning/openai/EditTest.java
+3-5Lines changed: 3 additions & 5 deletions b/‎client/src/test/java/com/theokanning/openai/EditTest.java
Copy file name to clipboardExpand all lines: client/src/test/java/com/theokanning/openai/EditTest.java
+3-5Lines changed: 3 additions & 5 deletions
@@ -1,9 +1,7 @@
 package com.theokanning.openai.edit;
 
-import lombok.AllArgsConstructor;
-import lombok.Builder;
-import lombok.Data;
-import lombok.NoArgsConstructor;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.*;
 
 /**
  * Given a prompt and an instruction, OpenAi will return an edited version of the prompt
@@ -25,8 +23,14 @@ public class EditRequest {
      * The instruction that tells the model how to edit the prompt.
      * For example, "Fix the spelling mistakes"
      */
+    @NonNull
     String instruction;
 
+    /**
+     * How many edits to generate for the input and instruction.
+     */
+    Integer n;
+
     /**
      * What sampling temperature to use. Higher values means the model will take more risks. Try 0.9 for more creative applications, and 0 (argmax sampling) for ones with a well-defined answer.
      *
@@ -39,5 +43,6 @@ public class EditRequest {
      *
      * We generally recommend altering this or {@link EditRequest#temperature} but not both.
      */
+    @JsonProperty("top_p")
     Double topP;
 }
@@ -26,4 +26,9 @@ public class EditResult {
      * A list of generated edits.
      */
     List<EditChoice> choices;
+
+    /**
+     * The API usage for this request
+     */
+    EditUsage usage;
 }
@@ -0,0 +1,27 @@
+package com.theokanning.openai.edit;
+
+import lombok.Data;
+
+/**
+ * An object containing the API usage for an edit request
+ *
+ * https://beta.openai.com/docs/api-reference/edits/create
+ */
+@Data
+public class EditUsage {
+
+    /**
+     * The number of prompt tokens consumed.
+     */
+    String promptTokens;
+
+    /**
+     * The number of completion tokens consumed.
+     */
+    String completionTokens;
+
+    /**
+     * The number of total tokens consumed.
+     */
+    String totalTokens;
+}
@@ -2,12 +2,10 @@
 
 import com.theokanning.openai.edit.EditRequest;
 import com.theokanning.openai.edit.EditResult;
-import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 
-@Disabled // disabled until edit example CURL works
 public class EditTest {
 
     String token = System.getenv("OPENAI_TOKEN");
@@ -20,8 +18,8 @@ void edit() {
                 .instruction("Fix the spelling mistakes")
                 .build();
 
-        EditResult result = service.createEdit("text-ada-001", request);
+        EditResult result = service.createEdit("text-davinci-edit-001", request);
 
-        assertEquals("What day of the week is it?", result.getChoices().get(0).getText());
+        assertNotNull(result.getChoices().get(0).getText());
     }
 }