uPj , UddlmZddlmZmZmZddlZddlmZm Z m Z m Z ddl m Z ddlmZdddd d d d d Zeed <ddeeefdedeeeffdZddededefdZGddZy)) Coroutine)AnyFinalcastN)AsyncAzureOpenAI AsyncOpenAI AzureOpenAIOpenAI)verbose_logger)LiteLLMFineTuningJobqueuedrunning succeededfailed cancelled)pending notRunningrrrcanceled canceling_AZURE_STATUS_MAPdatais_azurereturnc|s|S|j}|jdd|d<|jdg|d<|jd}|tvr t||d<|S)u Normalize Azure OpenAI FineTuningJob response to match OpenAI schema. Azure differences: - organization_id: null → "" - result_files: null → [] - status: mapped via _AZURE_STATUS_MAP organization_id result_filesstatus)copygetr)rr normalizedrs l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/openai/fine_tuning/handler.py_normalize_fine_tuning_job_dictr#sw   J~~'(0(* $%~~n%-%' >"NN8,F ""08 8 responsecJtdit|j|S)N)r)r r# model_dump)r%rs r"&_litellm_fine_tuning_job_from_responser)2s"  l"A(BUBUBWbj"k llr$ceZdZdZdfd Z ddedzdedzdeejzde dzd edzd e e ze ze zdzd ed edzd edzde e ze ze zdzfdZdede e zdefdZ dd edededzdedzd edzdeejzde dzd edzd e e ze ze zdzdeeeeefzfdZdede e zdefdZ dd edededzdedzd edzdeejzde dzd edzd e e ze ze zdzdeeeeefzfdZ dde e zdedzde dzfdZ d d ededzdedzd edzdeejzde dzd edzd e e ze ze zdzdedzde dzfdZdede e zdefdZ dd edededzdedzd edzdeejzde dzd edzd e e ze ze zdzdeeeeefzfdZxZS)!OpenAIFineTuningAPIz/ OpenAI methods to support for batches rNc"t|yN)super__init__)self __class__s r"r/zOpenAIFineTuningAPI.__init__;s r$api_keyapi_basetimeout max_retries organizationclient _is_async api_versionlitellm_paramsc t} d} |]i} | jD]*\} }| dk(s | dk(s| dk(r| dk(r||| d<#|&|| | <,|dur tdi| } | Stdi| } | S|} | S)Nr0r7r8r3base_urlTr')localsitemsrr )r0r2r3r4r5r6r7r8r9r: received_args openai_clientrkvs r"get_openai_clientz%OpenAIFineTuningAPI.get_openai_client>s &x VZ >D%++- 1;!x-1 3C*_'(D$]DG  D + 3d 3  !' #Mr$create_fine_tuning_job_datar@cK|jjjdi|d{}t|S7w)Nr') fine_tuningjobscreater))r0rDr@r%s r"acreate_fine_tuning_jobz+OpenAIFineTuningAPI.acreate_fine_tuning_job^s? !F 9 9 > > E E dHc dd5h??e *><>c h|j|||||| ||} | td|dur4t| ttfs td|j || St jd|tt| jjjdi|} t| S)Nr2r3r4r5r6r7r8r9jOpenAI client is not initialized. Make sure api_key is passed or OPENAI_API_KEY is set in the environment.T\OpenAI client is not an instance of AsyncOpenAI. Make sure you passed an AsyncOpenAI client.)rDr@z"creating fine tuning job, args= %sr')rC ValueError isinstancerrrIr debugrr rFrGrHr)) r0r8rDr2r3r9r4r5r6r7r@r%s r"create_fine_tuning_jobz*OpenAIFineTuningAPI.create_fine_tuning_jobgs^b]s]s#%#^t ^   |   mk;K-LM r//,G+0  AC^_M$v}5AAFFMMlPkl5h??r$fine_tuning_job_idcK|jjj|d{}t|S7wNrS)rFrGcancelr)r0rSr@r%s r"acancel_fine_tuning_jobz+OpenAIFineTuningAPI.acancel_fine_tuning_jobs< !. 9 9 > > E EYk E ll5h??mrJc h|j|||||| ||} | td|dur4t| ttfs td|j || St jd|tt| jjj|} t| S)NrLrMTrNrSr@z#canceling fine tuning job, args= %srV)rCrOrPrrrYr rQrr rFrGrWr) r0r8rSr2r3r9r4r5r6r7r@r%s r"cancel_fine_tuning_jobz*OpenAIFineTuningAPI.cancel_fine_tuning_jobs^b]s]s#%#^t ^   |   mk;K-LM r//#5+0  BDVWv}5AAFFMMasMt5h??r$afterlimitcpK|jjj||d{}|S7w)Nr^r_)rFrGlist)r0r@r^r_r%s r"alist_fine_tuning_jobsz*OpenAIFineTuningAPI.alist_fine_tuning_jobss8 !. 9 9 > > C C%W\ C ]]^s +646c @|j||||||||} | td|dur5t| ttfs td|j | | | St jd| | | jjj| | } | S)NrLrMTrN)r^r_r@z*list fine tuning job, after= %s, limit= %sra) rCrOrPrrrcr rQrFrGrb) r0r8r2r3r9r4r5r6r7r^r_r@r%s r"list_fine_tuning_jobsz)OpenAIFineTuningAPI.list_fine_tuning_jobss^b]s]s#%#^t ^   |   mk;K-LM r..+/  I5RWX'3388==EQV=Wr$cK|jjj|d{}t|S7wrU)rFrGretriever)rXs r"aretrieve_fine_tuning_jobz-OpenAIFineTuningAPI.aretrieve_fine_tuning_jobs< !. 9 9 > > G G[m G nn5h??orJc \|j|||||| ||} | td|dur.t| ts td|j || St j d|tt| jjj|} t| S)NrLrMTrNr[z"retrieving fine tuning job, id= %srV) rCrOrPrrhr rQrr rFrGrgr)r\s r"retrieve_fine_tuning_jobz,OpenAIFineTuningAPI.retrieve_fine_tuning_jobs^b]s]s#%#^t ^   |   m[9 r11#5+2  ACUVv}5AAFFOOcuOv5h??r$)rN)NFNNr-)NN)NNN)__name__ __module__ __qualname____doc__r/strfloathttpxTimeoutintr rr rbooldictrCr rIrrrRrYr]rcrerhrj __classcell__)r1s@r"r+r+6spPT"&&*t*&  4Z  Dj ${25EEL4Zt  +  +.> > E@@%)@#%55@  @&PT&@&@&*&@t &@ * &@ 4Z &@&&@4Z&@Dj&@${25EEL&@  #s4H*H I I&@P@@#%55@  @$PT&@&@ &@t &@ * &@ 4Z &@&&@4Z&@Dj&@${25EEL&@  #s4H*H I I&@V! "%55TzTz $PT  ((t(* ( 4Z ( & (4Z(Dj(${25EEL(Tz(Tz(T@@#%55@  @$PT&@&@ &@t &@ * &@ 4Z &@&&@4Z&@Dj&@${25EEL&@  #s4H*H I I&@r$r+)F)collections.abcrtypingrrrrqopenairrr r litellm._loggingr litellm.types.utilsr r__annotations__rurortr#r)r+r'r$r"r}s%## EE+45$sCx.DUYZ]_bZbUc6mSmDmUimj@j@r$