fix(tests): Fix import paths and update for google-genai API

tests/test_agent_capabilities.py

@@ -6,7 +6,7 @@ from unittest.mock import patch, MagicMock
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-import ai_client
+from src import ai_client
 
 def test_agent_capabilities_listing() -> None:
  # Mock credentials

tests/test_api_events.py

@@ -1,6 +1,6 @@
 from typing import Any
 from unittest.mock import MagicMock, patch
-import ai_client
+from src import ai_client
 
 class MockUsage:
  def __init__(self) -> None:

tests/test_ast_parser.py

@@ -4,7 +4,7 @@ from src.file_cache import ASTParser
 def test_ast_parser_initialization() -> None:
  """Verify that ASTParser can be initialized with a language string."""
  parser = ASTParser(language="python")
- assert parser.language == "python"
+ assert parser.language.name == "python"
 
 def test_ast_parser_parse() -> None:
  """Verify that the parse method returns a tree_sitter.Tree."""

tests/test_deepseek_infra.py

@@ -7,8 +7,8 @@ import sys
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-import ai_client
-import project_manager
+from src import ai_client
+from src import project_manager
 
 def test_credentials_error_mentions_deepseek(monkeypatch: pytest.MonkeyPatch) -> None:
  """

tests/test_gemini_metrics.py

@@ -7,7 +7,7 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
 # Import the necessary functions from ai_client, including the reset helper
-from ai_client import get_gemini_cache_stats, reset_session
+from src.ai_client import get_gemini_cache_stats, reset_session
 
 def test_get_gemini_cache_stats_with_mock_client() -> None:
  """

tests/test_logging_e2e.py

@@ -2,7 +2,7 @@ import pytest
 from typing import Any
 from pathlib import Path
 from datetime import datetime, timedelta
-import session_logger
+from src import session_logger
 from src.log_registry import LogRegistry
 from log_pruner import LogPruner
 

tests/test_mcp_perf_tool.py

@@ -6,7 +6,7 @@ from unittest.mock import patch
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src")))
 
-import mcp_client
+from src import mcp_client
 
 def test_mcp_perf_tool_retrieval() -> None:
  mock_metrics = {"fps": 60, "last_frame_time_ms": 16.6}

tests/test_session_logging.py

@@ -2,7 +2,7 @@ import pytest
 import tomllib
 from pathlib import Path
 from typing import Generator
-import session_logger
+from src import session_logger
 
 @pytest.fixture
 def temp_logs(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Generator[Path, None, None]: