)`iddlmZmZddlmZddlmZmZddlmZddl m Z ddl m Z dd l mZmZmZmZdd lmZd efd Zd efdZd efdZdS))datetimetimezone) HTTPStatus)RequestResponse) HTTPException)logger)get_session_manager)SESSION_DISABLED_ERROR_DETAILSESSION_DISABLED_LOG_MESSAGESageMakerSessionHeaderSessionRequestType)get_session_id_from_request request_typecb|tjkrtS|tjkrtSdS)zMap session request type to the appropriate handler function. Args: request_type: The type of session request Returns: Handler function for the request type, or None if no handler N)r NEW_SESSIONcreate_sessionCLOSE close_session)rs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/sagemaker/sessions/handlers.pyget_handler_for_request_typers3)555 +1 1 1t raw_requestcLKt|}t}|>tjtt t jjt | |tj d|dtt j jd|dtj|iS#t $rY}tjdt%|t t jjdt%|d}~wwxYw)aAClose an existing session and clean up its resources. Args: raw_request: FastAPI Request object containing session ID in headers Returns: Response with 200 status and closed session ID in headers Raises: HTTPException: If session closure fails with 424 FAILED_DEPENDENCY status N status_codedetailSession z closedrcontentheaderszFailed to close session: )rr r errorrrr BAD_REQUESTvaluer rinforOKrCLOSED_SESSION_ID Exception exceptionstrFAILED_DEPENDENCY)r session_idsession_manageres rrr#s;-[99J)++O 1222".40      %%j111 2z222333" +2z222+=*O       =SVV==>>>"4:7s1vv77     s!AC D# ADD#cKt}|>tjtt t jjt | }|j Jtj |j tjd}tjd|jdt%t jjd|jdt(j|jd|iS#t,$rY}tjd t1|t t jjd t1|d}~wwxYw) aOCreate a new stateful session with expiration tracking. Args: raw_request: FastAPI Request object (unused but part of handler signature) Returns: Response with 200 status, session ID and expiration in headers Raises: HTTPException: If session creation fails with 424 FAILED_DEPENDENCY status Nr)tzz%Y-%m-%dT%H:%M:%SZr z createdz ; Expires=r!zFailed to create session: )r r r$rrrr%r&r r expiration_tsr fromtimestamprutcstrftimer'r.rr(rNEW_SESSION_IDr*r+r,r-)rr/sessionr3r0s rrrGs*++O 1222".40     !0022$000 .  !hl   (' ( (   ;w1;;;<<<" +;w1;;;&5':L7g7gXe7g7g        >c!ff>>???"4:8A88     sB1D E'AE""E'N)rrhttprfastapirrfastapi.exceptionsrlogging_configr managerr modelsr rrrutilsrrrrrrrAs*''''''''%%%%%%%%,,,,,,$$$$$$(((((( /...../A"! W! ! ! ! H' g' ' ' ' ' ' r