fixing CI

Author: miguelg719

tests/unit/llm/test_llm_integration.py

@@ -6,6 +6,7 @@
 
 from stagehand.llm.client import LLMClient
 from tests.mocks.mock_llm import MockLLMClient, MockLLMResponse
+from stagehand.logging import StagehandLogger
 
 
 class TestLLMClientInitialization:
@@ -15,7 +16,8 @@ def test_llm_client_creation_with_openai(self):
         """Test LLM client creation with OpenAI provider"""
         client = LLMClient(
             api_key="test-openai-key",
-            default_model="gpt-4o"
+            default_model="gpt-4o",
+            stagehand_logger=StagehandLogger(),
         )
 
         assert client.default_model == "gpt-4o"
@@ -25,7 +27,8 @@ def test_llm_client_creation_with_anthropic(self):
         """Test LLM client creation with Anthropic provider"""
         client = LLMClient(
             api_key="test-anthropic-key",
-            default_model="claude-3-sonnet"
+            default_model="claude-3-sonnet",
+            stagehand_logger=StagehandLogger(),
         )
 
         assert client.default_model == "claude-3-sonnet"
@@ -35,7 +38,8 @@ def test_llm_client_with_custom_options(self):
         """Test LLM client with custom configuration options"""
         client = LLMClient(
             api_key="test-key",
-            default_model="gpt-4o-mini"
+            default_model="gpt-4o-mini",
+            stagehand_logger=StagehandLogger(),
         )
 
         assert client.default_model == "gpt-4o-mini"