Add Google to test_client.py

Author: standsleeping

tests/client/test_client.py

@@ -15,8 +15,10 @@ class TestClient(unittest.TestCase):
     @patch(
         "aisuite.providers.anthropic_provider.AnthropicProvider.chat_completions_create"
     )
+    @patch("aisuite.providers.google_provider.GoogleProvider.chat_completions_create")
     def test_client_chat_completions(
         self,
+        mock_google,
         mock_anthropic,
         mock_azure,
         mock_bedrock,
@@ -31,6 +33,7 @@ def test_client_chat_completions(
         mock_anthropic.return_value = "Anthropic Response"
         mock_groq.return_value = "Groq Response"
         mock_mistral.return_value = "Mistral Response"
+        mock_google.return_value = "Google Response"
 
         # Provider configurations
         provider_configs = {
@@ -50,6 +53,11 @@ def test_client_chat_completions(
             ProviderNames.MISTRAL: {
                 "api_key": "mistral-api-key",
             },
+            ProviderNames.GOOGLE: {
+                "project_id": "test_google_project_id",
+                "region": "us-west4",
+                "application_credentials": "test_google_application_credentials",
+            },
         }
 
         # Initialize the client
@@ -104,6 +112,14 @@ def test_client_chat_completions(
         self.assertEqual(mistral_response, "Mistral Response")
         mock_mistral.assert_called_once()
 
+        # Test Google model
+        google_model = ProviderNames.GOOGLE + ":" + "google-model"
+        google_response = client.chat.completions.create(
+            google_model, messages=messages
+        )
+        self.assertEqual(google_response, "Google Response")
+        mock_google.assert_called_once()
+
         # Test that new instances of Completion are not created each time we make an inference call.
         compl_instance = client.chat.completions
         next_compl_instance = client.chat.completions