Update art_explorer.py

art_explorer.py

@@ -1,7 +1,12 @@
 from typing import Optional, List, Dict
 import json
+import instructor
 from openai import OpenAI
 from prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
+from models import ExplorationResponse
+
+# Enable instructor
+instructor.patch()
 
 class ExplorationPathGenerator:
   def __init__(self, api_key: str):
@@ -25,13 +30,14 @@ class ExplorationPathGenerator:
               "previous_explorations": []
           }
 
-      formatted_prompt = format_exploration_prompt(
-          user_query=user_query,
-          selected_path=selected_path,
-          exploration_parameters=exploration_parameters
-      )
-
       try:
+          formatted_prompt = format_exploration_prompt(
+              user_query=user_query,
+              selected_path=selected_path,
+              exploration_parameters=exploration_parameters
+          )
+
+          # Use instructor to generate and validate the response
           response = self.client.chat.completions.create(
               model="mixtral-8x7b-32768",
               messages=[
@@ -45,19 +51,13 @@ class ExplorationPathGenerator:
                   }
               ],
               temperature=0.7,
-              max_tokens=2000
+              max_tokens=2000,
+              response_model=ExplorationResponse
           )
           
-          result = response.choices[0].message.content
-          try:
-              parsed_result = json.loads(result)
-              print("Generated response:", json.dumps(parsed_result, indent=2))  # Debug print
-              return parsed_result
-          except json.JSONDecodeError as e:
-              print(f"Error parsing JSON response: {e}")
-              print("Raw response:", result)  # Debug print
-              return DEFAULT_RESPONSE
+          # Convert to dict for JSON serialization
+          return response.model_dump()
               
       except Exception as e:
-          print(f"Error in API call: {e}")  # Debug print
+          print(f"Error in API call: {e}")
           return DEFAULT_RESPONSE