o 2j/@sPddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZddlmZmZmZeeZdZd Zd Zd Zd d hZGdddeeZGdddeZGddde ZGdddeZ GdddeZ!GdddeZ"GdddeZ#de$de$fddZ%Gd d!d!eeZ&Gd"d#d#e&Z'Gd$d%d%e&Z(dS)&N)ABC)Any)InferenceProviderMapping)RequestParameters_as_dict_as_url)BaseConversationalTaskBaseTextGenerationTaskTaskProviderHelper filter_none) get_sessionhf_raise_for_statusloggingtogetherzhttps://api.together.xyzg@queued in_progresscs<eZdZdZdeffdd Zdededefdd ZZS) TogetherTaskz"Base class for Together API tasks.taskcstjtt|ddS)N)providerbase_urlrsuper__init__ _PROVIDER _BASE_URL)selfr __class__h/var/www/html/whisper/venv/lib/python3.10/site-packages/huggingface_hub/inference/_providers/together.pyr(zTogetherTask.__init__ mapped_modelapi_keyreturncCst|jdkr dSdkrdSdkrdSdkr"dSdkr)d Sd kr0d Std |jd ) N text-to-imageimage-to-imagez/v1/images/generationstext-to-speechz/v1/audio/speechfeature-extractionz/v1/embeddings text-to-videoimage-to-videoz /v2/videoszUnsupported task 'z' for Together API.)r ValueError)rr"r#rrr _prepare_route+s      zTogetherTask._prepare_route)__name__ __module__ __qualname____doc__strrr, __classcell__rrrr r%srcs<eZdZfddZd deeBdedBdefddZZ S) TogetherTextGenerationTaskctjttddSN)rrrrrrr r:z#TogetherTextGenerationTask.__init__Nresponserequest_paramsr$cCs0t|dd}|d|d|dddS)Nchoicesrtext finish_reasonseed)r<r=)generated_textdetails)rget)rr8r9outputrrr get_response=s z'TogetherTextGenerationTask.get_responseN) r-r.r/rbytesdictrrrBr2rrrr r39s (r3c s>eZdZfddZdededededBffdd ZZS) TogetherConversationalTaskcr4r5rr6rrr rIr7z#TogetherConversationalTask.__init__inputs parametersprovider_mapping_infor$Ncstt|||}|durdS|d}t|tr8|ddkr8t|dtr8d|dvr8d|ddd|d<|S)Nresponse_formattype json_schemaschema)rKrM)r_prepare_payload_as_dictr@ isinstancerE)rrGrHrIpayloadrJrrr rNLs    z3TogetherConversationalTask._prepare_payload_as_dict r-r.r/rrrErrNr2rrrr rFHs rFc ZeZdZfddZdededededBfdd Zdd eeBd e dBdefd d Z Z S)TogetherTextToImageTaskctddS)Nr%rrr6rrr rcz TogetherTextToImageTask.__init__rGrHrIr$NcCs:|j}t|}d|vr|d|d<|dd|d|iS)Nnum_inference_stepsstepsbase64)promptrJmodel) provider_idr pop)rrGrHrIr"rrr rNfs z0TogetherTextToImageTask._prepare_payload_as_dictr8r9cCt|}t|dddSNdatarb64_jsonrrY b64decoderr8r9 response_dictrrr rBpz$TogetherTextToImageTask.get_responserC r-r.r/rrrErrNrDrrBr2rrrr rSbs  ( rSc rR)TogetherImageToImageTaskcrT)Nr&rUr6rrr rvrVz!TogetherImageToImageTask.__init__rGrHrIr$Nc Cs|j}t|dd}t|}|dd}d|vr|d|d<|}d|vo)d|v}|r0d |ind |gi} d|i| d d i|d |iS)Nz image/jpegdefault_mime_typerZrWrXkontextzflux.1 image_urlreference_imagesrJrYr[)r\rr r]lower) rrGrHrIr"rmrZlowered use_image_url image_fieldrrr rNys&  z1TogetherImageToImageTask._prepare_payload_as_dictr8r9cCr^r_rbrdrrr rBrfz%TogetherImageToImageTask.get_responserCrgrrrr rhus  (rhc rR)TogetherFeatureExtractionTaskcrT)Nr(rUr6rrr rrVz&TogetherFeatureExtractionTask.__init__rGrHrIr$NcC||jdt|SN)inputr[r\r rrGrHrIrrr rN z6TogetherFeatureExtractionTask._prepare_payload_as_dictr8r9cCsddt|dDS)NcSsg|]}|dqS) embeddingr).0itemrrr sz>TogetherFeatureExtractionTask.get_response..r`)rrr8r9rrr rBr!z*TogetherFeatureExtractionTask.get_responserCrgrrrr rss  ( rsc rR)TogetherTextToSpeechTaskcrT)Nr'rUr6rrr rrVz!TogetherTextToSpeechTask.__init__rGrHrIr$NcCrtrurwrxrrr rNs z1TogetherTextToSpeechTask._prepare_payload_as_dictr8r9cCs$t|tr|Stdt|jd)Nz1Expected raw audio bytes for text-to-speech, got .)rOrDr+rKr-r~rrr rBs z%TogetherTextToSpeechTask.get_responserCrgrrrr rs  ( rrHr$cCs\t|}d|vr|d|d<d|vr,|d}d|vr"|d|d<d|vr,|d|d<|S)zNMap HF inference-client conventions onto Together's video API parameter names.rWrX target_sizewidthheight)r r])rHrrrr _normalize_video_parameterss   rc@s0eZdZdZddeeBdedBdefddZdS) TogetherVideoTaskz>Base class for Together's asynchronous video generation tasks.Nr8r9r$c CsL|durtdt|}|d}|std|jd|}td|d}ttD]1}|dur9|tvr9n.t t t j||j d}t||}|d}|dur^|tvr^n q-tdtd |d kr~|d pqi} td | d pzd|dkrtd||dpid} | stdt | } t| | jS)NzEA `RequestParameters` object is required to poll Together video jobs.idz6No job ID found in Together video generation response./z+Generating video, polling for completion...status)headersuITimed out while waiting for Together video generation — aborting after z status pollsfailederrorz"Together video generation failed: messagez Unknown error completedz&Unexpected Together video job status: outputs video_urlz3No video URL found in completed Together video job.)r+rr@urlloggerinforange_VIDEO_MAX_POLL_ATTEMPTS_VIDEO_PENDING_STATUSEStimesleep_VIDEO_POLLING_INTERVALr rr json RuntimeErrorcontent) rr8r9jobjob_id status_urlr_status_responserrvideo_responserrr rBsH       zTogetherVideoTask.get_responserC) r-r.r/r0rDrErrrBrrrr rs$rc :eZdZfddZdededededBfdd ZZS) TogetherTextToVideoTaskcrT)Nr)rUr6rrr rrVz TogetherTextToVideoTask.__init__rGrHrIr$NcCrt)N)rZr[)r\rrxrrr rN ryz0TogetherTextToVideoTask._prepare_payload_as_dictrQrrrr r rc r) TogetherImageToVideoTaskcrT)Nr*rUr6rrr rrVz!TogetherImageToVideoTask.__init__rGrHrIr$NcCs$|jt|ddddgdt|S)Nz image/pngrifirst) input_imageframe)r[ frame_images)r\rrrxrrr rNs  z1TogetherImageToVideoTask._prepare_payload_as_dictrQrrrr rrr))rYrabcrtypingrhuggingface_hub.hf_apir!huggingface_hub.inference._commonrrr,huggingface_hub.inference._providers._commonrr r r huggingface_hub.utilsr r r get_loggerr-rrrrrrrr3rFrSrhrsrrErrrrrrrr s2    (3