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

config.toml

@@ -1,6 +1,6 @@
 [ai]
-provider = "gemini_cli"
+provider = "gemini"
-model = "gemini-2.0-flash"
+model = "gemini-2.5-flash-lite"
 temperature = 0.0
 max_tokens = 8192
 history_trunc_limit = 8000

project_history.toml

@@ -8,5 +8,5 @@ active = "main"
 
 [discussions.main]
 git_commit = ""
-last_updated = "2026-03-05T14:42:31"
+last_updated = "2026-03-05T16:50:57"
 history = []

src/ai_client.py

@@ -685,7 +685,7 @@ def _send_gemini(md_content: str, user_message: str, base_dir: str,
      temperature=_temperature,
      max_output_tokens=_max_tokens,
     )
-    resp = _gemini_chat.send_message(payload, config=config, stream=True)
+    resp = _gemini_chat.send_message_stream(payload, config=config)
     txt_chunks: list[str] = []
     calls = []
     usage = {}

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
+from src import ai_client
-import project_manager
+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_log_registry.py

@@ -4,7 +4,7 @@ import os
 from datetime import datetime, timedelta
 
 # This import is expected to fail with ImportError until log_registry.py is created.
-from log_registry import LogRegistry
+from src.log_registry import LogRegistry
 
 class TestLogRegistry(unittest.TestCase):
 

tests/test_logging_e2e.py

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

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_orchestrator_pm.py

@@ -2,8 +2,8 @@ import unittest
 from typing import Any
 from unittest.mock import patch
 import json
-import orchestrator_pm
+from src import orchestrator_pm
-import mma_prompts
+from src import mma_prompts
 
 class TestOrchestratorPM(unittest.TestCase):
 

tests/test_orchestrator_pm_history.py

@@ -3,7 +3,7 @@ from unittest.mock import patch, MagicMock
 import shutil
 import json
 from pathlib import Path
-import orchestrator_pm
+from src import orchestrator_pm
 
 class TestOrchestratorPMHistory(unittest.TestCase):
  def setUp(self) -> None:

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]:

tests/test_track_state_persistence.py

@@ -3,7 +3,7 @@ from datetime import datetime
 # Import the real models
 from src.models import TrackState, Metadata, Ticket
 # Import the persistence functions from project_manager
-from project_manager import save_track_state, load_track_state
+from src.project_manager import save_track_state, load_track_state
 
 def test_track_state_persistence(tmp_path) -> None:
  """