Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -7,7 +7,7 @@ from typing import Optional, List, Dict
 import requests
 from bs4 import BeautifulSoup
 from models.models import Message, Attachment
-
+from fastapi import WebSocket
 
 def get_messages(code: str) -> List[Message]:
     access_token = code