
    g>                       U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZm	Z	m
Z
mZmZmZmZmZmZmZ d dlmZmZmZmZmZ d dlZd dlZd dlZddlmZ ddlmZmZm Z m!Z! ddl"m#Z#m$Z$m%Z% dd	l&m'Z' dd
l(m)Z)m*Z*m+Z+m,Z, ddl-m.Z. erddl"m/Z/ ddl0m1Z1  ed      Z2 ed      Z3 ed      Z4 ejj                  e6      Z7de8d<    G d dee3         Z9 G d de:      Z;ddZ<ddZ= G d d      Z>y)    )annotationsN)
TYPE_CHECKINGAnyUnionGenericTypeVarCallableIteratorAsyncIteratorcastoverload)	Awaitable	ParamSpecoverride
deprecated
get_origin   )NoneType)is_givenextract_type_argis_annotated_typeis_type_alias_type)	BaseModelis_basemodeladd_request_id)RAW_RESPONSE_HEADER)StreamAsyncStreamis_stream_class_typeextract_stream_chunk_type)APIResponseValidationError)FinalRequestOptions)
BaseClientPR_Tzlogging.Loggerlogc                     e Zd ZU dZded<   ded<   ded<   ded	<   d
ed<   ded<   ded<   ded<   	 dd	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d'dZed(d       Zed)d       Z	ed*d       Z	ddd+dZ	ed,d       Z
ed-d       Zed.d       Zed/d       Zed0d       Zed1d        Zed0d!       Zed0d"       Zed2d#       Zed3d$       Zddd+d%Zed0d&       Zy)4LegacyAPIResponsea  This is a legacy class as it will be replaced by `APIResponse`
    and `AsyncAPIResponse` in the `_response.py` file in the next major
    release.

    For the sync client this will mostly be the same with the exception
    of `content` & `text` will be methods instead of properties. In the
    async client, all methods will be async.

    A migration script will be provided & the migration in general should
    be smooth.
    type[R]_cast_toBaseClient[Any, Any]_clientzdict[type[Any], Any]_parsed_by_typebool_stream1type[Stream[Any]] | type[AsyncStream[Any]] | None_stream_clsr"   _optionshttpx.Responsehttp_responseintretries_takenr   )r7   c               t    || _         || _        i | _        || _        || _        || _        || _        || _        y N)r+   r-   r.   r0   r2   r3   r5   r7   )selfrawcast_toclientstream
