uPj(UdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZmZdd lmZdd lmZdd lmZddlmZehdZe ed<dZe ed<eddddZ e ed<de!de"fdZ#de!de!fdZ$de"ddfdZ%de"de!ddfd Z&d!e"de!ddfd"Z'd#e"de!de"fd$Z(d%e!de)e"e ffd&Z*Gd'd(eZ+y))zJVideo generation for Hosted VLLM (vLLM-Omni OpenAI-compatible /v1/videos).N)Mapping)BufferedReader)MappingProxyType)Final)urlparse) FileTypes RequestFiles)ImageEditRequestUtils) SSRFError validate_url)OpenAIVideoConfig)get_secret_str)GenericLiteLLMParams) VideoCreateOptionalRequestParams> modelprompttimeout characters extra_body extra_query extra_headersinput_referencecustom_llm_provider_EXCLUDED_FORM_KEYS)image_referencevideo_referenceaudio_referencewidthheight num_framesfpsnum_inference_stepsguidance_scaleguidance_scale_2boundary_ratio flow_shifttrue_cfg_scaleseedgenerate_soundsound_durationnegative_promptenable_frame_interpolationframe_interpolation_expframe_interpolation_scaleframe_interpolation_model_pathlora extra_params aspect_ratio_VLLM_OMNI_VIDEO_PARAMS image_url video_url audio_url)rrr_REFERENCE_URL_KEYSvaluereturnct|tr|St|tr|rdSdSt|ttt frt j|St|S)Ntruefalse) isinstancestrboolrlisttuplejsondumps)r8s s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/hosted_vllm/videos/transformation.py_serialize_form_valuerEEsR% %v+G+%'4/0zz%  u:ct|ts|S|j}|r|ddvr|Stj|S)Nrz{[)r=r>striprBloads)r8strippeds rD _maybe_jsonrKOs> eS ! kkmH x{$. ::h rFurlct|jj}|dvry|dvrtd|dt |y)N)data)httphttpsz URL scheme 'z' is not allowed)rschemelowerr r )rLrRs rD_reject_unsafe_media_urlrTXsKSM((..0F  &&,vh.>?@@rFurl_keyitemct|tsy|j|}t|tr t |yyN)r=rgetr>rT)rUrVrLs rD_reject_unsafe_urls_in_itemrZas4 dG $'"C#s %rF field_namectj|}|yt|}t|tr|D]}t ||yt|t r t ||yyrX)r7rYrKr=r@rZr)r[r8rUparsedrVs rD_reject_unsafe_media_urlsr^isc(,,Z8G&F&$ 7D ' 6 7&'"#GV4#rFkeyc0t||t|SrX)r^rE)r_r8s rD _form_valueravsc5)  ''rF referencec|tj|}t|trd|j||ffSdd||ffS)Nrzinput_reference.png)r get_image_content_typer=rname)rb content_types rD_input_reference_filerg{sE/FFyQL)^,!INNI|#LMM  5y,O PPrFceZdZdZdedeffd Zdededede fdZ dd e ded edzd e dzde f d Z ded edzd e defdZ deded ede d e d e dee eeffdZxZS)HostedVLLMVideoConfigz vLLM-Omni videos API is OpenAI-compatible but requires multipart/form-data. https://docs.vllm.ai/projects/vllm-omni/en/latest/serving/videos_api/ rr9c2gt||tSrX)superget_supported_openai_paramsr3)selfr __class__s rDrlz1HostedVLLMVideoConfig.get_supported_openai_paramss' W 0 7 $  rFvideo_create_optional_params drop_paramsc\|jDcic] \}}| || c}}Scc}}wrX)items)rmrorrpr_r8s rDmap_openai_paramsz'HostedVLLMVideoConfig.map_openai_paramss9*F)K)K)M %3QVQbCJ   s ((Nheadersapi_keylitellm_paramscb| |jndxs|xstdxsd}i|dd|iS)NHOSTED_VLLM_API_KEYz fake-api-key AuthorizationzBearer )rur)rmrtrrurv resolved_keys rDvalidate_environmentz*HostedVLLMVideoConfig.validate_environmentsU(6'A^ # #t  34   F'E?gl^,DEErFapi_basec|xs td}| td|jd}|jdr|dS|dS)NHOSTED_VLLM_API_BASEztapi_base not set for Hosted VLLM videos API. Set via api_base parameter or HOSTED_VLLM_API_BASE environment variable/z/v1z/videosz /v1/videos)r ValueErrorrstripendswith)rmrr|rvresolved_api_basetrimmeds rDget_complete_urlz&HostedVLLMVideoConfig.get_complete_urlsh $,#U~>T/U  $Z +11#6   E "Yg& &*%%rFr$video_create_optional_request_paramsc ||d|jDcic]\}}|tvr| |t||c}}} |jd} | | d|fS| t | f|fScc}}w)N)rrr)rrrrarYrg) rmrrr|rrvrtr_r8rOrs rDtransform_video_create_requestz4HostedVLLMVideoConfig.transform_video_create_requests #G"L"L"NC11e6G[e,, "F!I!IJ[!\  "X% %+O<>HHs!A%)NN)__name__ __module__ __qualname____doc__r>r@rlrr?dictrsrr{rrAr r __classcell__)rns@rDriris     &F      #6: F F Ft F -t3 F  F&&*& & &"III I /3 I - II t\3& 'IrFri),rrBcollections.abcriortypesrtypingr urllib.parser httpx._typesrr litellm.images.utilsr $litellm.litellm_core_utils.url_utilsr r )litellm.llms.openai.videos.transformationr litellm.secret_managers.mainrlitellm.types.routerrlitellm.types.videos.mainr frozensetr__annotations__r3r7objectr>rErKrTrZr^rarArgrirrFrDrs.P #"!06HG75F&  U "6.&&&UC v & #$&&F&t& 5# 5f 5 5(S((C( QVQc9n0EQLI-LIrF