From b13181730bcee530b57d9ee99c1c89f61d129331 Mon Sep 17 00:00:00 2001
From: Rebecca Le <543859+sevenseacat@users.noreply.github.com>
Date: Fri, 6 Sep 2024 12:24:02 +0800
Subject: [PATCH] docs: Fix examples for builtin `before_action` and
 `after_action` preparations (#1439)

---
 lib/ash/resource/preparation/builtins.ex | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/lib/ash/resource/preparation/builtins.ex b/lib/ash/resource/preparation/builtins.ex
index 9b401458..27e63709 100644
--- a/lib/ash/resource/preparation/builtins.ex
+++ b/lib/ash/resource/preparation/builtins.ex
@@ -39,11 +39,12 @@ defmodule Ash.Resource.Preparation.Builtins do
   @doc ~S"""
   Directly attach a `before_action` function to the query.
 
-  See `Ash.Query.before_action/2` for more information.
+  This function will be called by `Ash.Query.before_action/2`,
+  with an additional `context` argument.
 
   ## Example
 
-      prepare before_action(fn query ->
+      prepare before_action(fn query, _context ->
         Logger.debug("About to execute query for #{query.action.name} on #{inspect(query.resource)}")
 
         query
@@ -63,11 +64,12 @@ defmodule Ash.Resource.Preparation.Builtins do
   @doc ~S"""
   Directly attach an `after_action` function to the query.
 
-  See `Ash.Query.after_action/2` for more information.
+  This function will be called by `Ash.Query.after_action/2`,
+  with an additional `context` argument.
 
   ## Example
 
-      prepare after_action(fn query, records ->
+      prepare after_action(fn query, records, _context ->
         Logger.debug("Query for #{query.action.name} on resource #{inspect(query.resource)} returned #{length(records)} records")
 
         {:ok, records}