
    2ix                       d dl mZ d dl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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Z0 G d de          Z1 G d de          Z2 G d d          Z3 G d d          Z4 G d d          Z5 G d d          Z6dS )    )annotationsN)UnionIterableOptional)Literal   )_legacy_response)BodyOmitQueryHeadersNotGivenomit	not_given)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dZed0d            Zed1d            Z ej        d          e	e	ddde
d	d2d            Z ej        d          ddde
dd3d             Z ej        d          e	ddde
d!d4d"            Z ej        d          e	e	e	e	e	ddde
d#	d5d-            Z ej        d          ddde
dd6d/            ZdS )7r#   4Build Assistants that can call models and use tools.returnMessagesWithRawResponsec                     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(   selfs    m/root/.openclaw/workspace/.venv-openai/lib/python3.11/site-packages/openai/resources/beta/threads/messages.pywith_raw_responsezMessages.with_raw_response"   s     't,,,    MessagesWithStreamingResponsec                     t          |           S 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
        )r0   r+   s    r-   with_streaming_responsez Messages.with_streaming_response,   s     -T222r/   >The Assistants API is deprecated in favor of the Responses APINattachmentsmetadataextra_headersextra_query
extra_bodytimeout	thread_idstrcontent-Union[str, Iterable[MessageContentPartParam]]roleLiteral['user', 'assistant']r6   ;Optional[Iterable[message_create_params.Attachment]] | Omitr7   Optional[Metadata] | Omitr8   Headers | Noner9   Query | Noner:   Body | Noner;   'float | httpx.Timeout | None | NotGivenr   c          
         |st          d|          ddi|pi }|                     d| dt          ||||dt          j                  t          ||||	          t                    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@   r6   r7   r8   r9   r:   r;   bodyoptionscast_to)
ValueError_postr   r   MessageCreateParamsr   r   
r,   r<   r>   r@   r6   r7   r8   r9   r:   r;   s
             r-   createzMessages.create5   s    ^  	geXaeefff&QM<ORQzz,	,,, & #. (	  &9  )+Q[el     
 
 	
r/   rP   
message_idc          	         |st          d|          |st          d|          ddi|pi }|                     d| d| t          ||||          t                    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
        rJ   9Expected a non-empty value for `message_id` but received rK   rL   rM   
