feat(llm): extract reasoning and tool calls from content blocks langchain v1

Author: Pouyanpi

nemoguardrails/actions/llm/utils.py

@@ -317,10 +317,26 @@ def _extract_and_remove_think_tags(response) -> Optional[str]:
 
 def _store_tool_calls(response) -> None:
     """Extract and store tool calls from response in context."""
-    tool_calls = getattr(response, "tool_calls", None)
+    tool_calls = _extract_tool_calls_from_content_blocks(response)
+    if not tool_calls:
+        tool_calls = _extract_tool_calls_from_attribute(response)
     tool_calls_var.set(tool_calls)
 
 
+def _extract_tool_calls_from_content_blocks(response) -> List | None:
+    if hasattr(response, "content_blocks"):
+        tool_calls = []
+        for block in response.content_blocks:
+            if block.get("type") == "tool_call":
+                tool_calls.append(block)
+        return tool_calls if tool_calls else None
+    return None
+
+
+def _extract_tool_calls_from_attribute(response) -> List | None:
+    return getattr(response, "tool_calls", None)
+
+
 def _store_response_metadata(response) -> None:
     """Store response metadata excluding content for metadata preservation.