Check for shared vector database location in cloud

WolframResearch · Dec 17, 2024 · 9a13296 · 9a13296
1 parent 53b74ef
commit 9a13296
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Source/Chatbook/PromptGenerators/VectorDatabases.wl b/Source/Chatbook/PromptGenerators/VectorDatabases.wl
@@ -58,6 +58,7 @@ $vectorDBDownloadURLs = AssociationMap[
 (*Paths*)
 $pacletVectorDBDirectory := FileNameJoin @ { $thisPaclet[ "Location" ], "Assets/VectorDatabases" };
 $localVectorDBDirectory  := ChatbookFilesDirectory[ "VectorDatabases" ];
+$cloudVectorDBDirectory  := PacletObject[ "Wolfram/NotebookAssistantCloudResources" ][ "AssetLocation", "VectorDatabases" ];
 
 (* ::**************************************************************************************************************:: *)
 (* ::Subsection::Closed:: *)
@@ -105,6 +106,7 @@ getVectorDBDirectory[ ] := Enclose[
     $vectorDBDirectory = SelectFirst[
         {
             $pacletVectorDBDirectory,
+            If[ $CloudEvaluation, $cloudVectorDBDirectory, Nothing ],
             $localVectorDBDirectory
         },
         vectorDBDirectoryQ,