add error handling for test case gen (#1351)

Author: zhang-lucy

13_sandboxes/test_case_generator.py

@@ -152,8 +152,12 @@ def generate(self, file_name: str) -> str:
             },
         )
         output = response.choices[0].message.content
-        output_contents = json.loads(output)["file_contents"]
-        return self.write_outputs(f"test_{file_name}", output_contents)
+        try:
+            output_contents = json.loads(output)["file_contents"]
+            return self.write_outputs(f"test_{file_name}", output_contents)
+        except Exception as e:
+            print(f"Error generating test file: {e}")
+            return None
 
 
 @app.function(