File: //usr/local/lib/python3.10/dist-packages/openai/__pycache__/_legacy_response.cpython-310.pyc
o
���gn? � @ s� U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZ d dlmZmZmZmZmZ d dlZd dlZd dlZddlm Z ddl m Z m Z m Z m!Z! ddl"m#Z#m$Z$m%Z% dd l&m'Z' dd
l(m)Z)m*Z*m+Z+m,Z, dd l-m.Z. er�dd l"m/Z/ dd
l0m1Z1 ed�Z2e d�Z3e d�Z4e�5e6�Z7de8d< G dd� de e3 �Z9G dd� de:�Z;d#dd �Z<d$d d �Z=G d!d"� d"�Z>dS )%� )� 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�_Tzlogging.Logger�logc @ sH e Zd ZU dZded< ded< ded< ded < d
ed < d ed
< ded< ded< dd�dKd d �ZedLd d ��ZedMd$d%��Z edNd'd%��Z d(d)�dOd,d%�Z edPd.d/��Z
edQd1d2��Z edRd3d4��Z edSd6d7��Z
edTd9d:��ZedUd<d=��ZedTd>d?��ZedTd@dA��ZedVdBdC��ZedWdEdF��Zd(d)�dOdGdH�ZedTdIdJ��Zd(S )X�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]�_clientzdict[type[Any], Any]�_parsed_by_type�bool�_stream�1type[Stream[Any]] | type[AsyncStream[Any]] | None� _stream_clsr! �_options�httpx.Response�
http_response�int�
retries_takenr )r5 �raw�cast_to�client�stream�
stream_cls�options�return�Nonec C s4 || _ || _i | _|| _|| _|| _|| _|| _d S �N)r) r+ r, r. r0 r1 r3 r5 )�selfr6 r7 r8 r9 r: r; r5 � r@ �B/usr/local/lib/python3.10/dist-packages/openai/_legacy_response.py�__init__F s
zLegacyAPIResponse.__init__�
str | Nonec C s | j j�d�S )Nz x-request-id)r3 �headers�get�r? r@ r@ rA �
request_idZ s z LegacyAPIResponse.request_id�to�type[_T]r% c C � d S r> r@ )r? rH r@ r@ rA �parse^ � zLegacyAPIResponse.parser$ c C rJ r> r@ rF r@ r@ rA rK a rL N�rH �type[_T] | None�R | _Tc C sx |dur|n| j }| j�|�}|dur|S | j|d�}t| jj�r'| j�|�}t|t�r2t || 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`
NrM )
r) r, rE �_parser r1 � post_parser�
isinstancer r rG r r$ )r? rH � cache_key�cached�parsedr@ r@ rA rK d s
�
httpx.Headersc C � | j jS r> )r3 rD rF r@ r@ rA rD � � zLegacyAPIResponse.headers�
httpx.Requestc C rW r> )r3 �requestrF r@ r@ rA � http_request� rX z LegacyAPIResponse.http_requestc C rW r> )r3 � status_coderF r@ r@ rA r\ � rX z LegacyAPIResponse.status_code� httpx.URLc C rW r> )r3 �urlrF r@ r@ rA r^ � rX zLegacyAPIResponse.url�strc C rW r> )r[ �methodrF r@ r@ rA r` � rX zLegacyAPIResponse.method�bytesc C rW )z�Return the binary response content.
NOTE: this will be removed in favour of `.read()` in the
next major version.
)r3 �contentrF r@ r@ rA rb � s zLegacyAPIResponse.contentc C rW )zyReturn the decoded response content.
NOTE: this will be turned into a method in the next major version.
)r3 �textrF r@ r@ rA rc � s zLegacyAPIResponse.textc C rW r> )r3 � http_versionrF r@ r@ rA rd � rX z LegacyAPIResponse.http_versionc C rW r> )r3 � is_closedrF r@ r@ rA re � rX zLegacyAPIResponse.is_closed�datetime.timedeltac C rW )zCThe time taken for the complete request/response cycle to complete.)r3 �elapsedrF r@ r@ rA rg � s zLegacyAPIResponse.elapsedc
C s. |d ur|n| j }t|�r|j}|rt|�rt|d�}t|�p |}| jrz|rGt|�s4tdt � dt
� ���t t |t
|dd�| jt t| j�d��S | jr]t t| jt
| j�| jt t| j�d��S t d| jj�}|d u rkt� �t t||| jt t| j�d��S |tu r�t td �S | j}|tkr�t t|j�S |tkr�t tt|j��S |tkr�t tt|j��S |tkr�t t|j�� dk�S t � |�r�t |t �r�t t||��S |t kr�t!d ��t � |�r�t |t"j#�r�|t"j#kr�t$d
��t t|�S t � |�r�t |t%�s�t |t&j%�r�td ��|t'u�r4|t(u�r4|t)u�r4|t*u�r4t |t%��s4t!d |� d
t%� dt)� dt(� dt*� dt� dt� dt"j#� d���|j+�,dd��-d�^}}|dk�r�t.|��rvz|�/� }W n t0�yl } zt1�2dt3| �| � W Y d } ~ nd } ~ ww | jj4|||d�S | jj5�r�t6|d|� d�|jd��|jS |�/� }| jj4|||d�S )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)r7 �responser8 r/ �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`zUnsupported type, expected z to be a subclass of z, �.z content-type�*�;zapplication/jsonz5Could not read JSON from response data due to %s - %s)�datar7 ri zMExpected Content-Type response header to be `application/json` but received `z
` instead.)ri �message�body)7r) r � __value__r r r r. r � TypeErrorr r r r% r r3 r r+ r0 r$ �_default_stream_cls�MissingStreamClassErrorr r_ rc r4 �floatr- �lower�inspect�isclass�
issubclass�HttpxBinaryResponseContentr' � RuntimeError�httpx�Response�
ValueErrorr �pydantic�object�list�dictr rD rE �splitr �json� Exceptionr& �debug�type�_process_response_data�_strict_response_validationr )
r? rH r7 �originr: ri � content_type�_rn �excr@ r@ rA rP � s�
�
��
��
��
�
�
���
�
�6�
���
� �zLegacyAPIResponse._parsec C s d| j � d| jj� d| j� d�S )Nz<APIResponse [� z] type=�>)r\ r3 �
reason_phraser) rF r@ r@ rA �__repr__T s zLegacyAPIResponse.__repr__)r6 r2 r7 r( r8 r* r9 r- r: r/ r; r! r5 r4 r< r= �r< rC )rH rI r< r% )r< r$ )rH rN r< rO )r< rV )r< rY )r< r4 )r< r] �r< r_ �r< ra )r< r- )r< rf )�__name__�
__module__� __qualname__�__doc__�__annotations__rB �propertyrG r rK rD r[ r\ r^ r` rb rc rd re rg rP r r� r@ r@ r@ rA r' - sX
� / r' c s e Zd Zd� fdd� Z� ZS )rt r< r= c s t � �d� d S )NzzThe `stream` argument was set to `True` but the `stream_cls` argument was not given. See `openai._streaming` for reference)�superrB rF �� __class__r@ rA rB Z s �z MissingStreamClassError.__init__�r< r= )r� r� r� rB �
__classcell__r@ r@ r� rA rt Y s rt �func�Callable[P, R]r< �!Callable[P, LegacyAPIResponse[R]]c � t �� �d
� fdd� �}|S ) ��Higher order function that takes one of our bound API methods and wraps it
to support returning the raw `APIResponse` object directly.
�args�P.args�kwargs�P.kwargsr< �LegacyAPIResponse[R]c s@ i t t|�d��p
i �}d|t< ||d< t tt � | i |���S �N�
extra_headersrj �r r rE r r' r$ �r� r� r� �r� r@ rA �wrappede s z(to_raw_response_wrapper.<locals>.wrappedN�r� r� r� r� r< r� �� functools�wraps�r� r� r@ r� rA �to_raw_response_wrapper` � r� �Callable[P, Awaitable[R]]�,Callable[P, Awaitable[LegacyAPIResponse[R]]]c r� ) r� r� r� r� r� r< r� c � sH �i t t|�d��p i �}d|t< ||d< t tt � | i |��I d H �S r� r� r� r� r@ rA r� v s
� z.async_to_raw_response_wrapper.<locals>.wrappedNr� r� r� r@ r� rA � async_to_raw_response_wrapperq r� r� c @ s e Zd ZU ded< d?dd�Zed@dd ��ZedAd d ��ZedBdd��ZedBdd��Z dCdd�Z
d@dd�Z dDdEd d �Z dDdFd d �Z
dGd!d"�ZdDdEd#d$�ZdHd'd(�Zed)�dd*�dId+d,��ZdJd-d.�Zd@d/d0�ZdDdKd2d3�ZdDdLd5d6�ZdMd7d8�ZdDdKd9d:�Zed)�dd*�dId;d<��ZdJd=d>�ZdS )Nrz r2 ri r< r= c C s
|| _ d S r> )ri )r? ri r@ r@ rA rB � �
z#HttpxBinaryResponseContent.__init__ra c C rW r> )ri rb rF r@ r@ rA rb � rX z"HttpxBinaryResponseContent.contentr_ c C rW r> )ri rc rF r@ r@ rA rc � rX z HttpxBinaryResponseContent.textrC c C rW r> )ri �encodingrF r@ r@ rA r� � rX z#HttpxBinaryResponseContent.encodingc C rW r> )ri �charset_encodingrF r@ r@ rA r� � rX z+HttpxBinaryResponseContent.charset_encodingr� r c K s | j jdi |��S )Nr@ )ri r� )r? r� r@ r@ rA r� � s z HttpxBinaryResponseContent.jsonc C �
| j �� S r> )ri �readrF r@ r@ rA r� � r� z HttpxBinaryResponseContent.readN�
chunk_size�
int | None�Iterator[bytes]c C � | j �|�S r> )ri �
iter_bytes�r? r� r@ r@ rA r� � � z%HttpxBinaryResponseContent.iter_bytes�
Iterator[str]c C r� r> )ri � iter_textr� r@ r@ rA r� � r� z$HttpxBinaryResponseContent.iter_textc C r� r> )ri �
iter_linesrF r@ r@ rA r� � r� z%HttpxBinaryResponseContent.iter_linesc C r� r> )ri �iter_rawr� r@ r@ rA r� � r� z#HttpxBinaryResponseContent.iter_raw�file�str | os.PathLike[str]c C sL t |dd��}| j�� D ]}|�|� q W d � dS 1 s w Y dS )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��openri r� �write)r? r� �frn r@ r@ rA �
write_to_file� s
�"�z(HttpxBinaryResponseContent.write_to_filez�Due to a bug, this method doesn't actually stream the response content, `.with_streaming_response.method()` should be used instead)r� c C sN t |dd��}| j�|�D ]}|�|� q
W d � d S 1 s w Y d S �Nr� r� r� )r? r� r� r� rn r@ r@ rA �stream_to_file� s
�"�z)HttpxBinaryResponseContent.stream_to_filec C r� r> )ri �closerF r@ r@ rA r� � r� z HttpxBinaryResponseContent.closec � � �| j �� I d H S r> )ri �areadrF r@ r@ rA r� � � �z HttpxBinaryResponseContent.aread�AsyncIterator[bytes]c � � �| j �|�S r> )ri � aiter_bytesr� r@ r@ rA r� � � � z&HttpxBinaryResponseContent.aiter_bytes�AsyncIterator[str]c � r� r> )ri �
aiter_textr� r@ r@ rA r� � r� z%HttpxBinaryResponseContent.aiter_textc � s �| j �� S r> )ri � aiter_linesrF r@ r@ rA r� � s �
z&HttpxBinaryResponseContent.aiter_linesc � r� r> )ri � aiter_rawr� r@ r@ rA r� � r� z$HttpxBinaryResponseContent.aiter_rawc � s� �t �|�}|jdd�I d H 4 I d H �"}| j�|�2 z3 d H W }|�|�I d H q6 W d �I d H d S 1 I d H s;w Y d S r� )�anyio�Pathr� ri r� r� )r? r� r� �pathr� rn r@ r@ rA �astream_to_file� s �
�.�z*HttpxBinaryResponseContent.astream_to_filec � r� r> )ri �acloserF r@ r@ rA r� � r� z!HttpxBinaryResponseContent.aclose)ri r2 r< r= r� r� r� )r� r r< r r> )r� r� r< r� )r� r� r< r� )r< r� )r� r� r< r= )r� r� r� r� r< r= r� )r� r� r< r� )r� r� r< r� )r< r� )r� r� r� r� rB r� rb rc r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r@ r@ r@ rA rz � sF
��
�� rz )r� r� r< r� )r� r� r< r� )?�
__future__r �osrw �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' rr rt r� r� rz r@ r@ r@ rA �<module> s> 0 .