more reliability in processing malformed content, but exeption introduced

test_valid_response.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+"""
+Test the error handling with a valid response.
+"""
+
+import sys
+import os
+import json
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from generate_prompts import JournalPromptGenerator
+
+def test_valid_response():
+    """Test with a valid JSON response."""
+
+    # Create a mock generator
+    generator = JournalPromptGenerator(config_path=".env")
+
+    # Create a valid response with 4 prompts (default num_prompts from settings)
+    valid_response = {
+        "newprompt0": "Write about a time when you felt truly at peace.",
+        "newprompt1": "Describe your ideal morning routine in detail.",
+        "newprompt2": "What are three things you're grateful for today?",
+        "newprompt3": "Reflect on a recent challenge and what you learned from it."
+    }
+
+    # Convert to JSON string
+    json_response = json.dumps(valid_response)
+
+    print("\n=== Test: Valid JSON response ===")
+    result = generator._parse_ai_response(json_response)
+    print(f"Number of prompts extracted: {len(result)}")
+
+    for i, prompt_dict in enumerate(result):
+        prompt_key = list(prompt_dict.keys())[0]
+        prompt_text = prompt_dict[prompt_key]
+        print(f"Prompt {i+1}: {prompt_text[:50]}...")
+
+    # Test with backticks
+    print("\n=== Test: Valid JSON response with backticks ===")
+    backticks_response = f"```json\n{json_response}\n```"
+    result = generator._parse_ai_response(backticks_response)
+    print(f"Number of prompts extracted: {len(result)}")
+
+    # Test with "json" prefix
+    print("\n=== Test: Valid JSON response with 'json' prefix ===")
+    json_prefix_response = f"json\n{json_response}"
+    result = generator._parse_ai_response(json_prefix_response)
+    print(f"Number of prompts extracted: {len(result)}")
+
+if __name__ == "__main__":
+    test_valid_response()
+