a
    f                     @  sl  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!m"Z" ddl#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l.m/Z/ ddgZ0G dd deZ1G dd deZ2G dd dZ3G dd dZ4G dd dZ5G dd dZ6d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)assistant_list_paramsassistant_create_paramsassistant_update_params)AsyncPaginatormake_request_options)	ChatModel)	Assistant)AssistantDeleted)AssistantToolParam)"AssistantResponseFormatOptionParam
AssistantsAsyncAssistantsc                   @  s>  e Zd ZeddddZeddddZeeeeeeeeeddded	d
dddddddddddddddddZdddeddddddddddZeeeeeeeeeedddedddddddddd dddddddd!d"d#Z	eeeeddded$ddd%d&ddddd'd(	d)d*Z
dddeddddddd+dd,d-ZdS ).r!   AssistantsWithRawResponsereturnc                 C  s   t | S N)r#   self r)   Y/var/www/ai-form-bot/venv/lib/python3.9/site-packages/openai/resources/beta/assistants.pywith_raw_response$   s    zAssistants.with_raw_responseAssistantsWithStreamingResponsec                 C  s   t | S r&   )r,   r'   r)   r)   r*   with_streaming_response(   s    z"Assistants.with_streaming_responseNdescriptioninstructionsmetadatanameresponse_formattemperaturetool_resourcestoolstop_pextra_headersextra_query
extra_bodytimeoutUnion[str, ChatModel]Optional[str] | NotGivenOptional[object] | NotGiven7Optional[AssistantResponseFormatOptionParam] | NotGivenOptional[float] | NotGiven:Optional[assistant_create_params.ToolResources] | NotGiven'Iterable[AssistantToolParam] | NotGivenHeaders | NoneQuery | NoneBody | None'float | httpx.Timeout | None | NotGivenr   modelr/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r%   c                C  sL   ddi|pi }| j dt|||||||||	|
d
tjt||||dtdS )  
        Create an assistant with a model and instructions.

        Args:
          model: ID of the model to use. You can use the
              [List models](https://platform.openai.com/docs/api-reference/models/list) API to
              see all of your available models, or see our
              [Model overview](https://platform.openai.com/docs/models/overview) for
              descriptions of them.

          description: The description of the assistant. The maximum length is 512 characters.

          instructions: The system instructions that the assistant uses. The maximum length is 256,000
              characters.

          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 maxium of 512
              characters long.

          name: The name of the assistant. The maximum length is 256 characters.

          response_format: Specifies the format that the model must output. Compatible with
              [GPT-4o](https://platform.openai.com/docs/models/gpt-4o),
              [GPT-4 Turbo](https://platform.openai.com/docs/models/gpt-4-turbo-and-gpt-4),
              and all GPT-3.5 Turbo models since `gpt-3.5-turbo-1106`.

              Setting to `{ "type": "json_schema", "json_schema": {...} }` enables Structured
              Outputs which guarantees the model will match your supplied JSON schema. Learn
              more in the
              [Structured Outputs guide](https://platform.openai.com/docs/guides/structured-outputs).

              Setting to `{ "type": "json_object" }` enables JSON mode, which guarantees the
              message the model generates is valid JSON.

              **Important:** when using JSON mode, you **must** also instruct the model to
              produce JSON yourself via a system or user message. Without this, the model may
              generate an unending stream of whitespace until the generation reaches the token
              limit, resulting in a long-running and seemingly "stuck" request. Also note that
              the message content may be partially cut off if `finish_reason="length"`, which
              indicates the generation exceeded `max_tokens` or the conversation exceeded the
              max context length.

          temperature: What sampling temperature to use, between 0 and 2. Higher values like 0.8 will
              make the output more random, while lower values like 0.2 will make it more
              focused and deterministic.

          tool_resources: A set of resources that are used by the assistant's tools. The resources are
              specific to the type of tool. For example, the `code_interpreter` tool requires
              a list of file IDs, while the `file_search` tool requires a list of vector store
              IDs.

          tools: A list of tool enabled on the assistant. There can be a maximum of 128 tools per
              assistant. Tools can be of types `code_interpreter`, `file_search`, or
              `function`.

          top_p: An alternative to sampling with temperature, called nucleus sampling, where the
              model considers the results of the tokens with top_p probability mass. So 0.1
              means only the tokens comprising the top 10% probability mass are considered.

              We generally recommend altering this or temperature but not both.

          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
        OpenAI-Betaassistants=v2/assistants
rH   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   bodyoptionscast_to)_postr   r   AssistantCreateParamsr   r   r(   rH   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r)   r)   r*   create,   s,    ZzAssistants.createrN   strassistant_idr8   r9   r:   r;   r%   c                C  sD   |st d|ddi|pi }| jd| t||||dtdS )K  
        Retrieves an assistant.

        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
        ;Expected a non-empty value for `assistant_id` but received rJ   rK   /assistants/rN   rQ   rR   
ValueError_getr   r   r(   rY   r8   r9   r:   r;   r)   r)   r*   retrieve   s    zAssistants.retriever/   r0   r1   rH   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   str | NotGiven:Optional[assistant_update_params.ToolResources] | NotGivenrY   r/   r0   r1   rH   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r%   c                C  sd   |st d|ddi|pi }| jd| t||||||||	|
|d
tjt||||dtdS )  Modifies an assistant.

        Args:
          description: The description of the assistant.

        The maximum length is 512 characters.

          instructions: The system instructions that the assistant uses. The maximum length is 256,000
              characters.

          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 maxium of 512
              characters long.

          model: ID of the model to use. You can use the
              [List models](https://platform.openai.com/docs/api-reference/models/list) API to
              see all of your available models, or see our
              [Model overview](https://platform.openai.com/docs/models/overview) for
              descriptions of them.

          name: The name of the assistant. The maximum length is 256 characters.

          response_format: Specifies the format that the model must output. Compatible with
              [GPT-4o](https://platform.openai.com/docs/models/gpt-4o),
              [GPT-4 Turbo](https://platform.openai.com/docs/models/gpt-4-turbo-and-gpt-4),
              and all GPT-3.5 Turbo models since `gpt-3.5-turbo-1106`.

              Setting to `{ "type": "json_schema", "json_schema": {...} }` enables Structured
              Outputs which guarantees the model will match your supplied JSON schema. Learn
              more in the
              [Structured Outputs guide](https://platform.openai.com/docs/guides/structured-outputs).

              Setting to `{ "type": "json_object" }` enables JSON mode, which guarantees the
              message the model generates is valid JSON.

              **Important:** when using JSON mode, you **must** also instruct the model to
              produce JSON yourself via a system or user message. Without this, the model may
              generate an unending stream of whitespace until the generation reaches the token
              limit, resulting in a long-running and seemingly "stuck" request. Also note that
              the message content may be partially cut off if `finish_reason="length"`, which
              indicates the generation exceeded `max_tokens` or the conversation exceeded the
              max context length.

          temperature: What sampling temperature to use, between 0 and 2. Higher values like 0.8 will
              make the output more random, while lower values like 0.2 will make it more
              focused and deterministic.

          tool_resources: A set of resources that are used by the assistant's tools. The resources are
              specific to the type of tool. For example, the `code_interpreter` tool requires
              a list of file IDs, while the `file_search` tool requires a list of vector store
              IDs.

          tools: A list of tool enabled on the assistant. There can be a maximum of 128 tools per
              assistant. Tools can be of types `code_interpreter`, `file_search`, or
              `function`.

          top_p: An alternative to sampling with temperature, called nucleus sampling, where the
              model considers the results of the tokens with top_p probability mass. So 0.1
              means only the tokens comprising the top 10% probability mass are considered.

              We generally recommend altering this or temperature but not both.

          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
        r[   rJ   rK   r\   
r/   r0   r1   rH   r2   r3   r4   r5   r6   r7   rN   rO   )r_   rS   r   r   AssistantUpdateParamsr   r   r(   rY   r/   r0   r1   rH   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r)   r)   r*   update   s0    \zAssistants.updateafterbeforelimitorderr8   r9   r:   r;   int | NotGiven!Literal['asc', 'desc'] | NotGivenzSyncCursorPage[Assistant]	rm   rn   ro   rp   r8   r9   r:   r;   r%   c          	      C  sF   ddi|pi }| j dtt t||||t||||dtjdtdS a  Returns a list of assistants.

        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.

          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   ro   rp   )r8   r9   r:   r;   query)pagerQ   rH   )_get_api_listr   r   r   r   r   AssistantListParams	r(   rm   rn   ro   rp   r8   r9   r:   r;   r)   r)   r*   list6  s(    +zAssistants.listr   c                C  sD   |st d|ddi|pi }| jd| t||||dtdS )H  
        Delete an assistant.

        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
        r[   rJ   rK   r\   rN   r]   r_   _deleter   r   ra   r)   r)   r*   deletew  s    zAssistants.delete__name__
