~iO"UddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZddlZddlmZmZmZddlmZddlmZmZdd lm Z dd l!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*dd l+m,Z,m-Z-m.Z.er dd l/m0Z0ddl1m2Z2edZ3e dZ4e dZ5e ddZ6e ddZ7e de e'e e(e fZ8ej9e:Z;deGdde=e4Z?Gd d!e@ZAGd"d#e ZBd4d(ZCd5d+ZDd6d-ZEd7d/ZFGd0d1e e6ZGGd2d3e e7ZHdS)8) annotationsN) TracebackType) AnyUnionGenericTypeVarIterator AsyncIteratorcastCallable Awaitable TYPE_CHECKINGoverload) ParamSpecoverride get_origin)NoneType)CLIENT_REQUEST_HEADERRAW_RESPONSE_HEADER)ArkError)is_givenextract_type_argis_annotated_typeis_type_alias_type)Stream AsyncStreamis_stream_class_typeextract_stream_chunk_type) BaseModel is_basemodeladd_request_id)RequestOptions) BaseClientPR_T _APIResponseTzArkAPIResponse[Any])bound_AsyncAPIResponseTzArkAsyncAPIResponse[Any]_DefaultStreamTzlogging.LoggerlogcfeZdZUded<ded<ded<ded<d ed <d ed <d ed<ded<d/dZed0dZed1dZed2dZed3d Z ed4d"Z ed0d#Z ed0d$Z ed5d&Z ed6d'Zed0d(Zd)d*d7d.Zd)S)8BaseAPIResponsetype[R]_cast_tor$_clientzdict[type[Any], Any]_parsed_by_typebool_is_sse_stream1type[Stream[Any]] | type[AsyncStream[Any]] | None _stream_clsstr _request_idr#_optionshttpx.Response http_responserawcast_toclientstream stream_clsoptionsreturnNonech||_||_i|_||_||_||_||_dSN)r0r1r2r4r6r;r9)selfr<r=r>r?r@rAs j/root/.openclaw/workspace/.venvs/ark-sdk/lib/python3.11/site-packages/volcenginesdkarkruntime/_response.py__init__zBaseAPIResponse.__init__Is=   !$%  cJ|jjtSrEr;headersgetrrFs rG request_idzBaseAPIResponse.request_id[s!)--.CDDDrI httpx.Headersc|jjSrE)r;rLrNs rGrLzBaseAPIResponse.headers_!))rI httpx.Requestc|jjSrE)r;requestrNs rG http_requestzBaseAPIResponse.http_requestcrRrIintc|jjSrE)r; status_coderNs rGrYzBaseAPIResponse.status_codegs!--rI httpx.URLc|jjSrE)r;urlrNs rGr\zBaseAPIResponse.urlks!%%rIc|jjSrE)rVmethodrNs rGr^zBaseAPIResponse.methodos ''rIc|jjSrE)r; http_versionrNs rGr`zBaseAPIResponse.http_versionss!..rIdatetime.timedeltac|jjSrE)r;elapsedrNs rGrczBaseAPIResponse.elapsedwrRrIc|jjSrE)r; is_closedrNs rGrezBaseAPIResponse.is_closed{s!++rIc \d|jjd|jd|jjd|jd S)N) __class____name__rYr; reason_phraser0rNs rG__repr__zBaseAPIResponse.__repr__s=z4>*zzd.>zzASAazzjnjwzzzzrINtorotype[_T] | NoneR | _Tcf ||n|j}t|r|j}|rt|rt |d}t |p|}|jrF|r{t|stdtdttt|t|d|jtt|jS|jrZtt$|t|j|jtt|jStd|jj}|t)tt$|||jtt|jS|t*urtt$dS|j}|t,krtt$|jS|t0krtt$|jS|t4kr'tt$t5|jS|t6kr'tt$t7|jS|t8kr0tt$|jdkSt=j|r)|j d krtt$||S|tBkrtEd t=j|rNtG|tHj%r4|tHj%krtMd tt$|St=j|r>tG|tNs)tG|tPj'rtd |tRur|tTurw|tVurn|tXuretG|tNsPtEd |dtNdtVdtTdtXdt*dt,dtHj%d|j-.dd/d^}}|dkrta|rr |1}|j2|||S#tf$r3} th5dtm| | Yd} ~ nd} ~ wwxYw|jS|1}|j2|||S)Nrz/Expected custom parse type to be a subclass of z or zUExpected custom stream type to be passed with a type argument, e.g. Stream[ChunkType])failure_message)r=responser>r5trueHttpxBinaryResponseContentz+Unexpected state - cast_to is `APIResponse`z:Subclasses of httpx.Response cannot be passed to `cast_to`z1Pydantic models must subclass our base model typezUnsupported type, expected z to be a subclass of z, .z content-type*;zapplication/json)datar=rtz5Could not read JSON from response data due to %s - %s)7r0r __value__rrrr4r TypeErrorrrr r'rr;rr1r6r&_default_stream_clsMissingStreamClassErrorrr7textbytescontentrWfloatr3lowerinspectisclassrkArkAPIResponse RuntimeError issubclasshttpxResponse ValueErrorr pydanticobjectlistdictrrLrMsplitr!json_process_response_data Exceptionr,debugtype) rFror=originr@rt content_type_rzexcs rG_parsezBaseAPIResponse._parses""DM g & & ('G  3(11 3&w22GG$$/  +  +B//#c&ccVaccB 9-D!!!"&!3#C66     $$ 9$:J K K!%!3#C66%C 0J!-/// #!/T\22  h  4== % c>>8=)) ) e  8+,, , c>>3x}--.. . e  5//00 0 d??8=..00F:;; ; OG $ $ . $@@@778,,-- - ^ # #LMM M ?6 " " %z&%.'I'I % %.(( P8$$ $ O   Qvy11  Q 68#566  QOPP P 6 ! !d""d""e##vy11$YgYYIYYY]YYaeYYinYYrzYYBYYHMHVYYY  $+//DDJJ3OO q - - -G$$  #==??D <>>! '!)?!IIOS  = }}|223   s R;; S8)S33S8)r<r:r=r/r>r$r?r3r@r5rAr#rBrCrBr7)rBrP)rBrS)rBrW)rBrZ)rBra)rBr3rorprBrq)rk __module__ __qualname____annotations__rHpropertyrOrLrVrYr\r^r`rcrerrmrrIrGr.r.>s))))BBBB!!!!    $EEEXE***X****X*...X.&&&X&(((X(///X/***X*,,,X,{{{X{/3V V V V V V V V rIr.cpeZdZeddZdddd Zdd Zdd ZddZddZ dddZ dddZ d dZ dS)!rrBr7cL|jjtdSNrKrNs rGrOzArkAPIResponse.request_id!)--.CRHHHrINrnrorprqc||n|j}|j|}||S|js|||}t |jjr|j|}t|trt||j ||j|<tt|S)aZReturns the rich python representation of this response's data. For lower-level control, see `.read()`, `.json()`, `.iter_bytes()`. You can customise the type that the response is parsed into through the `to` argument, e.g. ```py class MyModel(BaseModel): foo: str obj = response.parse(to=MyModel) print(obj.foo) ``` We support parsing: - `BaseModel` - `dict` - `list` - `Union` - `str` - `int` - `float` - `httpx.Response` Nrnr0r2rMr4readrrr9 post_parser isinstancer r"rOr r&rFro cache_keycachedparseds rGparsezArkAPIResponse.parse!s6.BBdm %)))44  M"  IIKKK## DM- . . 7]..v66F fi ( ( 4 64? 3 3 3*0Y'AvrIrc |jS#tj$r}t |d}~wwxYwrE)r;rrStreamConsumedStreamAlreadyConsumedrFrs rGrzArkAPIResponse.readNsL 3%**,, ,# 3 3 3'))s 2 3s>9>cB||jjSrErr;rrNs rGrzArkAPIResponse.textTs !&&rIrc\||jSrErr;rrNs rGrzArkAPIResponse.jsonXs$ !&&(((rIrCc8|jdSrE)r;closerNs rGrzArkAPIResponse.close\s   """""rI chunk_size int | NoneIterator[bytes]c#LK|j|D]}|VdSrE)r; iter_bytesrFrchunks rGrzArkAPIResponse.iter_bytes_s;'22:>>  EKKKK  rI Iterator[str]c#LK|j|D]}|VdSrE)r; iter_textrs rGrzArkAPIResponse.iter_textcs;'11*==  EKKKK  rIc#JK|jD]}|VdSrE)r; iter_linesrFrs rGrzArkAPIResponse.iter_linesgs9'2244  EKKKK  rIrrrBrrBrrBrCrE)rrrBr)rrrBr)rBr) rkrrrrOrrrrrrrrrrIrGrrs IIIXI.2++++++Z3333 ''''))))####rIrceZdZeddZeddZedd Zd d d dZd!dZddZd"dZ d#dZ d$d%dZ d$d&dZ d'dZ d S)(ArkAsyncAPIResponserBr7cL|jjtdSrrKrNs rGrOzArkAsyncAPIResponse.request_idmrrIrotype[_T]r'c KdSrEr)rFros rGrzArkAsyncAPIResponse.parseqs 25#rIr&c KdSrErrNs rGrzArkAsyncAPIResponse.parsets #rINrnrprqcK||n|j}|j|}||S|js|d{V||}t |jjr|j|}t|trt||j ||j|<tt|S)a4Returns the rich python representation of this response's data. For lower-level control, see `.read()`, `.json()`, `.iter_bytes()`. You can customise the type that the response is parsed into through the `to` argument, e.g. ```py class MyModel(BaseModel): foo: str obj = response.parse(to=MyModel) print(obj.foo) ``` We support parsing: - `BaseModel` - `dict` - `list` - `Union` - `str` - `httpx.Response` Nrnrrs rGrzArkAsyncAPIResponse.parsews2.BBdm %)))44  M" ))++       ## DM- . . 7]..v66F fi ( ( 4 64? 3 3 3*0Y'AvrIrcK |jd{VS#tj$r}t |d}~wwxYw)z,Read and return the binary response content.N)r;areadrrrrs rGrzArkAsyncAPIResponse.readsd 3+1133333333 3# 3 3 3())s 2  3s#AAAcRK|d{V|jjS)z3Read and decode the response content into a string.NrrNs rGrzArkAsyncAPIResponse.texts2iikk!&&rIrclK|d{V|jS)z*Read and decode the JSON response content.NrrNs rGrzArkAsyncAPIResponse.jsons;iikk!&&(((rIrCcHK|jd{VdS)zClose the response and release the connection. Automatically called if the response body is read to completion. N)r;acloserNs rGrzArkAsyncAPIResponse.closes5  '')))))))))))rIrrAsyncIterator[bytes]cZK|j|23d{V}|WV 6dS)z A byte-iterator over the decoded response content. This automatically handles gzip, deflate and brotli encoded responses. N)r; aiter_bytesrs rGrzArkAsyncAPIResponse.iter_bytessZ  -99*EE       %KKKKKFEE*AsyncIterator[str]cZK|j|23d{V}|WV 6dS)zA str-iterator over the decoded response content that handles both gzip, deflate, etc but also detects the content's string encoding. N)r; aiter_textrs rGrzArkAsyncAPIResponse.iter_textsZ  -88DD       %KKKKKEDDrcXK|j23d{V}|WV 6dS)z;Like `iter_text()` but will only yield chunks for each lineN)r; aiter_linesrs rGrzArkAsyncAPIResponse.iter_linessU-99;;       %KKKKK<;;s)r)rorrBr')rBr&rrrrrE)rrrBr)rrrBr)rBr)rkrrrrOrrrrrrrrrrrIrGrrls IIIXI555X5 ###X#37))))))V3333'''' )))) ****rIrc eZdZdfd ZxZS)r~rBrCcJtddS)NzTThe `stream` argument was set to `True` but the `stream_cls` argument was not given.superrH)rFrjs rGrHz MissingStreamClassError.__init__s,  b     rIrrkrrrH __classcell__rjs@rGr~r~s=          rIr~c eZdZdfd ZxZS)rrBrCcNd}t|dS)Na&Attempted to read or stream some content, but the content has already been streamed. This could be due to attempting to stream the response content more than once. You can fix this by manually accumulating the response content while streaming or by calling `.read()` before starting to stream.r)rFmessagerjs rGrHzStreamAlreadyConsumed.__init__s. A  !!!!!rIrrrs@rGrrs= " " " " " " " " " "rIrfuncCallable[P, R]rBCallable[P, ArkAPIResponse[R]]cHtjdfd }|S) Higher order function that takes one of our bound API methods and wraps it to support returning the raw `APIResponse` object directly. argsP.argskwargsP.kwargsrBArkAPIResponse[R]citt|dpi}d|t<||d<ttt |i|SN extra_headersr<)r rrMrrr&rrrrs rGwrappedz(to_raw_response_wrapper..wrappedsh) CO4455;) .3 )*"/N1%ttT'>>>>> NrICallable[P, Awaitable[R]].Callable[P, Awaitable[ArkAsyncAPIResponse[R]]]cHtjdfd }|S) rrrrrrBArkAsyncAPIResponse[R]cKitt|dpi}d|t<||d<ttt |i|d{VSr)r rrMrrr&rs rGrz.async_to_raw_response_wrapper..wrappeds) CO4455;) .3 )*"/'*$$2G2G2G,G,G,G,G,G,GHHHrI)rrrrrBrrrs` rGasync_to_raw_response_wrapperrsE_TIIIIII NrI6Callable[P, ResponseContextManager[ArkAPIResponse[R]]]cHtjdfd }|S) Higher order function that takes one of our bound API methods and wraps it to support streaming and returning the raw `APIResponse` object directly. rrrrrB)ResponseContextManager[ArkAPIResponse[R]]citt|dpi}d|t<||d<t jg|Ri|}t ttgttf|SNrr?) r rrMrrpartialResponseContextManagerr rr&rrr make_requestrs rGrz-to_streamed_response_wrapper..wrappeds) CO4455;) .6 )*"/ (?????? % "nQ//0, ? ?   rI)rrrrrBrrrs` rGto_streamed_response_wrapperr?_T      NrI@Callable[P, AsyncResponseContextManager[ArkAsyncAPIResponse[R]]]cHtjdfd }|S) rrrrrrB3AsyncResponseContextManager[ArkAsyncAPIResponse[R]]citt|dpi}d|t<||d<|i|}t tt t t|Sr )r rrMrAsyncResponseContextManagerr rr&r s rGrz3async_to_streamed_response_wrapper..wrapped3s) CO4455;) .6 )*"/tT,V,, * .q12L A A   rI)rrrrrBrrrs` rG"async_to_streamed_response_wrapperr,rrIc*eZdZdZddZddZddZdS)r Context manager for ensuring that a request is not made until it is entered and that the response will always be closed when the context manager exits request_funcCallable[[], _APIResponseT]rBrCc"||_d|_dSrE _request_func!_ResponseContextManager__response)rFrs rGrHzResponseContextManager.__init__Ms)04rIr(cB||_|jSrErrNs rG __enter__z ResponseContextManager.__enter__Qs,,..rIexc_typetype[BaseException] | NonerBaseException | Noneexc_tbTracebackType | NonecJ|j|jdSdSrE)rrrFr!rr$s rG__exit__zResponseContextManager.__exit__Us. ? & O ! ! # # # # # ' &rIN)rrrBrC)rBr(r!r"rr#r$r%rBrC)rkrr__doc__rHr r(rrIrGr r GsZ 5555$$$$$$rIr c*eZdZdZddZddZddZdS)rr api_requestAwaitable[_AsyncAPIResponseT]rBrCc"||_d|_dSrE _api_request&_AsyncResponseContextManager__response)rFr,s rGrHz$AsyncResponseContextManager.__init__es'59rIr*c8K|jd{V|_|jSrEr/rNs rG __aenter__z&AsyncResponseContextManager.__aenter__is* $ 1111111rIr!r"rr#r$r%cZK|j!|jd{VdSdSrE)r1rr's rG __aexit__z%AsyncResponseContextManager.__aexit__msD ? &/'')) ) ) ) ) ) ) ) ) ) ' &rIN)r,r-rBrC)rBr*r))rkrrr*rHr3r5rrIrGrr_sZ ::::******rIr)rrrBr)rrrBr)rrrBr)rrrBr)I __future__rdatetimerrloggingrtypesrtypingrrrrr r r r r rrrtyping_extensionsrrr_typesr _constantsrr _exceptionsr_utilsrrrr _streamingrrrr_modelsr r!r"_request_optionsr# _base_clientr$r%r&r'r(r*r+ getLoggerrkr,rr.rrr|r~rrrrrr rrrIrGrEs?#""""""                           ==========BBBBBBBB!!!!!!UUUUUUUUUUUU =<<<<<<<<<)000000(((((( IcNN GCLL WT]]/DEEE W19STTT'+5kRUFV9V3WXXX'g'111111[ [ [ [ [ gaj[ [ [ |MMMMM_Q'MMM`eeeee/!,eeeP     i    " " " " "H " " "&*66$$$$$W]3$$$0*****'*<"=*****rI