o 畢hho@shddlmZddlmZmZmZddlmZddlZddl m Z ddl m Z m Z mZmZmZddlmZmZdd lmZd d lmZmZmZmZmZmZdd lmZmZdd l m!Z!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddgZ1GdddeZ2GdddeZ3GdddZ4GdddZ5GdddZ6GdddZ7dS)) annotations)UnionIterableOptional)LiteralN)_legacy_response) NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property) CheckpointsAsyncCheckpointsCheckpointsWithRawResponseAsyncCheckpointsWithRawResponse CheckpointsWithStreamingResponse%AsyncCheckpointsWithStreamingResponse)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)AsyncPaginatormake_request_options)job_list_paramsjob_create_paramsjob_list_events_params) FineTuningJob)FineTuningJobEventJobs AsyncJobsc @eZdZed6ddZed7ddZed8d d Zeeeeed d d ed d9d#d$Zd d d ed%d:d'd(Z eed d d ed)d;d/d0Z d d d ed%d:d1d2Z eed d d ed)dOptional[int] | NotGivenr?Optional[str] | NotGivenr@rAHeaders | NonerB Query | NonerC Body | NonerD'float | httpx.Timeout | None | NotGivenr#c Cs6|jdt|||||||dtjt|| | | dtdS) Creates a fine-tuning job which begins the process of creating a new model from a given dataset. Response includes details of the enqueued job including job status and the name of the fine-tuned models once complete. [Learn more about fine-tuning](https://platform.openai.com/docs/guides/fine-tuning) Args: model: The name of the model to fine-tune. You can select one of the [supported models](https://platform.openai.com/docs/guides/fine-tuning/which-models-can-be-fine-tuned). training_file: The ID of an uploaded file that contains training data. See [upload file](https://platform.openai.com/docs/api-reference/files/create) for how to upload a file. Your dataset must be formatted as a JSONL file. Additionally, you must upload your file with the purpose `fine-tune`. The contents of the file should differ depending on if the model uses the [chat](https://platform.openai.com/docs/api-reference/fine-tuning/chat-input) or [completions](https://platform.openai.com/docs/api-reference/fine-tuning/completions-input) format. See the [fine-tuning guide](https://platform.openai.com/docs/guides/fine-tuning) for more details. hyperparameters: The hyperparameters used for the fine-tuning job. integrations: A list of integrations to enable for your fine-tuning job. seed: The seed controls the reproducibility of the job. Passing in the same seed and job parameters should produce the same results, but may differ in rare cases. If a seed is not specified, one will be generated for you. suffix: A string of up to 64 characters that will be added to your fine-tuned model name. For example, a `suffix` of "custom-model-name" would produce a model name like `ft:gpt-4o-mini:openai:custom-model-name:7p4lURel`. validation_file: The ID of an uploaded file that contains validation data. If you provide this file, the data is used to generate validation metrics periodically during fine-tuning. These metrics can be viewed in the fine-tuning results file. The same data should not be present in both train and validation files. Your dataset must be formatted as a JSONL file. You must upload your file with the purpose `fine-tune`. See the [fine-tuning guide](https://platform.openai.com/docs/guides/fine-tuning) for more details. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds /fine_tuning/jobsrErGr<r=r>r?r@rArBrCrDbodyoptionscast_to)_postrr!JobCreateParamsrr# r-rErGr<r=r>r?r@rArBrCrDr.r.r/create?s$Q  z Jobs.createrTfine_tuning_job_idcCs4|s td||jd|t||||dtdS) Get info about a fine-tuning job. [Learn more about fine-tuning](https://platform.openai.com/docs/guides/fine-tuning) Args: extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds AExpected a non-empty value for `fine_tuning_job_id` but received /fine_tuning/jobs/rTrWrX ValueError_getrr#r-r]rArBrCrDr.r.r/retrievesz Jobs.retrieveafterlimitrArBrCrDrhstr | NotGivenriint | NotGivenSyncCursorPage[FineTuningJob]c C2|jdttt||||t||dtjdtdSa List your organization's fine-tuning jobs Args: after: Identifier for the last job from the previous pagination request. limit: Number of fine-tuning jobs to retrieve. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rRrhrirArBrCrDquerypagerWrE) _get_api_listrr#rrr JobListParamsr-rhrirArBrCrDr.r.r/list" z Jobs.listcCs6|s td||jd|dt||||dtdS)W Immediately cancel a fine-tune job. Args: extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r_r`/cancelrTrarcrYrr#rer.r.r/cancels z Jobs.cancel"SyncCursorPage[FineTuningJobEvent]c CL|s td||jd|dttt||||t||dtjdtdSa Get status updates for a fine-tuning job. Args: after: Identifier for the last event from the previous pagination request. limit: Number of events to retrieve. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r_r`z/eventsrorprr)rcrtrr$rrr"JobListEventsParamsr-r]rhrirArBrCrDr.r.r/ list_events&  zJobs.list_events)r(r)r(r2)r(r7rErFrGrHr<rIr=rJr>rKr?rLr@rLrArMrBrNrCrOrDrPr(r# r]rHrArMrBrNrCrOrDrPr(r#)rhrjrirkrArMrBrNrCrOrDrPr(rl)r]rHrhrjrirkrArMrBrNrCrOrDrPr(r} __name__ __module__ __qualname__rr0r5r9r r\rfrwr|rr.r.r.r/r%'R    k&5%c @r')=r&r(rcCr)r*)rr+r,r.r.r/r0Kr1zAsyncJobs.checkpointsAsyncJobsWithRawResponsecCr3r4)rr,r.r.r/r5Or6zAsyncJobs.with_raw_responseAsyncJobsWithStreamingResponsecCr3r8)rr,r.r.r/r9Yr:z!AsyncJobs.with_streaming_responseNr;rErFrGrHr<rIr=rJr>rKr?rLr@rArMrBrNrCrOrDrPr#c sD|jdt|||||||dtjIdHt|| | | dtdIdHS)rQrRrSNrTrU)rYrr!rZrr#r[r.r.r/r\bs&Q   zAsyncJobs.createrTr]cs<|s td||jd|t||||dtdIdHS)r^r_r`rTraNrbrer.r.r/rfs zAsyncJobs.retrievergrhrjrirk=AsyncPaginator[FineTuningJob, AsyncCursorPage[FineTuningJob]]c Crmrn)rtrr#rrr rurvr.r.r/rwrxzAsyncJobs.listcs>|s td||jd|dt||||dtdIdHS)ryr_r`rzrTraNr{rer.r.r/r|s  zAsyncJobs.cancelGAsyncPaginator[FineTuningJobEvent, AsyncCursorPage[FineTuningJobEvent]]c Cr~r)rcrtrr$rrr"rrr.r.r/r:rzAsyncJobs.list_events)r(r)r(r)r(rrr)rhrjrirkrArMrBrNrCrOrDrPr(r)r]rHrhrjrirkrArMrBrNrCrOrDrPr(rrr.r.r.r/r&Jrc@$eZdZd ddZed dd Zd S) r2jobsr%r(NonecCP||_t|j|_t|j|_t|j|_t|j|_t|j|_dSr*)_jobsrto_raw_response_wrapperr\rfrwr|rr-rr.r.r/__init__n  zJobsWithRawResponse.__init__rcC t|jjSr*)rrr0r,r.r.r/r0 zJobsWithRawResponse.checkpointsNrr%r(r)r(rrrrrrr0r.r.r.r/r2m r2c@r) rrr&r(rcCrr*)rrasync_to_raw_response_wrapperr\rfrwr|rrr.r.r/rrz!AsyncJobsWithRawResponse.__init__rcCrr*)rrr0r,r.r.r/r0rz$AsyncJobsWithRawResponse.checkpointsNrr&r(r)r(rrr.r.r.r/rrrc@r) r7rr%r(rcCF||_t|j|_t|j|_t|j|_t|j|_t|j|_dSr*)rrr\rfrwr|rrr.r.r/r  z"JobsWithStreamingResponse.__init__rcCrr*)rrr0r,r.r.r/r0rz%JobsWithStreamingResponse.checkpointsNr)r(rrr.r.r.r/r7rr7c@r) rrr&r(rcCrr*)rrr\rfrwr|rrr.r.r/rrz'AsyncJobsWithStreamingResponse.__init__rcCrr*)rrr0r,r.r.r/r0rz*AsyncJobsWithStreamingResponse.checkpointsNr)r(rrr.r.r.r/rrr)8 __future__rtypingrrrtyping_extensionsrhttpxr_typesr r r r r _utilsrr_compatrr0rrrrrr _resourcerr _responserr paginationrr _base_clientrrtypes.fine_tuningr r!r"!types.fine_tuning.fine_tuning_jobr#'types.fine_tuning.fine_tuning_job_eventr$__all__r%r&r2rr7rr.r.r.r/s6       %%