Merge pull request #885 from WolframResearch/feature/enable-rag-for-n…

…ormal-chat-inputs Enable RelatedDocumentation RAG for some personas
WolframResearch · Nov 6, 2024 · bdf10af · bdf10af
2 parents e8ee9cd + dc45180
commit bdf10af
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 16 deletions.
diff --git a/LLMConfiguration/Personas/Birdnardo/LLMConfiguration.wl b/LLMConfiguration/Personas/Birdnardo/LLMConfiguration.wl
@@ -2,9 +2,8 @@
 	"Description"       -> "The one and only Birdnardo",
 	"DisplayName"    	-> Dynamic @ FEPrivate`FrontEndResource[ "ChatbookStrings", "PersonaNameBirdnardo" ],
 	"DocumentationLink" -> "https://resources.wolframcloud.com/PromptRepository/resources/Birdnardo/",
+	"PromptGenerators"  -> { "RelatedDocumentation" },
 	"Tools"             -> {
-		"DocumentationLookup",
-		"DocumentationSearcher",
 		"WolframAlpha",
 		"WolframLanguageEvaluator",
 		"WebSearcher",

diff --git a/LLMConfiguration/Personas/CodeAssistant/LLMConfiguration.wl b/LLMConfiguration/Personas/CodeAssistant/LLMConfiguration.wl
@@ -1,13 +1,8 @@
 <|
-    "BasePrompt"  -> { "Notebooks", "WolframLanguageStyle" },
-    "Description" -> "Help with writing and generating Wolfram Language code",
-    "DisplayName" -> Dynamic @ FEPrivate`FrontEndResource[ "ChatbookStrings", "PersonaNameCodeAssistant" ],
-    "Icon"        -> RawBoxes @ TemplateBox[ { }, "ChatIconCodeAssistant" ],
-    "Tools"       -> {
-        "DocumentationLookup",
-        "DocumentationSearcher",
-        "WolframAlpha",
-        "WolframLanguageEvaluator",
-        ParentList
-    }
+    "BasePrompt"       -> { "Notebooks", "WolframLanguageStyle" },
+    "Description"      -> "Help with writing and generating Wolfram Language code",
+    "DisplayName"      -> Dynamic @ FEPrivate`FrontEndResource[ "ChatbookStrings", "PersonaNameCodeAssistant" ],
+    "Icon"             -> RawBoxes @ TemplateBox[ { }, "ChatIconCodeAssistant" ],
+	"PromptGenerators" -> { "RelatedDocumentation" },
+    "Tools"            -> { "WolframAlpha", "WolframLanguageEvaluator", ParentList }
 |>
diff --git a/LLMConfiguration/Personas/CodeWriter/LLMConfiguration.wl b/LLMConfiguration/Personas/CodeWriter/LLMConfiguration.wl
@@ -2,7 +2,8 @@
     "BasePrompt"        -> { "WolframLanguageStyle" },
     "DisplayName"       -> Dynamic @ FEPrivate`FrontEndResource[ "ChatbookStrings", "PersonaNameCodeWriter" ],
     "Icon"              -> RawBoxes @ TemplateBox[ { }, "ChatIconCodeWriter" ],
-    "Tools"             -> { "DocumentationLookup", "DocumentationSearcher", ParentList },
+	"PromptGenerators"  -> { "RelatedDocumentation" },
+    "Tools"             -> { ParentList },
     "Description"       -> "AI code generation with minimal commentary",
     "DocumentationLink" -> "https://resources.wolframcloud.com/PromptRepository/resources/CodeWriter/"
 |>
diff --git a/LLMConfiguration/Personas/Wolfie/LLMConfiguration.wl b/LLMConfiguration/Personas/Wolfie/LLMConfiguration.wl
@@ -2,9 +2,8 @@
 	"Description"       -> "Wolfram's friendliest AI guide",
 	"DisplayName"       -> Dynamic @ FEPrivate`FrontEndResource[ "ChatbookStrings", "PersonaNameWolfie" ],
 	"DocumentationLink" -> "https://resources.wolframcloud.com/PromptRepository/resources/Wolfie/",
+	"PromptGenerators"  -> { "RelatedDocumentation" },
 	"Tools"             -> {
-		"DocumentationLookup",
-		"DocumentationSearcher",
 		"WolframAlpha",
 		"WolframLanguageEvaluator",
 		"WebSearcher",