Major update to support additional persona customizations

Developer/Resources/Styles.wl

@@ -133,7 +133,7 @@ Cell[
     CounterIncrements -> { "ChatInputCount" },
     Evaluatable       -> True,
     MenuSortingValue  -> 1543,
-    StyleKeyMapping   -> { "~" -> "ChatDelimiter", "'" -> "SideChat" },
+    StyleKeyMapping   -> { "~" -> "ChatDelimiter", "'" -> "SideChat", "=" -> "WolframAlphaShort", "*" -> "Item" },
     TaggingRules      -> <| "ChatNotebookSettings" -> <| |> |>,
 	CellDingbat -> Cell[
         BoxData @ DynamicBox @ ToBoxes[

LLMConfiguration/Personas/RawModel/LLMConfiguration.wl

@@ -2,5 +2,6 @@
     "BasePrompt"  -> None,
 	"DisplayName" -> "Raw Model",
     "Icon"        -> RawBoxes @ TemplateBox[ { }, "PersonaRawModel" ],
-    "Description" -> "No custom prompting, just the raw LLM"
+    "Description" -> "No custom prompting, just the raw LLM",
+    "Tools"       -> None
 |>

PacletInfo.wl

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

Source/Chatbook/Actions.wl

@@ -17,8 +17,8 @@ BeginPackage[ "Wolfram`Chatbook`Actions`" ];
 `WidgetSend;
 
 `$alwaysOpen;
-`$autoAssistMode;
 `$autoOpen;
+`$chatState;
 `$finalCell;
 `$lastCellObject;
 `$lastChatString;
@@ -34,6 +34,7 @@ BeginPackage[ "Wolfram`Chatbook`Actions`" ];
 `standardizeMessageKeys;
 `systemCredential;
 `toAPIKey;
+`withChatState;
 
 Begin[ "`Private`" ];
 
@@ -102,6 +103,7 @@ ChatbookAction[ "EvaluateChatInput"    , args___ ] := catchMine @ EvaluateChatIn
 ChatbookAction[ "ExclusionToggle"      , args___ ] := catchMine @ ExclusionToggle @ args;
 ChatbookAction[ "ExplodeDuplicate"     , args___ ] := catchMine @ ExplodeDuplicate @ args;
 ChatbookAction[ "ExplodeInPlace"       , args___ ] := catchMine @ ExplodeInPlace @ args;
+ChatbookAction[ "InsertCodeBelow",       args___ ] := catchMine @ insertCodeBelow @ args;
 ChatbookAction[ "InsertInlineReference", args___ ] := catchMine @ InsertInlineReference @ args;
 ChatbookAction[ "OpenChatBlockSettings", args___ ] := catchMine @ OpenChatBlockSettings @ args;
 ChatbookAction[ "OpenChatMenu"         , args___ ] := catchMine @ OpenChatMenu @ args;
@@ -404,11 +406,12 @@ EvaluateChatInput[ source: _CellObject | $Failed ] :=
         EvaluateChatInput @ evalCell /; chatInputCellQ @ evalCell
     ];
 
+(* TODO: resolve auto settings here and set as a Block symbol *)
 EvaluateChatInput[ evalCell_CellObject, nbo_NotebookObject ] :=
-    EvaluateChatInput[ evalCell, nbo, currentChatSettings @ evalCell ];
+    withChatState @ EvaluateChatInput[ evalCell, nbo, resolveAutoSettings @ currentChatSettings @ evalCell ];
 
 EvaluateChatInput[ evalCell_CellObject, nbo_NotebookObject, settings_Association? AssociationQ ] :=
-    withChatState @ Block[ { $autoAssistMode = False, $aborted = False },
+    withChatState @ Block[ { $AutomaticAssistance = False, $aborted = False },
         $lastCellObject     = None;
         $lastChatString     = None;
         $lastMessages       = None;
@@ -594,9 +597,15 @@ AIAutoAssist[ cell_CellObject ] :=
     ];
 
 AIAutoAssist[ cell_CellObject, nbo_NotebookObject ] := withChatState @
-    If[ autoAssistQ[ cell, nbo ],
-        Block[ { $autoAssistMode = True }, needsBasePrompt[ "AutoAssistant" ]; SendChat @ cell ],
-        Null
+    Module[ { settings },
+        settings = resolveAutoSettings @ currentChatSettings @ cell;
+        If[ autoAssistQ @ settings,
+            Block[ { $AutomaticAssistance = True },
+                needsBasePrompt[ "AutoAssistant" ];
+                SendChat[ cell, parentNotebook @ cell, settings ]
+            ],
+            Null
+        ]
     ];
 
 AIAutoAssist // endDefinition;
@@ -606,6 +615,9 @@ AIAutoAssist // endDefinition;
 (*autoAssistQ*)
 autoAssistQ // beginDefinition;
 
+autoAssistQ[ settings_Association ] :=
+    TrueQ @ settings[ "Assistance" ];
+
 autoAssistQ[ cell_CellObject, nbo_NotebookObject ] :=
     autoAssistQ @ cell;
 
@@ -1056,7 +1068,7 @@ excludeChatCells // endDefinition;
 WidgetSend // beginDefinition;
 
 WidgetSend[ cell_CellObject ] := withChatState @
-    Block[ { $alwaysOpen = True, cellPrint = cellPrintAfter @ cell, $finalCell = cell, $autoAssistMode = True },
+    Block[ { $alwaysOpen = True, cellPrint = cellPrintAfter @ cell, $finalCell = cell, $AutomaticAssistance = True },
         (* TODO: this is currently the only UI method to turn this back on *)
         CurrentValue[ parentNotebook @ cell, { TaggingRules, "ChatNotebookSettings", "Assistance" } ] = True;
         SendChat @ cell
@@ -1128,7 +1140,7 @@ SendChat[ evalCell_CellObject ] := SendChat[ evalCell, parentNotebook @ evalCell
 SendChat[ evalCell_CellObject, nbo_NotebookObject? queuedEvaluationsQ ] := Null;
 
 SendChat[ evalCell_CellObject, nbo_NotebookObject ] :=
-    SendChat[ evalCell, nbo, currentChatSettings @ evalCell ];
+    withChatState @ SendChat[ evalCell, nbo, resolveAutoSettings @ currentChatSettings @ evalCell ];
 
 SendChat[ evalCell_CellObject, nbo_NotebookObject, settings_Association? AssociationQ ] :=
     SendChat[ evalCell, nbo, settings, Lookup[ settings, "ShowMinimized", Automatic ] ];
@@ -1377,8 +1389,15 @@ $apiKeyDialogDescription := $apiKeyDialogDescription = Get @ FileNameJoin @ {
 withChatState // beginDefinition;
 withChatState // Attributes = { HoldFirst };
 
+(* TODO: create a `$CurrentChatSettings` symbol that's scoped here and defined as soon as settings are resolved *)
 withChatState[ eval_ ] :=
-    Block[ { $enableLLMServices },
+    Block[
+        {
+            $AutomaticAssistance = False,
+            $chatState           = True,
+            $enableLLMServices   = Automatic,
+            withChatState        = # &
+        },
         $ChatHandlerData = <| |>;
         withToolBox @ withBasePromptBuilder @ eval
     ];

Source/Chatbook/ChatHistory.wl

@@ -16,6 +16,7 @@ HoldComplete[
 Begin[ "`Private`" ];
 
 Needs[ "Wolfram`Chatbook`"              ];
+Needs[ "Wolfram`Chatbook`Actions`"      ];
 Needs[ "Wolfram`Chatbook`ChatMessages`" ];
 Needs[ "Wolfram`Chatbook`Common`"       ];
 Needs[ "Wolfram`Chatbook`FrontEnd`"     ];
@@ -66,7 +67,7 @@ GetChatHistory[ cell_CellObject, "CellObjects" ] :=
     catchMine @ getCellsInChatHistory @ cell;
 
 GetChatHistory[ cell_CellObject, property: $$historyProperty ] := catchMine @ Enclose[
-    Module[ { cells, data, as },
+    withChatState @ Module[ { cells, data, as },
         { cells, data } = Reap[ getCellsInChatHistory @ cell, $chatHistoryTag ];
         ConfirmMatch[ cells, { ___CellObject }, "CellObjects" ];
         as = ConfirmBy[ <| data, "CellObjects" -> cells |>, AssociationQ, "Data" ];