
    "hjX@                       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# 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/m0Z0 erddl$m1Z1 ddl2m3Z3  ed      Z4 ed      Z5 ed      Z6 ejn                  e8      Z9de:d<    G d dee5         Z; 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)http_response_types)	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   )r8   c               t    || _         || _        i | _        || _        || _        || _        || _        || _        y N)r,   r.   r/   r1   r3   r4   r6   r8   )selfrawcast_toclientstream
stream_clsoptionsr8   s           U/root/.hermes/SkillClaw/.venv/lib/python3.12/site-packages/openai/_legacy_response.py__init__zLegacyAPIResponse.__init__G   s@      !% *    c                L    | j                   j                  j                  d      S )Nzx-request-id)r6   headersgetr;   s    rB   
request_idzLegacyAPIResponse.request_id[   s    !!))--n==rD   c                    y r:    )r;   tos     rB   parsezLegacyAPIResponse.parse_   s    ,/rD   c                     y r:   rK   rH   s    rB   rM   zLegacyAPIResponse.parseb   s    rD   NrL   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`
        rO   )r,   r/   rG   _parser   r4   post_parser
isinstancer   r   rI   r   r&   )r;   rL   	cache_keycachedparseds        rB   rM   zLegacyAPIResponse.parsee   s    @ .Bdmm	%%)))4M#DMM--.]]..v6Ffi(64??3*0Y'AvrD   c                .    | j                   j                  S r:   )r6   rF   rH   s    rB   rF   zLegacyAPIResponse.headers       !!)))rD   c                .    | j                   j                  S r:   )r6   requestrH   s    rB   http_requestzLegacyAPIResponse.http_request   rX   rD   c                .    | j                   j                  S r:   )r6   status_coderH   s    rB   r]   zLegacyAPIResponse.status_code   s    !!---rD   c                .    | j                   j                  S r:   )r6   urlrH   s    rB   r_   zLegacyAPIResponse.url   s    !!%%%rD   c                .    | j                   j                  S r:   )r[   methodrH   s    rB   ra   zLegacyAPIResponse.method   s      '''rD   c                .    | j                   j                  S )zReturn the binary response content.

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

        NOTE: this will be turned into a method in the next major version.
        )r6   textrH   s    rB   re   zLegacyAPIResponse.text   s     !!&&&rD   c                .    | j                   j                  S r:   )r6   http_versionrH   s    rB   rg   zLegacyAPIResponse.http_version   s    !!...rD   c                .    | j                   j                  S r:   )r6   	is_closedrH   s    rB   ri   zLegacyAPIResponse.is_closed   s    !!+++rD   c                .    | j                   j                  S )zCThe time taken for the complete request/response cycle to complete.)r6   elapsedrH   s    rB   rk   zLegacyAPIResponse.elapsed   s     !!)))rD   c               	   ||n| j                   }t        |      r|j                  }|rt        |      rt	        |d      }t        |      xs |}| j                  rW|rut        |      st        dt         dt               t        t         |t        |d      | j                  t        t        | j                         | j"                              S | j$                  rbt        t&        | j%                  t        | j$                        | j                  t        t        | j                         | j"                              S t        d| j                   j(                        }|
t+               t        t&         ||| j                  t        t        | j                         | j"                              S |t,        u rt        t&        d       S | j                  }|t.        k(  rt        t&        |j0                        S |t2        k(  r#t        t&        t3        |j0                              S |t4        k(  r#t        t&        t5        |j0                              S |t6        k(  r+t        t&        |j0                  j9                         dk(        S t;        j<                  |      r&t?        |t@              rt        t&         ||            S |tB        k(  rtE        d	      tG               }t;        j<                  |      r+t?        ||      r||vrtI        d
      t        t&        |      S t;        j<                  |      r5t?        |tJ              s%t?        |tL        jJ                        rt        d      |tN        urr|tP        urj|tR        urb|tT        urZt?        |tJ              sJtE        d| dtJ         dtR         dtP         dtT         dt,         dt.         dtV        jX                   d      |jZ                  j]                  dd      j_                  d      ^}}|ja                  d      sxtc        |      r/	 |je                         }	| j                   jg                  |	||      S | j                   jp                  rts        |d| d|j0                        |j0                  S |je                         }	| j                   jg                  |	||      S # th        $ r*}
tj        jm                  dto        |
      |
       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>   rA   r2   truez+Unexpected state - cast_to is `APIResponse`zASubclasses of HTTP response classes 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*;json)datar=   rn   z5Could not read JSON from response data due to %s - %szMExpected Content-Type response header to be `application/json` but received `z
