^j/ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZddlmZmZmZej*eZdZd Zd Zd Zd d hZGddeeZGddeZGdde ZGddeZ GddeZ!GddeZ"GddeZ#de$de$fdZ%Gdd eeZ&Gd!d"e&Z'Gd#d$e&Z(y)%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.taskc:t|tt|y)N)providerbase_urlrsuper__init__ _PROVIDER _BASE_URL)selfr __class__s x/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/together.pyrzTogetherTask.__init__(s )idK mapped_modelapi_keyreturnc|jxxdk(ryxdk(ryxdk(ryxdk(ryxdk(ry xd k(ry td |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+seii33/$43/4!)%' 3#$43#4-dii[8KLMMr")__name__ __module__ __qualname____doc__strrr. __classcell__r s@r!rr%s2,LSL N3 N N Nr"rc@eZdZfdZddeezdedzdefdZxZ S)TogetherTextGenerationTaskc8t|ttyN)rrrrr s r!rz#TogetherTextGenerationTask.__init__: )i@r"Nresponserequest_paramsr%cvt|dd}|d|jd|jdddS)Nchoicesrtext finish_reasonseed)rArB)generated_textdetails)rget)rr<r=outputs r! get_responsez'TogetherTextGenerationTask.get_response=sD(#I.q1$Vn!'O!< 6*  r"N) r/r0r1rbytesdictrrrGr4r5s@r!r7r79s/A UT\ CTW[C[ gj r"r7c @eZdZfdZdededededzffd ZxZS)TogetherConversationalTaskc8t|ttyr9rr:s r!rz#TogetherConversationalTask.__init__Ir;r"inputs parametersprovider_mapping_infor%Nct||||}|y|jd}t|trH|jddk(r4t|jdtrd|dvrd|ddd|d<|S)Nresponse_formattype json_schemaschema)rSrU)r_prepare_payload_as_dictrE isinstancerJ)rrNrOrPpayloadrRr s r!rVz3TogetherConversationalTask._prepare_payload_as_dictLs'26:G\] ?"++&78  -##F+}<?..}=tDOM::&)-8B*G% &r" r/r0r1rrrJrrVr4r5s@r!rLrLHs7A'+D\ r"rLc ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherTextToImageTaskc$t|dy)Nr'rrr:s r!rz TogetherTextToImageTask.__init__c )r"rNrOrPr%Ncv|j}t|}d|vr|jd|d<|dd|d|iS)Nnum_inference_stepsstepsbase64)promptrRmodel) provider_idr pop)rrNrOrPr#s r!rVz0TogetherTextToImageTask._prepare_payload_as_dictfsM-88  , J .",..1F"GJw  XccWVbccr"r<r=cTt|}tj|dddSNdatarb64_jsonrrb b64decoderr<r= response_dicts r!rGz$TogetherTextToImageTask.get_responsep, *  f 5a 8 DEEr"rH r/r0r1rrrJrrVrIrrGr4r5s@r!r[r[bs`*dd'+dD\d dFUT\FCTW[C[FgjFr"r[c ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherImageToImageTaskc$t|dy)Nr(r]r:s r!rz!TogetherImageToImageTask.__init__v )*r"rNrOrPr%Nc|j}t|d}t|}|jdd}d|vr|jd|d<|j }d|vxrd|v}|rd |ind |gi} d|i| d d i|d |iS)Nz image/jpegdefault_mime_typercr`rakontextzflux.1 image_urlreference_imagesrRrbrd)rerr rflower) rrNrOrPr#rzrclowered use_image_url image_fields r!rVz1TogetherImageToImageTask._prepare_payload_as_dictys-88 FlC !, "- J .",..1F"GJw $$&!W,DW1D BO{I&>VhktjuUv f   x   \   r"r<r=cTt|}tj|dddSrhrkrms r!rGz%TogetherImageToImageTask.get_responseror"rHrpr5s@r!rrrrus[+  '+ D\  >FUT\FCTW[C[FgjFr"rrc ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherFeatureExtractionTaskc$t|dy)Nr*r]r:s r!rz&TogetherFeatureExtractionTask.__init__s -.r"rNrOrPr%Nc6||jdt|SN)inputrdrer rrNrOrPs r!rVz6TogetherFeatureExtractionTask._prepare_payload_as_dicts)*66 *%  r"r<r=cJt|dDcgc]}|d c}Scc}w)Nri embedding)r)rr<r=items r!rGz*TogetherFeatureExtractionTask.get_responses%.6x.@.HId[!IIIs rHrpr5s@r!rrs[/  '+ D\  JUT\JCTW[C[JgjJr"rc ^eZdZfdZdededededzfdZd deezd e dzdefd Z xZ S) TogetherTextToSpeechTaskc$t|dy)Nr)r]r:s r!rz!TogetherTextToSpeechTask.__init__rtr"rNrOrPr%Nc6||jdt|Srrrs r!rVz1TogetherTextToSpeechTask._prepare_payload_as_dicts)*66 *%  r"r<r=cjt|tr|Stdt|jd)Nz1Expected raw audio bytes for text-to-speech, got .)rWrIr-rSr/)rr<r=s r!rGz%TogetherTextToSpeechTask.get_responses3 h &OLTRZ^MdMdLeefghhr"rHrpr5s@r!rrs[+    '+  D\    iUT\iCTW[C[igjir"rrOr%ct|}d|vr|jd|d<d|vr)|jd}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_parametersrsrZ(J *(nn-BC 7 " nn]3 k !"-g"6Jw  { "#.x#8Jx r"c2eZdZdZddeezdedzdefdZy)TogetherVideoTaskz>Base class for Together's asynchronous video generation tasks.Nr<r=r%c| tdt|}|jd}|s td|jd|}tj d|jd}t tD]}| |tvrntjttj||j}t||j}|jd}|||tvsntdtd |d k(r6|jd xsi} t!d | jd xsd|dk7rt!d||jdxsijd} | s tdtj| } t| | j"S)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_responses  !de ex UV V'**+1VH5  AB"/0 A!f4K&K JJ. /)m// NDZDZ/[O  0!&&(CWWX&F!f4K&K &&>%?}N  X GGG$*E!CEIIiDXDk\kClmn n [ !GzRS SWWY'-222;? RS S$**95N+%%%r"rH) r/r0r1r2rIrJrrrGr"r!rrs+H-&UT\-&CTW[C[-&gj-&r"rc <eZdZfdZdededededzfdZxZS)TogetherTextToVideoTaskc$t|dy)Nr+r]r:s r!rz TogetherTextToVideoTask.__init__r^r"rNrOrPr%Nc6||jdt|S)N)rcrd)rerrs r!rVz0TogetherTextToVideoTask._prepare_payload_as_dict s)*66 **5  r"rYr5s@r!rrs1*  '+ D\  r"rc <eZdZfdZdededededzfdZxZS)TogetherImageToVideoTaskc$t|dy)Nr,r]r:s r!rz!TogetherImageToVideoTask.__init__rtr"rNrOrPr%NcT|jt|dddgdt|S)Nz image/pngrvfirst) input_imageframe)rd frame_images)rerrrs r!rVz1TogetherImageToVideoTask._prepare_payload_as_dicts9+66-4V{-[fmno **5  r"rYr5s@r!rrs1+    '+  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  ;  LK   H %  & $]3N%sN(  !7  !74FlF&%F|%FPJLJ"i|i, D T 0& c0&f  /   0 r"