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@neZdZdefddZddddejedejede e fd d Z dd d e dejede e fd dZdS)RawModelsClientclient_wrappercC ||_dSN_client_wrapperselfrrX/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/manage/v1/models/raw_client.py__init__ zRawModelsClient.__init__Ninclude_outdatedrequest_optionsrrreturnc Cs|jjjd|jjdd|i|d}z@d|jkrdkr3nnttt t| d}t ||dWS|jd krMt t |jttjt tj| dd | }Wntyet|jt |j|jd wt|jt |j|d ) a Returns metadata on all the latest public models. To retrieve custom models, use Get Project Models. Parameters ---------- 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/modelsGETrbase_urlmethodparamsr,type_object_responsedataheadersbody status_coder0r1)r httpx_clientrequestget_environmentbaser3typingcastrr jsonrr dictr0Anyrrtextrrr _response_data_response_jsonrrrlistsD    zRawModelsClient.listrmodel_idc Cs|jjjdt||jjd|d}z@d|jkr dkr5nntt t t | d}t ||dWS|jdkrOt t|jttjt tj| dd | }Wntygt|jt|j|jd wt|jt|j|d ) a Returns metadata for a specific public model Parameters ---------- 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 v1/models/r!r#r$rr&r'r(r+r.r/r2)rr4r5r r6r7r3r8r9r r r:rr r;r0r<rrr=rrDrr?r@rArrrgetKs@    zRawModelsClient.get)__name__ __module__ __qualname__rrr8Optionalboolr rrrBstrr rHrrrrr$ 7rc@r)AsyncRawModelsClientrcCrrrrrrrrrzAsyncRawModelsClient.__init__Nrrrrc s|jjjd|jjdd|i|dIdH}z@d|jkr"dkr7nnttt t| d}t ||d WS|jd krQt t |jttjt tj| dd | }Wntyit|jt |j|jd wt|jt |j|d ) a Returns metadata on all the latest public models. To retrieve custom models, use Get Project Models. Parameters ---------- 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!rr"Nr&r'r(r+r.r/r2)rr4r5r6r7r3r8r9rr r:rr r;r0r<rrr=r>rrrrBsF     zAsyncRawModelsClient.listrCrDc s|jjjdt||jjd|dIdH}z@d|jkr$dkr9nntt t t | d}t ||dWS|jd krSt t|jttjt tj| dd | }Wntykt|jt|j|jd wt|jt|j|d ) a Returns metadata for a specific public model Parameters ---------- 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 rEr!rFNr&r'r(r+r.r/r2)rr4r5r r6r7r3r8r9r r r:rr r;r0r<rrr=rGrrrrHsB     zAsyncRawModelsClient.get)rIrJrKrrr8rLrMr rrrBrNr rHrrrrrPrOrP)r8 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_v1responserrrPrrrrs        n