Update art_explorer.py

art_explorer.py

@@ -1,5 +1,4 @@
 from typing import Optional, List, Dict
-import json
 import instructor
 from openai import OpenAI
 from prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
@@ -7,11 +6,16 @@ from models import ExplorationResponse
 
 class ExplorationPathGenerator:
   def __init__(self, api_key: str):
+      # Create a client without instructor
+      client = OpenAI(
+          base_url="https://api.groq.com/openai/v1",
+          api_key=api_key
+      )
+      # Create an instructor client specifically for GROQ
       self.client = instructor.patch(
-          OpenAI(
-              base_url="https://api.groq.com/openai/v1",
-              api_key=api_key
-          )
+          client=client,
+          mode="groq",
+          validate_responses=True
       )
 
   def generate_exploration_path(
@@ -36,8 +40,10 @@ class ExplorationPathGenerator:
               exploration_parameters=exploration_parameters
           )
 
+          # Use the GROQ-specific completion
           response = self.client.chat.completions.create(
               model="mixtral-8x7b-32768",
+              response_model=ExplorationResponse,
               messages=[
                   {
                       "role": "system",
@@ -49,8 +55,7 @@ class ExplorationPathGenerator:
                   }
               ],
               temperature=0.7,
-              max_tokens=2000,
-              response_model=ExplorationResponse
+              max_tokens=2000
           )
           
           return response.model_dump()