collect some error

chatbot_simulator.py

@@ -136,9 +136,15 @@ Example Output Format:
         # Parse and update the user state
         updated_state = json_repair.loads(updated_state)
 
+        if isinstance(updated_state, dict):
+            pass
+        else:
+            print(updated_state)
+            print(type(updated_state))
+
         try:
             if updated_state['task_completed'].lower() == 'true':
-                return f"Task completed! You took {self.prompt_count} steps."
+                print(f"Task completed! You took {self.prompt_count} steps.")
         except:
             updated_state['task_completed'] = 'False'