o
    #`^hj                     @  sT  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gZ,G dd deZ-G dd deZ.G dd dZ/G dd dZ0G dd dZ1G dd dZ2d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)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
        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(   d/home/air/shanriGPT/back/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']r1   ?Optional[Iterable[message_create_params.Attachment]] | NotGivenr2   Optional[object] | NotGivenr3   Headers | Noner4   Query | Noner5   Body | Noner6   '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. Keys
              can be a maximum of 64 characters long and values can be a maximum of 512
              characters long.

          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r9   r;   r1   r2   r3   r4   r5   r6   bodyoptionscast_to)
ValueError_postr   r   MessageCreateParamsr   r   
r'   r7   r9   r;   r1   r2   r3   r4   r5   r6   r(   r(   r)   create4   s$   ,
	zMessages.createrJ   
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
        rD   9Expected a non-empty value for `message_id` but received rE   rF   rG   
/messages/rJ   rM   rN   rO   _getr   r   r'   rT   r7   r3   r4   r5   r6   r(   r(   r)   retrievet      zMessages.retriever2   r3   r4   r5   r6   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. Keys
              can be a maximum of 64 characters long and values can be a maximum of 512
              characters long.

          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
        rD   rW   rE   rF   rG   rX   r2   rJ   rK   )rO   rP   r   r   MessageUpdateParamsr   r   r'   rT   r7   r2   r3   r4   r5   r6   r(   r(   r)   update   s   zMessages.update	afterbeforelimitorderrun_idr3   r4   r5   r6   re   str | NotGivenrf   rg   int | NotGivenrh   !Literal['asc', 'desc'] | NotGivenri   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 	af  
        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,
              ending 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
        rD   rE   rF   rG   rH   )re   rf   rg   rh   ri   )r3   r4   r5   r6   query)pagerM   model)rO   _get_api_listr   r   r   r   r   MessageListParamsr'   r7   re   rf   rg   rh   ri   r3   r4   r5   r6   r(   r(   r)   list   .   .
zMessages.listr   c                C  rU   )	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
        rD   rW   rE   rF   rG   rX   rJ   rY   rO   _deleter   r   r\   r(   r(   r)   delete  r^   zMessages.delete)r"   r#   )r"   r,   r7   r8   r9   r:   r;   r<   r1   r=   r2   r>   r3   r?   r4   r@   r5   rA   r6   rB   r"   r   rT   r8   r7   r8   r3   r?   r4   r@   r5   rA   r6   rB   r"   r   rT   r8   r7   r8   r2   r>   r3   r?   r4   r@   r5   rA   r6   rB   r"   r   )r7   r8   re   rj   rf   rj   rg   rk   rh   rl   ri   rj   r3   r?   r4   r@   r5   rA   r6   rB   r"   rm   rT   r8   r7   r8   r3   r?   r4   r@   r5   rA   r6   rB   r"   r   __name__
__module____qualname__r   r*   r.   r	   rS   r]   rc   rv   r{   r(   r(   r(   r)   r       L    	G*0Nc                
   @  r!   )<r    r"   AsyncMessagesWithRawResponsec                 C  r$   r%   )r   r&   r(   r(   r)   r*   3  r+   zAsyncMessages.with_raw_response"AsyncMessagesWithStreamingResponsec                 C  r$   r-   )r   r&   r(   r(   r)   r.   =  r/   z%AsyncMessages.with_streaming_responseNr0   r7   r8   r9   r:   r;   r<   r1   r=   r2   r>   r3   r?   r4   r@   r5   rA   r6   rB   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 )
rC   rD   rE   rF   rG   rH   rI   NrJ   rK   )rO   rP   r   r   rQ   r   r   rR   r(   r(   r)   rS   F  s&   ,
	zAsyncMessages.createrJ   rT   c                  d   |s
t d||st d|ddi|pi }| jd| d| t||||dtdI d	H S )
rV   rD   rW   rE   rF   rG   rX   rJ   rY   NrZ   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 )r`   rD   rW   rE   rF   rG   rX   r2   NrJ   rK   )rO   rP   r   r   ra   r   r   rb   r(   r(   r)   rc     s   zAsyncMessages.updaterd   re   rj   rf   rg   rk   rh   rl   ri   1AsyncPaginator[Message, AsyncCursorPage[Message]]c       	         C  rn   ro   )rO   rs   r   r   r   r   r   rt   ru   r(   r(   r)   rv     rw   zAsyncMessages.listr   c                  r   )
rx   rD   rW   rE   rF   rG   rX   rJ   rY   Nry   r\   r(   r(   r)   r{     r   zAsyncMessages.delete)r"   r   )r"   r   r|   r}   r~   )r7   r8   re   rj   rf   rj   rg   rk   rh   rl   ri   rj   r3   r?   r4   r@   r5   rA   r6   rB   r"   r   r   r   r(   r(   r(   r)   r    2  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_wrapperrS   r]   rc   rv   r{   r'   r   r(   r(   r)   __init__E      
z MessagesWithRawResponse.__init__Nr   r   r"   r   r   r   r   r   r(   r(   r(   r)   r#   D      r#   c                   @  r   )	r   r   r    r"   r   c                 C  r   r   )r   r   async_to_raw_response_wrapperrS   r]   rc   rv   r{   r   r(   r(   r)   r   Z  r   z%AsyncMessagesWithRawResponse.__init__Nr   r    r"   r   r   r(   r(   r(   r)   r   Y  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   rS   r]   rc   rv   r{   r   r(   r(   r)   r   o      
z&MessagesWithStreamingResponse.__init__Nr   r   r(   r(   r(   r)   r,   n  r   r,   c                   @  r   )	r   r   r    r"   r   c                 C  r   r   )r   r   rS   r]   rc   rv   r{   r   r(   r(   r)   r     r   z+AsyncMessagesWithStreamingResponse.__init__Nr   r   r(   r(   r(   r)   r     r   r   )3
__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.beta.threads.message_deletedr   -types.beta.threads.message_content_part_paramr   __all__r   r    r#   r   r,   r   r(   r(   r(   r)   <module>   s6       