L i+>dUddlmZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlZddlmZddlmZmZdd lmZmZmZdd lmZmZdd lmZdd lmZdd lm Z m!Z!ddlm"Z"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1ejde3Z4de5d<dZ6e deejnejpfZ9e dee ee!efZ:d!dZ;d"dZ<Gdde&e9e:fZ=Gdde=ejne efe'Z>Gdd e=ejpe!efe(Z?y)#) annotationsN)AnyUnionMappingTypeVar)Selfoverride) _exceptions)Beta AsyncBeta) NOT_GIVENTimeoutNotGiven)is_dictis_given) model_copy) __version__)Stream AsyncStream)AnthropicErrorAPIStatusError)DEFAULT_MAX_RETRIES BaseClient SyncAPIClientAsyncAPIClientFinalRequestOptionsAWSEventStreamDecoder)Messages AsyncMessages) CompletionsAsyncCompletionszlogging.Loggerlogzbedrock-2023-05-31 _HttpxClientT)bound_DefaultStreamTc4t|d}t|jr}|jjdtt |j rH|j jd}|r+|jjd|jd|jdvr|jdk(rt|js td |jjd d}tjjt!|d }|jjd d}|r d|d|_ n d|d|_ |jj#dr t%d|jdk(r t%d|S)NT)deepanthropic_versionzanthropic-betaanthropic_beta,> /v1/complete /v1/messages/v1/messages?beta=truepostz>GKK OO...."44??1k$$   w'rJ rrrUrryrzr~r{rset_default_headersrset_default_query _extra_kwargsc | | td| | td|j}| i|| }n| | }|j}| i|| }n| | }|jd|xs |j|xs |j |xs |j |xs |j|xs |jt|tr |jn||t|r|n |j||d | Sz~ Create a new client instance re-using the same options given to the current client with optional overriding. zPThe `default_headers` and `set_default_headers` arguments are mutually exclusivezLThe `default_query` and `set_default_query` arguments are mutually exclusive) rrrUrryrzr~r{rrro ValueError_custom_headers _custom_queryrrrrUrry isinstancerrzrr{rgrrrUrryrzr~r{rrrrrr9paramss rHcopyzAnthropicBedrock.copy(  &+>+Jop p  $):)Fkl l&&  &44O4G ,)G##  $00-0F  *&Ft~~  )@T-@-@)@T-@-@!4T__/I43I3I.$.w$ADLLw#'/ '< $BRBR#     rJ)r str | NonerrrUrrrrrrystr | httpx.URL | Nonerz'float | httpx.Timeout | None | NotGivenr{intrMapping[str, str] | NonerMapping[str, object] | Noner~httpx.Client | NonerboolrkNonerkr rFrrkrrz httpx.Requestrkr)rrrrrUrrrryrrz!float | Timeout | None | NotGivenr~rr{int | NotGivenrrrrrrrrrMapping[str, Any]rkrrlrmrn__annotations__rrrr rrIrr with_options __classcell__rs@rHrqrqs J&*%)!%"&(,+/;D.4859+/-2-3-"3-#3- 3-  3- & 3-)3-93-3-23-33-)3-,&*-3-. /3-j''))((*&*%)!%(,+/5>+/&/488<599=+-2 #2 # 2  2 & 2 )2 32 )2 $2 22 62 32 72 )2 !2 lLrJrqc"eZdZUded<ded<ded<ddddddeeddddf dfd Zedd Zedd Z edd Z dddddededdddid ddZ e Z xZ S)AsyncAnthropicBedrockr"rrr$rsrrtNFc `||_||_| tn||_||_||_|t jjd}|d|jd}t |)t|||| | | | t||_ t||_t!||_yrv)rrrWrUrrrNrOr:rrrr"rrr$rsrrtrs rHrzAsyncAnthropicBedrock.__init__s0-,-7-?-/Z&!2  zz~~&BCH  1$//1B.QH #*&#(C  &d+ +D1dO rJctSrrrs rHrz'AsyncAnthropicBedrock._make_sse_decoderKrrJc Kt|Swrrrs rHrIz&AsyncAnthropicBedrock._prepare_optionsOs((s c fKddlm}|jj}||jt |j |j|j|j|j|jxsd|j| }|jj|ywrrrs rHrz&AsyncAnthropicBedrock._prepare_requestSs+||~$$&">>GKK OO...."44??1k$$   w'sB/B1rc | | td| | td|j}| i|| }n| | }|j}| i|| }n| | }|jd|xs |j|xs |j |xs |j |xs |j|xs |jt|tr |jn||t|r|n |j||d | Srrrs rHrzAsyncAnthropicBedrock.copyfrrJ)rrrrrUrrrrrryrrzrr{rrrrrr~httpx.AsyncClient | Nonerrrkrrrr)rrrrrUrrrryrrzrr~rr{rrrrrrrrrrrrkrrrs@rHrrs!! O&*%)!%"&(,+/;D.485904-2-3$"3$#3$ 3$  3$ & 3$)3$93$3$23$33$.3$,&*-3$. /3$j''))((*&*%)!%(,+/5>04&/488<599=+-2 #2 # 2  2 & 2 )2 32 .2 $2 22 62 32 72 )2 !2 lLrJr)rErrkr)rkrC)@ __future__rrNlogging urllib.parser@typingrrrrtyping_extensionsrr httpxr _betar r_typesrrr_utilsrr_compatr_versionr _streamingrrrr _base_clientrrrrr_stream_decoderr resources.messagesr!r"resources.completionsr#r$ getLoggerrlr%rr8Client AsyncClientr&r(rIrWrYrqrrorJrHrs" //, "22'!#-939B'g''1^1&uU\\5CTCT5T/UV +5kRUFV9V3WX@.#E =/#AB#ELK(vc{)BC]K\K-e.?.?SAQ.QRTbKrJ