diff --git a/README.md b/README.md new file mode 100644 index 00000000..c4356fd3 --- /dev/null +++ b/README.md @@ -0,0 +1,18 @@ +# OpenAI - ChatGPT IntelliJ Plugin +ChatGPT as your copilot to level up your developer experience. This is the perfect assistant for any programmer who wants to improve their coding skills and make more efficient use of the time. + +## Installation +The plugin is available from [JetBrains Marketplace](https://plugins.jetbrains.com/plugin/21056-openai--chatgpt). +You can install it directly from your IDE via the `File | Settings/Preferences | Plugins` screen. +On the `Marketplace` tab simply search for `openai` and select the `OpenAI - ChatGPT` suggestion: + +![marketplace](docs/assets/marketplace.png) + +## Available features +The main features of this plugin include: +- Ask ChatGPT: Ask free-form text questions. +- Find bugs: Analyze and find bugs in your code. +- Add tests: Write tests for you. +- Refactor: Refactor your code. +- Optimize: Add suggestions to your code to improve. +- Explain: Explain the selected code. diff --git a/docs/assets/marketplace.png b/docs/assets/marketplace.png new file mode 100644 index 00000000..15bfdc9e Binary files /dev/null and b/docs/assets/marketplace.png differ diff --git a/src/main/java/ee/carlrobert/chatgpt/client/ApiClient.java b/src/main/java/ee/carlrobert/chatgpt/client/ApiClient.java index 774cd8de..8e47a383 100644 --- a/src/main/java/ee/carlrobert/chatgpt/client/ApiClient.java +++ b/src/main/java/ee/carlrobert/chatgpt/client/ApiClient.java @@ -6,6 +6,8 @@ import ee.carlrobert.chatgpt.ide.settings.SettingsState; import java.net.URI; import java.net.http.HttpClient; import java.net.http.HttpRequest; +import java.net.http.HttpResponse.BodySubscriber; +import java.net.http.HttpResponse.ResponseInfo; import java.util.ArrayList; import java.util.List; import java.util.Map; @@ -21,27 +23,44 @@ public final class ApiClient { private ApiClient() { } - public void getCompletionsAsync(String prompt, Consumer onMessage) { - try { - var query = new StringBuilder( - "You are ChatGPT, a large language model trained by OpenAI.\n"); - for (var entry : queries) { - query.append("User:\n") - .append(entry.getKey()) + public static ApiClient getInstance() { + if (instance == null) { + instance = new ApiClient(); + } + return instance; + } + + public void getCompletionsAsync(String userPrompt, Consumer onMessage) { + var prompt = buildCompletePrompt(userPrompt); + this.client.sendAsync(buildHttpRequest(prompt), respInfo -> subscribe(respInfo, userPrompt, onMessage)); + } + + public void clearQueries() { + queries.clear(); + } + + private String buildCompletePrompt(String prompt) { + var basePrompt = new StringBuilder("You are ChatGPT, a large language model trained by OpenAI.\n"); + queries.forEach(query -> + basePrompt.append("User:\n") + .append(query.getKey()) .append("<|im_end|>\n") .append("\n") .append("ChatGPT:\n") - .append(entry.getValue()) + .append(query.getValue()) .append("<|im_end|>\n") - .append("\n"); - } - query.append("User:\n") - .append(prompt) - .append("<|im_end|>\n") - .append("\n") - .append("ChatGPT:\n"); + .append("\n")); + basePrompt.append("User:\n") + .append(prompt) + .append("<|im_end|>\n") + .append("\n") + .append("ChatGPT:\n"); + return basePrompt.toString(); + } - var req = HttpRequest.newBuilder() + private HttpRequest buildHttpRequest(String prompt) { + try { + return HttpRequest.newBuilder() .uri(URI.create("https://api.openai.com/v1/completions")) .header("Accept", "text/event-stream") .header("Content-Type", "application/json") @@ -51,46 +70,33 @@ public final class ApiClient { .writeValueAsString(Map.of( "model", "text-davinci-003", "stop", List.of("<|im_end|>"), - "prompt", query.toString(), + "prompt", prompt, "max_tokens", 1000, "temperature", 1.0, "stream", true )))) .build(); - - this.client.sendAsync(req, respInfo -> - { - if (respInfo.statusCode() == 200) { - return new Subscriber((messageData -> - onMessage.accept(messageData.getChoices().get(0).getText())), - (finalMsg) -> queries.add(Map.entry(prompt, finalMsg))); - } else if (respInfo.statusCode() == 401) { - onMessage.accept("Incorrect API key provided.\n" + - "You can find your API key at https://platform.openai.com/account/api-keys."); - throw new IllegalArgumentException(); - } else if (respInfo.statusCode() == 429) { - onMessage.accept("You exceeded your current quota, please check your plan and billing details."); - throw new RuntimeException("Insufficient quota"); - } else { - onMessage.accept("Something went wrong. Please try again later."); - clearQueries(); - throw new RuntimeException(); - } - }); } catch (JsonProcessingException e) { + throw new RuntimeException("Unable to serialize request payload", e); + } + } + + private BodySubscriber subscribe(ResponseInfo responseInfo, String userPrompt, Consumer onMessage) { + if (responseInfo.statusCode() == 200) { + return new Subscriber((messageData -> + onMessage.accept(messageData.getChoices().get(0).getText())), + (finalMsg) -> queries.add(Map.entry(userPrompt, finalMsg))); + } else if (responseInfo.statusCode() == 401) { + onMessage.accept("Incorrect API key provided.\n" + + "You can find your API key at https://platform.openai.com/account/api-keys."); + throw new IllegalArgumentException(); + } else if (responseInfo.statusCode() == 429) { + onMessage.accept("You exceeded your current quota, please check your plan and billing details."); + throw new RuntimeException("Insufficient quota"); + } else { onMessage.accept("Something went wrong. Please try again later."); - throw new RuntimeException(e); + clearQueries(); + throw new RuntimeException(); } } - - public static ApiClient getInstance() { - if (instance == null) { - instance = new ApiClient(); - } - return instance; - } - - public void clearQueries() { - queries.clear(); - } } diff --git a/src/main/java/ee/carlrobert/chatgpt/client/Subscriber.java b/src/main/java/ee/carlrobert/chatgpt/client/Subscriber.java index 502494fb..f22e3c82 100644 --- a/src/main/java/ee/carlrobert/chatgpt/client/Subscriber.java +++ b/src/main/java/ee/carlrobert/chatgpt/client/Subscriber.java @@ -74,7 +74,7 @@ public class Subscriber implements HttpResponse.BodySubscriber { var data = extractMessageData(message.split("\n")); var choice = data.getChoices().get(0); if ("stop".equals(choice.getFinish_reason())) { - onComplete(); + // onComplete(); } else { msgBuilder.append(choice.getText()); }