Access AgentRunResult.usage as a property in common.ai logging

providers/common/ai/src/airflow/providers/common/ai/utils/logging.py

@@ -36,7 +36,7 @@
 
 def log_run_summary(logger: Logger | logging.Logger, result: AgentRunResult[Any]) -> None:
     """Log model name, token usage, and tool call sequence from an agent run."""
-    usage = result.usage()
+    usage = result.usage
     model_name = getattr(result.response, "model_name", "unknown")
     logger.info(
         "::group::LLM run complete: model=%s, requests=%s, tool_calls=%s, "

providers/common/ai/tests/unit/common/ai/conftest.py

@@ -34,14 +34,12 @@ def isolate_hook_lineage_collector(hook_lineage_collector):
 def make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary.
 
-    Returns a MagicMock with .output, .usage(), .response, and .all_messages()
+    Returns a MagicMock with .output, .usage, .response, and .all_messages()
     configured so that log_run_summary can read them without error.
     """
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/decorators/test_agent.py

@@ -44,9 +44,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/decorators/test_llm.py

@@ -28,9 +28,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/decorators/test_llm_branch.py

@@ -30,9 +30,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/decorators/test_llm_file_analysis.py

@@ -27,7 +27,7 @@
 def _make_mock_run_result(output):
     mock_result = MagicMock(spec=["output", "usage", "response", "all_messages"])
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
+    mock_result.usage = MagicMock(
         spec=["requests", "tool_calls", "input_tokens", "output_tokens", "total_tokens"],
         requests=1,
         tool_calls=0,

providers/common/ai/tests/unit/common/ai/decorators/test_llm_schema_compare.py

@@ -32,9 +32,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/decorators/test_llm_sql.py

@@ -28,9 +28,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/operators/test_agent.py

@@ -48,9 +48,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/operators/test_llm.py

@@ -57,9 +57,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/operators/test_llm_branch.py

@@ -29,9 +29,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/operators/test_llm_file_analysis.py

@@ -46,7 +46,7 @@ class Summary(BaseModel):
 def _make_mock_run_result(output):
     mock_result = MagicMock(spec=["output", "usage", "response", "all_messages"])
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
+    mock_result.usage = MagicMock(
         spec=["requests", "tool_calls", "input_tokens", "output_tokens", "total_tokens"],
         requests=1,
         tool_calls=0,

providers/common/ai/tests/unit/common/ai/operators/test_llm_schema_compare.py

@@ -35,9 +35,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/operators/test_llm_sql.py

@@ -37,9 +37,7 @@ def _make_mock_run_result(output):
     """Create a mock AgentRunResult compatible with log_run_summary."""
     mock_result = MagicMock()
     mock_result.output = output
-    mock_result.usage.return_value = MagicMock(
-        requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0
-    )
+    mock_result.usage = MagicMock(requests=1, tool_calls=0, input_tokens=0, output_tokens=0, total_tokens=0)
     mock_result.response = MagicMock(model_name="test-model")
     mock_result.all_messages.return_value = []
     return mock_result

providers/common/ai/tests/unit/common/ai/utils/test_logging.py

@@ -44,7 +44,7 @@ def _make_mock_result(model_name="gpt-5", tool_names=None, usage_kwargs=None):
         "total_tokens": 3359,
     }
     result = MagicMock()
-    result.usage.return_value = MagicMock(**usage_kwargs)
+    result.usage = MagicMock(**usage_kwargs)
     result.response = MagicMock(model_name=model_name)
 
     messages: list = []