Update art_explorer.py

art_explorer.py

@@ -1,65 +1,44 @@
-from typing import Optional, List, Dict
-import os
+# art_explorer.py
 import instructor
-from groq import Groq
+from openai import OpenAI
 from pydantic import BaseModel
-from prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
-from models import ExplorationResponse
+from typing import List, Optional
+import json
 
-class ExplorationPathGenerator:
-    def __init__(self, api_key: str):
-        # Initialize GROQ client
-        self.client = Groq(
-            api_key=api_key
-        )
-        # Patch with instructor using TOOLS mode
-        self.client = instructor.from_groq(
-            self.client, 
-            mode=instructor.Mode.TOOLS
-        )
+class ExplorationNode(BaseModel):
+  title: str
+  description: str
+  connections: List[str]
+  depth: int
 
-    def generate_exploration_path(
-        self, 
-        user_query: str, 
-        selected_path: Optional[List[Dict[str, str]]] = None,
-        exploration_parameters: Optional[Dict] = None
-    ) -> Dict:
-        if selected_path is None:
-            selected_path = []
-        if exploration_parameters is None:
-            exploration_parameters = {
-                "depth": 5,
-                "domain": None,
-                "previous_explorations": []
-            }
+class ExplorationPath(BaseModel):
+  nodes: List[ExplorationNode]
+  context: str
 
-        try:
-            formatted_prompt = format_exploration_prompt(
-                user_query=user_query,
-                selected_path=selected_path,
-                exploration_parameters=exploration_parameters
-            )
+class ExplorationPathGenerator:
+  def __init__(self, api_key: str):
+      # Initialize OpenAI client with GROQ's base URL
+      self.client = OpenAI(
+          api_key=api_key,
+          base_url="https://api.groq.com/openai/v1"
+      )
+      # Patch the client with instructor
+      self.client = instructor.patch(self.client)
 
-            response = self.client.chat.completions.create(
-                model="mixtral-8x7b-32768",
-                response_model=ExplorationResponse,
-                messages=[
-                    {
-                        "role": "system",
-                        "content": SYSTEM_PROMPT
-                    },
-                    {
-                        "role": "user",
-                        "content": formatted_prompt
-                    }
-                ],
-                temperature=0.7,
-                max_tokens=2000
-            )
-            
-            return response.model_dump()
-                
-        except Exception as e:
-            print(f"Error in API call: {e}")
-            print(f"Error details: {str(e)}")
-            return DEFAULT_RESPONSE
+  def generate_path(self, query: str, max_depth: int = 3) -> ExplorationPath:
+      try:
+          system_prompt = """You are an art history expert creating exploration paths. 
+          Generate a structured exploration path based on the query, with connected concepts."""
+          
+          response = self.client.chat.completions.create(
+              model="mixtral-8x7b-32768",
+              response_model=ExplorationPath,
+              messages=[
+                  {"role": "system", "content": system_prompt},
+                  {"role": "user", "content": f"Generate an exploration path for: {query}. Maximum depth: {max_depth}"}
+              ]
+          )
+          return response
+      except Exception as e:
+          print(f"Error generating exploration path: {str(e)}")
+          raise