[*] formatting fixes

Author: kapulkin

aisuite/async_client.py

@@ -51,4 +51,6 @@ async def create(self, model: str, messages: list, **kwargs):
             raise ValueError(f"Could not load provider for '{provider_key}'.")
 
         # Delegate the chat completion to the correct provider's async implementation
-        return await provider.chat_completions_create_async(model_name, messages, **kwargs) 
+        return await provider.chat_completions_create_async(
+            model_name, messages, **kwargs
+        )

aisuite/provider.py

@@ -23,6 +23,7 @@ def chat_completions_create_async(self, model, messages):
         """Abstract method for async chat completion calls, to be implemented by each provider."""
         pass
 
+
 class ProviderFactory:
     """Factory to dynamically load provider instances based on naming conventions."""
 

tests/client/test_client.py

@@ -176,7 +176,7 @@ def test_invalid_model_format_in_create(monkeypatch):
             "aisuite.providers.fireworks_provider.FireworksProvider.chat_completions_create_async",
             "fireworks",
             "fireworks-model",
-        )
+        ),
     ],
 )
 @pytest.mark.asyncio
@@ -194,7 +194,9 @@ async def test_async_client_chat_completions(
         ]
 
         model_str = f"{provider}:{model}"
-        model_response = await client.chat.completions.create(model_str, messages=messages)
+        model_response = await client.chat.completions.create(
+            model_str, messages=messages
+        )
         assert model_response == expected_response
 
 

tests/client/test_prerelease.py

@@ -37,9 +37,10 @@ def get_test_async_models() -> List[str]:
     return [
         "anthropic:claude-3-5-sonnet-20240620",
         "mistral:open-mistral-7b",
-        "openai:gpt-3.5-turbo"
+        "openai:gpt-3.5-turbo",
     ]
 
+
 def get_test_messages() -> List[Dict[str, str]]:
     """Return the test messages to send to each model."""
     return [

tests/providers/test_mistral_provider.py

@@ -74,4 +74,4 @@ async def test_mistral_provider_async():
             temperature=chosen_temperature,
         )
 
-        assert response.choices[0].message.content == response_text_content
+        assert response.choices[0].message.content == response_text_content