Update routes/account_management.py

routes/account_management.py

@@ -6,9 +6,9 @@ router = APIRouter()
 
 
 @router.post("/sign-in-up")
-def handle_sign_in_up(query: Query):
+def handle_sign_in_up(input: Input):
     customer_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'customer_id'), None)
+        (entity['value'] for entity in input.entities if entity['entity'] == 'customer_id'), None)
     if customer_id:
         customer = get_customer(customer_id)
         if not customer:
@@ -21,13 +21,13 @@ def handle_sign_in_up(query: Query):
 
 
 @router.post("/update-account")
-def handle_update_account(query: Query):
+def handle_update_account(input: Input):
     customer_id = next(
-        (entity['value'] for entity in query.entities if entity['entity'] == 'customer_id'), None)
-    updates = {entity['entity']: entity['value'] for entity in query.entities}
+        (entity['value'] for entity in input.entities if entity['entity'] == 'customer_id'), None)
+    updates = {entity['entity']: entity['value'] for entity in input.entities}
     if customer_id:
         update_customer(customer_id, updates)
         return {"message": "Account information updated successfully"}
     else:
         raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND, detail="Customer ID not provided.")
+            status_code=status.HTTP_404_NOT_FOUND, detail="Customer ID not provided.")