default to empty dict if no generation_info

jupyterlab · Sep 23, 2024 · 86b77be · 86b77be
1 parent 3977976
commit 86b77be
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 {}