formatting

Author: patricebechard

src/agentlab/llm/chat_api.py

@@ -110,7 +110,9 @@ def make_model(self):
 class AzureModelArgs(BaseModelArgs):
     """Serializable object for instantiating a generic chat model with an Azure model."""
 
-    deployment_name: str = None  # NOTE: deployment_name is deprecated for Azure OpenAI and won't be used.
+    deployment_name: str = (
+        None  # NOTE: deployment_name is deprecated for Azure OpenAI and won't be used.
+    )
 
     def make_model(self):
         return AzureChatModel(
@@ -405,9 +407,13 @@ def __init__(
         log_probs=False,
     ):
         api_key = api_key or os.getenv("AZURE_OPENAI_API_KEY")
-        assert api_key, "AZURE_OPENAI_API_KEY has to be defined in the environment when using AzureChatModel"
+        assert (
+            api_key
+        ), "AZURE_OPENAI_API_KEY has to be defined in the environment when using AzureChatModel"
         endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
-        assert endpoint, "AZURE_OPENAI_ENDPOINT has to be defined in the environment when using AzureChatModel"
+        assert (
+            endpoint
+        ), "AZURE_OPENAI_ENDPOINT has to be defined in the environment when using AzureChatModel"
 
         if deployment_name is not None:
             logging.info(

tests/llm/test_chat_api.py

@@ -21,7 +21,9 @@
 
 @pytest.mark.pricy
 @pytest.mark.skipif(skip_tests, reason="Skipping on remote as Azure is pricy")
-@pytest.mark.skipif(not os.getenv("AZURE_OPENAI_API_KEY"), reason="Skipping as Azure API key not set")
+@pytest.mark.skipif(
+    not os.getenv("AZURE_OPENAI_API_KEY"), reason="Skipping as Azure API key not set"
+)
 def test_api_model_args_azure():
     model_args = AzureModelArgs(
         model_name="gpt-4.1-nano",
@@ -66,7 +68,9 @@ def test_api_model_args_openai():
 
 @pytest.mark.pricy
 @pytest.mark.skipif(skip_tests, reason="Skipping on remote as Anthropic is pricy")
-@pytest.mark.skipif(not os.getenv("ANTHROPIC_API_KEY"), reason="Skipping as Anthropic API key not set")
+@pytest.mark.skipif(
+    not os.getenv("ANTHROPIC_API_KEY"), reason="Skipping as Anthropic API key not set"
+)
 def test_api_model_args_anthropic():
     model_args = AnthropicModelArgs(
         model_name="claude-3-haiku-20240307",