Merge pull request #972 from WolframResearch/feature/InstallVectorDatabases

Added `InstallVectorDatabases` as exported symbol

Author: rhennigan

PacletInfo.wl

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

Source/Chatbook/Main.wl

@@ -66,6 +66,7 @@ BeginPackage[ "Wolfram`Chatbook`" ];
 `GetExpressionURIs;
 `GetFocusedNotebook;
 `InlineTemplateBoxes;
+`InstallVectorDatabases;
 `InvalidateServiceCache;
 `ListSavedChats;
 `LoadChat;
@@ -232,6 +233,7 @@ $ChatbookProtectedNames = "Wolfram`Chatbook`" <> # & /@ {
     "GetExpressionURIs",
     "GetFocusedNotebook",
     "InlineTemplateBoxes",
+    "InstallVectorDatabases",
     "ListSavedChats",
     "LoadChat",
     "LogChatTiming",

Source/Chatbook/PromptGenerators/VectorDatabases.wl

@@ -65,6 +65,21 @@ $$vectorDatabase = HoldPattern[ _VectorDatabaseObject? System`Private`ValidQ ];
 $vectorDBSearchCache = <| |>;
 $embeddingCache      = <| |>;
 
+(* ::**************************************************************************************************************:: *)
+(* ::Section::Closed:: *)
+(*InstallVectorDatabases*)
+InstallVectorDatabases // beginDefinition;
+
+InstallVectorDatabases[ ] := catchMine @ Enclose[
+    Success[
+        "VectorDatabasesInstalled",
+        <| "Location" -> ConfirmBy[ getVectorDBDirectory[ ], vectorDBDirectoryQ, "Location" ] |>
+    ],
+    throwInternalFailure
+];
+
+InstallVectorDatabases // endExportedDefinition;
+
 (* ::**************************************************************************************************************:: *)
 (* ::Section::Closed:: *)
 (*Vector Database Utilities*)