Merge pull request #46 from DefangLabs/linda-trigger-rebuild-config

Use config for trigger rebuild token

Author: commit111

.github/workflows/deploy.yaml

@@ -34,13 +34,14 @@ jobs:
       - name: Deploy
         uses: DefangLabs/[email protected]
         with:
-          config-env-vars: ASK_TOKEN OPENAI_API_KEY SECRET_KEY SEGMENT_WRITE_KEY DISCORD_APP_ID DISCORD_TOKEN DISCORD_PUBLIC_KEY
+          config-env-vars: ASK_TOKEN OPENAI_API_KEY REBUILD_TOKEN SECRET_KEY SEGMENT_WRITE_KEY DISCORD_APP_ID DISCORD_TOKEN DISCORD_PUBLIC_KEY
           mode: production
           provider: aws
 
         env:
           ASK_TOKEN: ${{ secrets.ASK_TOKEN }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          REBUILD_TOKEN: ${{ secrets.REBUILD_TOKEN }}
           SECRET_KEY: ${{ secrets.SECRET_KEY }}
           SEGMENT_WRITE_KEY: ${{ secrets.SEGMENT_WRITE_KEY }}
           DISCORD_APP_ID: ${{ secrets.DISCORD_APP_ID }}

app/app.py

@@ -91,7 +91,7 @@ def v1_ask():
 @csrf.exempt
 def trigger_rebuild():
     token = request.args.get('token')
-    if token != 'b75c82e0-2f47-4fcb-8b02-d66932803885':
+    if token != os.getenv('REBUILD_TOKEN'):
         return jsonify({"error": "Unauthorized"}), 401
     try:
         print("Running get_knowledge_base.py script...")

compose.dev.yaml

@@ -8,6 +8,7 @@ services:
     environment:
       ASK_TOKEN: asktoken
       FLASK_APP: app.py
+      REBUILD_TOKEN: rebuildtoken
       SECRET_KEY: supersecret
       SEGMENT_WRITE_KEY: ${SEGMENT_WRITE_KEY} # Set your Segment write key here or in the .env file
       SESSION_COOKIE_SECURE: 0

compose.yaml

@@ -15,6 +15,7 @@ services:
       ASK_TOKEN:
       FLASK_APP: app.py
       DEBUG: 0
+      REBUILD_TOKEN:
       SECRET_KEY:
       SEGMENT_WRITE_KEY:
       SESSION_COOKIE_SECURE: 1