Merge pull request #94 from DL4DS/Farid-patch

apps/ai_tutor/chainlit_app.py

@@ -40,10 +40,6 @@ AGENT = "Agent"
 YOU = "User"
 ERROR = "Error"
 
-COOLDOWN_TIME = config_manager.get_config().dict()["token_config"]["cooldown_time"]
-TOKENS_LEFT = config_manager.get_config().dict()["token_config"]["tokens_left"]
-REGEN_TIME = config_manager.get_config().dict()["token_config"]["regen_time"]
-
 # set config
 config = config_manager.get_config().dict()
 
@@ -397,7 +393,11 @@ class Chatbot:
         if user.metadata["tokens_left"] <= 0 and "admin" not in user.metadata["role"]:
             current_datetime = get_time()
             cooldown, cooldown_end_time = await check_user_cooldown(
-                user, current_datetime, COOLDOWN_TIME, TOKENS_LEFT, REGEN_TIME
+                user,
+                current_datetime,
+                self.config["token_config"]["cooldown_time"],
+                self.config["token_config"]["tokens_left"],
+                self.config["token_config"]["regen_time"],
             )
             if cooldown:
                 # get time left in cooldown