/messages/rP   rS   rT   rU   _getr   r   r,   rZ   r<   r8   r9   r:   r;   s          r-   retrievezMessages.retrievex   s    2  	geXaeefff 	igYcgghhh&QM<ORQyy9	99Z99(+Q[el     
 
 	
r/   r7   r8   r9   r:   r;   c          
         |st          d|          |st          d|          ddi|pi }|                     d| d| t          d|it          j                  t          ||||          t          	          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
        rJ   r]   rK   rL   rM   r^   r7   rP   rQ   )rU   rV   r   r   MessageUpdateParamsr   r   r,   rZ   r<   r7   r8   r9   r:   r;   s           r-   updatezMessages.update   s    B  	geXaeefff 	igYcgghhh&QM<ORQzz9	99Z99 *h!79N9bcc(+Q[el     
 
 	
r/   	afterbeforelimitorderrun_idr8   r9   r:   r;   rk   
str | Omitrl   rm   
int | Omitrn   Literal['asc', 'desc'] | Omitro   SyncCursorPage[Message]c       	            |st          d|          ddi|pi }|                     d| dt          t                   t	          |||	|
t          |||||dt          j                            t                    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
        rJ   rK   rL   rM   rN   )rk   rl   rm   rn   ro   )r8   r9   r:   r;   query)pagerS   model)rU   _get_api_listr   r   r   r   r   MessageListParamsr,   r<   rk   rl   rm   rn   ro   r8   r9   r:   r;   s              r-   listzMessages.list   s    ^  	geXaeefff&QM<ORQ!!,	,,,((+'%%!&"(!&!&"(  (9	 	    ' " 
 
 	
r/   r!   c          	         |st          d|          |st          d|          ddi|pi }|                     d| d| t          ||||          t                    S )	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
        rJ   r]   rK   rL   rM   r^   rP   r_   rU   _deleter   r!   rb   s          r-   deletezMessages.delete  s    2  	geXaeefff 	igYcgghhh&QM<ORQ||9	99Z99(+Q[el   #  
 
 	
r/   )r'   r(   )r'   r0   r<   r=   r>   r?   r@   rA   r6   rB   r7   rC   r8   rD   r9   rE   r:   rF   r;   rG   r'   r   rZ   r=   r<   r=   r8   rD   r9   rE   r:   rF   r;   rG   r'   r   rZ   r=   r<   r=   r7   rC   r8   rD   r9   rE   r:   rF   r;   rG   r'   r   )r<   r=   rk   rp   rl   rp   rm   rq   rn   rr   ro   rp   r8   rD   r9   rE   r:   rF   r;   rG   r'   rs   rZ   r=   r<   r=   r8   rD   r9   rE   r:   rF   r;   rG   r'   r!   __name__
__module____qualname____doc__r   r.   r3   typing_extensions
deprecatedr   r   rY   rc   ri   r|   r    r/   r-   r#   r#      s       >>- - - _- 3 3 3 _3 "!"bcc TX.2 )-$("&;D@
 @
 @
 @
 @
 dc@
D "!"bcc )-$("&;D#
 #
 #
 #
 #
 dc#
J "!"bcc /3 )-$("&;D,
 ,
 ,
 ,
 ,
 dc,
\ "!"bcc
 !! /3! )-$("&;DE
 E
 E
 E
 E
 dcE
N "!"bcc )-$("&;D#
 #
 #
 #
 #
 dc#
 #
 #
r/   c                     e Zd ZdZed0d            Zed1d            Z ej        d          e	e	ddde
d	d2d            Z ej        d          ddde
dd3d             Z ej        d          e	ddde
d!d4d"            Z ej        d          e	e	e	e	e	ddde
d#	d5d-            Z ej        d          ddde
dd6d/            ZdS )7r$   r&   r'   AsyncMessagesWithRawResponsec                     t          |           S r*   )r   r+   s    r-   r.   zAsyncMessages.with_raw_response?  s     ,D111r/   "AsyncMessagesWithStreamingResponsec                     t          |           S r2   )r   r+   s    r-   r3   z%AsyncMessages.with_streaming_responseI  s     2$777r/   r4   Nr5   r<   r=   r>   r?   r@   rA   r6   rB   r7   rC   r8   rD   r9   rE   r:   rF   r;   rG   r   c          
        K   |st          d|          ddi|pi }|                     d| dt          ||||dt          j                   d{V t          ||||	          t          	           d{V S )
rI   rJ   rK   rL   rM   rN   rO   NrP   rQ   )rU   rV   r   r   rW   r   r   rX   s
             r-   rY   zAsyncMessages.createR  s      ^  	geXaeefff&QM<ORQZZ,	,,,,& #. (	  &9        )+Q[el      
 
 
 
 
 
 
 
 	
