From 321fb488cf78f7f8f0c6083ca551f9478d48e3f0 Mon Sep 17 00:00:00 2001
From: Ju4tCode <42488585+yanyongyu@users.noreply.github.com>
Date: Wed, 12 Mar 2025 03:06:07 +0000
Subject: [PATCH 1/2] :bug: fix return type of async function

---
 nonebot/drivers/websockets.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/nonebot/drivers/websockets.py b/nonebot/drivers/websockets.py
index 72251300003a..65dcb6c008f7 100644
--- a/nonebot/drivers/websockets.py
+++ b/nonebot/drivers/websockets.py
@@ -17,10 +17,11 @@
     description: nonebot.drivers.websockets 模块
 """
 
-from collections.abc import AsyncGenerator, Coroutine
+from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 from functools import wraps
 import logging
+from types import CoroutineType
 from typing import TYPE_CHECKING, Any, Callable, TypeVar, Union
 from typing_extensions import ParamSpec, override
 
@@ -47,8 +48,8 @@
 
 
 def catch_closed(
-    func: Callable[P, Coroutine[Any, Any, T]],
-) -> Callable[P, Coroutine[Any, Any, T]]:
+    func: Callable[P, CoroutineType[Any, Any, T]],
+) -> Callable[P, CoroutineType[Any, Any, T]]:
     @wraps(func)
     async def decorator(*args: P.args, **kwargs: P.kwargs) -> T:
         try:

From fb73c0c7304b0303e5b03bbe3ab0c012de729424 Mon Sep 17 00:00:00 2001
From: Ju4tCode <42488585+yanyongyu@users.noreply.github.com>
Date: Wed, 12 Mar 2025 03:16:01 +0000
Subject: [PATCH 2/2] :bug: fix type object subscript

---
 nonebot/drivers/websockets.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nonebot/drivers/websockets.py b/nonebot/drivers/websockets.py
index 65dcb6c008f7..ff9601b05b9d 100644
--- a/nonebot/drivers/websockets.py
+++ b/nonebot/drivers/websockets.py
@@ -48,8 +48,8 @@
 
 
 def catch_closed(
-    func: Callable[P, CoroutineType[Any, Any, T]],
-) -> Callable[P, CoroutineType[Any, Any, T]]:
+    func: Callable[P, "CoroutineType[Any, Any, T]"],
+) -> Callable[P, "CoroutineType[Any, Any, T]"]:
     @wraps(func)
     async def decorator(*args: P.args, **kwargs: P.kwargs) -> T:
         try: