refactor(all): ♻️ 🤖 💫 Automated AI magic at work

Author: natehouk

.coverage.Nates-Mac-mini.local.70143.XrMSrUyx .coverage.Nates-Mac-mini.local.72960.XBgMWIlx

@@ -0,0 +1 @@
+bun x lint-staged

.husky/pre-commit

@@ -0,0 +1 @@
+bun x devmoji --edit --lint

.husky/prepare-commit-msg

@@ -269,7 +269,7 @@ def search_audio(
         """Search audio analyses with natural language."""
         logger.info("Searching audio with query: {}", query)
 
-        return audio_index.search(query, n_results, filters)
+        return audio_index.search_audio(query, n_results, filters)
 
 # Create global instance
 ak = AudioKit()

audiokit/__init__.py

@@ -0,0 +1,4 @@
+export default {
+  '**/*.ts': () => 'tsc --noEmit',
+  '**/*.{ts,js,md,svelte,css,html,json}': ['eslint --fix', 'prettier --write --list-different'],
+};

cleaned.wav

@@ -42,4 +42,16 @@ def setup_test_env():
     os.environ["AUDIOKIT_LOG_LEVEL"] = "DEBUG"  # Ensure DEBUG level
 
     # Initialize logging
-    get_logger() 
+    get_logger()
+
+@pytest.fixture(autouse=True)
+def change_to_temp_dir(tmp_path, monkeypatch):
+    """
+    Automatically change the current working directory to a temporary subdirectory
+    within the test's tmp_path. This ensures that any files (e.g. .wav files) generated
+    by the tests using relative paths are created in this folder rather than in the project root.
+    """
+    generated_dir = tmp_path / "generated_audio"
+    generated_dir.mkdir()
+    monkeypatch.chdir(generated_dir)
+    print("Current working directory changed to:", os.getcwd()) 

lint-staged.config.mjs

@@ -7,7 +7,6 @@
 from time import time, sleep
 from rich.console import Console  # import Rich's Console to enable recording
 import re
-import io
 
 # Use the package logger consistently
 logger = get_logger(__name__)
@@ -23,38 +22,35 @@ def setup_config():
     os.environ.pop("AUDIOKIT_LOG_LEVEL", None)
     os.environ.pop("AUDIOKIT_LOG_FILE", None)
 
-runner = CliRunner()
-
-class AccumulatingStream(io.StringIO):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.accumulated = []
-
-    def write(self, s):
-        self.accumulated.append(s)
-        return super().write(s)
+@pytest.fixture(autouse=True)
+def patch_console(monkeypatch):
+    """
+    Monkeypatch the global console in audiokit.cli to use a recording Console.
+    """
+    import audiokit.cli as cli  # import the module that created the console
+    recording_console = Console(force_terminal=True, record=True)
+    monkeypatch.setattr(cli, "console", recording_console)
+    return recording_console
 
-    def get_accumulated(self):
-        return ''.join(self.accumulated)
+runner = CliRunner()
 
 def strip_ansi(text: str) -> str:
     # Regular expression for ANSI escape sequences
     ansi_escape = re.compile(r'\x1B\[[0-?]*[ -/]*[@-~]')
     return ansi_escape.sub('', text)
 
-def test_cli_analyze(sample_audio_path):
+def test_cli_analyze(sample_audio_path, patch_console):
     """Test CLI analyze command"""
     logger.info("Testing CLI analyze command")
 
-    # Create an accumulating stream and override the runner's output stream method.
-    accum_stream = AccumulatingStream()
-    original_method = runner._get_output_stream
-    runner._get_output_stream = lambda: accum_stream
     result = runner.invoke(
         app, ["analyze", str(sample_audio_path)], catch_exceptions=False
     )
-    runner._get_output_stream = original_method
-    final_output = accum_stream.get_accumulated()
+    # Allow a brief pause to let all output be recorded.
+    sleep(0.2)
+    final_output = patch_console.export_text(clear=False)
+    if not final_output:
+        final_output = result.stdout
 
     logger.debug("CLI analyze command rendered output: {}", final_output)
     logger.debug("CLI analyze command exit code: {}", result.exit_code)
@@ -75,13 +71,9 @@ def test_cli_analyze(sample_audio_path):
     assert "0.85" in final_output_clean         # Guitar
     assert "0.90" in final_output_clean         # Drums
 
-def test_cli_process(sample_audio_path, tmp_path):
+def test_cli_process(sample_audio_path, tmp_path, patch_console):
     """Test CLI process command"""
     logger.info("Testing CLI process command")
-    # Create an accumulating stream and override the runner's output stream method.
-    accum_stream = AccumulatingStream()
-    original_method = runner._get_output_stream
-    runner._get_output_stream = lambda: accum_stream
     result = runner.invoke(
         app, [
             "process", 
@@ -91,8 +83,10 @@ def test_cli_process(sample_audio_path, tmp_path):
         ],
         catch_exceptions=False
     )
-    runner._get_output_stream = original_method
-    final_output = accum_stream.get_accumulated()
+    sleep(0.2)
+    final_output = patch_console.export_text(clear=False)
+    if not final_output:
+        final_output = result.stdout
 
     logger.debug("CLI process command rendered output: {}", final_output)
     logger.debug("CLI process command exit code: {}", result.exit_code)

tests/conftest.py

@@ -2,6 +2,7 @@
 from audiokit import ak
 from audiokit.core.indexing import audio_index
 from audiokit.core.logging import get_logger
+import json
 
 logger = get_logger(__name__)
 
@@ -13,7 +14,7 @@ def test_index_data(sample_audio_path):
 
     results = audio_index.search_audio("test", 1)
     assert len(results) > 0
-    assert results[0]["content"] == test_data
+    assert json.loads(results[0]["content"]) == test_data
 
 def test_search_audio(sample_audio_path):
     """Test audio search"""

tests/test_cli.py

@@ -0,0 +1,14 @@
+import pytest
+import os
+
+# This fixture automatically changes the working directory for each test.
+# Any output files (e.g. .wav files) generated by the tested code using relative paths
+# will be created in a temporary directory instead of in the project root.
+@pytest.fixture(autouse=True)
+def change_to_temp_dir(tmp_path, monkeypatch):
+    # Optionally, you can create a subdirectory (e.g., "generated_audio") inside tmp_path:
+    generated_dir = tmp_path / "generated_audio"
+    generated_dir.mkdir()
+    monkeypatch.chdir(generated_dir)
+    # Debug: print the current working directory (remove or comment in production)
+    print("Current working directory changed to:", os.getcwd())