)`iddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z d efd Zd ee d efd ZdS)) HTTPStatus)Optional)Request) HTTPException)logger)SessionManager)SageMakerSessionHeader raw_requestc\|jsdS|jtjS)zExtract the session ID from the request headers. Args: raw_request: FastAPI Request object Returns: Session ID string if present in headers, None otherwise N)headersgetr SESSION_ID)r s /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/sagemaker/sessions/utils.pyget_session_id_from_requestr s/  t   " "#9#D E EEsession_managerct|}|T|Ptjdtjdt t jjdtjddS| |}|S)aRetrieve the session associated with the request. Args: session_manager: SessionManager instance, or None if sessions disabled raw_request: FastAPI Request object containing session ID in headers Returns: Session instance if found, None if no session ID in request Raises: HTTPException: If session header is present but sessions are not enabled (400 BAD_REQUEST) ValueError: If session_id is not found in the session manager (propagated from get_session) Nz0Invalid payload. stateful sessions not enabled, z header not supported) status_codedetail) rrerrorr rrr BAD_REQUESTvalue get_session)rr session_idsessions rrrs-[99J  ! L{CYCd{{{    &28CJ`JkCCC t))*55G NrN)httprtypingrfastapirfastapi.exceptionsrlogging_configrmanagerr modelsr rrrrr&s,,,,,,$$$$$$######****** FW F F F F.!9r