Merge pull request #1038 from WolframResearch/1001-need-better-error-…

…handling-for-when-cloud-is-unreachable Bugfix: Add error handling for unreachable cloud when checking vector DB downloads
WolframResearch · Jan 21, 2025 · 9f93261 · 9f93261
2 parents b2b4077 + a7dcd9f
commit 9f93261
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/Source/Chatbook/PromptGenerators/VectorDatabases.wl b/Source/Chatbook/PromptGenerators/VectorDatabases.wl
@@ -429,8 +429,22 @@ cleanupLegacyVectorDBFiles // endDefinition;
 (* ::Subsubsection::Closed:: *)
 (*getDownloadSize*)
 getDownloadSize // beginDefinition;
+
 getDownloadSize[ url_String ] := getDownloadSize @ CloudObject @ url;
-getDownloadSize[ obj: $$cloudObject ] := FileByteCount @ obj;
+getDownloadSize[ obj: $$cloudObject ] := getDownloadSize[ obj, FileByteCount @ obj ];
+getDownloadSize[ obj_, size_Integer ] := size;
+
+getDownloadSize[ obj_, $Failed ] := throwFailureToChatOutput @ Failure[
+    "CloudDownloadError",
+    <|
+        "MessageTemplate"   :> Chatbook::CloudDownloadError,
+        "MessageParameters" -> { },
+        "CloudObject"       -> obj,
+        "Evaluation"        -> HoldForm @ FileByteCount @ obj,
+        "Results"           -> $Failed
+    |>
+];
+
 getDownloadSize // endDefinition;
 
 (* ::**************************************************************************************************************:: *)