Apply ruff formatting to parallel tool calling tests

Author: fede-kamel

libs/oci/tests/integration_tests/chat_models/test_parallel_tool_calling_integration.py

@@ -82,9 +82,9 @@ def test_parallel_tool_calling_enabled():
     logging.info("\nQuery: 'What's the weather in New York City?'")
 
     start_time = time.time()
-    response = chat_with_tools.invoke([
-        HumanMessage(content="What's the weather in New York City?")
-    ])
+    response = chat_with_tools.invoke(
+        [HumanMessage(content="What's the weather in New York City?")]
+    )
     elapsed_time = time.time() - start_time
 
     logging.info(f"\nResponse time: {elapsed_time:.2f}s")
@@ -135,9 +135,9 @@ def test_parallel_tool_calling_disabled():
     logging.info("\nQuery: 'What's the weather in New York City?'")
 
     start_time = time.time()
-    response = chat_with_tools.invoke([
-        HumanMessage(content="What's the weather in New York City?")
-    ])
+    response = chat_with_tools.invoke(
+        [HumanMessage(content="What's the weather in New York City?")]
+    )
     elapsed_time = time.time() - start_time
 
     logging.info(f"\nResponse time: {elapsed_time:.2f}s")
@@ -181,14 +181,14 @@ def test_bind_tools_override():
     # Override with True in bind_tools
     chat_with_tools = chat.bind_tools(
         [get_weather, get_population],
-        parallel_tool_calls=True  # Override to enable
+        parallel_tool_calls=True,  # Override to enable
     )
 
     logging.info("\nQuery: 'What's the weather and population of Tokyo?'")
 
-    response = chat_with_tools.invoke([
-        HumanMessage(content="What's the weather and population of Tokyo?")
-    ])
+    response = chat_with_tools.invoke(
+        [HumanMessage(content="What's the weather and population of Tokyo?")]
+    )
 
     logging.info(f"\nResponse content: {response.content}")
     logging.info(f"Tool calls count: {len(response.tool_calls)}")
@@ -219,17 +219,14 @@ def test_cohere_model_error():
     )
 
     # Try to enable parallel tool calls with Cohere (should fail)
-    chat_with_tools = chat.bind_tools(
-        [get_weather],
-        parallel_tool_calls=True
-    )
+    chat_with_tools = chat.bind_tools([get_weather], parallel_tool_calls=True)
 
     logging.info("\nAttempting to use parallel_tool_calls with Cohere model...")
 
     try:
-        _ = chat_with_tools.invoke([
-            HumanMessage(content="What's the weather in Paris?")
-        ])
+        _ = chat_with_tools.invoke(
+            [HumanMessage(content="What's the weather in Paris?")]
+        )
         logging.info("❌ TEST FAILED: Should have raised ValueError")
         return False
     except ValueError as e:
@@ -313,6 +310,7 @@ def main():
     except Exception as e:
         logging.info(f"\n❌ ERROR: {e}")
         import traceback
+
         traceback.print_exc()
         return 1
 

libs/oci/tests/unit_tests/chat_models/test_parallel_tool_calling.py

@@ -1,4 +1,5 @@
 """Unit tests for parallel tool calling feature."""
+
 from unittest.mock import MagicMock
 
 import pytest
@@ -14,7 +15,7 @@ def test_parallel_tool_calls_class_level():
     llm = ChatOCIGenAI(
         model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
         parallel_tool_calls=True,
-        client=oci_gen_ai_client
+        client=oci_gen_ai_client,
     )
     assert llm.parallel_tool_calls is True
 
@@ -24,8 +25,7 @@ def test_parallel_tool_calls_default_false():
     """Test that parallel_tool_calls defaults to False."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
     assert llm.parallel_tool_calls is False
 
@@ -35,8 +35,7 @@ def test_parallel_tool_calls_bind_tools_explicit_true():
     """Test parallel_tool_calls=True in bind_tools."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
 
     def tool1(x: int) -> int:
@@ -47,10 +46,7 @@ def tool2(x: int) -> int:
         """Tool 2."""
         return x * 2
 
-    llm_with_tools = llm.bind_tools(
-        [tool1, tool2],
-        parallel_tool_calls=True
-    )
+    llm_with_tools = llm.bind_tools([tool1, tool2], parallel_tool_calls=True)
 
     assert llm_with_tools.kwargs.get("is_parallel_tool_calls") is True
 
@@ -60,18 +56,14 @@ def test_parallel_tool_calls_bind_tools_explicit_false():
     """Test parallel_tool_calls=False in bind_tools."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
 
     def tool1(x: int) -> int:
         """Tool 1."""
         return x + 1
 
-    llm_with_tools = llm.bind_tools(
-        [tool1],
-        parallel_tool_calls=False
-    )
+    llm_with_tools = llm.bind_tools([tool1], parallel_tool_calls=False)
 
     # When explicitly False, should not set the parameter
     assert "is_parallel_tool_calls" not in llm_with_tools.kwargs
@@ -84,7 +76,7 @@ def test_parallel_tool_calls_bind_tools_uses_class_default():
     llm = ChatOCIGenAI(
         model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
         parallel_tool_calls=True,  # Set class default
-        client=oci_gen_ai_client
+        client=oci_gen_ai_client,
     )
 
     def tool1(x: int) -> int:
@@ -105,7 +97,7 @@ def test_parallel_tool_calls_bind_tools_overrides_class_default():
     llm = ChatOCIGenAI(
         model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
         parallel_tool_calls=True,  # Set class default to True
-        client=oci_gen_ai_client
+        client=oci_gen_ai_client,
     )
 
     def tool1(x: int) -> int:
@@ -124,8 +116,7 @@ def test_parallel_tool_calls_passed_to_oci_api_meta():
     """Test that is_parallel_tool_calls is passed to OCI API for Meta models."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
 
     def get_weather(city: str) -> str:
