Skip to content

Commit e2abcff

Browse files
Allow extra text to be added after custom prompts.
Signed-off-by: mydeveloperplanet <[email protected]>
1 parent bfab356 commit e2abcff

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/main/java/com/devoxx/genie/service/ChatPromptExecutor.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -222,20 +222,21 @@ private Optional<String> handleCustomCommand(@NotNull ChatMessageContext chatMes
222222
.findFirst();
223223

224224
if (matchingPrompt.isPresent()) {
225-
return processMatchingPrompt(chatMessageContext, promptOutputPanel, matchingPrompt.get());
225+
return processMatchingPrompt(chatMessageContext, promptOutputPanel, prompt, matchingPrompt.get());
226226
} else {
227227
LOG.debug("No matching command prompt found");
228228
return Optional.of(prompt);
229229
}
230230
}
231231

232232
private Optional<String> processMatchingPrompt(@NotNull ChatMessageContext chatMessageContext,
233-
PromptOutputPanel promptOutputPanel, @NotNull CustomPrompt matchingPrompt) {
233+
PromptOutputPanel promptOutputPanel, @NotNull String prompt, @NotNull CustomPrompt matchingPrompt) {
234234
if (matchingPrompt.getName().equalsIgnoreCase(HELP_COMMAND)) {
235235
promptOutputPanel.showHelpText();
236236
return Optional.empty();
237237
}
238238
chatMessageContext.setCommandName(matchingPrompt.getName());
239-
return Optional.of(matchingPrompt.getPrompt());
239+
String customPrompt = matchingPrompt.getPrompt() + " " + prompt.replaceFirst(COMMAND_PREFIX + matchingPrompt.getName(), "");
240+
return Optional.of(customPrompt);
240241
}
241242
}

0 commit comments

Comments
 (0)