37 lines
1.4 KiB
Java
37 lines
1.4 KiB
Java
package example;
|
|
|
|
import com.theokanning.openai.OpenAiService;
|
|
import com.theokanning.openai.completion.CompletionRequest;
|
|
import com.theokanning.openai.engine.Engine;
|
|
import com.theokanning.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("\nGetting available engines...");
|
|
service.getEngines().forEach(System.out::println);
|
|
|
|
System.out.println("\nGetting ada engine...");
|
|
Engine ada = service.getEngine("ada");
|
|
System.out.println(ada);
|
|
|
|
System.out.println("\nCreating completion...");
|
|
|
|
CompletionRequest completionRequest = CompletionRequest.builder()
|
|
.prompt("Somebody once told me the world is gonna roll me")
|
|
.echo(true)
|
|
.build();
|
|
service.createCompletion("ada", completionRequest).getChoices().forEach(System.out::println);
|
|
|
|
System.out.println("\nSearching documents...");
|
|
SearchRequest searchRequest = SearchRequest.builder()
|
|
.documents(Arrays.asList("Water", "Earth", "Electricity", "Fire"))
|
|
.query("Pikachu")
|
|
.build();
|
|
service.search("ada", searchRequest).forEach(System.out::println);
|
|
}
|
|
} |