__module____qualname__r   r+   r-   r	   rV   rb   rk   rz   r~   r)   r)   r)   r*   r!   #   sh   .x&0y"Gc                   @  s>  e Zd ZeddddZeddddZeeeeeeeeeddded	d
dddddddddddddddddZdddeddddddddddZeeeeeeeeeedddedddddddddd dddddddd!d"d#Z	eeeeddded$ddd%d&ddddd'd(	d)d*Z
dddeddddddd+dd,d-ZdS ).r"   AsyncAssistantsWithRawResponser$   c                 C  s   t | S r&   )r   r'   r)   r)   r*   r+     s    z!AsyncAssistants.with_raw_response$AsyncAssistantsWithStreamingResponsec                 C  s   t | S r&   )r   r'   r)   r)   r*   r-     s    z'AsyncAssistants.with_streaming_responseNr.   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   r   rG   c                  sX   ddi|pi }| j dt|||||||||	|
d
tjI dH t||||dtdI dH S )rI   rJ   rK   rL   rM   NrN   rO   )rS   r   r   rT   r   r   rU   r)   r)   r*   rV     s,    ZzAsyncAssistants.createrN   rW   rX   c                  sJ   |st d|ddi|pi }| jd| t||||dtdI dH S )rZ   r[   rJ   rK   r\   rN   r]   Nr^   ra   r)   r)   r*   rb     s    zAsyncAssistants.retrieverc   rd   re   rf   c                  sp   |st d|ddi|pi }| jd| t||||||||	|
|d
tjI dH t||||dtdI dH S )	rg   r[   rJ   rK   r\   rh   NrN   rO   )r_   rS   r   r   ri   r   r   rj   r)   r)   r*   rk   7  s0    \zAsyncAssistants.updaterl   rq   rr   z5AsyncPaginator[Assistant, AsyncCursorPage[Assistant]]rs   c          	      C  sF   ddi|pi }| j dtt t||||t||||dtjdtdS rt   )rw   r   r   r   r   r   rx   ry   r)   r)   r*   rz     s(    +zAsyncAssistants.listr   c                  sJ   |st d|ddi|pi }| jd| t||||dtdI dH S )r{   r[   rJ   rK   r\   rN   r]   Nr|   ra   r)   r)   r*   r~     s    zAsyncAssistants.deleter   r)   r)   r)   r*   r"     sh   .x&0y"Gc                   @  s   e Zd ZdddddZdS )r#   r!   None
assistantsr%   c                 C  sP   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S r&   )_assistantsr   Zto_raw_response_wrapperrV   rb   rk   rz   r~   r(   r   r)   r)   r*   __init__  s     z"AssistantsWithRawResponse.__init__Nr   r   r   r   r)   r)   r)   r*   r#     s   r#   c                   @  s   e Zd ZdddddZdS )r   r"   r   r   c                 C  sP   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S r&   )r   r   Zasync_to_raw_response_wrapperrV   rb   rk   rz   r~   r   r)   r)   r*   r   '  s     z'AsyncAssistantsWithRawResponse.__init__Nr   r)   r)   r)   r*   r   &  s   r   c                   @  s   e Zd ZdddddZdS )r,   r!   r   r   c                 C  sF   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S r&   )r   r   rV   rb   rk   rz   r~   r   r)   r)   r*   r   <  s     z(AssistantsWithStreamingResponse.__init__Nr   r)   r)   r)   r*   r,   ;  s   r,   c                   @  s   e Zd ZdddddZdS )r   r"   r   r   c                 C  sF   || _ t|j| _t|j| _t|j| _t|j| _t|j| _d S r&   )r   r   rV   rb   rk   rz   r~   r   r)   r)   r*   r   Q  s     z-AsyncAssistantsWithStreamingResponse.__init__Nr   r)   r)   r)   r*   r   P  s   r   )7
__future__r   typingr   r   r   typing_extensionsr   Zhttpx r   _typesr	   r
   r   r   r   Z_utilsr   r   _compatr   Z	_resourcer   r   	_responser   r   Z
paginationr   r   Z
types.betar   r   r   Z_base_clientr   r   Ztypes.chat_modelr   Ztypes.beta.assistantr   Ztypes.beta.assistant_deletedr   Ztypes.beta.assistant_tool_paramr   Z1types.beta.assistant_response_format_option_paramr    __all__r!   r"   r#   r   r,   r   r)   r)   r)   r*   <module>   s8     y  y