From 86b77be6d296f952ac7831699f55698c6f1dc57b Mon Sep 17 00:00:00 2001 From: "David L. Qiu" Date: Mon, 23 Sep 2024 15:57:09 -0700 Subject: [PATCH] default to empty dict if no generation_info --- packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py b/packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py index fa4407af..145cab31 100644 --- 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 {}