` instead.)rn   messagebody):r,   r   	__value__r   r   r   r1   r    	TypeErrorr   r   r   r'   r!   r6   r   r.   r4   r3   r&   _default_stream_clsMissingStreamClassErrorr   strre   r7   floatr0   lowerinspectisclass
issubclassHttpxBinaryResponseContentr*   RuntimeErrorr   
ValueErrorr   pydanticobjectlistdictr   httpxResponserF   rG   splitendswithr   rs   _process_response_data	Exceptionr(   debugtype_strict_response_validationr"   )r;   rL   r=   originr@   rn   response_typescontent_type_rt   excs              rB   rQ   zLegacyAPIResponse._parse   s   "DMM g&''G (1&w2GG$/<<+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 MM	  h4= %%c>8==))c>3x}}-..e5/00d?8==..0F:;;??6"z&:T'U78,--&&LMM,.??
0 n, !dee8$$ OO vy168#5#56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$$V,G$	#==?D  <<>>! '!) ?   ||770%klxky  zD  E!  == }}||22 3 
 	
/ ! gIIUW[\_W`beffgs   .S 	S; S66S;c                j    d| j                    d| j                  j                   d| j                   dS )Nz<APIResponse [ z] type=>)r]   r6   reason_phraser,   rH   s    rB   __repr__zLegacyAPIResponse.__repr__Y  s;     0 0143E3E3S3S2TT[\`\i\i[jjkllrD   )r<   r5   r=   r+   r>   r-   r?   r0   r@   r2   rA   r#   r8   r7   returnNoner   z
str | None)rL   ztype[_T]r   r'   )r   r&   )rL   ztype[_T] | Noner   zR | _T)r   zhttpx.Headers)r   zhttpx.Request)r   r7   )r   z	httpx.URLr   r{   r   bytes)r   r0   )r   zdatetime.timedelta)__name__
__module____qualname____doc____annotations__rC   propertyrI   r   rM   rF   r[   r]   r_   ra   rc   re   rg   ri   rk   rQ   r   r   rK   rD   rB   r*   r*   .   s   
 !!))MBB!!!!M + + 	+
 %+ + F+ %+ + 
+( > > / / -1 -^ * * * * . . & & ( ( * * ' ' / / , , * * /3 Q
f m mrD   r*   c                        e Zd Zd fdZ xZS )rz   c                $    t         |   d       y )NzzThe `stream` argument was set to `True` but the `stream_cls` argument was not given. See `openai._streaming` for reference)superrC   )r;   	__class__s    rB   rC   z MissingStreamClassError.__init___  s     I	
rD   r   r   )r   r   r   rC   __classcell__)r   s   @rB   rz   rz   ^  s    
 
rD   rz   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_headersro   r   r   rG   r   r*   r&   argskwargsr   funcs      rB   wrappedz(to_raw_response_wrapper.<locals>.wrappedj  sU    (ZDfjj6Q,R,XVX(Z-3)*"/%a($*?*?@@rD   r   zP.argsr   zP.kwargsr   zLegacyAPIResponse[R]	functoolswrapsr   r   s   ` rB   to_raw_response_wrapperr   e  s)    
 __TA A NrD   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      rB   r   z.async_to_raw_response_wrapper.<locals>.wrapped{  s_     (ZDfjj6Q,R,XVX(Z-3)*"/%a(d0Ef0E*EFF*Es   AAA
	Ar   r   r   s   ` rB   async_to_raw_response_wrapperr   v  s)    
 __TG G NrD   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   r5   rn   c                    || _         y r:   )rn   )r;   rn   s     rB   rC   z#HttpxBinaryResponseContent.__init__  s	     rD   c                .    | j                   j                  S r:   )rn   rc   rH   s    rB   rc   z"HttpxBinaryResponseContent.content  s    }}$$$rD   c                .    | j                   j                  S r:   )rn   re   rH   s    rB   re   zHttpxBinaryResponseContent.text  s    }}!!!rD   c                .    | j                   j                  S r:   )rn   encodingrH   s    rB   r   z#HttpxBinaryResponseContent.encoding  s    }}%%%rD   c                .    | j                   j                  S r:   )rn   charset_encodingrH   s    rB   r   z+HttpxBinaryResponseContent.charset_encoding  s    }}---rD   c                :     | j                   j                  di |S )NrK   )rn   rs   )r;   r   s     rB   rs   zHttpxBinaryResponseContent.json  s    !t}}!!+F++rD   c                6    | j                   j                         S r:   )rn   readrH   s    rB   r   zHttpxBinaryResponseContent.read  s    }}!!##rD   Nc                8    | j                   j                  |      S r:   )rn   
iter_bytesr;   
chunk_sizes     rB   r   z%HttpxBinaryResponseContent.iter_bytes  s    }}''
33rD   c                8    | j                   j                  |      S r:   )rn   	iter_textr   s     rB   r   z$HttpxBinaryResponseContent.iter_text  s    }}&&z22rD   c                6    | j                   j                         S r:   )rn   
iter_linesrH   s    rB   r   z%HttpxBinaryResponseContent.iter_lines  s    }}''))rD   c                8    | j                   j                  |      S r:   )rn   iter_rawr   s     rB   r   z#HttpxBinaryResponseContent.iter_raw  s    }}%%j11rD   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openrn   r   write)r;   filefrt   s       rB   write_to_filez(HttpxBinaryResponseContent.write_to_file  sL     $T" 	a002 	 	 	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   rt   s        rB   stream_to_filez)HttpxBinaryResponseContent.stream_to_file  sN     $T" 	a00< 	 	 	s   2A

Ac                6    | j                   j                         S r:   )rn   closerH   s    rB   r   z HttpxBinaryResponseContent.close  s    }}""$$rD   c                R   K   | j                   j                          d {   S 7 wr:   )rn   areadrH   s    rB   r   z HttpxBinaryResponseContent.aread  s     ]]((****   '%'c                @   K   | j                   j                  |      S wr:   )rn   aiter_bytesr   s     rB   r   z&HttpxBinaryResponseContent.aiter_bytes  s     }}((44   c                @   K   | j                   j                  |      S wr:   )rn   
aiter_textr   s     rB   r   z%HttpxBinaryResponseContent.aiter_text  s     }}''
33r   c                >   K   | j                   j                         S wr:   )rn   aiter_linesrH   s    rB   r   z&HttpxBinaryResponseContent.aiter_lines  s     }}((**s   c                @   K   | j                   j                  |      S wr:   )rn   	aiter_rawr   s     rB   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   rn   r   r   )r;   r   r   pathr   rt   s         rB   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r:   )rn   acloserH   s    rB   r   z!HttpxBinaryResponseContent.aclose  s     ]]))++++r   )rn   r5   r   r   r   r   r   )r   r   r   r   r:   )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   rC   r   rc   re   r   r   rs   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rK   rD   rB   r   r     s5   ! % % " " & & . .,$43*2$ 
   	M "&	$ 	
 
%+54+3  	M "&		$$	$ 		$
 
	$	$,rD   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]]])A
__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   _httpx2r   _modelsr   r   r   
_constantsr   
_streamingr   r   r    r!   _exceptionsr"   r#   _base_clientr$   r%   r&   r'   	getLoggerr   r(   r   r*   rx   rz   r   r   r   rK   rD   rB   <module>r     s    " 	       U T     U U ( < < + \ \ 3,( cNCLT]'g''1^ 1mm
 mm`	
i 
""f, f,rD   