From f4339be4265ff4b275e9097dbe525cd7b50d8c52 Mon Sep 17 00:00:00 2001
From: Zach Daniel <zach@zachdaniel.dev>
Date: Mon, 29 Jan 2024 15:56:48 -0500
Subject: [PATCH] chore: fix stream option taking (again)

---
 lib/ash/actions/update/bulk.ex | 3 +--
 lib/ash/api/api.ex             | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/lib/ash/actions/update/bulk.ex b/lib/ash/actions/update/bulk.ex
index 9c89e038..6c20a680 100644
--- a/lib/ash/actions/update/bulk.ex
+++ b/lib/ash/actions/update/bulk.ex
@@ -39,8 +39,7 @@ defmodule Ash.Actions.Update.Bulk do
     case fully_atomic_changeset do
       {:not_atomic, _} ->
         read_opts =
-          opts
-          |> Keyword.take(Ash.Api.stream_opts())
+          Keyword.take(opts, Ash.Api.stream_opt_keys())
 
         read_opts =
           if stream_batch_size = opts[:stream_batch_size] do
diff --git a/lib/ash/api/api.ex b/lib/ash/api/api.ex
index 8584fdc5..ace428a5 100644
--- a/lib/ash/api/api.ex
+++ b/lib/ash/api/api.ex
@@ -199,7 +199,7 @@ defmodule Ash.Api do
                )
 
   @doc false
-  def stream_opts, do: @stream_opts
+  def stream_opt_keys, do: Keyword.keys(@stream_opts)
 
   @doc """
   Streams the results of a query.