Implemented missing API for issue #366: No Retrieve file content api (#…

…367)
TheoKanning · Oct 22, 2023 · 4420830 · 4420830
1 parent 442cf13
commit 4420830
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/client/src/main/java/com/theokanning/openai/client/OpenAiApi.java b/client/src/main/java/com/theokanning/openai/client/OpenAiApi.java
@@ -89,6 +89,10 @@ public interface OpenAiApi {
     @GET("/v1/files/{file_id}")
     Single<File> retrieveFile(@Path("file_id") String fileId);
 
+    @Streaming
+    @GET("/v1/files/{file_id}/content")
+    Single<ResponseBody> retrieveFileContent(@Path("file_id") String fileId);
+
     @POST("/v1/fine_tuning/jobs")
     Single<FineTuningJob> createFineTuningJob(@Body FineTuningJobRequest request);
 

diff --git a/service/src/main/java/com/theokanning/openai/service/OpenAiService.java b/service/src/main/java/com/theokanning/openai/service/OpenAiService.java
@@ -173,6 +173,10 @@ public File retrieveFile(String fileId) {
         return execute(api.retrieveFile(fileId));
     }
 
+    public ResponseBody retrieveFileContent(String fileId) {
+        return execute(api.retrieveFileContent(fileId));
+    }
+
     public FineTuningJob createFineTuningJob(FineTuningJobRequest request) {
         return execute(api.createFineTuningJob(request));
     }

diff --git a/service/src/test/java/com/theokanning/openai/service/FileTest.java b/service/src/test/java/com/theokanning/openai/service/FileTest.java
@@ -7,6 +7,9 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.TestMethodOrder;
 
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 
@@ -52,6 +55,13 @@ void retrieveFile() {
 
     @Test
     @Order(4)
+    void retrieveFileContent() throws IOException {
+        String fileBytesToString = service.retrieveFileContent(fileId).string();
+        assertEquals(Files.readString(Path.of(filePath)), fileBytesToString);
+    }
+
+    @Test
+    @Order(5)
     void deleteFile() {
         DeleteResult result = service.deleteFile(fileId);
         assertTrue(result.isDeleted());