@@ -139,22 +130,19 @@ def get_weather(city: str) -> str:
         [HumanMessage(content="What's the weather?")],
         stop=None,
         stream=False,
-        **llm_with_tools.kwargs
+        **llm_with_tools.kwargs,
     )
 
     # Verify is_parallel_tool_calls is in the request
-    assert hasattr(request.chat_request, 'is_parallel_tool_calls')
+    assert hasattr(request.chat_request, "is_parallel_tool_calls")
     assert request.chat_request.is_parallel_tool_calls is True
 
 
 @pytest.mark.requires("oci")
 def test_parallel_tool_calls_cohere_raises_error():
     """Test that Cohere models raise error for parallel tool calls."""
     oci_gen_ai_client = MagicMock()
-    llm = ChatOCIGenAI(
-        model_id="cohere.command-r-plus",
-        client=oci_gen_ai_client
-    )
+    llm = ChatOCIGenAI(model_id="cohere.command-r-plus", client=oci_gen_ai_client)
 
     def tool1(x: int) -> int:
         """Tool 1."""
@@ -168,7 +156,7 @@ def tool1(x: int) -> int:
             [HumanMessage(content="test")],
             stop=None,
             stream=False,
-            **llm_with_tools.kwargs
+            **llm_with_tools.kwargs,
         )
 
 
@@ -179,7 +167,7 @@ def test_parallel_tool_calls_cohere_class_level_raises_error():
     llm = ChatOCIGenAI(
         model_id="cohere.command-r-plus",
         parallel_tool_calls=True,  # Set at class level
-        client=oci_gen_ai_client
+        client=oci_gen_ai_client,
     )
 
     def tool1(x: int) -> int:
@@ -194,18 +182,15 @@ def tool1(x: int) -> int:
             [HumanMessage(content="test")],
             stop=None,
             stream=False,
-            **llm_with_tools.kwargs
+            **llm_with_tools.kwargs,
         )
 
 
 @pytest.mark.requires("oci")
 def test_version_filter_llama_3_0_blocked():
     """Test that Llama 3.0 models are blocked from parallel tool calling."""
     oci_gen_ai_client = MagicMock()
-    llm = ChatOCIGenAI(
-        model_id="meta.llama-3-70b-instruct",
-        client=oci_gen_ai_client
-    )
+    llm = ChatOCIGenAI(model_id="meta.llama-3-70b-instruct", client=oci_gen_ai_client)
 
     def tool1(x: int) -> int:
         """Tool 1."""
@@ -220,10 +205,7 @@ def tool1(x: int) -> int:
 def test_version_filter_llama_3_1_blocked():
     """Test that Llama 3.1 models are blocked from parallel tool calling."""
     oci_gen_ai_client = MagicMock()
-    llm = ChatOCIGenAI(
-        model_id="meta.llama-3.1-70b-instruct",
-        client=oci_gen_ai_client
-    )
+    llm = ChatOCIGenAI(model_id="meta.llama-3.1-70b-instruct", client=oci_gen_ai_client)
 
     def tool1(x: int) -> int:
         """Tool 1."""
@@ -239,8 +221,7 @@ def test_version_filter_llama_3_2_blocked():
     """Test that Llama 3.2 models are blocked from parallel tool calling."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-3.2-11b-vision-instruct",
-        client=oci_gen_ai_client
+        model_id="meta.llama-3.2-11b-vision-instruct", client=oci_gen_ai_client
     )
 
     def tool1(x: int) -> int:
@@ -256,10 +237,7 @@ def tool1(x: int) -> int:
 def test_version_filter_llama_3_3_blocked():
     """Test that Llama 3.3 models are blocked from parallel tool calling."""
     oci_gen_ai_client = MagicMock()
-    llm = ChatOCIGenAI(
-        model_id="meta.llama-3.3-70b-instruct",
-        client=oci_gen_ai_client
-    )
+    llm = ChatOCIGenAI(model_id="meta.llama-3.3-70b-instruct", client=oci_gen_ai_client)
 
     def tool1(x: int) -> int:
         """Tool 1."""
@@ -275,8 +253,7 @@ def test_version_filter_llama_4_allowed():
     """Test that Llama 4 models are allowed parallel tool calling."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
 
     def tool1(x: int) -> int:
@@ -294,10 +271,7 @@ def test_version_filter_other_models_allowed():
     oci_gen_ai_client = MagicMock()
 
     # Test with xAI Grok
-    llm_grok = ChatOCIGenAI(
-        model_id="xai.grok-4-fast",
-        client=oci_gen_ai_client
-    )
+    llm_grok = ChatOCIGenAI(model_id="xai.grok-4-fast", client=oci_gen_ai_client)
 
     def tool1(x: int) -> int:
         """Tool 1."""
@@ -313,8 +287,7 @@ def test_version_filter_supports_parallel_tool_calls_method():
     """Test the _supports_parallel_tool_calls method directly."""
     oci_gen_ai_client = MagicMock()
     llm = ChatOCIGenAI(
-        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8",
-        client=oci_gen_ai_client
+        model_id="meta.llama-4-maverick-17b-128e-instruct-fp8", client=oci_gen_ai_client
     )
 
     # Test various model IDs