Add fine-tuning api models to json test (#379)

TheoKanning · Oct 22, 2023 · 44cba2e · 44cba2e
1 parent a4f2df8
commit 44cba2e
Show file tree

Hide file tree

Showing 5 changed files with 47 additions and 2 deletions.
diff --git a/api/src/main/java/com/theokanning/openai/fine_tuning/FineTuningEvent.java b/api/src/main/java/com/theokanning/openai/fine_tuning/FineTuningEvent.java
@@ -15,6 +15,11 @@ public class FineTuningEvent {
      */
     String object;
 
+    /**
+     * The ID of the fine-tuning event.
+     */
+    String id;
+
     /**
      * The creation time in epoch seconds.
      */

diff --git a/api/src/test/java/com/theokanning/openai/JsonTest.java b/api/src/test/java/com/theokanning/openai/JsonTest.java
@@ -12,10 +12,11 @@
 import com.theokanning.openai.embedding.EmbeddingResult;
 import com.theokanning.openai.engine.Engine;
 import com.theokanning.openai.file.File;
+import com.theokanning.openai.fine_tuning.FineTuningEvent;
+import com.theokanning.openai.fine_tuning.FineTuningJob;
+import com.theokanning.openai.fine_tuning.FineTuningJobRequest;
 import com.theokanning.openai.finetune.FineTuneEvent;
 import com.theokanning.openai.finetune.FineTuneResult;
-import com.theokanning.openai.image.CreateImageEditRequest;
-import com.theokanning.openai.image.CreateImageRequest;
 import com.theokanning.openai.image.ImageResult;
 import com.theokanning.openai.model.Model;
 import com.theokanning.openai.moderation.ModerationRequest;
@@ -43,6 +44,9 @@ public class JsonTest {
             File.class,
             FineTuneEvent.class,
             FineTuneResult.class,
+            FineTuningEvent.class,
+            FineTuningJob.class,
+            FineTuningJobRequest.class,
             ImageResult.class,
             TranscriptionResult.class,
             TranslationResult.class,

diff --git a/api/src/test/resources/fixtures/FineTuningEvent.json b/api/src/test/resources/fixtures/FineTuningEvent.json
@@ -0,0 +1,8 @@
+{
+  "object": "fine_tuning.job.event",
+  "id": "ft-event-ddTJfwuMVpfLXseO0Am0Gqjm",
+  "created_at": 1692407401,
+  "level": "info",
+  "message": "Fine tuning job successfully completed",
+  "type": "message"
+}
diff --git a/api/src/test/resources/fixtures/FineTuningJob.json b/api/src/test/resources/fixtures/FineTuningJob.json
@@ -0,0 +1,19 @@
+{
+  "id": "ftjob-abc123",
+  "object": "fine_tuning.job",
+  "model": "davinci-002",
+  "status": "succeeded",
+  "hyperparameters": {
+    "n_epochs": 4
+  },
+  "created_at": 1692661014,
+  "finished_at": 1692661190,
+  "fine_tuned_model": "ft:davinci-002:my-org:custom_suffix:7q8mpxmy",
+  "organization_id": "org-123",
+  "training_file": "file-abc123",
+  "result_files": [
+    "file-abc123"
+  ],
+  "validation_file": "validation-file",
+  "trained_tokens": 5768
+}
diff --git a/api/src/test/resources/fixtures/FineTuningJobRequest.json b/api/src/test/resources/fixtures/FineTuningJobRequest.json
@@ -0,0 +1,9 @@
+{
+  "model": "davinci-002",
+  "validation_file": "file-abc123",
+  "training_file": "file-abc123",
+  "hyperparameters": {
+    "n_epochs": 4
+  },
+  "suffix": "test"
+}