Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -16,7 +16,7 @@ def get_company_type(company_name:str,websocket:WebSocket)->str:
     print(company_types_dict["louis vuitton"])
     return company_types_dict.get(company_name.lower(), 'Others')
 
-def get_messages(code: str,brand_name: Optional[str] = None):
+def get_messages(code: str,brand_name: Optional[str] = None,websocket:WebSocket):
     access_token = code
     g_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR subject:aankoopbon OR subject:reçu OR subject:invoice OR subject:invoices OR category:purchases) has:attachment'
     if brand_name is not None: