diff --git a/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py b/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py index 072b0bc23..b6925c1cf 100644 --- a/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py +++ b/packages/jupyter-ai/jupyter_ai/chat_handlers/base.py @@ -147,9 +147,9 @@ def reply( agent_msg = AgentChatMessage( id=uuid4().hex, time=time.time(), - body=response, # append info about error to the string + body=response, reply_to=human_msg.id if human_msg else "", - # error_type=APIAuthenticationError + show_edit_settings=show_edit_settings, ) for handler in self._root_chat_handlers.values():