_j/ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZddlmZmZmZejeZdZd Zd Zd Zd d hZGddeeZGddeZGdde ZGddeZ GddeZ!GddeZ"GddeZ#de$de$fdZ%Gdd eeZ&Gd!d"e&Z'Gd#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_progressc<eZdZdZdeffd ZdededefdZxZS) TogetherTaskz"Base class for Together API tasks.taskcdttt|dS)N)providerbase_urlrsuper__init__ _PROVIDER _BASE_URL)selfr __class__s j/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/huggingface_hub/inference/_providers/together.pyrzTogetherTask.__init__(s( )idKKKKK mapped_modelapi_keyreturnc|jxxdkrn xdkrnndSxdkrdSxdkrdSxdkrn xd krnnd 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$s r!_prepare_routezTogetherTask._prepare_route+si3#3#3#3#3#3#3#3#3#3//!!!!!))%%%%%'' #3#3#3#3#3#3#3#3#|LdiLLLMMMr")__name__ __module__ __qualname____doc__strrr. __classcell__r s@r!rr%s,,LSLLLLLL N3 N N N N N N N N N Nr"rc@eZdZfdZddeezdedzdefdZxZ S)TogetherTextGenerationTaskcbtttdSN)rrrrr s r!rz#TogetherTextGenerationTask.__init__:& )i@@@@@r"Nresponserequest_paramsr%ct|dd}|d|d|dddS)Nchoicesrtext finish_reasonseed)rArB)generated_textdetails)rget)rr<r=outputs r! get_responsez'TogetherTextGenerationTask.get_response=sW(##I.q1$Vn!'O!&>VhktjuUv f   x    \    r"r<r=clt|}tj|dddSrhrkrms r!rGz%TogetherImageToImageTask.get_responseror"rHrpr5s@r!rrrrus+++++  '+ D\     >FFUT\FCTW[C[FgjFFFFFFFFr"rrc ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherFeatureExtractionTaskcJtddS)Nr*r]r:s r!rz&TogetherFeatureExtractionTask.__init__s" -.....r"rNrOrPr%Nc4||jdt|SN)inputrdrer rrNrOrPs r!rVz6TogetherFeatureExtractionTask._prepare_payload_as_dicts.*6  *%%  r"r<r=c@dt|dDS)Ncg|] }|d S) embedding).0items r! z>TogetherFeatureExtractionTask.get_response..sIIId[!IIIr"ri)rrr<r=s r!rGz*TogetherFeatureExtractionTask.get_responses#IIhx.@.@.HIIIIr"rHrpr5s@r!rrs/////  '+ D\     JJUT\JCTW[C[JgjJJJJJJJJr"rc ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherTextToSpeechTaskcJtddS)Nr)r]r:s r!rz!TogetherTextToSpeechTask.__init__rtr"rNrOrPr%Nc4||jdt|Srrrs r!rVz1TogetherTextToSpeechTask._prepare_payload_as_dicts.*6  *%%  r"r<r=czt|tr|Stdt|jd)Nz1Expected raw audio bytes for text-to-speech, got .)rWrIr-rSr/rs r!rGz%TogetherTextToSpeechTask.get_responses= h & & OgTRZ^^Mdggghhhr"rHrpr5s@r!rrs+++++    '+  D\       iiUT\iCTW[C[igjiiiiiiiir"rrOr%ct|}d|vr|d|d<d|vr3|d}d|vr |d|d<d|vr |d|d<|S)zNMap HF inference-client conventions onto Together's video API parameter names.r`ra target_sizewidthheight)r rf)rOrs r!_normalize_video_parametersrsZ((J **(nn-BCC 7 "" nn]33 k ! !"-g"6Jw  { " "#.x#8Jx r"c4eZdZdZddeezdedzdefdZdS)TogetherVideoTaskz>Base class for Together's asynchronous video generation tasks.Nr<r=r%c$|tdt|}|d}|std|jd|}td|d}t tD]}| |tvrntj tt||j }t||}|d}| |tvrntdtd |d kr>|d pi} t!d | d pd|dkrt!d||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-rrEurlloggerinforange_VIDEO_MAX_POLL_ATTEMPTS_VIDEO_PENDING_STATUSEStimesleep_VIDEO_POLLING_INTERVALr rrjson RuntimeErrorcontent) rr<r=jobjob_id status_urlr_status_responserrvideo_responses r!rGzTogetherVideoTask.get_responses3  !dee ex   WUVV V'*55V55  ABBB""/00  A!f4K&K&K J. / / /)mm// NDZ/[[O  0 0 0!&&((CWWX&&F!f4K&K&KN&>NNN  X  GGG$$*EmEIIiDXDXDk\kmmnn n [ RRRSS SWWY''-222;??  TRSS S$**955N+++%%r"rH) r/r0r1r2rIrJrrrGrr"r!rrsTHH-&-&UT\-&CTW[C[-&gj-&-&-&-&-&-&r"rc <eZdZfdZdededededzfdZxZS)TogetherTextToVideoTaskcJtddS)Nr+r]r:s r!rz TogetherTextToVideoTask.__init__r^r"rNrOrPr%Nc4||jdt|S)N)rcrd)rerrs r!rVz0TogetherTextToVideoTask._prepare_payload_as_dict s.*6  **55  r"rYr5s@r!rrsp*****  '+ D\         r"rc <eZdZfdZdededededzfdZxZS)TogetherImageToVideoTaskcJtddS)Nr,r]r:s r!rz!TogetherImageToVideoTask.__init__rtr"rNrOrPr%NcZ|jt|dddgdt|S)Nz image/pngrvfirst) input_imageframe)rd frame_images)rerrrs r!rVz1TogetherImageToVideoTask._prepare_payload_as_dictsG+6-4V{-[-[-[fmnno  **55  r"rYr5s@r!rrsp+++++    '+  D\           r"r))rbrabcrtypingrhuggingface_hub.hf_apir!huggingface_hub.inference._commonrrr,huggingface_hub.inference._providers._commonr r r r huggingface_hub.utilsr rr get_loggerr/rrrrrrrr7rLr[rrrrrJrrrrrr"r!rs ;;;;;;  LKKKKKKKKK  H % %  & $]3NNNNN%sNNN(      !7    !74FFFFFlFFF&%F%F%F%F%F|%F%F%FPJJJJJLJJJ"iiiii|iii, D T    0&0&0&0&0& c0&0&0&f      /         0     r"