Fix test

jupyterlab · Dec 12, 2024 · b237e96 · b237e96
1 parent fe3ee7d
commit b237e96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/jupyterlab-chat/jupyterlab_chat/ychat.py b/python/jupyterlab-chat/jupyterlab_chat/ychat.py
@@ -109,11 +109,11 @@ def get_messages(self) -> list[Message]:
         message_dicts = self._get_messages()
         return [Message(**message_dict) for message_dict in message_dicts]
 
-    def _get_message_by_index(self, index: int) -> dict:
+    def _get_message_by_index(self, index: int):
         """
         Return a message from its index.
         """
-        return self._ymessages[index].to_py()
+        return self._ymessages[index]
 
     def _get_messages(self) -> list[dict]:
         """