Fix mypy

Author: brichet

python/jupyterlab-chat/jupyterlab_chat/ychat.py

@@ -370,7 +370,13 @@ async def _set_timestamp(self, msg_idx: int, timestamp: float):
             # should be the last one.
             # The next() function below return the index of the first message with a timestamp inferior of the
             # current one, starting from the end of the list.
-            new_idx = len(self._ymessages) - next((i for i, v in enumerate(self._get_messages()[::-1]) if v.get("time") < timestamp), len(self._ymessages))
+            new_idx = len(self._ymessages) - 1
+            for i, v in enumerate(self._get_messages()[::-1]):
+                new_idx = len(self._ymessages) - i
+                t = v.get("time")
+                if t is not None and (t < timestamp):
+                    break
+
             if msg_idx != new_idx:
                 message = self._ymessages.pop(msg_idx)
                 self._ymessages.insert(new_idx, message)