Cut public .append_message_chunk() method; rename context manager method

Author: cpsievert

shiny/ui/_chat.py

@@ -590,46 +590,8 @@ async def append_message(
             icon=icon,
         )
 
-    async def append_message_chunk(self, message_chunk: Any):
-        """
-        Append a message chunk to the current message stream.
-
-        Append a chunk of message content to either a stream started with
-        `.message_stream()` or an active `.append_message_stream()`.
-
-        Parameters
-        ----------
-        message_chunk
-            A message chunk to inject.
-
-        Note
-        ----
-        A useful pattern for displaying tool calls in a chat is for the tools to display
-        content using an "inner" `.message_stream()` while the response generation is
-        happening in an "outer" `.append_message_stream()`. This allows the inner stream
-        to display "ephemeral" content, then eventually show a final state with
-        `.append_message_chunk(operation="replace")`.
-
-        Raises
-        ------
-        ValueError
-            If there is active stream (i.e., no `.message_stream()` or
-            `.append_message_stream()`)
-        """
-        stream_id = self._current_stream_id
-        if stream_id is None:
-            raise ValueError(
-                "Can't .append_message_chunk() without an active message stream. "
-                "Use .message_stream() or .append_message_stream() to start one."
-            )
-
-        return await self._append_message_chunk(
-            message_chunk,
-            stream_id=stream_id,
-        )
-
     @asynccontextmanager
-    async def message_stream(self):
+    async def append_message_context(self):
         """
         Message stream context manager.
 
@@ -639,13 +601,13 @@ async def message_stream(self):
         Yields
         ------
         :
-            A `MessageStream` instance with a method for `.append()`ing message chunks
-            and a method for `.restore()`ing the stream back to it's initial state.
+            A `MessageStream` class instance, which has a method for `.append()`ing
+            and `.restore()`ing the stream back to it's initial state.
 
         Note
         ----
         A useful pattern for displaying tool calls in a chat interface is for the
-        tool to display using `.message_stream()` while the the response generation
+        tool to display using `.append_message_context()` while the the response generation
         is happening through `.append_message_stream()`. This allows the inner stream
         to display "ephemeral" content, then eventually show a final state
         with `.append_message_chunk(operation="replace")`.
@@ -1493,7 +1455,9 @@ def chat_ui(
 
 
 class MessageStream:
-    """"""
+    """
+    An object to yield from a `.append_message_context()` context manager.
+    """
 
     def __init__(self, chat: Chat, stream_id: str):
         self._chat = chat

tests/playwright/shiny/components/chat/message-stream/app.py

@@ -37,15 +37,15 @@ async def _():
         " Progress: 50%",
         " Progress: 100%",
     ]
-    async with chat.message_stream() as stream:
+    async with chat.append_message_context() as stream:
         for chunk in chunks:
             await stream.append(chunk)
             await asyncio.sleep(SLEEP_TIME)
         await stream.restore()
         await stream.append("Completed stream 1 ✅")
 
 
-# TODO: add test here for nested .message_stream()
+# TODO: add test here for nested .append_message_context()
 
 
 @reactive.effect
@@ -69,7 +69,7 @@ async def mock_tool():
         " Progress: 50%",
         " Progress: 100%",
     ]
-    async with chat.message_stream() as stream:
+    async with chat.append_message_context() as stream:
         for chunk in chunks:
             await stream.append(chunk)
             await asyncio.sleep(SLEEP_TIME)