update websocket middleware assignments.

Author: tjholm

nitric/faas.py

@@ -733,7 +733,7 @@ def bucket_notification(
         self.__bucket_notification_handler = compose_middleware(*handlers)
         return self
 
-    def websocket(self, *handlers: WebsocketMiddleware) -> FunctionServer:
+    def websocket(self, *handlers: WebsocketHandler | WebsocketMiddleware) -> FunctionServer:
         """
         Register one or more Websocket Trigger Handlers or Middleware.