refactor: Remove GOOGLE provider

It's deprecated on inference gateway.

Signed-off-by: Eden Reich <[email protected]>

Author: edenreich

inference_gateway/client.py

@@ -10,7 +10,6 @@ class Provider(str, Enum):
     OLLAMA = "ollama"
     GROQ = "groq"
     OPENAI = "openai"
-    GOOGLE = "google"
     CLOUDFLARE = "cloudflare"
     COHERE = "cohere"
 
@@ -68,7 +67,8 @@ def list_models(self) -> List[ProviderModels]:
         return response.json()
 
     def generate_content(self, provider: Provider, model: str, messages: List[Message]) -> Dict:
-        payload = {"model": model, "messages": [msg.to_dict() for msg in messages]}
+        payload = {"model": model, "messages": [
+            msg.to_dict() for msg in messages]}
 
         response = self.session.post(
             f"{self.base_url}/llms/{provider.value}/generate", json=payload

tests/test_client.py

@@ -24,7 +24,8 @@ def test_client_initialization():
     assert client.base_url == "http://test-api"
     assert "Authorization" not in client.session.headers
 
-    client_with_token = InferenceGatewayClient("http://test-api", token="test-token")
+    client_with_token = InferenceGatewayClient(
+        "http://test-api", token="test-token")
     assert "Authorization" in client_with_token.session.headers
     assert client_with_token.session.headers["Authorization"] == "Bearer test-token"
 
@@ -42,7 +43,8 @@ def test_list_models(mock_get, client, mock_response):
 @patch("requests.Session.post")
 def test_generate_content(mock_post, client, mock_response):
     """Test content generation"""
-    messages = [Message(Role.SYSTEM, "You are a helpful assistant"), Message(Role.USER, "Hello!")]
+    messages = [Message(Role.SYSTEM, "You are a helpful assistant"), Message(
+        Role.USER, "Hello!")]
 
     mock_post.return_value = mock_response
     response = client.generate_content(Provider.OPENAI, "gpt-4", messages)
@@ -86,7 +88,6 @@ def test_provider_enum():
     assert Provider.OPENAI == "openai"
     assert Provider.OLLAMA == "ollama"
     assert Provider.GROQ == "groq"
-    assert Provider.GOOGLE == "google"
     assert Provider.CLOUDFLARE == "cloudflare"
     assert Provider.COHERE == "cohere"