Update helper.py

helper.py

@@ -3,16 +3,11 @@ import uuid
 from datetime import datetime, timedelta
 import random
 import string
-
-github_username_zed_userid_list = [
-    ("aaa", 123123),
-]
-
+import os
 
 def generate_random_tuple():
     """
     Generate a random tuple containing a string and an integer.
-
     Returns:
         tuple: A tuple containing:
             - str: A random string of length 6-12 consisting of ASCII letters and digits.
@@ -26,28 +21,14 @@ def generate_random_tuple():
     
     return (random_string, random_int)
 
-
 def create_jwt(github_user_login: str, user_id: int) -> str:
     """
     Create a JSON Web Token (JWT) for a given GitHub user.
-
     Args:
         github_user_login (str): The GitHub username of the user.
         user_id (int): The user's ID.
-
     Returns:
         str: A JWT encoded string containing user information and authentication details.
-
-    Note:
-        The token has a lifetime of 1 hour and includes the following claims:
-        - iat: Issued at time
-        - exp: Expiration time
-        - jti: Unique token identifier
-        - userId: User's ID
-        - githubUserLogin: GitHub username
-        - isStaff: Boolean indicating staff status (default: False)
-        - hasLlmClosedBetaFeatureFlag: Boolean for LLM closed beta feature (default: False)
-        - plan: User's plan (default: "Free")
     """
     LLM_TOKEN_LIFETIME = timedelta(hours=1)
     now = datetime.utcnow()
@@ -63,4 +44,6 @@ def create_jwt(github_user_login: str, user_id: int) -> str:
         "plan": "Free"
     }
 
-    return jwt.encode(payload, 'llm-secret', algorithm='HS256')
+    # 使用环境变量获取密钥，如果没有设置，则使用默认值
+    secret_key = os.environ.get('JWT_SECRET_KEY', 'llm-secret')
+    return jwt.encode(payload, secret_key, algorithm='HS256')