Merge pull request #1036 from WolframResearch/1007-stopchat-can-fail-if-container-for-streaming-content-has-not-been-fully-initialized

Bugfix: Prevent failure in `StopChat` when container hasn't been fully initialized

Author: rhennigan

PacletInfo.wl

@@ -1,7 +1,7 @@
 PacletObject[ <|
     "Name"           -> "Wolfram/Chatbook",
     "PublisherID"    -> "Wolfram",
-    "Version"        -> "2.1.0",
+    "Version"        -> "2.1.1",
     "WolframVersion" -> "14.1+",
     "Description"    -> "Wolfram Notebooks + LLMs",
     "License"        -> "MIT",

Source/Chatbook/Actions.wl

@@ -739,8 +739,8 @@ StopChat[ cell0_CellObject ] := Enclose[
         settings = ConfirmMatch[ currentChatSettings @ cell, _Association|_Missing, "ChatNotebookSettings" ];
         If[ MissingQ @ settings, finish[ ] ];
         removeTask @ Lookup[ settings, "Task", None ];
-        container = ConfirmMatch[ Lookup[ settings, "Container", None ], _Association|None, "Container" ];
-        If[ container === None, finish[ ] ];
+        container = ConfirmMatch[ Lookup[ settings, "Container", None ], _Association|_Symbol, "Container" ];
+        If[ MatchQ[ container, _Symbol ], finish[ ] ];
         content = ConfirmMatch[ Lookup[ container, "FullContent" ], _String|$$progressIndicator, "Content" ];
         FinishDynamic[ ];
         Block[ { createFETask = # & }, writeReformattedCell[ settings, content, cell ] ]