fix: make check happy

Signed-off-by: Chojan Shang <psiace@apache.org>

Author: PsiACE

tests/conftest.py

@@ -44,6 +44,7 @@
     McpServerStdio,
     PermissionOption,
     ResourceContentBlock,
+    SessionInfoUpdate,
     SseMcpServer,
     TextContentBlock,
     ToolCallProgress,
@@ -129,6 +130,10 @@ def __init__(self) -> None:
         self.notifications: list[SessionNotification] = []
         self.ext_calls: list[tuple[str, dict]] = []
         self.ext_notes: list[tuple[str, dict]] = []
+        self._agent_conn = None
+
+    def on_connect(self, conn) -> None:
+        self._agent_conn = conn
 
     def queue_permission_cancelled(self) -> None:
         self.permission_outcomes.append(RequestPermissionResponse(outcome=DeniedOutcome(outcome="cancelled")))
@@ -167,7 +172,8 @@ async def session_update(
         | ToolCallProgress
         | AgentPlanUpdate
         | AvailableCommandsUpdate
-        | CurrentModeUpdate,
+        | CurrentModeUpdate
+        | SessionInfoUpdate,
         **kwargs: Any,
     ) -> None:
         self.notifications.append(SessionNotification(session_id=session_id, update=update, field_meta=kwargs or None))