sammyjava
diff --git a/‎example/src/main/java/example/OpenAiApiExample.java
Copy file name to clipboardExpand all lines: example/src/main/java/example/OpenAiApiExample.java
+11-3Lines changed: 11 additions & 3 deletions b/‎example/src/main/java/example/OpenAiApiExample.java
Copy file name to clipboardExpand all lines: example/src/main/java/example/OpenAiApiExample.java
+11-3Lines changed: 11 additions & 3 deletions
diff --git a/‎openai-api/src/main/java/openai/search/SearchRequest.java
Copy file name to clipboard
+11Lines changed: 11 additions & 0 deletions b/‎openai-api/src/main/java/openai/search/SearchRequest.java
Copy file name to clipboard
+11Lines changed: 11 additions & 0 deletions
diff --git a/‎openai-api/src/main/java/openai/search/SearchResult.java
Copy file name to clipboard
+10Lines changed: 10 additions & 0 deletions b/‎openai-api/src/main/java/openai/search/SearchResult.java
Copy file name to clipboard
+10Lines changed: 10 additions & 0 deletions
diff --git a/‎openai-client/src/main/java/openai/OpenAiApi.java
Copy file name to clipboardExpand all lines: openai-client/src/main/java/openai/OpenAiApi.java
+5Lines changed: 5 additions & 0 deletions b/‎openai-client/src/main/java/openai/OpenAiApi.java
Copy file name to clipboardExpand all lines: openai-client/src/main/java/openai/OpenAiApi.java
+5Lines changed: 5 additions & 0 deletions
diff --git a/‎openai-client/src/main/java/openai/OpenAiService.java
Copy file name to clipboardExpand all lines: openai-client/src/main/java/openai/OpenAiService.java
+6Lines changed: 6 additions & 0 deletions b/‎openai-client/src/main/java/openai/OpenAiService.java
Copy file name to clipboardExpand all lines: openai-client/src/main/java/openai/OpenAiService.java
+6Lines changed: 6 additions & 0 deletions
@@ -3,24 +3,32 @@
 import openai.OpenAiService;
 import openai.completion.CompletionRequest;
 import openai.engine.Engine;
+import openai.search.SearchRequest;
+
+import java.util.Arrays;
 
 class OpenAiApiExample {
     public static void main(String... args) {
         String token = System.getenv("OPENAI_TOKEN");
         OpenAiService service = new OpenAiService(token);
 
-        System.out.println("Getting available engines...");
+        System.out.println("\nGetting available engines...");
         service.getEngines().forEach(System.out::println);
 
-        System.out.println("Getting ada engine...");
+        System.out.println("\nGetting ada engine...");
         Engine ada = service.getEngine("ada");
         System.out.println(ada);
 
-        System.out.println("Creating completion...");
+        System.out.println("\nCreating completion...");
         CompletionRequest completionRequest = new CompletionRequest();
         completionRequest.setPrompt("Somebody once told me the world is gonna roll me");
         completionRequest.setEcho(true);
         service.createCompletion("ada", completionRequest).getChoices().forEach(System.out::println);
 
+        System.out.println("\nSearching documents...");
+        SearchRequest searchRequest = new SearchRequest();
+        searchRequest.setDocuments(Arrays.asList("Water", "Earth", "Electricity", "Fire"));
+        searchRequest.setQuery("Pikachu");
+        service.search("ada", searchRequest).forEach(System.out::println);
     }
 }
@@ -0,0 +1,11 @@
+package openai.search;
+
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+public class SearchRequest {
+    List<String> documents;
+    String query;
+}
@@ -0,0 +1,10 @@
+package openai.search;
+
+import lombok.Data;
+
+@Data
+public class SearchResult {
+    Integer document;
+    String object;
+    Double score;
+}
@@ -4,6 +4,8 @@
 import openai.completion.CompletionRequest;
 import openai.completion.CompletionResult;
 import openai.engine.Engine;
+import openai.search.SearchRequest;
+import openai.search.SearchResult;
 import retrofit2.http.Body;
 import retrofit2.http.GET;
 import retrofit2.http.POST;
@@ -19,4 +21,7 @@ public interface OpenAiApi {
 
     @POST("/v1/engines/{engine_id}/completions")
     Single<CompletionResult> createCompletion(@Path("engine_id") String engineId, @Body CompletionRequest request);
+
+    @POST("/v1/engines/{engine_id}/search")
+    Single<OpenAiResponse<SearchResult>> search(@Path("engine_id") String engineId, @Body SearchRequest request);
 }
@@ -9,6 +9,8 @@
 import openai.completion.CompletionRequest;
 import openai.completion.CompletionResult;
 import openai.engine.Engine;
+import openai.search.SearchRequest;
+import openai.search.SearchResult;
 import retrofit2.Retrofit;
 import retrofit2.adapter.rxjava2.RxJava2CallAdapterFactory;
 import retrofit2.converter.jackson.JacksonConverterFactory;
@@ -52,4 +54,8 @@ public Engine getEngine(String engineId) {
     public CompletionResult createCompletion(String engineId, CompletionRequest request) {
         return api.createCompletion(engineId, request).blockingGet();
     }
+
+    public List<SearchResult> search(String engineId, SearchRequest request) {
+        return api.search(engineId, request).blockingGet().data;
+    }
 }