o i:&@sddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZGd ddZGdddZdS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)construct_type)BadRequestError)GetModelV1Response)ListModelsV1Responsec @veZdZdefddZddddedejedeje d e e fd d Z dd ded edeje d e e fddZdS)RawModelsClientclient_wrappercC ||_dSN_client_wrapperselfrra/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/manage/v1/projects/models/raw_client.py__init__ zRawModelsClient.__init__Ninclude_outdatedrequest_options project_idrrreturnc Cs|jjjdt|d|jjdd|i|d}z@d|jkr$dkr9nntt t t | d}t ||d WS|jd krSt t|jttjt tj| dd | }Wntykt|jt|j|jd wt|jt|j|d ) a# Returns metadata on all the latest models that a specific project has access to, including non-public models Parameters ---------- project_id : str The unique identifier of the project include_outdated : typing.Optional[bool] returns non-latest versions of models request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListModelsV1Response] A list of models v1/projects//modelsGETrbase_urlmethodparamsr,type_object_responsedataheadersbody status_coder2r3)r httpx_clientrequestr get_environmentbaser5typingcastrr jsonrr dictr2Anyrrtextrrrr _response_data_response_jsonrrrlistsD    zRawModelsClient.listrmodel_idc Cs|jjjdt|dt||jjd|d}z@d|jkr%dkr:nntt t t | d}t ||dWS|jd krTt t|jttjt tj| dd | }Wntylt|jt|j|jd wt|jt|j|d ) a Returns metadata for a specific model Parameters ---------- project_id : str The unique identifier of the project model_id : str The specific UUID of the model request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetModelV1Response] A model object that can be either STT or TTS r!/models/r#r%r&rr(r)r*r-r0r1r4)rr6r7r r8r9r5r:r;r r r<rr r=r2r>rrr?rrrFrrArBrCrrrgetRs@   zRawModelsClient.get)__name__ __module__ __qualname__rrstrr:Optionalboolr rrrDr rJrrrrr. >rc @r)AsyncRawModelsClientrcCrrrrrrrrrzAsyncRawModelsClient.__init__Nrrrrr c s|jjjdt|d|jjdd|i|dIdH}z@d|jkr(dkr=nntt t t | d }t ||d WS|jd krWt t|jttjt tj| d d | }Wntyot|jt|j|jd wt|jt|j|d )a( Returns metadata on all the latest models that a specific project has access to, including non-public models Parameters ---------- project_id : str The unique identifier of the project include_outdated : typing.Optional[bool] returns non-latest versions of models request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListModelsV1Response] A list of models r!r"r#rr$Nr(r)r*r-r0r1r4)rr6r7r r8r9r5r:r;rr r<rr r=r2r>rrr?r@rrrrDsF     zAsyncRawModelsClient.listrErFc s|jjjdt|dt||jjd|dIdH}z@d|jkr)dkr>nntt t t | d}t ||d WS|jd krXt t|jttjt tj| dd | }Wntypt|jt|j|jd wt|jt|j|d ) a Returns metadata for a specific model Parameters ---------- project_id : str The unique identifier of the project model_id : str The specific UUID of the model request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetModelV1Response] A model object that can be either STT or TTS r!rGr#rHNr(r)r*r-r0r1r4)rr6r7r r8r9r5r:r;r r r<rr r=r2r>rrr?rIrrrrJsB    zAsyncRawModelsClient.get)rKrLrMrrrNr:rOrPr rrrDr rJrrrrrRrQrR)r: json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.unchecked_base_modelr errors.bad_request_errorr types.get_model_v1responser types.list_models_v1responserrrRrrrrs        x