[ENG-850] Large files break logger

examples/logging/large_image_upload.py

@@ -0,0 +1,18 @@
+import openai
+from magentic import SystemMessage, chatprompt
+from magentic.vision import UserImageMessage
+
+from log10.load import log10
+
+
+log10(openai)
+
+with open("./examples/logging/large_image.png", "rb") as f:
+    image_bytes = f.read()
+
+
+@chatprompt(SystemMessage("What's in the following screenshot?"), UserImageMessage(image_bytes))
+def _llm() -> str: ...
+
+
+_llm()

log10/load.py

@@ -705,7 +705,11 @@ def wrapper(*args, **kwargs):
                     while result_queue.empty():
                         pass
                     result = result_queue.get()
-                    completionID = result["completionID"]
+
+                    if result is None:
+                        return output
+
+                    completionID = result.get("completionID")
                     last_completion_response_var.set(result)
 
             with timed_block("result call duration (sync)"):