Merge pull request #972 from WolframResearch/feature/InstallVectorDat…

…abases Added `InstallVectorDatabases` as exported symbol
WolframResearch · Dec 6, 2024 · 87764c2 · 87764c2
2 parents fce190d + b0b2f4f
commit 87764c2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/PacletInfo.wl b/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",

diff --git a/Source/Chatbook/Main.wl b/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",

diff --git a/Source/Chatbook/PromptGenerators/VectorDatabases.wl b/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*)