Refactor

Author: natehouk

.coverage.Nates-Mac-mini.local.19889.XVElrfOx

@@ -19,4 +19,23 @@ To trigger a manual build:
    PINECONE_API_KEY=your-api-key
    PINECONE_INDEX_NAME=audiokit-brain
    ```
-3. Ensure the `.env` file is not committed to version control
+3. Ensure the `.env` file is not committed to version control
+
+## Testing Requirements
+
+To run the tests, you'll need:
+
+1. OpenRouter API Key
+   - Create an account at https://openrouter.ai
+   - Generate an API key
+   - Add it to your `.env` file as `OPENROUTER_API_KEY`
+   - Choose a model and add it as `OPENROUTER_MODEL` (e.g., `openai/gpt-4`)
+
+2. Pinecone API Key
+   - Create an account at https://www.pinecone.io
+   - Generate an API key
+   - Add it to your `.env` file as `PINECONE_API_KEY`
+
+3. Create a Pinecone index
+   - Create an index in your Pinecone dashboard
+   - Add the index name to your `.env` file as `PINECONE_INDEX_NAME`

DEVELOPMENT.md

@@ -37,6 +37,10 @@ def __init__(self):
         logger.debug(f"Pinecone API key: {'*' * 8}{self.pinecone_api_key[-4:]}" if self.pinecone_api_key else "No API key found")
         self.pinecone_index_name = self.get("PINECONE_INDEX_NAME")
 
+        # OpenRouter configuration
+        self.openrouter_api_key = os.getenv("OPENROUTER_API_KEY")
+        self.openrouter_model = os.getenv("OPENROUTER_MODEL", "openai/gpt-4")
+        
         # Create necessary directories
         self._setup_directories()
         logger.debug("Configuration initialized")

audiokit/core/config.py

@@ -12,6 +12,8 @@
 
 from llama_index.core import VectorStoreIndex, Document, StorageContext
 from llama_index.vector_stores.pinecone import PineconeVectorStore
+from llama_index.llms import OpenRouter
+from llama_index.embeddings import OpenRouterEmbedding
 
 from .logging import get_logger
 from .exceptions import IndexingError, ConfigurationError
@@ -49,10 +51,13 @@ def __init__(self):
             storage_context = StorageContext.from_defaults(vector_store=vector_store)
 
             logger.debug("Initializing vector store index")
+            llm = OpenRouter(api_key=config.openrouter_api_key, model=config.openrouter_model)
+            embed_model = OpenRouterEmbedding(api_key=config.openrouter_api_key, model=config.openrouter_model)
+            
             self.index = VectorStoreIndex.from_documents(
-                [],
-                storage_context=storage_context,
-                show_progress=True
+                documents=[],
+                llm=llm,
+                embed_model=embed_model
             )
 
             logger.info("Audio index initialized successfully")

audiokit/core/indexing.py

@@ -48,6 +48,8 @@ addopts = "-v --cov=audiokit --cov-report=term-missing"
 testpaths = [
     "tests",
 ]
+log_cli = true
+log_level = "INFO"
 
 [tool.mypy]
 python_version = "3.12"

pyproject.toml

@@ -9,14 +9,28 @@
 import pytest
 from pathlib import Path
 from dotenv import load_dotenv
+from loguru import logger
 
 # Load environment variables before any tests run
 load_dotenv()
 
 @pytest.fixture(scope="session", autouse=True)
-def load_env():
-    """Ensure environment variables are loaded for all tests."""
-    pass
+def log_api_keys():
+    """Log API keys used in tests."""
+    logger.debug("log_api_keys fixture is running")
+    from audiokit.core.config import config
+    logger.info("Using Pinecone API key: {}", config.pinecone_api_key)
+    logger.info("Using Pinecone index: {}", config.pinecone_index_name)
+    logger.info("Using OpenRouter API key: {}", config.openrouter_api_key)
+    yield
+
+@pytest.fixture(scope="session", autouse=True)
+def setup_openrouter():
+    """Setup OpenRouter configuration for tests."""
+    from audiokit.core.config import config
+    if not config.openrouter_api_key:
+        pytest.skip("OpenRouter API key is required for these tests")
+    yield
 
 @pytest.fixture
 def fixture_dir() -> Path: