added option to

Author: Dema-koder

src/main/java/org/example/project/service/ChatGPTService.java

@@ -32,20 +32,20 @@ public ChatGPTService(ApplicationConfig config) {
                 .init();
     }
 
-    public String sendMessage(String prompt, String question) {
+    public String sendMessage(String prompt, String question, String model) {
         log.info("Question: {}", question);
 
         Message system = Message.ofSystem(prompt);
         Message message = Message.of(question);
         messageHistory.add(system);
         messageHistory.add(message);
 
-        return sendMessagesToChatGPT();
+        return sendMessagesToChatGPT(model);
     }
 
-    private String sendMessagesToChatGPT(){
+    private String sendMessagesToChatGPT(String model){
         ChatCompletion chatCompletion = ChatCompletion.builder()
-                .model(ChatCompletion.Model.GPT4oMini)
+                .model(model)
                 .messages(messageHistory)
                 .maxTokens(3000)
                 .temperature(0.9)

src/main/java/org/example/project/service/DialogMode.java

@@ -4,5 +4,6 @@ public enum DialogMode {
     MAIN,
     ADD_NOTE,
     WAIT_DEPARTURE_PLACE,
+    WAIT_GPT_VERSION,
     WAIT_QUESTION,
 }

src/main/java/org/example/project/service/TelegramBot.java

@@ -1,5 +1,6 @@
 package org.example.project.service;
 
+import com.plexpt.chatgpt.entity.chat.ChatCompletion;
 import lombok.extern.slf4j.Slf4j;
 import org.example.project.configuration.ApplicationConfig;
 import org.example.project.domain.Users;
@@ -41,6 +42,8 @@ public class TelegramBot extends TelegramLongPollingBot {
 
     private DialogMode dialogMode;
 
+    private String gptVersion = "";
+
     static final String HELP_TEXT = "This bot is created by Demyan Zverev and for his own purpose\n\n" +
             "You can execute from the main menu on the left or by typing command: \n\n" +
             "Type /start to see a welcome message\n\n" +
@@ -76,6 +79,7 @@ public void onUpdateReceived(Update update) {
             String messageText = update.getMessage().getText();
             long chatId = update.getMessage().getChatId();
 
+
             switch(dialogMode) {
                 case MAIN:
                     switch (messageText) {
@@ -114,25 +118,50 @@ public void onUpdateReceived(Update update) {
                 case WAIT_DEPARTURE_PLACE:
                     sendBusSchedule(chatId, messageText);
                     break;
+                case WAIT_GPT_VERSION:
+                    gptVersion = gptVersionReceived(chatId, messageText);
+                    break;
                 case WAIT_QUESTION:
-                    questionToGptReceived(chatId, messageText);
+                    questionToGptReceived(chatId, messageText, gptVersion);
                     break;
             }
         }
     }
 
     // TODO: Изменить взаимодействие с ChatGPT в боте
-    private void questionToGptReceived(long chatId, String question) {
+    private void questionToGptReceived(long chatId, String question, String model) {
         log.info("Send message to ChatGPT");
-        String answer = chatGPTService.sendMessage("", question);
+        String answer = chatGPTService.sendMessage("", question, model);
         sendMessage(chatId, answer);
         dialogMode = DialogMode.MAIN;
     }
 
     private void gptReceived(long chatId) {
+        String answer = """
+                Choose version of ChatGPT:
+                1) GPT 3.5 Turbo
+                2) GPT 4
+                3) GPT 4o mini
+                4) GPT 4o
+                Print only digit""";
+        sendMessage(chatId, answer);
+        dialogMode = DialogMode.WAIT_GPT_VERSION;
+    }
+
+    private String gptVersionReceived(long chatId, String gptVersion) {
         String answer = "Write your question:";
         sendMessage(chatId, answer);
         dialogMode = DialogMode.WAIT_QUESTION;
+        return whichGptVersion(gptVersion);
+    }
+
+    private String whichGptVersion(String version) {
+        return switch (version) {
+            case "2" -> ChatCompletion.Model.GPT4;
+            case "3" -> ChatCompletion.Model.GPT4oMini;
+            case "4" -> ChatCompletion.Model.GPT4o;
+            default -> ChatCompletion.Model.GPT_3_5_TURBO;
+        };
     }
 
     private void getIP(long chatId) {