style: fix import sorting and formatting

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: UIPath-Harshit

tests/agent/tools/test_datafabric_query_tool.py

@@ -7,7 +7,9 @@
     _normalize_sql,
     _validate_sql,
 )
-from uipath_langchain.agent.tools.datafabric_tool.models import DataFabricExecuteSqlInput
+from uipath_langchain.agent.tools.datafabric_tool.models import (
+    DataFabricExecuteSqlInput,
+)
 
 
 def test_normalize_sql_strips_single_trailing_semicolon():
@@ -17,8 +19,7 @@ def test_normalize_sql_strips_single_trailing_semicolon():
 
 def test_validate_sql_rejects_multiple_statements():
     assert (
-        _validate_sql("SELECT 1; SELECT 2")
-        == "Multiple SQL statements are not allowed"
+        _validate_sql("SELECT 1; SELECT 2") == "Multiple SQL statements are not allowed"
     )
 
 

tests/agent/tools/test_datafabric_tool.py

@@ -25,7 +25,9 @@ async def test_datafabric_handler_returns_single_terminal_tool_message():
     handler._compiled = _FakeCompiledGraph(  # type: ignore[assignment]
         {
             "messages": [
-                ToolMessage(content="{'records': [1], 'total_count': 1}", tool_call_id="1")
+                ToolMessage(
+                    content="{'records': [1], 'total_count': 1}", tool_call_id="1"
+                )
             ]
         }
     )
@@ -61,7 +63,10 @@ async def test_datafabric_handler_aggregates_multiple_terminal_tool_messages():
     assert "Multiple SQL queries executed successfully." in result
     assert "Result 1:" in result
     assert "Result 2:" in result
-    assert "{'records': [{'id': 1}], 'total_count': 1, 'sql_query': 'SELECT ...'}" in result
+    assert (
+        "{'records': [{'id': 1}], 'total_count': 1, 'sql_query': 'SELECT ...'}"
+        in result
+    )
     assert (
         "{'records': [{'name': 'Acme'}], 'total_count': 1, 'sql_query': 'SELECT ...'}"
         in result
@@ -77,7 +82,9 @@ async def test_datafabric_handler_prefers_terminal_ai_message():
     handler._compiled = _FakeCompiledGraph(  # type: ignore[assignment]
         {
             "messages": [
-                ToolMessage(content="{'records': [], 'total_count': 0}", tool_call_id="1"),
+                ToolMessage(
+                    content="{'records': [], 'total_count': 0}", tool_call_id="1"
+                ),
                 AIMessage(content="I could not find any matching rows."),
             ]
         }