a f g@s8ddlZddlZddlmZmZmZmZddlmZddlm Z GdddeZ ddZ d d Z d d gfd dZdN)AnyCallable NamedTupleOptional)assert_has_pandas)pandasc@seZdZUeed<dZeeed<dZeeed<dZee e ge fed<dZ eeed<dZ ee e ge fed<dZ eeed<dS) RemediationnameN immediate_msg necessary_msg necessary_fn optional_msg optional_fn error_msg)__name__ __module__ __qualname__str__annotations__r rr r rrr rrrrT/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/openai/validators.pyr s rcCs8d}t||krdnd}dt|d|}td|dS)z This validator will only print out the number of examples and recommend to the user to increase the number of examples if less than 100. dz. In general, we recommend having at least a few hundred examples. We've found that performance tends to linearly increase for every doubling of the number of examplesz - Your file contains z prompt-completion pairs num_examplesr r )lenr)dfZ MIN_EXAMPLESZoptional_suggestionr rrrnum_examples_validators rcsddd}d}d}d}|jvrndd|jDvrbfdd}|}dd }d d }n d d }td ||||dS)z[ This validator will ensure that the necessary column is present in the dataframe. cs2fdd|jD}|j|didd|S)Ncs g|]}t|kr|qSrrlower.0ccolumnrr )zInecessary_column_validator..lower_case_column..rT)columnsZinplace)r'renamer)rr$colsrr#rlower_case_column(sz5necessary_column_validator..lower_case_columnNcSsg|]}t|qSrrr rrrr%3r&z.necessary_column_validator..cs |SNr)rr*necessary_columnrrlower_case_column_creator5sz=necessary_column_validator..lower_case_column_creatorz - The `z ` column/key should be lowercasezLower case column name to ``z^` column/key is missing. Please make sure you name your columns/keys appropriately, then retryr-)r r r r r)r'r)rr-r r r rr.rr,rnecessary_column_validator#s(   r0prompt completioncsg}d}d}d}t|jdkrfdd|jD}d}|D]8fdd|D}t|dkr:|dd d 7}q:d ||}d |}fd d}td|||dS)zK This validator will remove additional columns from the dataframe. Ncsg|]}|vr|qSrrr fieldsrrr%Rr&z/additional_column_validator..rcsg|]}|vr|qSrrr )acrrr%Ur&rz9 WARNING: Some of the additional columns/keys contain `z<` in their name. These will be ignored, and the column/key `z`` will be used instead. This could also result from a duplicate column/key in the provided file.zh - The input file should contain exactly two columns/keys per row. Additional columns/keys present are: z Remove additional columns/keys: cs|Sr+rxr4rrr [sz1additional_column_validator..necessary_fnZadditional_columnr r r r )rr'r)rr5Zadditional_columnsr r r warn_messageZdupsr)r6r5radditional_column_validatorIs(   r;csd}d}d}|dds2|r|dk|B}|j|}dd|}fdd}d t|d d }td |||d S)zA This validator will ensure that no completion is empty. NcSs|dkS)Nrrr7rrrnr&z+non_empty_field_validator..rz - `z?` column/key should not contain empty strings. These are rows: cs||dkjgdS)NrZsubset)Zdropnar7fieldrrr ssz/non_empty_field_validator..necessary_fnRemove z rows with empty sZempty_r9)applyanyZisnull reset_indexindextolistrr)rr?r r r Z empty_rowsZ empty_indexesrr>rnon_empty_field_validatorfs& rGcs|jd}|j|}d}d}d}t|dkrrdt|ddd|}dt|d }fd d }td |||d S)zY This validator will suggest to the user to remove duplicate rows if they exist. r=Nr - There are z duplicated -z sets. These are rows: r@z duplicate rowscs |jdS)Nr=)Zdrop_duplicatesr7r4rrrsz.duplicated_rows_validator..optional_fnduplicated_rowsr r r r)Z duplicatedrDrErFrjoinr)rr5rJZduplicated_indexesr r rrr4rduplicated_rows_validators    rMcs|d}d}d}t|}|dkrldd|tdkrldtdd}d td }fd d }td |||dS)zW This validator will suggest to the user to remove examples that are too long. Nopen-ended generationcSs$|jdddd}|j|S)NcSst|jt|jdkS)Ni')rr1r2r7rrrr<r&zClong_examples_validator..get_long_indexes..)Zaxis)rBrDrErF)d long_examplesrrrget_long_indexessz1long_examples_validator..get_long_indexesrrHz. examples that are very long. These are rows: zf For conditional generation, and for classification the examples shouldn't be longer than 2048 tokens.r@z long examplescs8|}|kr.tjdt|d|d||S)NzeThe indices of the long examples has changed as a result of a previously applied recommendation. The z? long examples to be dropped are now at the following indices:  )sysstdoutwriterdrop)r8Zlong_indexes_to_droprRZ long_indexesrrrs z,long_examples_validator..optional_fnrQrK)infer_task_typerr)rr r rft_typerrXrlong_examples_validators"  r[c sxd}d}d}d}dgd}|D]@}|dkr@|jjdr@q |jjj|ddrXq |qbq dd}t|}|d krtd d Sd d t|jdd} |j| krd| d}td |dS| dkr>| dd} d| d}t | dkr|d|d7}|jjdt |  jj| ddrB|d| d7}nd}| dkrfd|d}fdd}td||||d S)!z This validator will suggest to add a common suffix to the prompt if one doesn't already exist in case of classification or conditional generation. Nz ### => ) ->z ### z === z --- z ===> z ---> r\rSFregex\nrN common_suffixr cSs|d|7<|SNr1rr8suffixrrr add_suffixsz2common_prompt_suffix_validator..add_suffixrdxfixzAll prompts are identical: `zt` Consider leaving the prompts blank if you want to do open-ended generation, otherwise ensure prompts are differentr rrz - All prompts end with suffix `r/ R. This suffix seems very long. Consider replacing with a shorter suffix, such as `z5 WARNING: Some of your prompts contain the suffix `zZ` more than once. We strongly suggest that you review your prompts and add a unique suffixa - Your data does not contain a common separator at the end of your prompts. Having a separator string appended to the end of the prompt makes it clearer to the fine-tuned model where the completion should begin. See https://platform.openai.com/docs/guides/fine-tuning/preparing-your-dataset for more detail and examples. If you intend to do open-ended generation, then you should leave the prompts emptyzAdd a suffix separator `z` to all promptscs |Sr+rr7reZsuggested_suffixrrrsz3common_prompt_suffix_validator..optional_fncommon_completion_suffixr r r rr) r1rcontainsrCreplacerYrget_common_xfixallr) rrr r rsuffix_options suffix_optiondisplay_suggested_suffixrZr`common_suffix_new_line_handledrrkrcommon_prompt_suffix_validators\         rvcsd}d}d}d}t|jdddkr0tddSdd |jkrPtddSdkrd d }|tkr|d 7}d d}fdd}td|||dS)zd This validator will suggest to remove a common prefix from the prompt if a long one exist. Nprefixrfr common_prefixracSs|djt|d|d<|Srbrr)r8rxrrrremove_common_prefixsz.remove_common_prefixz" - All prompts start with prefix `r/z. Fine-tuning doesn't require the instruction specifying the task, or a few-shot example scenario. Most of the time you should only add the input data into the prompt, and the desired output into the completionRemove prefix `z` from all promptscs |Sr+rr7ryr{rrr(sz3common_prompt_prefix_validator..optional_fnZcommon_prompt_prefixrK)rpr1rrqrrZMAX_PREFIX_LENr r rrr}rcommon_prompt_prefix_validator s,     rcsd}t|jddtdko(ddkt|kr@tddSdd |jkr`tddSd d }d d }fdd}td|||dS)zh This validator will suggest to remove a common prefix from the completion if a long one exist. rxrfr ryracSs2|djt|d|d<|r.d|d|d<|S)Nr2rrz)r8rx ws_prefixrrrr{>sz@common_completion_prefix_validator..remove_common_prefixz& - All completions start with prefix `z_`. Most of the time you should only add the output data into the completion, without any prefixr|z` from all completionscs |Sr+rr7ryr{rrrrLsz7common_completion_prefix_validator..optional_fnZcommon_completion_prefixrK)rpr2rrrqr~rrr"common_completion_prefix_validator3s"     rc shd}d}d}d}t|}|dks(|dkr2tddSt|jdd}|j|krld|d |d }td|d Sd gd }|D]$}|jjj|ddrq||qq|dd} dd|dkr.|dd} d| d }t |dkr|d| d 7}|jjdt | jj|ddr2|d|d7}nd}|dkrVd| d}fdd}td||||d S)!z This validator will suggest to add a common suffix to the completion if one doesn't already exist in case of classification or conditional generation. NrNclassificationr`rardrfz All completions are identical: `zJ` Ensure completions are different, otherwise the model will just repeat `r/rhz [END]) rS.z ENDz***z+++z&&&z$$$z@@@z%%%Fr]rSr_cSs|d|7<|SNr2rrcrrrre}sz6common_completion_suffix_validator..add_suffixrz$ - All completions end with suffix `rirjz9 WARNING: Some of your completions contain the suffix `zU` more than once. We suggest that you review your completions and add a unique endingaH - Your data does not contain a common ending at the end of your completions. Having a common ending string appended to the end of the completion makes it clearer to the fine-tuned model where the completion should end. See https://platform.openai.com/docs/guides/fine-tuning/preparing-your-dataset for more detail and examples.zAdd a suffix ending `z` to all completionscs |Sr+rr7rkrrrsz7common_completion_suffix_validator..optional_fnrlrm) rYrrpr2rqrrnrCror) rrr r rrZr`rrrsrtrurrkr"common_completion_suffix_validatorWsV          rcCs\dd}d}d}d}|jjdddks@|jjdddkrLd}d}|}td |||d S) z This validator will suggest to add a space at the start of the completion if it doesn't already exist. This helps with tokenization. cSs|ddd|d<|S)Nr2cSs|ddkrdnd|S)Nrrrrr7rrrr<r&zLcompletions_space_start_validator..add_space_start..)rBr7rrradd_space_startsz:completions_space_start_validator..add_space_startNrOrrz - The completion should start with a whitespace character (` `). This tends to produce better results due to the tokenization we use. See https://platform.openai.com/docs/guides/fine-tuning/preparing-your-dataset for more detailsz=Add a whitespace character to the beginning of the completionZcompletion_space_startrK)r2rnuniquevaluesr)rrr rr rrr!completions_space_start_validators,rcsnfdd}|dd}|dd}|d|krjtddd d d d |d SdS)zt This validator will suggest to lowercase the column values, if more than a third of letters are uppercase. cs|j|<|Sr+rr7r#rr lower_casesz(lower_case_validator..lower_casecSstdd|DS)Ncss"|]}|r|rdVqdSrON)isalphaisupperr rrr r&9lower_case_validator....sumr7rrrr<r&z&lower_case_validator..cSstdd|DS)Ncss"|]}|r|rdVqdSr)rislowerr rrrrr&rrr7rrrr<r&r3rz - More than a third of your `z%` column/key is uppercase. Uppercase zs tends to perform worse than a mixture of case encountered in normal language. We recommend to lower case the data if that makes sense in your domain. See https://platform.openai.com/docs/guides/fine-tuning/preparing-your-dataset for more detailsz'Lowercase all your data in column/key `r/rKN)rBrr)rr$rZ count_upperZ count_lowerrr#rlower_case_validators    rc Cstd}d}d}d}d}tj|rzJ|dsH|dr|drZdnd\}}d|d}d|d }tj||td  d }n|d rd }d}t |} | j } t | dkr|d7}tj |td d }n|dr\d}d}t|d>} | } tjdd| dD|td d }Wdn1sN0Yn|drtj|dtd d }t |dkrrd}d}tj|td d }nn|dr2zFtj|dtd d }t |dkrtj|td d }nd }d}Wn(ty.tj|td d }Yn0n@d!}d"|vrb|d#|d$|d"d%d&7}n|d#|d'7}WnBttfy|d"d%}d(|d)|d*|d+}Yn0n d,|d-}td.|||d/}||fS)0z This function will read a file saved in .csv, .json, .txt, .xlsx or .tsv format using pandas. - for .xlsx it will read the first sheet - for .txt it will assume completions and split on newline Nz.csvz.tsv)ZCSV,)ZTSV z= - Based on your file extension, your file is formatted as a z filez Your format `z` will be converted to `JSONL`)sepdtyperz.xlsxzH - Based on your file extension, your file is formatted as an Excel filez/Your format `XLSX` will be converted to `JSONL`rOz - Your Excel file contains more than one sheet. Please either save as csv or ensure all data is present in the first sheet. WARNING: Reading only the first sheet...)rz.txtz9 - Based on your file extension, you provided a text filez.Your format `TXT` will be converted to `JSONL`rcSsg|] }d|gqS)rr)r!linerrrr%r&z#read_any_format..rS)r'r.jsonlT)linesrz^ - Your JSONL file appears to be in a JSON format. Your file will be converted to JSONL formatz/Your format `JSON` will be converted to `JSONL`z.jsonz^ - Your JSON file appears to be in a JSONL format. Your file will be converted to JSONL formatz]Your file must have one of the following extensions: .CSV, .TSV, .XLSX, .TXT, .JSON or .JSONLrz Your file `z` ends with the extension `.z` which is not supported.z` is missing a file extension.z Your file `z!` does not appear to be in valid z9 format. Please ensure your file is formatted as a valid z file.zFile z does not exist.read_any_format)r r r r)rospathisfilerendswithpdZread_csvrZfillnaZ ExcelFileZ sheet_namesrZ read_excelopenreadZ DataFramesplit read_json ValueError TypeErrorupperr) fnamer5 remediationr r rrZfile_extension_str separatorZxlsZsheetsfcontentrrrrs     ( "  rcCs,t|}d}|dkr d|d}td|dS)z This validator will infer the likely fine-tuning format of the data, and display it to the user if it is classification. It will also suggest to use ada and explain train/validation split benefits. NrzK - Based on your data it seems like you're trying to fine-tune a model for z - For classification, we recommend you try one of the faster and cheaper models, such as `ada` - For classification, you can estimate the expected model performance by keeping a held out dataset, which is not used for trainingrr)rYr)rrZr rrrformat_inferrer_validator7s  rcCsb|jdur2tjd|jd|jdtd|jdurJtj|j|jdur^||}|S)zs This function will apply a necessary remediation to a dataframe, or print an error message if one exists. Nz ERROR in z validator: z Aborting...rO) rrTstderrrVr exitr rUr )rrrrrapply_necessary_remediationCs     rcCs.tj||r tjddStdkS)NzY Tn)rTrUrVinputr) input_text auto_acceptrrraccept_suggestionSs   rcCs\d}d|jd}|jdur4t||r4||}d}|jdurTtjd|jd||fS)zc This function will apply an optional remediation to a dataframe, based on the user input. Fz- [Recommended] z [Y/n]: NTz- [Necessary] rS)r rrr rTrUrV)rrroptional_appliedrrrrapply_optional_remediation[s    rcCsjt|}d}|dkr&t|}|d}n|jdd}|d}dd}||d }tjd |d d S) z? Estimate the time it'll take to fine-tune the dataset g?rg ףp= ?T)rEg|?5^?cSsh|dkrt|ddS|dkr4t|dddS|dkrPt|dddSt|dddSdS) N<r3z secondsiz minutesiQz hoursz days)round)timerrr format_timewsz.estimate_fine_tuning_time..format_timez:Once your model starts training, it'll approximately take z~ to train a `curie` model, and less for `ada` and `babbage`. Queue will approximately take half an hour per job ahead of you. N)rYrZ memory_usagerrTrUrV)r ft_formatZ expected_timersizerZ time_stringrrrestimate_fine_tuning_timejs   rcsf|r ddgndg}d}|dkr*d|dndfdd|D}td d |DsX|S|d 7}qdS) NZ_train_validrrz ()cs,g|]$}tjdd|dqS)rZ _preparedr)rrsplitext)r!rdrZ index_suffixrrr%sz!get_outfnames..css|]}tj|VqdSr+)rrr)r!rrrrrr&z get_outfnames..rO)rC)rrsuffixesiZcandidate_fnamesrrr get_outfnamess rcCs.|j}d}|dkr&|jjd}||fS)Nr3r)r2rZ value_countsrE)r n_classes pos_classrrrget_classification_hyperparamss  rc Cst|}t|jdd}t|jdd}d}d}|dkrBt||rBd}d} |dd } |dd } t| d krvd | d nd} d }|s|stj d|d| d| d| d t |nt||rtt ||} |rt| dkrd| d vrd| dvsJd}t t||t t|d}|j|dd}||j}|ddgj| d dddd|ddgj| dddddt|\}}| d7} |dkr| d|d7} n| d |7} n0t| dksJ|ddgj| d dddd|rd!ndd"d#| }|rd$| ddnd}t| d kr,dn d%| d}tj d&|d'| d d|| d(|| d t |n tj d)d*S)+aQ This function will write out a dataframe to a file, if the user would like to proceed, and also offer a fine-tuning command with the newly created file. For classification it will optionally ask the user if they would like to split the data into train/valid files, and modify the suggested command to include the valid set. rdrfFzQ- [Recommended] Would you like to split into training and validation set? [Y/n]: rTrrSr_rz Make sure to include `stop=["z;"]` so that the generated texts ends at the expected place.z@ Your data will be written to a new JSONL file. Proceed [Y/n]: zK You can use your file for fine-tuning: > openai api fine_tunes.create -t ""ue After you’ve fine-tuned a model, remember that your prompt has to end with the indicator string `zX` for the model to start generating completions, rather than continuing with the prompt.r3ZtrainZvalidrOig?*)rZ random_stater1r2records)rZorientZ force_asciiz! --compute_classification_metricsz" --classification_positive_class "z --classification_n_classes rAz to `z` and `z -v "ucAfter you’ve fine-tuned a model, remember that your prompt has to end with the indicator string `z Wrote modified filezd` Feel free to take a look! Now use that file when fine-tuning: > openai api fine_tunes.create -t "z z#Aborting... did not write the file N)rYrpr1r2rrorrTrUrVrrmaxintsamplerWrEto_jsonrrL)rrZany_remediationsrrZcommon_prompt_suffixrlrrZadditional_paramsZ%common_prompt_suffix_new_line_handledZ)common_completion_suffix_new_line_handledZoptional_ending_stringfnamesZMAX_VALID_EXAMPLESZn_trainZdf_trainZdf_validrrZ files_stringZ valid_stringZseparator_reminderrrrwrite_out_filesr      (           ( rcCs>d}t|jjdkrdSt|jt||kr:dSdS)z> Infer the likely fine-tuning task type from the data rrNrzconditional generation)rr1rrr2unique)rZCLASSIFICATION_THRESHOLDrrrrYs rYrdcCsld}|dkr$|jt|d dn|jdt|d}|dkrJqhq||jdkr\qhq|jd}q|S)zQ Finds the longest common suffix or prefix of all the values in a series rrdrONr)rrrr)ZseriesrgZ common_xfixZ common_xfixesrrrrps   rpcCs2tddddtttttddddtttt t gS)NcSs t|dSrbr0r7rrrr<r&z get_validators..cSs t|dSrrr7rrrr<r&cSs t|dSrbrr7rrrr<r&cSs t|dSrrr7rrrr<r&) rr;rGrrMr[rvrrrrrrrrget_validatorss rc Csg}|dur|||D](}||}|dur||t||}qtdd|D}tdd|D} d} |rtjd|D]}t|||\}} | p| } qn tjd| p| } |||| |dS)NcSs$g|]}|jdus|jdur|qSr+)r r r!rrrrr%6s  z$apply_validators..cSsg|]}|jdur|qSr+)r rrrrr%>s Fz? Based on the analysis we will perform the following actions: z No remediations found. )appendrrCrTrUrVr) rrrZ validatorsrZwrite_out_file_funcZoptional_remediationsZ validatorZ&any_optional_or_necessary_remediationsZany_necessary_appliedZany_optional_appliedrZ!any_optional_or_necessary_appliedrrrapply_validators$s>     r)r2)rd)%rrTtypingrrrrZopenai.datalib.pandas_helperrrrrrr0r;rGrMr[rvrrrrrrrrrrrrrrrYrprrrrrrs<   & (L'$L\ M