o
    ?Hh~l                     @  s`  d dl mZ d dlmZmZmZ d dlmZ d dlZddl	m
Z
 ddlmZmZmZmZmZ ddlmZmZ dd	lmZ dd
lmZmZ ddlmZmZ ddlmZmZ ddlm Z m!Z! ddl"m#Z#m$Z$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+ ddl,m-Z- ddgZ.G dd deZ/G dd deZ0G dd dZ1G dd dZ2G dd dZ3G dd dZ4dS )    )annotations)UnionIterableOptional)LiteralN   )_legacy_response)	NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)AsyncPaginatormake_request_options)message_list_paramsmessage_create_paramsmessage_update_params)Message)Metadata)MessageDeleted)MessageContentPartParamMessagesAsyncMessagesc                
   @     e Zd Zed5ddZed6ddZeeddded	d7ddZdddedd8d!d"Zeddded#d9d$d%Z	eeeeeddded&	d:d0d1Z
dddedd;d3d4ZdS )<r    returnMessagesWithRawResponsec                 C     t | S a  
        This property can be used as a prefix for any HTTP method call to return
        the raw response object instead of the parsed content.

        For more information, see https://www.github.com/openai/openai-python#accessing-raw-response-data-eg-headers
        )r$   self r)   f/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/openai/resources/beta/threads/messages.pywith_raw_response"      zMessages.with_raw_responseMessagesWithStreamingResponsec                 C  r%   z
        An alternative to `.with_raw_response` that doesn't eagerly read the response body.

        For more information, see https://www.github.com/openai/openai-python#with_streaming_response
        )r-   r'   r)   r)   r*   with_streaming_response,      z Messages.with_streaming_responseNattachmentsmetadataextra_headersextra_query
extra_bodytimeout	thread_idstrcontent-Union[str, Iterable[MessageContentPartParam]]roleLiteral['user', 'assistant']r2   ?Optional[Iterable[message_create_params.Attachment]] | NotGivenr3   Optional[Metadata] | NotGivenr4   Headers | Noner5   Query | Noner6   Body | Noner7   'float | httpx.Timeout | None | NotGivenr   c          
   	   C  sZ   |s	t d|ddi|pi }| jd| dt||||dtjt||||	dtdS )	/  
        Create a message.

        Args:
          content: The text contents of the message.

          role:
              The role of the entity that is creating the message. Allowed values include:

              - `user`: Indicates the message is sent by an actual user and should be used in
                most cases to represent user-generated messages.
              - `assistant`: Indicates the message is generated by the assistant. Use this
                value to insert messages from the assistant into the conversation.

          attachments: A list of files attached to the message, and the tools they should be added to.

          metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful
              for storing additional information about the object in a structured format, and
              querying for objects via API or the dashboard.

              Keys are strings with a maximum length of 64 characters. Values are strings with
              a maximum length of 512 characters.

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        8Expected a non-empty value for `thread_id` but received OpenAI-Betaassistants=v2	/threads/	/messagesr:   r<   r2   r3   r4   r5   r6   r7   bodyoptionscast_to)
ValueError_postr   r   MessageCreateParamsr   r   
r(   r8   r:   r<   r2   r3   r4   r5   r6   r7   r)   r)   r*   create5   s$   .
	zMessages.createrK   
message_idc                C  \   |s	t d||st d|ddi|pi }| jd| d| t||||dtdS )	G  
        Retrieve a message.

        Args:
          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rE   9Expected a non-empty value for `message_id` but received rF   rG   rH   
/messages/rK   rN   rO   rP   _getr   r   r(   rU   r8   r4   r5   r6   r7   r)   r)   r*   retrievew      zMessages.retriever3   r4   r5   r6   r7   c             	   C  sj   |s	t d||st d|ddi|pi }| jd| d| td|itjt||||dtd	S )
  
        Modifies a message.

        Args:
          metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful
              for storing additional information about the object in a structured format, and
              querying for objects via API or the dashboard.

              Keys are strings with a maximum length of 64 characters. Values are strings with
              a maximum length of 512 characters.

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rE   rX   rF   rG   rH   rY   r3   rK   rL   )rP   rQ   r   r   MessageUpdateParamsr   r   r(   rU   r8   r3   r4   r5   r6   r7   r)   r)   r*   update   s    zMessages.update	afterbeforelimitorderrun_idr4   r5   r6   r7   rf   str | NotGivenrg   rh   int | NotGivenri   !Literal['asc', 'desc'] | NotGivenrj   SyncCursorPage[Message]c       	         C  b   |s	t d|ddi|pi }| jd| dtt t|||	|
t|||||dtjdtdS 	ah  
        Returns a list of messages for a given thread.

        Args:
          after: A cursor for use in pagination. `after` is an object ID that defines your place
              in the list. For instance, if you make a list request and receive 100 objects,
              ending with obj_foo, your subsequent call can include after=obj_foo in order to
              fetch the next page of the list.

          before: A cursor for use in pagination. `before` is an object ID that defines your place
              in the list. For instance, if you make a list request and receive 100 objects,
              starting with obj_foo, your subsequent call can include before=obj_foo in order
              to fetch the previous page of the list.

          limit: A limit on the number of objects to be returned. Limit can range between 1 and
              100, and the default is 20.

          order: Sort order by the `created_at` timestamp of the objects. `asc` for ascending
              order and `desc` for descending order.

          run_id: Filter messages by the run ID that generated them.

          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rE   rF   rG   rH   rI   )rf   rg   rh   ri   rj   )r4   r5   r6   r7   query)pagerN   model)rP   _get_api_listr   r   r   r   r   MessageListParamsr(   r8   rf   rg   rh   ri   rj   r4   r5   r6   r7   r)   r)   r*   list   .   .
zMessages.listr   c                C  rV   )	F  
        Deletes a message.

        Args:
          extra_headers: Send extra headers

          extra_query: Add additional query parameters to the request

          extra_body: Add additional JSON properties to the request

          timeout: Override the client-level default timeout for this request, in seconds
        rE   rX   rF   rG   rH   rY   rK   rZ   rP   _deleter   r   r]   r)   r)   r*   delete  r_   zMessages.delete)r#   r$   )r#   r-   r8   r9   r:   r;   r<   r=   r2   r>   r3   r?   r4   r@   r5   rA   r6   rB   r7   rC   r#   r   rU   r9   r8   r9   r4   r@   r5   rA   r6   rB   r7   rC   r#   r   rU   r9   r8   r9   r3   r?   r4   r@   r5   rA   r6   rB   r7   rC   r#   r   )r8   r9   rf   rk   rg   rk   rh   rl   ri   rm   rj   rk   r4   r@   r5   rA   r6   rB   r7   rC   r#   rn   rU   r9   r8   r9   r4   r@   r5   rA   r6   rB   r7   rC   r#   r   __name__
__module____qualname__r   r+   r/   r	   rT   r^   rd   rw   r|   r)   r)   r)   r*   r    !   L    	I*2Nc                
   @  r"   )<r!   r#   AsyncMessagesWithRawResponsec                 C  r%   r&   )r   r'   r)   r)   r*   r+   8  r,   zAsyncMessages.with_raw_response"AsyncMessagesWithStreamingResponsec                 C  r%   r.   )r   r'   r)   r)   r*   r/   B  r0   z%AsyncMessages.with_streaming_responseNr1   r8   r9   r:   r;   r<   r=   r2   r>   r3   r?   r4   r@   r5   rA   r6   rB   r7   rC   r   c          
   	     sh   |s
t d|ddi|pi }| jd| dt||||dtjI dH t||||	dtd	I dH S )
rD   rE   rF   rG   rH   rI   rJ   NrK   rL   )rP   rQ   r   r   rR   r   r   rS   r)   r)   r*   rT   K  s&   .
	zAsyncMessages.createrK   rU   c                  d   |s
t d||st d|ddi|pi }| jd| d| t||||dtdI d	H S )
rW   rE   rX   rF   rG   rH   rY   rK   rZ   Nr[   r]   r)   r)   r*   r^        zAsyncMessages.retriever`   c             	     sx   |s
t d||st d|ddi|pi }| jd| d| td|itjI dH t||||d	td
I dH S )ra   rE   rX   rF   rG   rH   rY   r3   NrK   rL   )rP   rQ   r   r   rb   r   r   rc   r)   r)   r*   rd     s    zAsyncMessages.updatere   rf   rk   rg   rh   rl   ri   rm   rj   1AsyncPaginator[Message, AsyncCursorPage[Message]]c       	         C  ro   rp   )rP   rt   r   r   r   r   r   ru   rv   r)   r)   r*   rw     rx   zAsyncMessages.listr   c                  r   )
ry   rE   rX   rF   rG   rH   rY   rK   rZ   Nrz   r]   r)   r)   r*   r|   '  r   zAsyncMessages.delete)r#   r   )r#   r   r}   r~   r   )r8   r9   rf   rk   rg   rk   rh   rl   ri   rm   rj   rk   r4   r@   r5   rA   r6   rB   r7   rC   r#   r   r   r   r)   r)   r)   r*   r!   7  r   c                   @     e Zd ZdddZdS )	r$   messagesr    r#   Nonec                 C  P   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S N)	_messagesr   to_raw_response_wrapperrT   r^   rd   rw   r|   r(   r   r)   r)   r*   __init__N      
z MessagesWithRawResponse.__init__Nr   r    r#   r   r   r   r   r   r)   r)   r)   r*   r$   M      r$   c                   @  r   )	r   r   r!   r#   r   c                 C  r   r   )r   r   async_to_raw_response_wrapperrT   r^   rd   rw   r|   r   r)   r)   r*   r   c  r   z%AsyncMessagesWithRawResponse.__init__Nr   r!   r#   r   r   r)   r)   r)   r*   r   b  r   r   c                   @  r   )	r-   r   r    r#   r   c                 C  F   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S r   )r   r   rT   r^   rd   rw   r|   r   r)   r)   r*   r   x      
z&MessagesWithStreamingResponse.__init__Nr   r   r)   r)   r)   r*   r-   w  r   r-   c                   @  r   )	r   r   r!   r#   r   c                 C  r   r   )r   r   rT   r^   rd   rw   r|   r   r)   r)   r*   r     r   z+AsyncMessagesWithStreamingResponse.__init__Nr   r   r)   r)   r)   r*   r     r   r   )5
__future__r   typingr   r   r   typing_extensionsr   httpx r   _typesr	   r
   r   r   r   _utilsr   r   _compatr   	_resourcer   r   	_responser   r   
paginationr   r   _base_clientr   r   types.beta.threadsr   r   r   types.beta.threads.messager   types.shared_params.metadatar   "types.beta.threads.message_deletedr   -types.beta.threads.message_content_part_paramr   __all__r    r!   r$   r   r-   r   r)   r)   r)   r*   <module>   s8       