Upload app.py

app.py

@@ -68,15 +68,12 @@ async def book_ticket(request: Request):
         "message": f"Ticket booked for '{movie_name}'. Username: {username}, Time: {time}, Price: {price}"
     }
 
-class AgentActionContent(BaseModel):
-    email: str = None
-    from_number: str = None
-    to_number: str
-
 
 class AgentAction(BaseModel):
     name: str
-    content: AgentActionContent
+    email: str = None
+    from_number: str = None
+    to_number: str = None
 
 class Payload(BaseModel):
     agent_action_id: int
@@ -86,19 +83,19 @@ class Payload(BaseModel):
 async def generate_otp(payload: Payload, token: str = Depends(verify_token)):
     otp = random.randint(10000, 99999)
 
-    if payload.agent_action.content.from_number is None:
-        print(f"[AMSService][generate_otp] Generating OTP for email: {payload.agent_action.content.email}")
+    if payload.agent_action.from_number is None:
+        print(f"[AMSService][generate_otp] Generating OTP for email: {payload.agent_action.email}")
         result = {
             "status": "success",
             "verification_code": otp,
-            "message": f"OTP generated for email: {payload.agent_action.content.email}"
+            "message": f"OTP generated for email: {payload.agent_action.email}"
         }
     else:
-        print(f"[AMSService][generate_otp] Generating OTP for phone number: {payload.agent_action.content.from_number}")
+        print(f"[AMSService][generate_otp] Generating OTP for phone number: {payload.agent_action.from_number}")
         result = {
             "status": "success",
             "verification_code": otp,
-            "message": f"OTP generated for phone number: {payload.agent_action.content.from_number}"
+            "message": f"OTP generated for phone number: {payload.agent_action.from_number}"
         }
 
     return result