i! ddlmZddlmZmZmZmZmZddlm Z  ddl m Z n#e $rd dZ YnwxYwd dZ d Zd Zd Zd Zd ZedZd!dZd"dZd#dZd$dZd%dZdS)&) annotations)LiteralUnionget_argsOptionalAny)Headers)get_current_appreturn Optional[Any]cdSNrZ/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fal_client/_headers.pyr r strcBtx}|j|jSdS)z7Get the current request if we are running in a fal app.N)r current_request)apps r_current_fal_app_requestrs)   -#2E2Q"" 4rzX-Fal-Request-TimeoutzX-Fal-Request-Timeout-TypezX-Fal-Runner-HintzX-Fal-Queue-Priority)normallowtimeoutUnion[int, float]headersdict[str, str]Nonec t|}n #t$rtd|wxYw|tkrtdtdt||t<dS)zV Validates the timeout and adds the timeout header to the headers dictionary. zTimeout must be a number, got zTimeout must be greater than z secondsN)float ValueErrorMIN_REQUEST_TIMEOUT_SECONDSstrREQUEST_TIMEOUT_HEADER)rrs radd_timeout_headerr$"sE.. EEEC'CCDDDE--- Q,G Q Q Q   '*'llG "###s/hintr"c||t<dSr)RUNNER_HINT_HEADER)r%rs radd_hint_headerr(3s"&G rpriorityPrioritycvtt}||vrtd|d|d||t<dS)a& Validates the priority and adds the priority header to the headers dictionary. Args: priority: Queue priority, must be "normal" or "low". headers: Headers dictionary to add the priority header to. Raises: ValueError: If priority is not a valid value. zPriority must be one of z, got ''N)rr*r QUEUE_PRIORITY_HEADER)r)rvalid_prioritiess radd_priority_headerr/7sZ ))''' K'7 K K K K K   &.G !"""rcptx}r#|jdx}r ||d<dSdSdSNzx-fal-cdn-token)rrget)rrequest cdn_tokens radd_fal_app_context_headersr5JsX*,,,w3++,=>> >9 3)2G% & & &33 3 3rresponse_headersr cptx}r#|dx}r||jd<dSdSdSr1)rr2r)r6r3r4s rhandle_response_headersr8PsX*,,,w;(,,->?? ?9 ;1:GO- . . .;; ; ;rN)r r )rrrrr r)r%r"rrr r)r)r*rrr r)rrr r)r6r r r) __future__rtypingrrrrrhttpxr fal.refr ImportErrorrr!r#REQUEST_TIMEOUT_TYPE_HEADERr'r-r*r$r(r/r5r8rrrr?sV""""""::::::::::::::'''''''  1:(. ? #3333"''''....&3333 ;;;;;;s # //