stream_clsoptionsr7   s           L/var/www/openai/venv/lib/python3.12/site-packages/openai/_legacy_response.py__init__zLegacyAPIResponse.__init__F   s@      !% *    c                L    | j                   j                  j                  d      S )Nzx-request-id)r5   headersgetr:   s    rA   
request_idzLegacyAPIResponse.request_idZ   s    !!))--n==rC   c                    y r9    )r:   tos     rA   parsezLegacyAPIResponse.parse^   s    ,/rC   c                     y r9   rJ   rG   s    rA   rL   zLegacyAPIResponse.parsea   s    rC   NrK   c                  ||n| j                   }| j                  j                  |      }||S | j                  |      }t	        | j
                  j                        r| j
                  j                  |      }t        |t              rt        || j                         || j                  |<   t        t        |      S )a  Returns the rich python representation of this response's data.

        NOTE: For the async client: this will become a coroutine in the next major version.

        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
        from openai import BaseModel


        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`
        rN   )r+   r.   rF   _parser   r3   post_parser
isinstancer   r   rH   r   r%   )r:   rK   	cache_keycachedparseds        rA   rL   zLegacyAPIResponse.parsed   s    @ .Bdmm	%%)))4M#DMM--.]]..v6Ffi(64??3*0Y'AvrC   c                .    | j                   j                  S r9   )r5   rE   rG   s    rA   rE   zLegacyAPIResponse.headers       !!)))rC   c                .    | j                   j                  S r9   )r5   requestrG   s    rA   http_requestzLegacyAPIResponse.http_request   rW   rC   c                .    | j                   j                  S r9   )r5   status_coderG   s    rA   r\   zLegacyAPIResponse.status_code   s    !!---rC   c                .    | j                   j                  S r9   )r5   urlrG   s    rA   r^   zLegacyAPIResponse.url   s    !!%%%rC   c                .    | j                   j                  S r9   )rZ   methodrG   s    rA   r`   zLegacyAPIResponse.method   s      '''rC   c                .    | j                   j                  S )zReturn the binary response content.

        NOTE: this will be removed in favour of `.read()` in the
        next major version.
        )r5   contentrG   s    rA   rb   zLegacyAPIResponse.content   s     !!)))rC   c                .    | j                   j                  S )zyReturn the decoded response content.

        NOTE: this will be turned into a method in the next major version.
        )r5   textrG   s    rA   rd   zLegacyAPIResponse.text   s     !!&&&rC   c                .    | j                   j                  S r9   )r5   http_versionrG   s    rA   rf   zLegacyAPIResponse.http_version   s    !!...rC   c                .    | j                   j                  S r9   )r5   	is_closedrG   s    rA   rh   zLegacyAPIResponse.is_closed   s    !!+++rC   c                .    | j                   j                  S )zCThe time taken for the complete request/response cycle to complete.)r5   elapsedrG   s    rA   rj   zLegacyAPIResponse.elapsed   s     !!)))rC   c               	   ||n| j                   }t        |      r|j                  }|rt        |      rt	        |d      }| j
                  r6|rjt        |      st        dt         dt               t        t         |t        |d      | j                  t        t        | j                                    S | j                   rWt        t"        | j!                  t        | j                         | j                  t        t        | j                                    S t        d| j                  j$                        }|
t'               t        t"         ||| j                  t        t        | j                                    S |t(        u rt        t"        d       S | j                  }|t*        k(  rt        t"        |j,                        S |t.        k(  r#t        t"        t/        |j,                              S |t0        k(  r#t        t"        t1        |j,                              S |t2        k(  r+t        t"        |j,                  j5                         dk(        S t7        |      xs |}t9        j:                  |      r&t=        |t>              rt        t"         ||            S |t@        k(  rtC        d	      t9        j:                  |      rHt=        |tD        jF                        r.|tD        jF                  k7  rtI        d
      t        t"        |      S t9        j:                  |      r5t=        |tJ              s%t=        |tL        jJ                        rt        d      |tN        urr|tP        urj|tR        urb|tT        urZt=        |tJ              sJtC        d| dtJ         dtR         dtP         dtT         dt(         dt*         dtD        jF                   d      |jV                  jY                  dd      j[                  d      ^}}|dk7  rxt]        |      r/	 |j_                         }| j                  ja                  |||      S | j                  jj                  rtm        |d| d|j,                        |j,                  S |j_                         }| j                  ja                  |||      S # tb        $ r*}	td        jg                  dti        |	      |	       Y d }	~	d }	~	ww xY w)Nr   z/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=   r1   truez+Unexpected state - cast_to is `APIResponse`z:Subclasses of httpx.Response cannot be passed to `cast_to`zVPydantic models must subclass our base model type, e.g. `from openai import BaseModel`zUnsupported type, expected z to be a subclass of z, .zcontent-type*;zapplication/json)datar<   rm   z5Could not read JSON from response data due to %s - %szMExpected Content-Type response header to be `application/json` but received `z
` instead.)rm   messagebody)7r+   r   	__value__r   r   r0   r   	TypeErrorr   r   r   r&   r    r5   r   r-   r2   r%   _default_stream_clsMissingStreamClassErrorr   strrd   r6   floatr/   lowerr   inspectisclass
issubclassHttpxBinaryResponseContentr)   RuntimeErrorhttpxResponse
ValueErrorr   pydanticobjectlistdictr   rE   rF   splitr   json_process_response_data	Exceptionr'   debugtype_strict_response_validationr!   )
r:   rK   r<   r?   rm   origincontent_type_rr   excs
             rA   rP   zLegacyAPIResponse._parse   s|   "DMM g&''G (1&w2G<<+B/#&UV\U]]abman$opp 9 -D! "&!3!3#C6
 
 $$ 9$:J:J K!%!3!3#C6 %   QSWS_S_SsSstJ!-//#!//T\\2  h4= %%c>8==))c>3x}}-..e5/00d?8==..0F:;;G$/??6"z&:T'U78,--&&LMM??6"z&%..'I %..( #]__8$$??6":fi+HZX^`h`r`rMstuu 6!dNdNeOvy1-gY6KI;VXY]X^^`ae`ffhinhooqrzq{{}  B  ~C  CG  HM  HV  HV  GW  WX  Y  $++//DJJ3Oq--G$	#==?D  <<>>! '!) ?   ||770%klxky  zD  E!  == }}||22 3 
 	
/ ! gIIUW[\_W`beffgs   R. .	S!7 SS!c                j    d| j                    d| j                  j                   d| j                   dS )Nz<APIResponse [ z] type=>)r\   r5   reason_phraser+   rG   s    rA   __repr__zLegacyAPIResponse.__repr__L  s;     0 0143E3E3S3S2TT[\`\i\i[jjkllrC   )r;   r4   r<   r*   r=   r,   r>   r/   r?   r1   r@   r"   r7   r6   returnNoner   z
str | None)rK   ztype[_T]r   r&   )r   r%   )rK   ztype[_T] | Noner   zR | _T)r   zhttpx.Headers)r   zhttpx.Request)r   r6   )r   z	httpx.URLr   ry   r   bytes)r   r/   )r   zdatetime.timedelta)__name__
__module____qualname____doc____annotations__rB   propertyrH   r   rL   rE   rZ   r\   r^   r`   rb   rd   rf   rh   rj   rP   r   r   rJ   rC   rA   r)   r)   -   s   
 !!))MBB!!!!M + + 	+
 %+ + F+ %+ + 
+( > > / / -1 -^ * * * * . . & & ( ( * * ' ' / / , , * * /3 E
N m mrC   r)   c                        e Zd Zd fdZ xZS )rx   c                $    t         |   d       y )NzzThe `stream` argument was set to `True` but the `stream_cls` argument was not given. See `openai._streaming` for reference)superrB   )r:   	__class__s    rA   rB   z MissingStreamClassError.__init__R  s     I	
rC   r   r   )r   r   r   rB   __classcell__)r   s   @rA   rx   rx   Q  s    
 
rC   rx   c                D     t        j                         d fd       }|S )Higher order function that takes one of our bound API methods and wraps it
    to support returning the raw `APIResponse` object directly.
    c                     i t        t        |j                  d            xs i }d|t        <   ||d<   t        t        t
            | i |      S Nextra_headersrn   r   r   rF   r   r)   r%   argskwargsr   funcs      rA   wrappedz(to_raw_response_wrapper.<locals>.wrapped]  sU    (ZDfjj6Q,R,XVX(Z-3)*"/%a($*?*?@@rC   r   zP.argsr   zP.kwargsr   zLegacyAPIResponse[R]	functoolswrapsr   r   s   ` rA   to_raw_response_wrapperr   X  s)    
 __TA A NrC   c                D     t        j                         d fd       }|S )r   c                    K   i t        t        |j                  d            xs i }d|t        <   ||d<   t        t        t
            | i | d {         S 7 wr   r   r   s      rA   r   z.async_to_raw_response_wrapper.<locals>.wrappedn  s_     (ZDfjj6Q,R,XVX(Z-3)*"/%a(d0Ef0E*EFF*Es   AAA
	Ar   r   r   s   ` rA   async_to_raw_response_wrapperr   i  s)    
 __TG G NrC   c                  H   e Zd ZU ded<   ddZedd       Zedd       Zedd       Zedd       Z	ddZ
dd	Zd d!dZd d"dZd#dZd d!dZ	 	 	 	 d$dZ ed      d
d	 	 	 	 	 d%d       Zd&dZddZd d'dZd d(dZd)dZd d'dZ ed      d
d	 	 	 	 	 d%d       Zd&dZy
)*r   r4   rm   c                    || _         y r9   )rm   )r:   rm   s     rA   rB   z#HttpxBinaryResponseContent.__init__}  s	     rC   c                .    | j                   j                  S r9   )rm   rb   rG   s    rA   rb   z"HttpxBinaryResponseContent.content  s    }}$$$rC   c                .    | j                   j                  S r9   )rm   rd   rG   s    rA   rd   zHttpxBinaryResponseContent.text  s    }}!!!rC   c                .    | j                   j                  S r9   )rm   encodingrG   s    rA   r   z#HttpxBinaryResponseContent.encoding  s    }}%%%rC   c                .    | j                   j                  S r9   )rm   charset_encodingrG   s    rA   r   z+HttpxBinaryResponseContent.charset_encoding  s    }}---rC   c                :     | j                   j                  di |S )NrJ   )rm   r   )r:   r   s     rA   r   zHttpxBinaryResponseContent.json  s    !t}}!!+F++rC   c                6    | j                   j                         S r9   )rm   readrG   s    rA   r   zHttpxBinaryResponseContent.read  s    }}!!##rC   Nc                8    | j                   j                  |      S r9   )rm   
iter_bytesr:   
chunk_sizes     rA   r   z%HttpxBinaryResponseContent.iter_bytes  s    }}''
33rC   c                8    | j                   j                  |      S r9   )rm   	iter_textr   s     rA   r   z$HttpxBinaryResponseContent.iter_text  s    }}&&z22rC   c                6    | j                   j                         S r9   )rm   
iter_linesrG   s    rA   r   z%HttpxBinaryResponseContent.iter_lines  s    }}''))rC   c                8    | j                   j                  |      S r9   )rm   iter_rawr   s     rA   r   z#HttpxBinaryResponseContent.iter_raw  s    }}%%j11rC   c                    t        |d      5 }| j                  j                         D ]  }|j                  |        	 ddd       y# 1 sw Y   yxY w)av  Write the output to the given file.

        Accepts a filename or any path-like object, e.g. pathlib.Path

        Note: if you want to stream the data to the file instead of writing
        all at once then you should use `.with_streaming_response` when making
        the API request, e.g. `client.with_streaming_response.foo().stream_to_file('my_filename.txt')`
        wbmodeNopenrm   r   write)r:   filefrr   s       rA   write_to_filez(HttpxBinaryResponseContent.write_to_file  s>     $T"a002 3 #""s   1A		AzDue to a bug, this method doesn't actually stream the response content, `.with_streaming_response.method()` should be used instead)r   c                   t        |d      5 }| j                  j                  |      D ]  }|j                  |        	 d d d        y # 1 sw Y   y xY wNr   r   r   )r:   r   r   r   rr   s        rA   stream_to_filez)HttpxBinaryResponseContent.stream_to_file  s@     $T"a00< = #""s   2A

Ac                6    | j                   j                         S r9   )rm   closerG   s    rA   r   z HttpxBinaryResponseContent.close  s    }}""$$rC   c                R   K   | j                   j                          d {   S 7 wr9   )rm   areadrG   s    rA   r   z HttpxBinaryResponseContent.aread  s     ]]((****   '%'c                @   K   | j                   j                  |      S wr9   )rm   aiter_bytesr   s     rA   r   z&HttpxBinaryResponseContent.aiter_bytes  s     }}((44   c                @   K   | j                   j                  |      S wr9   )rm   
aiter_textr   s     rA   r   z%HttpxBinaryResponseContent.aiter_text  s     }}''
33r   c                >   K   | j                   j                         S wr9   )rm   aiter_linesrG   s    rA   r   z&HttpxBinaryResponseContent.aiter_lines  s     }}((**s   c                @   K   | j                   j                  |      S wr9   )rm   	aiter_rawr   s     rA   r   z$HttpxBinaryResponseContent.aiter_raw  s     }}&&z22r   c               V  K   t        j                  |      }|j                  d       d {   4 d {   }| j                  j	                  |      2 3 d {   }|j                  |       d {    #7 K7 D7 "7 6 d d d       d {  7   y # 1 d {  7  sw Y   y xY wwr   )anyioPathr   rm   r   r   )r:   r   r   pathr   rr   s         rA   astream_to_filez*HttpxBinaryResponseContent.astream_to_file  s      zz$----"mm77
C $dggdm## .-$# D .----s   +B)A9B)A;B)BBA=BB3A?4B9B);B)=B?BBB)BB)B&BB&"B)c                R   K   | j                   j                          d {   S 7 wr9   )rm   acloserG   s    rA   r   z!HttpxBinaryResponseContent.aclose  s     ]]))++++r   )rm   r4   r   r   r   r   r   )r   r   r   r   r9   )r   
int | Noner   zIterator[bytes])r   r   r   Iterator[str])r   r   )r   str | os.PathLike[str]r   r   )r   r   r   r   r   r   r   )r   r   r   zAsyncIterator[bytes])r   r   r   AsyncIterator[str])r   r   )r   r   r   r   rB   r   rb   rd   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rJ   rC   rA   r   r   z  s5   ! % % " " & & . .,$43*2$ 
   	M "&	$ 	
 
%+54+3  	M "&		$$	$ 		$
 
	$	$,rC   r   )r   zCallable[P, R]r   z!Callable[P, LegacyAPIResponse[R]])r   zCallable[P, Awaitable[R]]r   z,Callable[P, Awaitable[LegacyAPIResponse[R]]])?
__future__r   osr|   loggingdatetimer   typingr   r   r   r   r   r	   r
   r   r   r   typing_extensionsr   r   r   r   r   r   r   r   _typesr   _utilsr   r   r   r   _modelsr   r   r   
_constantsr   
_streamingr   r   r   r    _exceptionsr!   r"   _base_clientr#   r$   r%   r&   	getLoggerr   r'   r   r)   rv   rx   r   r   r   rJ   rC   rA   <module>r     s    " 	       U T     U U < < + \ \ 3,( cNCLT]'g''1^ 1am
 amH	
i 
""f, f,rC   