default to empty dict if no generation_info

jupyterlab · Sep 25, 2024 · 037ba3d · 037ba3d
1 parent 9c3f169
commit 037ba3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py b/packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py
@@ -23,4 +23,4 @@ def on_llm_end(self, response: LLMResult, **kwargs) -> None:
         if not (len(response.generations) and len(response.generations[0])):
             return
 
-        self.jai_metadata = response.generations[0][0].generation_info
+        self.jai_metadata = response.generations[0][0].generation_info or {}