Merge pull request #31 from nalbam/main

Refactor personal and system messages in .env.example
awskrug · Sep 26, 2024 · e9d068f · e9d068f
2 parents 927b988 + a277f61
commit e9d068f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/.env.example b/.env.example
@@ -12,4 +12,5 @@ MODEL_ID_IMAGE="stability.stable-diffusion-xl-v1"
 
 ALLOWED_CHANNEL_IDS="C000000,C000001"
 
-SYSTEM_MESSAGE="너는 AWSKRUG(AWS Korea User Group)에서 친절하게 도움을 주는 구름이(Gurumi)야."
+PERSONAL_MESSAGE="너는 AWSKRUG(AWS Korea User Group)의 친절하고 전문적인 AI 비서 구루미(Gurumi)야."
+SYSTEM_MESSAGE="참고한 링크가 있다면 링크도 알려주세요."
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
@@ -14,10 +14,11 @@ env:
   BOT_CURSOR: ${{ vars.BOT_CURSOR }}
   DYNAMODB_TABLE_NAME: ${{ vars.DYNAMODB_TABLE_NAME }}
   KNOWLEDGE_BASE_ID: ${{ vars.KNOWLEDGE_BASE_ID }}
+  MAX_LEN_SLACK: ${{ vars.MAX_LEN_SLACK }}
   MODEL_ID_TEXT: ${{ vars.MODEL_ID_TEXT }}
   PERSONAL_MESSAGE: ${{ vars.PERSONAL_MESSAGE }}
-  SYSTEM_MESSAGE: ${{ vars.SYSTEM_MESSAGE }}
   SLACK_SAY_INTERVAL: ${{ vars.SLACK_SAY_INTERVAL }}
+  SYSTEM_MESSAGE: ${{ vars.SYSTEM_MESSAGE }}
 
   AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
   AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
@@ -57,12 +58,13 @@ jobs:
           echo "BOT_CURSOR=${BOT_CURSOR}" >> .env
           echo "DYNAMODB_TABLE_NAME=${DYNAMODB_TABLE_NAME}" >> .env
           echo "KNOWLEDGE_BASE_ID=${KNOWLEDGE_BASE_ID}" >> .env
+          echo "MAX_LEN_SLACK=${MAX_LEN_SLACK}" >> .env
           echo "MODEL_ID_TEXT=${MODEL_ID_TEXT}" >> .env
+          echo "PERSONAL_MESSAGE=${PERSONAL_MESSAGE}" >> .env
           echo "SLACK_BOT_TOKEN=${SLACK_BOT_TOKEN}" >> .env
+          echo "SLACK_SAY_INTERVAL=${SLACK_SAY_INTERVAL}" >> .env
           echo "SLACK_SIGNING_SECRET=${SLACK_SIGNING_SECRET}" >> .env
-          echo "PERSONAL_MESSAGE=${PERSONAL_MESSAGE}" >> .env
           echo "SYSTEM_MESSAGE=${SYSTEM_MESSAGE}" >> .env
-          echo "SLACK_SAY_INTERVAL=${SLACK_SAY_INTERVAL}" >> .env
 
       - name: Deploy to AWS Lambda 🚀
         run: npx serverless deploy --region us-east-1
diff --git a/handler.py b/handler.py
@@ -28,7 +28,7 @@
 
 # Amazon Bedrock Model ID
 ANTHROPIC_VERSION = os.environ.get("ANTHROPIC_VERSION", "bedrock-2023-05-31")
-ANTHROPIC_TOKENS = int(os.environ.get("ANTHROPIC_TOKENS", 1024))
+ANTHROPIC_TOKENS = int(os.environ.get("ANTHROPIC_TOKENS", 2000))
 
 MODEL_ID_TEXT = os.environ.get("MODEL_ID_TEXT", "anthropic.claude-3")
 MODEL_ID_IMAGE = os.environ.get("MODEL_ID_IMAGE", "stability.stable-diffusion-xl")
@@ -42,7 +42,7 @@
 )
 SYSTEM_MESSAGE = os.environ.get("SYSTEM_MESSAGE", "None")
 
-MAX_LEN_SLACK = int(os.environ.get("MAX_LEN_SLACK", 2500))
+MAX_LEN_SLACK = int(os.environ.get("MAX_LEN_SLACK", 3000))
 MAX_LEN_BEDROCK = int(os.environ.get("MAX_LEN_BEDROCK", 4000))
 
 SLACK_SAY_INTERVAL = float(os.environ.get("SLACK_SAY_INTERVAL", 0))
@@ -426,7 +426,7 @@ def conversation(say: Say, thread_ts, query, channel, client_msg_id):
         # Send the prompt to Bedrock
         message = invoke_claude_3(prompt)
 
-        print("conversation: message: {}".format(message))
+        # print("conversation: message: {}".format(message))
 
         # Update the message in Slack
         chat_update(say, channel, thread_ts, latest_ts, message)