o i@s|ddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z d efd d Zd ee d efddZdS)) HTTPStatus)Optional)Request) HTTPException)logger)SessionManager)SageMakerSessionHeader raw_requestcCs|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 rn/home/ubuntu/.local/lib/python3.10/site-packages/model_hosting_container_standards/sagemaker/sessions/utils.pyget_session_id_from_request s rsession_managercCsXt|}|dur%|dur#tdtjdttjjdtjdddS| |}|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_idsessionrrrrs   rN)httprtypingrfastapirfastapi.exceptionsrlogging_configrmanagerr modelsr rrrrrrs