File: //usr/local/lib/python3.10/dist-packages/openai/cli/_api/__pycache__/completions.cpython-310.pyc
o
���g � @ s� d dl mZ d dlZd dlmZmZmZ d dlmZ d dl m
Z
d dl m Z ddl
mZ d d
lmZmZ d d l
mZ dd lmZ dd
lmZ d dlmZ erTd dlmZ ddd�ZG dd� de�ZG dd� d�Z dS )� )� annotationsN)�
TYPE_CHECKING�Optional�cast)�ArgumentParser)�partial)�
Completion� )�
get_client� )� NOT_GIVEN�
NotGivenOr)�is_given)�CLIError)� BaseModel)�Stream)�_SubParsersAction� subparser�!_SubParsersAction[ArgumentParser]�return�Nonec C s | � d�}|jddddd� |jddd d
� |jd d d
d� |jdddtd� |jdddtd� |jdddtd� |jdddtd� |jd d td� |jd d td� |jd d!d
d� |jd"d#td� |jd$d%td� |jd&d'd
� |jd(d)d
� |jd*d+d
� |jtjtd,� d S )-Nzcompletions.createz-mz--modelzThe model to useT)�help�requiredz-pz--promptz#An optional prompt to complete from)r z--streamz Stream tokens as they're ready.�
store_true)r �actionz-Mz --max-tokensz(The maximum number of tokens to generate)r �typez-tz
--temperaturez�What sampling temperature to use. Higher values means the model will take more risks. Try 0.9 for more creative applications, and 0 (argmax sampling) for ones with a well-defined answer.
Mutually exclusive with `top_p`.z-Pz--top_pa An alternative to sampling with temperature, called nucleus sampling, where the 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.
Mutually exclusive with `temperature`.z-nz--nz5How many sub-completions to generate for each prompt.z
--logprobsa Include the log probabilities on the `logprobs` most likely tokens, as well the chosen tokens. So for example, if `logprobs` is 10, the API will return a list of the 10 most likely tokens. If `logprobs` is 0, only the chosen tokens will have logprobs returned.z --best_ofz�Generates `best_of` completions server-side and returns the 'best' (the one with the highest log probability per token). Results cannot be streamed.z--echoz2Echo back the prompt in addition to the completionz--frequency_penaltyz�Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.z--presence_penaltyz�Positive values penalize new tokens based on whether they appear in the text so far, increasing the model's likelihood to talk about new topics.z--suffixz:The suffix that comes after a completion of inserted text.z--stopz3A stop sequence at which to stop generating tokens.z--userzbA unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse.)�func�
args_model)�
add_parser� add_argument�int�float� set_defaults�CLICompletions�create�CLICompletionCreateArgs)r �sub� r' �F/usr/local/lib/python3.10/dist-packages/openai/cli/_api/completions.py�register sx
����������r) c @ s� e Zd ZU ded< dZded< dZded< eZd ed
< eZd ed < eZ d ed
< eZ
ded< eZ d ed< eZ d ed< eZ
ded< eZd ed< eZd ed< eZded< eZded< eZded< dS )r% �str�modelF�bool�streamNz
Optional[str]�promptzNotGivenOr[int]�nzNotGivenOr[str]�stop�userzNotGivenOr[bool]�echo�suffix�best_ofzNotGivenOr[float]�top_p�logprobs�
max_tokens� temperature�presence_penalty�frequency_penalty)�__name__�
__module__� __qualname__�__annotations__r- r. r r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r' r' r' r( r% ] s
r% c @ s6 e Zd Zeddd��Zedd d
��Zedd
d��ZdS )r# �argsr% r r c C s� t | j�r| jdkr| jrtd��tt� jj| j| j| j | j
| j | j | j
| j| j| j| j| j| j| jd�}| jrEt�ttt |dd���S t�|� �S )N� z6Can't stream completions with n>1 with the current CLI)r/ r2 r0 r1 r+ r5 r. r3 r4 r6 r7 r8 r9 r: T)r- )r r/ r- r r r
� completionsr$ r2 r0 r1 r+ r5 r. r3 r4 r6 r7 r8 r9 r: r# �_stream_creater r r �_create)r? � make_requestr' r' r( r$ q s0 �� zCLICompletions.create�
completionr c C sj t | j�dk}| jD ](}|rtj�d�|j�� tj�|j� |s'|j�d�s-tj�d� tj� � q
d S )Nr@ z===== Completion {} =====
�
)
�len�choices�sys�stdout�write�format�index�text�endswith�flush)rE �should_print_header�choicer' r' r( rC � s
�zCLICompletions._creater- �Stream[Completion]c C s� | D ]5}t |j�dk}t|jdd� d�D ]"}|r"tj�d�|j�� tj�|j� |r1tj�d� tj� � qqtj�d� d S )Nr@ c S s | j S )N)rM )�cr' r' r( �<lambda>� s z/CLICompletions._stream_create.<locals>.<lambda>)�keyz ===== Chat Completion {} =====
rF )
rG rH �sortedrI rJ rK rL rM rN rP )r- rE rQ rR r' r' r( rB � s � z CLICompletions._stream_createN)r? r% r r )rE r r r )r- rS r r )r; r< r= � staticmethodr$ rC rB r' r' r' r( r# p s
r# )r r r r ) �
__future__r rI �typingr r r �argparser � functoolsr �openai.types.completionr �_utilsr
�_typesr r
r �_errorsr �_modelsr �
_streamingr r r) r% r# r' r' r' r( �<module> s"
H