Skip to content

Commit 2dfbb0f

Browse files
committed
update AI response footer to include model
1 parent 0763cbe commit 2dfbb0f

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

application/src/main/java/org/togetherjava/tjbot/features/chatgpt/ChatGptCommand.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
* which it will respond with an AI generated answer.
2626
*/
2727
public final class ChatGptCommand extends SlashCommandAdapter {
28+
private static final ChatGptModel CHAT_GPT_MODEL = ChatGptModel.HIGH_QUALITY;
2829
public static final String COMMAND_NAME = "chatgpt";
2930
private static final String QUESTION_INPUT = "question";
3031
private static final int MAX_MESSAGE_INPUT_LENGTH = 200;
@@ -83,7 +84,7 @@ public void onModalSubmitted(ModalInteractionEvent event, List<String> args) {
8384
String question = event.getValue(QUESTION_INPUT).getAsString();
8485

8586
Optional<String> chatgptResponse = chatGptService.ask(question,
86-
"You may use markdown syntax for the response", ChatGptModel.HIGH_QUALITY);
87+
"You may use markdown syntax for the response", CHAT_GPT_MODEL);
8788
if (chatgptResponse.isPresent()) {
8889
userIdToAskedAtCache.put(event.getMember().getId(), Instant.now());
8990
}
@@ -96,7 +97,8 @@ public void onModalSubmitted(ModalInteractionEvent event, List<String> args) {
9697
String response = chatgptResponse.orElse(errorResponse);
9798
SelfUser selfUser = event.getJDA().getSelfUser();
9899

99-
MessageEmbed responseEmbed = helper.generateGptResponseEmbed(response, selfUser, question);
100+
MessageEmbed responseEmbed =
101+
helper.generateGptResponseEmbed(response, selfUser, question, CHAT_GPT_MODEL);
100102

101103
event.getHook().sendMessageEmbeds(responseEmbed).queue();
102104
}

application/src/main/java/org/togetherjava/tjbot/features/help/HelpSystemHelper.java

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
*/
5757
public final class HelpSystemHelper {
5858
private static final Logger logger = LoggerFactory.getLogger(HelpSystemHelper.class);
59+
private static final ChatGptModel CHAT_GPT_MODEL = ChatGptModel.FAST;
5960

6061
static final Color AMBIENT_COLOR = new Color(255, 255, 165);
6162

@@ -144,7 +145,7 @@ RestAction<Message> constructChatGptAttempt(ThreadChannel threadChannel,
144145
String context =
145146
"Category %s on a Java Q&A discord server. You may use markdown syntax for the response"
146147
.formatted(matchingTag.getName());
147-
chatGptAnswer = chatGptService.ask(question, context, ChatGptModel.FAST);
148+
chatGptAnswer = chatGptService.ask(question, context, CHAT_GPT_MODEL);
148149

149150
if (chatGptAnswer.isEmpty()) {
150151
return useChatGptFallbackMessage(threadChannel);
@@ -169,7 +170,8 @@ RestAction<Message> constructChatGptAttempt(ThreadChannel threadChannel,
169170
answer = answer.substring(0, responseCharLimit);
170171
}
171172

172-
MessageEmbed responseEmbed = generateGptResponseEmbed(answer, selfUser, originalQuestion);
173+
MessageEmbed responseEmbed =
174+
generateGptResponseEmbed(answer, selfUser, originalQuestion, CHAT_GPT_MODEL);
173175
return post.flatMap(_ -> threadChannel.sendMessageEmbeds(responseEmbed)
174176
.addActionRow(generateDismissButton(componentIdInteractor, messageId.get())));
175177
}
@@ -182,8 +184,9 @@ RestAction<Message> constructChatGptAttempt(ThreadChannel threadChannel,
182184
* @param title The title for the MessageEmbed.
183185
* @return A MessageEmbed that contains response generated by AI.
184186
*/
185-
public MessageEmbed generateGptResponseEmbed(String answer, SelfUser selfUser, String title) {
186-
String responseByGptFooter = "- AI generated response";
187+
public MessageEmbed generateGptResponseEmbed(String answer, SelfUser selfUser, String title,
188+
ChatGptModel model) {
189+
String responseByGptFooter = "- AI generated response using %s model".formatted(model);
187190

188191
int embedTitleLimit = MessageEmbed.TITLE_MAX_LENGTH;
189192
String capitalizedTitle = Character.toUpperCase(title.charAt(0)) + title.substring(1);

0 commit comments

Comments
 (0)