feat: Switch to the custom chat bot feature and improve the suggestions

Author: dobairoland

action.yml

@@ -11,11 +11,25 @@ inputs:
     description: Issue title.
     required: false
     default: ''
+  bot_behavior_setup:
+    required: false
+    description: Sets the behavior of the bot who should know where it is being deployed.
+    default: >
+      You are an Engineer tasked to solve Github issues. You are professional, knowledgeable, helpful and always
+      friendly. You try to understand the given issue (bug report, feature request) and provide several ideas which
+      the customer can try until an Espressif Engineer handles the issue.
   bot_query_prefix:
-    description: The issue message can be prefixed with this string before evaluation.
+    description: Instructions to the bot how to respond to issues.
     required: false
     default: >
-      The customer opened a Github issue with the following information. Please help and I'll post your answer there.
+      Write a detailed reaction to the Github issue. Please notify the customer if you suggest to try something which
+      can do irreversible damage (e.g. writing eFuses, enabling security features). Check if the bug report have
+      enough information for reproducing the issue. Feature requests should give detailed justification of the request
+      including why existing solutions are insufficient. Ask the customer to provide more information if needed.
+      Try to provide links to documentation, troubleshooting guides or other Espressif documents if you used
+      knowledge from them. If possible then link related and similar issues. Suggest to open a new Github issue only
+      in the case the customer is opened it for the wrong Github repository. Provide the output in Github Markdown
+      format.
   github_comments:
     description: If set to false, the bot will not post a reply to the issue.
     default: true
@@ -69,13 +83,19 @@ runs:
       run: |
         source venv/bin/activate
         cat << 'EOF' > input.txt
+        The issue posted in the Github repository: ${{ inputs.github_repository }}
+        Github issue title: ${{ inputs.title }}
+        Github issue text: ${{ inputs.in_msg }}
+        EOF
+        cat << 'EOF' > bot_instructions.txt
         ${{ inputs.bot_query_prefix }}
-        ${{ inputs.title }}
-        ${{ inputs.in_msg }}
+        EOF
+        cat << 'EOF' > bot_behavior.txt
+        ${{ inputs.bot_behavior_setup }}
         EOF
         printf "%s\n" "${{ inputs.prefix_out_msg }}" > output.txt
         printf "\n---\n" >> output.txt
-        python bot_action/bot_action.py input.txt >> output.txt || exit 1
+        python bot_action/bot_action.py --behavior bot_behavior.txt --instructions bot_instructions.txt input.txt >> output.txt || exit 1
 
     - name: Archive artifacts
       uses: actions/upload-artifact@v4
@@ -85,6 +105,8 @@ runs:
         path: |
           output.txt
           input.txt
+          bot_instructions.txt
+          bot_behavior.txt
 
     - name: Comment
       if: ${{ inputs.github_comments == 'true' }}

bot_action/bot_action.py

@@ -8,10 +8,25 @@
 import json
 import os
 import requests
+import sys
 
+from pprint import pprint
 
-def get_suggestion(issue_body: str) -> str:
-    payload = json.dumps({'integration_id': os.environ['BOT_INTEGRATION_ID'], 'query': issue_body})
+
+def get_suggestion(query: str, instructions: str, behavior: str) -> str:
+    payload = json.dumps(
+        {
+            'integration_id': os.environ['BOT_INTEGRATION_ID'],
+            'messages': [
+                {'role': 'system', 'content': behavior},
+                {'role': 'context'},
+                {'role': 'user', 'content': instructions},
+                {'role': 'query', 'content': query},
+            ],
+        }
+    )
+
+    pprint(payload, stream=sys.stderr)
 
     headers = {'content-type': 'application/json', 'X-API-KEY': os.environ['BOT_API_KEY']}
     r = requests.post(os.environ['BOT_API_ENDPOINT'], data=payload, headers=headers)
@@ -30,12 +45,17 @@ def get_suggestion(issue_body: str) -> str:
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument('input_file', type=str, default=None)
+    parser.add_argument('query_file', type=str, default=None)
+    parser.add_argument('--instructions', type=str, required=True)
+    parser.add_argument('--behavior', type=str, required=True)
     args = parser.parse_args()
 
-    if args.input_file:
-        with open(args.input_file, 'r', encoding='utf-8') as f:
-            print(get_suggestion(f.read()))
+    with (
+        open(args.query_file, 'r', encoding='utf-8') as q,
+        open(args.instructions, 'r', encoding='utf-8') as inst,
+        open(args.behavior, 'r', encoding='utf-8') as behav,
+    ):
+        print(get_suggestion(q.read(), inst.read(), behav.read()))
 
 
 if __name__ == '__main__':