Update helper.py

helper.py

@@ -1,34 +1,28 @@
 import jwt
 import uuid
 from datetime import datetime, timedelta
-import random
-import string
-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.
-            - int: A random 6-digit integer between 100000 and 999999.
-    """
-    # Generate a random string of length 6-12
-    random_string = ''.join(random.choices(string.ascii_letters + string.digits, k=random.randint(6, 12)))
-    
-    # Generate a random 6-digit integer
-    random_int = random.randint(100000, 999999)
-    
-    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()
@@ -44,6 +38,4 @@ def create_jwt(github_user_login: str, user_id: int) -> str:
         "plan": "Free"
     }
 
-    # 使用环境变量获取密钥，如果没有设置，则使用默认值
-    secret_key = os.environ.get('JWT_SECRET_KEY', 'llm-secret')
-    return jwt.encode(payload, secret_key, algorithm='HS256')
+    return jwt.encode(payload, 'llm-secret', algorithm='HS256')