Refactor

Author: natehouk

.env

@@ -1,6 +1,8 @@
 # Pinecone configuration
-PINECONE_API_KEY=your_pinecone_api_key
-PINECONE_ENV=your_pinecone_environment
+PINECONE_API_KEY=pcsk_44gsvi_TjLpjYMZ6Whmo2xZuH3G7NwWUGT1D57Hjks22fp7egzEAnK6z8q2xmCQ7rSRUAo
+PINECONE_INDEX_NAME=audiokit-brain
+PINECONE_DIMENSION=1536
+PINECONE_METRIC=cosine
 
 # Logging configuration
 AUDIOKIT_LOG_LEVEL=INFO

audiokit/core/config.py

@@ -10,11 +10,11 @@
 from typing import Optional, Dict, Any
 from dotenv import load_dotenv
 
-from .logging import get_logger
-
 # Load environment variables from .env file
 load_dotenv()
 
+# Import logger after other imports to avoid circular dependency
+from .logging import get_logger
 logger = get_logger(__name__)
 
 class Config:
@@ -32,6 +32,11 @@ def __init__(self):
         self.soundcharts_app_id = self.get("SOUNDCHARTS_APP_ID")
         self.soundcharts_api_key = self.get("SOUNDCHARTS_API_KEY")
 
+        # Pinecone configuration
+        self.pinecone_api_key = self.get("PINECONE_API_KEY")
+        self.pinecone_environment = self.get("PINECONE_ENVIRONMENT", "us-west1-gcp")
+        self.pinecone_index_name = self.get("PINECONE_INDEX_NAME", "audiokit-index")
+        
         # Create necessary directories
         self._setup_directories()
         logger.debug("Configuration initialized")

audiokit/core/indexing.py

@@ -10,9 +10,6 @@
 from pathlib import Path
 from datetime import datetime
 
-import pinecone
-from dotenv import load_dotenv
-
 from llama_index.core import VectorStoreIndex, Document, StorageContext
 from llama_index.vector_stores.pinecone import PineconeVectorStore
 
@@ -28,33 +25,21 @@ class AudioIndex:
     def __init__(self):
         """Initialize the audio index."""
         try:
-            # Initialize Pinecone
-            pinecone.init(
-                api_key=config.get("PINECONE_API_KEY"),
-                environment=config.get("PINECONE_ENV")
+            # Initialize Pinecone vector store through LlamaIndex
+            vector_store = PineconeVectorStore(
+                api_key=config.pinecone_api_key,
+                environment=config.pinecone_environment,
+                index_name=config.pinecone_index_name
             )
 
-            # Get or create index
-            index_name = "audiokit"
-            if index_name not in pinecone.list_indexes():
-                pinecone.create_index(
-                    name=index_name,
-                    dimension=1536,  # Default for OpenAI embeddings
-                    metric="cosine"
-                )
-            
-            # Setup vector store
-            pinecone_index = pinecone.Index(index_name)
-            vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
-            
             # Create storage context
             storage_context = StorageContext.from_defaults(vector_store=vector_store)
 
             # Initialize vector store index
             self.index = VectorStoreIndex.from_documents(
                 [],
                 storage_context=storage_context,
-                show_progress=True  # Show progress in the method
+                show_progress=True
             )
 
             logger.info("Audio index initialized successfully")

audiokit/core/logging.py

@@ -10,11 +10,14 @@
 from typing import Optional, Dict, Any
 
 from loguru import logger
-from .config import config
 
 # Remove default handler
 logger.remove()
 
+def get_logger(name: str = None):
+    """Get a logger instance."""
+    return logger.bind(name=name)
+
 def setup_logging(
     log_level: str = "INFO",
     log_file: Optional[str] = None,
@@ -66,18 +69,6 @@ def setup_logging(
             enqueue=True
         )
 
-def get_logger(name: str) -> "logger":
-    """
-    Get a contextualized logger instance.
-    
-    Args:
-        name: Name of the module (usually __name__)
-        
-    Returns:
-        Loguru logger instance with context
-    """
-    return logger.bind(name=name)
-
 # Configure default exception handling
 @logger.catch(onerror=lambda _: sys.exit(1))
 def handle_exception(exc_type, exc_value, exc_traceback):