这是indexloc提供的服务，不要输入任何密码
Skip to content

BAEL-7806: Baeldung Tutor #16685

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 20, 2024
Merged

BAEL-7806: Baeldung Tutor #16685

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions libraries-ai/pom.xml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,12 +28,28 @@
<artifactId>neuroph</artifactId>
<version>${neuroph.version}</version>
</dependency>
<dependency>
<groupId>com.theokanning.openai-gpt3-java</groupId>
<artifactId>api</artifactId>
<version>${theokanning.gpt}</version>
</dependency>
<dependency>
<groupId>com.theokanning.openai-gpt3-java</groupId>
<artifactId>client</artifactId>
<version>${theokanning.gpt}</version>
</dependency>
<dependency>
<groupId>com.theokanning.openai-gpt3-java</groupId>
<artifactId>service</artifactId>
<version>${theokanning.gpt}</version>
</dependency>
</dependencies>

<properties>
<stanford-corenlp.version>4.5.3</stanford-corenlp.version>
<opennlp-tools.version>2.1.1</opennlp-tools.version>
<neuroph.version>2.92</neuroph.version>
<theokanning.gpt>0.18.2</theokanning.gpt>
</properties>

</project>
67 changes: 67 additions & 0 deletions libraries-ai/src/main/java/baeldungassistant/BaeldungLearningAssistant.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
package baeldungassistant;

import static com.theokanning.openai.utils.TikTokensUtil.ModelEnum.GPT_3_5_TURBO_0301;

import com.theokanning.openai.completion.chat.ChatCompletionRequest;
import com.theokanning.openai.completion.chat.ChatCompletionResult;
import com.theokanning.openai.completion.chat.ChatMessage;
import com.theokanning.openai.completion.chat.ChatMessageRole;
import com.theokanning.openai.service.OpenAiService;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

class BaeldungLearningAssistant {

private static final String OPENAI_TOKEN = "OPENAI_TOKEN";
private static final String PROMPT = "You're helping me to create a curriculum"
+ "to learn programming."
+ "I want to use Baedlung website as the base."
+ "I will tell you the topic,"
+ "and you should return me the list of articles"
+ "and tutorials with links."
+ "Order the articles from beginner to more advanced,"
+ "so I can learn them one-by-one."
+ "Use only the articles from www.baeldung.com.";
private static final int MAX_TOKENS = 900;
private static final String GREETING = "Hello!\nWhat do you want to learn?\n";
private static final String MODEL = GPT_3_5_TURBO_0301.getName();

public static void main(String[] args) {
String token = System.getenv(OPENAI_TOKEN);
OpenAiService service = new OpenAiService(token);

List<ChatMessage> messages = new ArrayList<>();
ChatMessage systemMessage = new ChatMessage(ChatMessageRole.SYSTEM.value(), PROMPT);
messages.add(systemMessage);

System.out.print(GREETING);
Scanner scanner = new Scanner(System.in);
ChatMessage firstMsg = new ChatMessage(ChatMessageRole.USER.value(), scanner.nextLine());
messages.add(firstMsg);

while (true) {
ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest
.builder()
.model(MODEL)
.maxTokens(MAX_TOKENS)
.messages(messages)
.build();
ChatCompletionResult result = service.createChatCompletion(chatCompletionRequest);
long usedTokens = result.getUsage().getTotalTokens();
ChatMessage response = result.getChoices().get(0).getMessage();

messages.add(response);

System.out.println(response.getContent());
System.out.println("Total tokens used: " + usedTokens);
System.out.print("Anything else?\n");
String nextLine = scanner.nextLine();
if (nextLine.equalsIgnoreCase("exit")) {
System.exit(0);
}
messages.add(new ChatMessage(ChatMessageRole.USER.value(), nextLine));
}
}

}