fix(test): patch google.adk.models.lite_llm module-level imports for proper mocking

Change-Id: Ia5a211bb889e7f62986fa1a163ba47ad14a69f8d
Co-developed-by: Cursor <[email protected]>

Author: OhYee

tests/unittests/integration/test_integration.py

@@ -68,6 +68,19 @@ async def fake_acompletion(*args, **kwargs):
         monkeypatch.setattr("litellm.completion", fake_completion)
         monkeypatch.setattr("litellm.acompletion", fake_acompletion)
 
+        # Also patch the module-level imports in google.adk.models.lite_llm
+        # Google ADK imports acompletion at module level, so we need to patch
+        # there as well to ensure the mock is used in all contexts
+        try:
+            import google.adk.models.lite_llm as lite_llm_module
+
+            monkeypatch.setattr(
+                lite_llm_module, "acompletion", fake_acompletion
+            )
+            monkeypatch.setattr(lite_llm_module, "completion", fake_completion)
+        except ImportError:
+            pass  # google.adk not installed, skip patching
+
     def _setup_respx(self):
         """Setup respx to intercept all httpx requests to mock base URL"""