OpenAI-J/example/src/main/java/example/OpenAiApiExample.java
Theo Kanning 5d74cfde5a Add Completion api
Might add a builder or something to make setting specific parameters easier
2020-09-14 21:56:31 -05:00

26 lines
924 B
Java

package example;
import openai.OpenAiService;
import openai.completion.CompletionRequest;
import openai.engine.Engine;
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...");
service.getEngines().forEach(System.out::println);
System.out.println("Getting ada engine...");
Engine ada = service.getEngine("ada");
System.out.println(ada);
System.out.println("Creating 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);
}
}