r/   rP   rZ   c          	        K   |st          d|          |st          d|          ddi|pi }|                     d| d| t          ||||          t                     d	{V S )
r\   rJ   r]   rK   rL   rM   r^   rP   r_   Nr`   rb   s          r-   rc   zAsyncMessages.retrieve  s      2  	geXaeefff 	igYcgghhh&QM<ORQYY9	99Z99(+Q[el     
 
 
 
 
 
 
 
 	
r/   rd   c          
       K   |st          d|          |st          d|          ddi|pi }|                     d| d| t          d|it          j                   d{V t          ||||	          t          
           d{V S )rf   rJ   r]   rK   rL   rM   r^   r7   NrP   rQ   )rU   rV   r   r   rg   r   r   rh   s           r-   ri   zAsyncMessages.update  s      B  	geXaeefff 	igYcgghhh&QM<ORQZZ9	99Z99,j(-CEZEnoooooooo(+Q[el      
 
 
 
 
 
 
 
 	
r/   rj   rk   rp   rl   rm   rq   rn   rr   ro   1AsyncPaginator[Message, AsyncCursorPage[Message]]c       	            |st          d|          ddi|pi }|                     d| dt          t                   t	          |||	|
t          |||||dt          j                            t                    S ru   )rU   ry   r   r   r   r   r   rz   r{   s              r-   r|   zAsyncMessages.list  s    ^  	geXaeefff&QM<ORQ!!,	,,, )(+'%%!&"(!&!&"(  (9	 	    ' " 
 
 	
r/   r!   c          	        K   |st          d|          |st          d|          ddi|pi }|                     d| d| t          ||||          t                     d	{V S )
r~   rJ   r]   rK   rL   rM   r^   rP   r_   Nr   rb   s          r-   r   zAsyncMessages.delete2  s      2  	geXaeefff 	igYcgghhh&QM<ORQ\\9	99Z99(+Q[el   # " 
 
 
 
 
 
 
 
 	
r/   )r'   r   )r'   r   r   r   r   )r<   r=   rk   rp   rl   rp   rm   rq   rn   rr   ro   rp   r8   rD   r9   rE   r:   rF   r;   rG   r'   r   r   r   r   r/   r-   r$   r$   <  s       >>2 2 2 _2 8 8 8 _8 "!"bcc TX.2 )-$("&;D@
 @
 @
 @
 @
 dc@
D "!"bcc )-$("&;D#
 #
 #
 #
 #
 dc#
J "!"bcc /3 )-$("&;D,
 ,
 ,
 ,
 ,
 dc,
\ "!"bcc
 !! /3! )-$("&;DE
 E
 E
 E
 E
 dcE
N "!"bcc )-$("&;D#
 #
 #
 #
 #
 dc#
 #
 #
r/   c                      e Zd ZddZdS )r(   messagesr#   r'   Nonec                @   || _         t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        d S N)	_messagesr	   to_raw_response_wrapperrY   rc   ri   r|   r   r,   r   s     r-   __init__z MessagesWithRawResponse.__init__Z  s    ! 4  	 4!  	 4  	 4  		 4  	r/   Nr   r#   r'   r   r   r   r   r   r   r/   r-   r(   r(   Y  (        
 
 
 
 
 
r/   r(   c                      e Zd ZddZdS )r   r   r$   r'   r   c                @   || _         t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        t          j        |j                  | _        d S r   )r   r	   async_to_raw_response_wrapperrY   rc   ri   r|   r   r   s     r-   r   z%AsyncMessagesWithRawResponse.__init__y  s    ! :  	 :!  	 :  	 :  		 :  	r/   Nr   r$   r'   r   r   r   r/   r-   r   r   x  r   r/   r   c                      e Zd ZddZdS )r0   r   r#   r'   r   c                   || _         t          |j                  | _        t          |j                  | _        t          |j                  | _        t          |j                  | _        t          |j                  | _        d S r   )r   r   rY   rc   ri   r|   r   r   s     r-   r   z&MessagesWithStreamingResponse.__init__  s    ! )  	 )!  	 )  	 )  		 )  	r/   Nr   r   r   r/   r-   r0   r0     r   r/   r0   c                      e Zd ZddZdS )r   r   r$   r'   r   c                   || _         t          |j                  | _        t          |j                  | _        t          |j                  | _        t          |j                  | _        t          |j                  | _        d S r   )r   r   rY   rc   ri   r|   r   r   s     r-   r   z+AsyncMessagesWithStreamingResponse.__init__  s    ! /  	 /!  	 /  	 /  		 /  	r/   Nr   r   r   r/   r-   r   r     r   r/   r   )7
__future__r   r   typingr   r   r   r   httpx r	   _typesr
   r   r   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   r0   r   r   r/   r-   <module>r      s   # " " " " "     , , , , , , , , , , % % % % % %  ! ! ! ! ! ! L L L L L L L L L L L L L L L L L L = = = = = = = = ' ' ' ' ' ' ; ; ; ; ; ; ; ; Z Z Z Z Z Z Z Z : : : : : : : :        e d d d d d d d d d 2 2 2 2 2 2 5 5 5 5 5 5 A A A A A A U U U U U U
'Z
 Z
 Z
 Z
 Z
 Z
 Z
 Z
zZ
 Z
 Z
 Z
 Z
$ Z
 Z
 Z
z
 
 
 
 
 
 
 
>
 
 
 
 
 
 
 
>
 
 
 
 
 
 
 
>
 
 
 
 
 
 
 
 
 
r/   