o RWj؉@sUddlmZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m ZGddde Zed d d Zd`ddZdaddZddgfdbddZdcddddZddgfdbddZd`d d!Zd`d"d#Zd`d$d%Zd`d&d'Zd`d(d)Zd`d*d+Zded/d0Zddgfdfd3d4Zd`d5d6Zdgd8d9Zdhd=d>Z did@dAZ!djdCdDZ"dkdFdGZ#dldIdJZ$dmdLdMZ%dndNdOZ&dodpdSdTZ'dUZ(dVe)dW<dqdYdZZ*drd^d_Z+dS)s) annotationsN)AnyTypeVarCallableOptional NamedTuple) TypeAlias)pandasc@s^eZdZUded<dZded<dZded<dZded<dZded <dZded <dZ ded <dS) RemediationstrnameNz Optional[str] immediate_msg necessary_msgzOptional[Callable[[Any], Any]] necessary_fn optional_msg optional_fn error_msg) __name__ __module__ __qualname____annotations__rrrrrrrre/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/openai/lib/_validators.pyr s      r OptionalDataFrameTzOptional[pd.DataFrame])bounddf pd.DataFramereturncCs8d}t||kr dnd}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 )r MIN_EXAMPLESoptional_suggestionrrrrnum_examples_validators  r&necessary_columnr csdddd}d}d}d}|jvr9d d |jDvr3dfd d }|}d d}dd}ndd}td||||dS)z[ This validator will ensure that the necessary column is present in the dataframe. rrcolumnrrcs2fdd|jD}|j|didd|S)Ncs g|] }t|kr|qSrr lower.0cr(rr -s zInecessary_column_validator..lower_case_column..rT)columnsinplace)r0renamer*)rr(colsrr.rlower_case_column,sz5necessary_column_validator..lower_case_columnNcSsg|]}t|qSrr)r+rrrr/7z.necessary_column_validator..c |SNr)rr4r'rrlower_case_column_creator9 z=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 rrrr)rrr(rrr)rrrr)r0r )rr'rrrrr9rr8rnecessary_column_validator's&    r<prompt completionfields list[str]csg}d}d}d}t|jdkrMfdd|jD}d}|D]fdd|D}t|dkr9|dd d 7}qd ||}d |}dfdd }td|||dS)zK This validator will remove additional columns from the dataframe. Nr csg|]}|vr|qSrrr+r?rrr/Ur5z/additional_column_validator..r csg|]}|vr|qSrrr+)acrrr/Xr5rz9 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: xrrcs|Sr7rrCrArrr^sz1additional_column_validator..necessary_fnadditional_columnr rrrrCrrr)r#r0r )rr?additional_columnsrrr warn_messagedupsr)rBr?radditional_column_validatorKs*  rKfieldcsd}d}d}|dds|rH|dk|B}|j|}dd|}dfd d }d t|d d}td|||dS)zA This validator will ensure that no completion is empty. NcSs|dkS)Nr rrDrrrqsz+non_empty_field_validator..r z - `z?` column/key should not contain empty strings. These are rows: rCrrcs||dkjgdS)Nr subset)dropnarDrLrrrvsz/non_empty_field_validator..necessary_fnRemove z rows with empty sempty_rFrG)applyanyisnull reset_indexindextolistr#r )rrLrrr empty_rows empty_indexesrrQrnon_empty_field_validatoris&r]cs|jd}|j|}d}d}d}t|dkr:dt|ddd|}dt|d }dfd d }td|||dS)zY This validator will suggest to the user to remove duplicate rows if they exist. rNNr - There are z duplicated -z sets. These are rows: rRz duplicate rowsrCrrcs |jdS)NrN)drop_duplicatesrDrArrr z.duplicated_rows_validator..optional_fnduplicated_rowsr rrrrG) duplicatedrXrYrZr#joinr )rr?rbduplicated_indexesrrrrrArduplicated_rows_validators   rgcsd}d}d}t|}|dkr8ddd|td kr8d td d }d td}dfdd }td|||dS)zW This validator will suggest to the user to remove examples that are too long. Nopen-ended generationdrrrcSs$|jdddd}|j|S)NcSst|jt|jdkS)Ni')r#r=r>rDrrrrMr5zClong_examples_validator..get_long_indexes..)axis)rUrXrYrZ)ri long_examplesrrrget_long_indexessz1long_examples_validator..get_long_indexesrr^z. examples that are very long. These are rows: zf For conditional generation, and for classification the examples shouldn't be longer than 2048 tokens.rRz long examplesrCcs8|}|krtjdt|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:  )sysstdoutwriter#drop)rClong_indexes_to_droprm long_indexesrrrs  z,long_examples_validator..optional_fnrlrc)rirrrrG)infer_task_typer#r )rrrrft_typerrtrlong_examples_validators"  rxc spd}d}d}d}dgd}|D]}|dkr |jjdr q|jjj|ddr,q|dd}t|}|d krBtd d Sd$ddt|jdd} |j| krad| d}td |dS| dkr| dd} d| d}t | dkr|d|d7}|jjdt |  jj| ddr|d| d7}nd}| dkrd|d}d%fd d! }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 ---> ryrnFregex\nrh common_suffixr rCrsuffixrcS|d|7<|SNr=rrCrrrr add_suffixz2common_prompt_suffix_validator..add_suffixxfixzAll prompts are identical: `zt` Consider leaving the prompts blank if you want to do open-ended generation, otherwise ensure prompts are differentr rr z - 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 promptscr6r7rrDrsuggested_suffixrrrr:z3common_prompt_suffix_validator..optional_fncommon_completion_suffixr rrrrrCrrrrrrG) r=r containsrVreplacervr get_common_xfixallr#) rrrrrsuffix_options suffix_optiondisplay_suggested_suffixrwr}common_suffix_new_line_handledrrrcommon_prompt_suffix_validatorsT        & rcsd}d}d}d}t|jdddkrtddSdd d |jkr)tddSdkrKd d}|tkrK|d7}dd}dfdd }td|||dS)zd This validator will suggest to remove a common prefix from the prompt if a long one exist. Nprefixrr common_prefixr~rCrrcSs|djt|d|d<|Srr r#)rCrrrrremove_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 |Sr7rrDrrrrr!r:z3common_prompt_prefix_validator..optional_fncommon_prompt_prefixrc)rCrrrrrrG)rr=r rr#rMAX_PREFIX_LENrrrrrrcommon_prompt_prefix_validators,      rcsd}t|jddtdkoddkt|kr tddSdd d |jkr1tddSdd}dd}dfdd }td|||dS)zh This validator will suggest to remove a common prefix from the completion if a long one exist. rrr rr~rCr ws_prefixrcSs4|djt|d|d<|rd|d|d<|S)Nr>rr)rCrrrrrr7sz@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 prefixrz` from all completionscs |Sr7rrDrrrrrrEraz7common_completion_prefix_validator..optional_fncommon_completion_prefixrcN)rCrrrrrrrrG)rr>r#r rrrrr"common_completion_prefix_validator,s"      rc sbd}d}d}d}t|}|dks|dkrtddSt|jdd}|j|kr6d|d |d }td|d Sd gd }|D]}|jjj|ddrLq>|dd} d$dd|dkr|dd} d| d }t |dkrx|d| d 7}|jjdt | jj|ddr|d|d7}nd}|dkrd| d}d%fd d! }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. Nrhclassificationr}r~rrz All completions are identical: `zJ` Ensure completions are different, otherwise the model will just repeat `r;rz [END]) rn.z ENDz***z+++z&&&z$$$z@@@z%%%Frzrnr|rCrrcSrNr>rrrrrrvrz6common_completion_suffix_validator..add_suffixr z$ - All completions end with suffix `rrz9 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 completionscr6r7rrDrrrrr:z7common_completion_suffix_validator..optional_fnrrrrG) rvr rr>rr rrVrr#) rrrrrrwr}rrrrrrr"common_completion_suffix_validatorPsN        & rcCs^ddd}d}d}d}|jjdddks!|jjddd kr'd }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. rCrrcSs|ddd|d<|S)Nr>cSs|dr d|Sd|S)Nrr ) startswith)rSrrrrMzLcompletions_space_start_validator..add_space_start..)rUrDrrradd_space_startsz:completions_space_start_validator..add_space_startNrjrrz - 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 completioncompletion_space_startrcrG)r>r nuniquevaluesr )rrrrrrrr!completions_space_start_validators ,rr(rRemediation | Nonecspdfdd }|dd}|dd}|d |kr6td d d d dd|dSdS)zt This validator will suggest to lowercase the column values, if more than a third of letters are uppercase. rCrrcs|j|<|Sr7r)rDr.rr lower_casesz(lower_case_validator..lower_casecStdd|DS)Ncs$|] }|r|rdVqdSrjN)isalphaisupperr+rrr "9lower_case_validator....sumrDrrrrMz&lower_case_validator..cSr)Ncsrr)rislowerr+rrrrrrrrDrrrrMrr rz - 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;rcNrG)rUrr )rr(r count_upper count_lowerrr.rlower_case_validators  rfname'tuple[pd.DataFrame | None, Remediation]c Csd}d}d}d}d}tj|rQz|ds!|drF|dr*dnd\}}d|d}d|d }tj||td d }n|d rnd }d}t |} | j } t | dkrc|d7}tj |tdd }n|drd}d}t |d} | } tjdd| dD|tdd }Wdn1swYn|drtj|dtdd }t |dkrd}d}tj|tdd }na n_|drz"tj|dtdd }t |dkrtj|tdd }nd }d}Wn4ty tj|tdd }Yn!wd!}d"|vr&|d#|d$|d"d%d&7}n|d#|d'7}Wn'ttfyP|d"d%}d(|d)|d*|d+}Ynwd,|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)CSV,)TSV z= - Based on your file extension, your file is formatted as a z filez Your format `z` will be converted to `JSONL`)sepdtyper z.xlsxzH - Based on your file extension, your file is formatted as an Excel filez/Your format `XLSX` will be converted to `JSONL`rjz - 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)r r)r,linerrrr/sz#read_any_format..rn)r0r.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 rrr)ospathisfiler*endswithpdread_csvr fillna ExcelFile sheet_namesr# read_excelopenread DataFramesplit read_json ValueError TypeErrorupperr ) rr? remediationrrrrfile_extension_str separatorxlssheetsfcontentrrrrs        " rcCs,t|}d}|dkrd|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 trainingr!r")rvr )rrwrrrrformat_inferrer_validators   rrcCsb|jdurtjd|jd|jdtd|jdur%tj|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...rj) rrostderrrqr exitrrpr)rrrrrapply_necessary_remediation(s     r input_text auto_acceptboolcCs.tj||rtjddStdkS)NzY Tn)rorprqinputr*)rrrrraccept_suggestion6s  rtuple[pd.DataFrame, bool]cCsjd}d|jd}|jdur!t||r!|jdusJ||}d}|jdur1tjd|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] rn)rrrrrorprq)rrroptional_appliedrrrrapply_optional_remediation>s    rNonecCslt|}d}|dkrt|}|d}n |jdd}|d}dd d }||d }tjd|ddS)z? Estimate the time it'll take to fine-tune the dataset g?rg ףp= ?T)rYg|?5^?timefloatrr cSsd|dkr t|ddS|dkrt|dddS|dkr(t|dddSt|dddS) N<r z secondsiz minutesiQz hoursz days)round)rrrr format_time]sz.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)rrrr )rvr# memory_usagerrorprq)r ft_format expected_timer!sizer time_stringrrrestimate_fine_tuning_timePs    rrcsd|rddgndg}d} |dkrd|dndfdd |D}td d |Ds-|S|d 7}q ) N_train_validr rTz ()cs,g|]}tjdd|dqS)r _preparedr)rrsplitext)r,rr index_suffixrrr/rs,z!get_outfnames..css|] }tj|VqdSr7)rrr)r,rrrrrssz get_outfnames..rj)rV)rrsuffixesicandidate_fnamesrr r get_outfnamesmsrtuple[int, object]cCs.|j}d}|dkr|jjd}||fS)Nr r)r>r value_countsrY)r n_classes pos_classrrrget_classification_hyperparamsxs rany_remediationsc Cst|}t|jdd}t|jdd}d}d}|dkr!t||r!d}d} |dd } |dd } t| d kr;d | d nd} d }|s\|s\tj d|d| d| d| d t |dSt||r:t ||} |rt| dkr{d| d vr{d| dvs}Jd}t t||t t|d}|j|dd}||j}|ddgj| d ddddd|ddgj| ddddddt|\}}| d7} |dkr| d |d7} n | d!|7} nt| dksJ|ddgj| d ddddd|rd"ndd#d$| }|r d%| ddnd}t| d krdnd&| d}tj d'|d(| d d|| d)|| d t |dStj d*dS)+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. rrFzQ- [Recommended] Would you like to split into training and validation set? [Y/n]: rTr rnr|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.r trainvalidrjig?*)r random_stater=r>recordsN)rorient force_asciiindentz! --compute_classification_metricsz" --classification_positive_class "z --classification_n_classes rSz 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 )rvrr=r>rrr#rorprqrrmaxintsamplerrrYto_jsonrre)rrrrrcommon_prompt_suffixrrradditional_params%common_prompt_suffix_new_line_handled)common_completion_suffix_new_line_handledoptional_ending_stringfnamesMAX_VALID_EXAMPLESn_traindf_traindf_validrr files_string valid_stringseparator_reminderrrrwrite_out_filesn        (      ( r1cCs>d}t|jjdkrdSt|jt||krdSdS)z> Infer the likely fine-tuning task type from the data rrhrzconditional generation)rr=r r#r>unique)rCLASSIFICATION_THRESHOLDrrrrvs rvrseriesrcCsnd} |dkr|jt|d dn |jdt|d}|dkr' |S||jdkr1 |S|jd}q)zQ Finds the longest common suffix or prefix of all the values in a series r TrrjNr)r r#rr)r5r common_xfix common_xfixesrrrrs4  rz,Callable[[pd.DataFrame], Remediation | None]r Validatorlist[Validator]cCs2tddddtttttddddtttt t gS)NcS t|dSrr<rDrrrrM z get_validators..cSr:rr;rDrrrrMr<cSr:rrrDrrrrMr<cSr:rr=rDrrrrMr<) r&rKr]rrgrxrrrrrrrrrget_validatorss r> validatorswrite_out_file_funcCallable[..., Any]c Csg}|dur |||D]}||}|dur!||t||}q tdd|D}tdd|D} d} |rPtjd|D]}t|||\}} | pM| } q@ntjd| pY| } |||| |dS)NcSs$g|]}|jdus|jdur|qSr7)rrr,rrrrr/s z$apply_validators..cSsg|] }|jdur|qSr7)rrBrrrr/rFz? Based on the analysis we will perform the following actions: z No remediations found. )appendrrVrorprqr) rrrr?rr@optional_remediations validator&any_optional_or_necessary_remediationsany_necessary_appliedany_optional_appliedr!any_optional_or_necessary_appliedrrrapply_validatorss6       rJ)rrrr )rrr'r rr )rrr?r@rr )r>)rrrLr rr )rrr(rrr)rr r?r@rr)rrrr rr)rr rrrr)rrrr rrrr)rrrr)rr rrrr@)rrrr) rrrr rrrrrr)rrrr )r)r5rrr rr )rr9)rrrr rrr?r9rrr@rArr), __future__rrrotypingrrrrrtyping_extensionsr_extrasr rr rr&r<rKr]rgrxrrrrrrrrrrrrrrr1rvrr8rr>rJrrrrsF    $   % D ' $ D  Y      K