fix error: unfound keys

app.py

@@ -68,7 +68,7 @@ def chat(user_input, history):
         valid_response = simulator_app.chat_interaction(user_input, history)
         return valid_response
     else:
-        invalid_response = "Invalid input. Please use the format: <Number>. <Description>: query"
+        invalid_response = "Invalid input. Please use the format: Number. Description: query"
         return invalid_response
 
 

chatbot_simulator.py

@@ -1,6 +1,8 @@
 from openai import OpenAI
 import json_repair
 from transformers import AutoTokenizer
+from openai import RateLimitError
+import time
 
 
 class ChatbotSimulation:
@@ -149,7 +151,8 @@ Example Output Format:
         # Parse and update the user state
         updated_state = json_repair.loads(updated_state)
 
-        while not isinstance(updated_state, dict):
+        required_keys = {'current_page', 'last_page', 'task_completed'}
+        while not isinstance(updated_state, dict) or not required_keys.issubset(updated_state.keys()):
             transform_prompt = f""""
             Transform {updated_state} to a properly formate JSON file.
             Example Output Format: