Merge pull request #1318 from WolframResearch/feature/minor-fixes

Miscellaneous Fixes

Author: rhennigan

Source/Chatbook/Formatting.wl

@@ -340,11 +340,14 @@ makeResultCell0[ thoughtsOpener[ thoughts_String ] ] :=
         If[ $thinkingEnd === None, $thinkingEnd = AbsoluteTime[ ] ];
         seconds = If[ NumberQ @ $thinkingStart && NumberQ @ $thinkingEnd, Round[ $thinkingEnd - $thinkingStart ] ];
         label = If[ NumberQ @ seconds, trStringTemplate[ "FormattingThinkingComplete" ][ <| "time" -> ToString @ seconds |> ], tr @ "FormattingThinkingCompleteFallback" ];
-        Cell[
-            BoxData @ templateBox[ { StringTrim @ thoughts, label }, "ThoughtsOpener" ],
-            "ThoughtsOpener",
-            Background -> None
-        ]
+        {
+            Cell[
+                BoxData @ templateBox[ { StringTrim @ thoughts, ToBoxes @ label }, "ThoughtsOpener" ],
+                "ThoughtsOpener",
+                Background -> None
+            ],
+            "\n"
+        }
     ];
 
 makeResultCell0[ speechCell[ speech_String ] ] :=

Source/Chatbook/PromptGenerators/RelatedDocumentation.wl

@@ -12,6 +12,7 @@ Needs[ "Wolfram`Chatbook`Common`"                  ];
 (*Configuration*)
 $snippetType                   = "Text";
 $documentationSnippetVersion  := $snippetVersion;
+$snippetFetchBatchSize         = 25;
 $baseURL                       = "https://www.wolframcloud.com/obj/wolframai-content/DocumentationSnippets";
 $documentationSnippetBaseURL  := URLBuild @ { $baseURL, $documentationSnippetVersion, $snippetType };
 $documentationMarkdownBaseURL := URLBuild @ { $baseURL, $documentationSnippetVersion, "Markdown" };
@@ -103,6 +104,7 @@ $RelatedDocumentationSources = Automatic;
 (*RelatedDocumentation*)
 RelatedDocumentation // beginDefinition;
 RelatedDocumentation // Options = {
+    "DownloadBatchSize" -> $snippetFetchBatchSize,
     "FilteredCount"     -> Automatic,
     "FilterResults"     -> Automatic,
     "LLMEvaluator"      -> Automatic,
@@ -173,10 +175,12 @@ RelatedDocumentation[ All, "URIs", Automatic, opts: OptionsPattern[ ] ] := catch
 RelatedDocumentation[ prompt: $$prompt, "Snippets", Automatic, opts: OptionsPattern[ ] ] := catchMine @ Enclose[
     ConfirmMatch[
         (* TODO: filter results *)
-        DeleteMissing @ makeDocSnippets @ vectorDBSearch[
-            getSources[ prompt, OptionValue[ "Sources" ], OptionValue[ "MaxSources" ] ],
-            prompt,
-            "Results"
+        Block[ { $snippetFetchBatchSize = OptionValue[ "DownloadBatchSize" ] },
+            DeleteMissing @ makeDocSnippets @ vectorDBSearch[
+                getSources[ prompt, OptionValue[ "Sources" ], OptionValue[ "MaxSources" ] ],
+                prompt,
+                "Results"
+            ]
         ],
         { ___String },
         "Snippets"
@@ -252,7 +256,8 @@ RelatedDocumentation[ prompt_, "Prompt", n_Integer, opts: OptionsPattern[ ] ] :=
                 OptionValue[ "PromptHeader" ],
                 $$unspecified :> $usePromptHeader
             ],
-            $RelatedDocumentationSources = getSources[ prompt, OptionValue[ "Sources" ], OptionValue[ "MaxSources" ] ]
+            $RelatedDocumentationSources = getSources[ prompt, OptionValue[ "Sources" ], OptionValue[ "MaxSources" ] ],
+            $snippetFetchBatchSize = OptionValue[ "DownloadBatchSize" ]
         },
         relatedDocumentationPrompt[
             ensureChatMessages @ prompt,
@@ -1177,6 +1182,9 @@ fetchDocumentationSnippets // beginDefinition;
 
 fetchDocumentationSnippets[ { } ] := { };
 
+fetchDocumentationSnippets[ uris: { __String } ] /; Length @ uris > $snippetFetchBatchSize :=
+    Flatten[ fetchDocumentationSnippets /@ TakeDrop[ uris, $snippetFetchBatchSize ] ];
+
 fetchDocumentationSnippets[ uris: { __String } ] := Enclose[
      Module[ { count, text, $results, tasks },
         count = Length @ uris;

Source/Chatbook/Sandbox.wl

@@ -86,7 +86,21 @@ $$outputForm := $$outputForm = Alternatives @@ $OutputForms;
 
 $nlpData = <| |>;
 
-$$simpleTemplateBoxName = "Entity"|"EntityClass"|"EntityProperty"|"DateObject"|"Quantity";
+$$simpleTemplateBoxName = Alternatives[
+    "DateObject",
+    "Entity",
+    "EntityClass",
+    "EntityProperty",
+    "Quantity",
+    "QuantityMixedUnit1",
+    "QuantityMixedUnit2",
+    "QuantityMixedUnit3",
+    "QuantityMixedUnit4",
+    "QuantityMixedUnit5",
+    "QuantityMixedUnit6",
+    "QuantityPostfix",
+    "QuantityPrefix"
+];
 
 (* ::**************************************************************************************************************:: *)
 (* ::Section::Closed:: *)

Source/Chatbook/SendChat.wl

@@ -1316,6 +1316,8 @@ $llmAutoCorrectRules := $llmAutoCorrectRules = Flatten @ {
     "\n<|image_sentinel|>\n" :> "\n",
     "<|image_sentinel|>" :> "",
     "paclet:ref/ResourceFunction/" :> "https://resources.wolframcloud.com/FunctionRepository/resources/",
+    "paclet:ref/resource-function/" :> "https://resources.wolframcloud.com/FunctionRepository/resources/",
+    StartOfLine ~~ "/functions." -> "/",
     $longNameCharacters
 };
 

Source/Chatbook/Serialization.wl

@@ -1682,7 +1682,7 @@ boxToString[ TemplateBox[ KeyValuePattern[ "shortenedBoxes" -> boxes_ ], "Output
     boxToString @ boxes;
 
 boxToString[ TemplateBox[ { size_ }, "OutputSizeLimit`Skeleton", ___ ] ] :=
-    " \[LeftSkeleton]" <> boxToString @ size <> "\[RightSkeleton] ";
+    " <<" <> boxToString @ size <> ">> ";
 
 (* Row *)
 boxToString[ TemplateBox[ args_, "RowDefault", ___ ] ] := boxToString @ args;