cli,bot: allow setting system message

Author: kezhenxu94

.github/workflows/chatgpt-bot.yaml

@@ -24,23 +24,28 @@ jobs:
           path: ~/.chatgpt
           key: conversations-${{ github.event.issue.number }}-${{ github.run_id }}-${{ github.run_attempt }}
           restore-keys: conversations-${{ github.event.issue.number }}-
-      - uses: kezhenxu94/chatgpt-java/chatgpt-cli@v0.9.0
+      - uses: kezhenxu94/chatgpt-java/chatgpt-cli@v0.10.0
       - name: Question
         if: github.event_name == 'issues'
         env:
           QUESTION: ${{ github.event.issue.title }}
-        run: echo "QUESTION=${QUESTION#'Q:'}" >> $GITHUB_ENV
+          SYSTEM: ${{ github.event.issue.body }}
+        run: |
+          echo "QUESTION=${QUESTION#'Q:'}" >> $GITHUB_ENV
+          echo "SCRIPT=.github/workflows/chatgpt-issue-created.sh" >> $GITHUB_ENV
       - name: Question
         if: github.event_name == 'issue_comment'
         env:
           QUESTION: ${{ github.event.comment.body }}
-        run: echo "QUESTION=${QUESTION#"Q:"}" >> $GITHUB_ENV
+        run: |
+          echo "QUESTION=${QUESTION#"Q:"}" >> $GITHUB_ENV
+          echo "SCRIPT=.github/workflows/chatgpt-issue-comment.sh" >> $GITHUB_ENV
       - name: Answer
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           ISSUE: ${{ github.event.issue.number }}
         run: |
-          envsubst < .github/workflows/chatgpt-script.sh > chatgpt-script.sh
+          envsubst < ${{ env.SCRIPT }} > chatgpt-script.sh
           chatgpt -Dspring.banner.location=file://$(pwd)/.github/workflows/banner.txt @chatgpt-script.sh | tee -a answer.txt
           
           gh issue comment $ISSUE -F answer.txt

.github/workflows/chatgpt-script.sh renamed to .github/workflows/chatgpt-issue-comment.sh

@@ -1,2 +1,2 @@
 load 'issue-$ISSUE'
-ask '$QUESTION'
+ask "\"$QUESTION\""

.github/workflows/chatgpt-issue-created.sh

@@ -0,0 +1,3 @@
+load 'issue-$ISSUE'
+system "\"$SYSTEM\""
+ask "\"$QUESTION\""

build.gradle

@@ -4,6 +4,6 @@ plugins {
 
 allprojects {
     group = 'io.github.kezhenxu94'
-    version = '0.8.0'
+    version = '0.10.0'
     sourceCompatibility = '8'
 }

chatgpt-cli/src/main/java/io/github/kezhenxu94/chatgpt/cli/commands/ConversationCommands.java

@@ -2,8 +2,12 @@
 
 import io.github.kezhenxu94.chatgpt.ChatGPT;
 import io.github.kezhenxu94.chatgpt.Conversation;
+import io.github.kezhenxu94.chatgpt.message.Message;
+import io.github.kezhenxu94.chatgpt.message.Role;
+import org.springframework.shell.Availability;
 import org.springframework.shell.standard.ShellComponent;
 import org.springframework.shell.standard.ShellMethod;
+import org.springframework.shell.standard.ShellMethodAvailability;
 
 @ShellComponent
 public class ConversationCommands {
@@ -15,13 +19,26 @@ public ConversationCommands(ChatGPT chatGPT) {
     this.chatGPT = chatGPT;
   }
 
-  @ShellMethod("Load a conversation by ID")
+  @ShellMethod(
+      "Set the system message as an instruction to guide the ChatGPT in following conversations")
+  @ShellMethodAvailability("isConversationNull")
+  public String system(String system) {
+    if (conversation == null) {
+      conversation = chatGPT.newConversation(system);
+    } else if (conversation.messages().stream().allMatch(it -> it.role() != Role.SYSTEM)) {
+      conversation.messages().add(0, Message.ofSystem(system));
+    }
+    return "System message set to: " + system;
+  }
+
+  @ShellMethod(
+      "Load a conversation by ID, if there is no conversation with the ID, a new one will be created with the ID")
   public String load(String id) throws Exception {
     conversation = chatGPT.loadConversation(id);
     return "You are now at conversation: " + id;
   }
 
-  @ShellMethod("Ask a question")
+  @ShellMethod("Ask a question in the current conversation")
   public String ask(String question) throws Exception {
     if (conversation == null) {
       conversation = chatGPT.newConversation();
@@ -30,4 +47,12 @@ public String ask(String question) throws Exception {
     final var answer = conversation.ask(question);
     return answer.content().trim();
   }
+
+  Availability isConversationNull() {
+    return conversation == null
+            || conversation.messages().stream().allMatch(it -> it.role() != Role.SYSTEM)
+        ? Availability.available()
+        : Availability.unavailable(
+            "you are already in a conversation, system message can only be set in new conversation.");
+  }
 }