Make frame render functions return the frame.

livebook-dev · Oct 3, 2024 · 743cb75 · 743cb75
1 parent e714459
commit 743cb75
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lib/kino/frame.ex b/lib/kino/frame.ex
@@ -78,11 +78,13 @@ defmodule Kino.Frame do
       updates are never a part of frame history
 
   """
-  @spec render(t(), term(), keyword()) :: :ok
+  @spec render(t(), term(), keyword()) :: t()
   def render(frame, term, opts \\ []) do
     opts = Keyword.validate!(opts, [:to, :temporary])
     destination = update_destination_from_opts!(opts)
-    GenServer.call(frame.pid, {:render, term, destination}, :infinity)
+    with :ok <- GenServer.call(frame.pid, {:render, term, destination}, :infinity) do
+      frame
+    end
   end
 
   defp update_destination_from_opts!(opts) do
@@ -118,11 +120,13 @@ defmodule Kino.Frame do
       updates are never a part of frame history
 
   """
-  @spec append(t(), term(), keyword()) :: :ok
+  @spec append(t(), term(), keyword()) :: t()
   def append(frame, term, opts \\ []) do
     opts = Keyword.validate!(opts, [:to, :temporary])
     destination = update_destination_from_opts!(opts)
-    GenServer.call(frame.pid, {:append, term, destination}, :infinity)
+    with :ok <- GenServer.call(frame.pid, {:append, term, destination}, :infinity) do
+      frame
+    end
   end
 
   @doc """