U zDe @sddlZdddddddddd dd dd dd ddd dddddddddZeedddZdeeedddZeedddZdS)NPython) full_nameindentcppz int main())rrmainZJavazpublic static void mainZcsharpzpublic static void MainZPHPZ TypeScriptZ JavaScriptZBash)pythonrjavacsphptsjssh)questionlangcsdd|dD|dkrrfddttDd}|dd}dd|}||fSdd d}ddd}||fS) NcSsg|]}|r|qS)strip).0xrra/weka-jd/prod/deepseek/permanent/yangdejian/DeepSeek-Coder/Evaluation/HumanEval/utils/instruct.py *sz%get_function_name.. rcsg|]}|dr|qS)zdef ) startswith)riZ func_linesrrr-s(r{)rsplitlowerrangelenjoin)rrZfunc_idx func_name func_prefixrrrget_function_name)s r%F)example lang_codeverbosec Cs |d}|d|d}|d}t|}|d}|d}zztd|d|tjtjBd } |r|td || |d dr|d | kr| |d } | d| } t ||\} } zt|  | } d }| |d kr| | |d d kr|d 7}qz| dd |d}Wnt | }YnXWn@d } z| dd |d}Wnt | }YnXYnX| | |}|dkr|dd |d7}| d|d}||d<WnJtk r}z*td ||||dd||d<W5d}~XYnX|S)Ntask_idoutputZgpt_completionpromptrrz```z (.*?)```rz>>> Task: {} {}r r})r r r generationzIFailed to extract code block with error `{}`: >>> Task: {} >>> Output: {})getrlanguge_settingsrefindallrDOTALL IGNORECASEprintformatindexr%rindexr! Exception)r&r'r(r)r*rZsettingrrZ code_blockZ main_startr#r$startendbodyr/exrrrextract_generation_code6sT &     &r?ZlangugercCsd||S)Nz Please help me to complete the function. Use the given packages only and DO NOT refer any new package. Please return all completed function in a codeblock. Here is the given code to do completion: ```{} {} ``` )rr7Zlower@rrrbuild_deepseekcoder_instructionksrA)F)r2r1strr%boolr?rArrrrs>& 5