@g ddlmZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl mZd d lmZmZd d l mZdd lmZdd lmZd dlmZerddlmZddZGddeZGddZdS)) annotationsN) TYPE_CHECKINGOptionalcast)ArgumentParser)partial) Completion) get_client) NOT_GIVEN NotGivenOr)is_given)CLIError) BaseModel)Stream)_SubParsersAction subparser!_SubParsersAction[ArgumentParser]returnNonec|d}|dddd|ddd |d d d |dddt|dddt|dddt|dddt|ddt|ddt|d d!d |d"d#t|d$d%t|d&d' |d(d) |d*d+ |t jt,dS)-Nzcompletions.createz-mz--modelzThe model to useT)helprequiredz-pz--promptz#An optional prompt to complete from)rz--streamzStream tokens as they're ready. store_true)ractionz-Mz --max-tokensz(The maximum number of tokens to generate)rtypez-tz --temperaturezWhat 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_paAn 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 --logprobsaInclude 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_ofzGenerates `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_penaltyzPositive 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_penaltyzPositive 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_argumentintfloat set_defaultsCLICompletionscreateCLICompletionCreateArgs)rsubs O/var/www/piapp/venv/lib/python3.11/site-packages/openai/cli/_api/completions.pyregisterr*st   3 4 4C   T:,QRRRZ&GP\]]]T>0Zadeee $  6   D   T  d   A m  `  Z&bcccX$YZZZ q ./1 with the current CLI)r2r5r3r4r.r8r1r6r7r9r:r;r<r=T)r0)rr2r0rrr completionsr&r5r3r4r.r8r1r6r7r9r:r;r<r=r%_stream_createrrr _create)rD make_requests r)r&zCLICompletions.createqs DF   U t{ STT T LL $ +f**;;L](!2"4   $ ; !00VJ'T)B)B)BCC  %%llnn555r+ completionr ct|jdk}|jD]}|r7tjd|jtj|j|s|jdstjdtj dS)NrFz===== Completion {} =====  ) lenchoicessysstdoutwriteformatindextextendswithflush)rKshould_print_headerchoices r)rIzCLICompletions._creates!*"4559 (  F" U   !>!E!Efl!S!STTT J  V[ ) ) )" '&+*>*>t*D*D '   &&& J        r+r0Stream[Completion]c|D]}t|jdk}t|jdD]}|r7tjd|jtj|j|rtjdtj tjddS)NrFc|jS)N)rT)cs r)z/CLICompletions._stream_create..s17r+)keyz===== Chat Completion {} ===== rM) rNrOsortedrPrQrRrSrTrUrW)r0rKrXrYs r)rHzCLICompletions._stream_creates # #J"%j&8"9"9A"=  !39J9JKKK # #&^J$$%G%N%Nv|%\%\]]]   ---&+J$$T***   """" # r+N)rDr'rr)rKr rr)r0rZrr)r>r?r@ staticmethodr&rIrHrBr+r)r%r%psr666\6<   \ \r+r%)rrrr) __future__rrPtypingrrrargparser functoolsropenai.types.completionr _utilsr _typesr rr_errorsr_modelsr _streamingrrr*r'r%rBr+r)rlsy"""""" 0000000000######......++++++++ +******EUEUEUEUP55555i555&==========r+