uPjdUddlmZmZddlmZddlmZddlmZmZm Z m Z ddl Z ddl m Z ddlmZmZddlZddlmZdd lmZdd lmZdd lmZdd lmZmZmZmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(er ddl)m*Z+e+Z,neZ,GddeZ-GddedZ.Gddee/Z0de jbde.fdZ2dZ3e e4d<d Z5e e6e6e7e8fd!fe4d"<d#e/de8dzfd$Z9d%e8dzde:dzfd&Z;d'e.de:dzfd(Zy)-)MappingSequence)datetime)MappingProxyType) TYPE_CHECKINGAnyFinalLiteralN) RequestFiles)ReadOnly TypedDict)RUNWAYML_DEFAULT_API_VERSION)encode_url_path_segment)BaseLLMException)BaseVideoConfig)AsyncHTTPHandler HTTPHandler_get_httpx_clientget_async_httpx_client)get_secret_str)GenericLiteLLMParams) VideoCreateOptionalRequestParams VideoObject)encode_video_id_with_providerextract_original_video_id)Loggingc eZdZy) RunwayMLErrorN)__name__ __module__ __qualname__p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/runwayml/videos/transformation.pyrr%sr#rceZdZUeeed<eeed<eeed<eeed<eeeezed<eeed<eeed<eeed<eeeefed <y ) _RunwayTaskResponseidstatus createdAt completedAtoutput failureCodefailureprogress estimatedCostN) rr r!r str__annotations__rfloatrr"r#r$r&r&)sp  SM}# Xc]S( ))# c]uoGCJ/00r#r&F)totalcTeZdZUeeed<eeded<eeed<eeed<y)_VideoObjectDatar'videoobjectr( created_atN)rr r!r r0r1r intr"r#r$r5r55s0  WW% && SM r#r5) extra_items raw_responsereturnc&|j}|SN)json)r; response_datas r$_parse_runway_task_responserA<s0<0A0A0CM r#g{Gz?_USD_PER_CREDIT))i' 480p)i`720p)i =1080p._RESOLUTION_AREA_TIERSratioct|trd|vry|jd\}}}|jr|jsyt |t |zt fdt DdS)N:c34K|]\}}|ks |ywr>r").0 thresholdlabelareas r$ z'_ratio_to_resolution..QsZ+9ePYIYZs 4k) isinstancer0 partitionisdigitr9nextrF)rG width_str_ height_strrNs @r$_ratio_to_resolutionrXJsm eS !S%5$s3Iq*    J$6$6$8i.3z?2D Z/EZ\` aar#secondsc>|sy t|S#t$rYywxYwr>)r2 ValueError)rYs r$_duration_secondsr\Ts) W~ s  r@c|jd}t|tsy|jd}t|ttfsyt |t zS)Nr/credits)getrQrr9r2rB)r@estimated_costr^s r$_estimated_cost_usdra]sO)--o>N ng .#'' 2G gU| , >O ++r#r.c Xtdtdtt|dzS)Ndr)minmaxroundr2)r.s r$_progress_percentrggs$ sC5x3!678 99r#c&eZdZdZfdZdedefdZdedede de fdZ d=d e ded ed zd e d zde f d Z deded zd e defdZdededede d e d e dee eeffdZdeeefdefdZ d=dedej,deded zde d zdef dZdedefdZded zdefdZ d>deded e d e ded zdeee ff dZd e eefdefd!Zdej,dede fd"Z!dej,dede fd#Z" d>dededed e d e d$eeefd zdeee ffd%Z# d>dej,deded zdefd&Z$ d?ded e d e d'ed zd(ed zd)ed zd*eeefd zdeee ffd+Z% d>dej,deded zde eeffd,Z&deded e d e deee ff d-Z'dej,dedefd.Z(deded e d e deee ff d/Z) d>dej,deded zdefd0Z*d1efd2Z+d3Z,d4Z-d5Z. d@d6Z/ d=d7Z0 d>d8Z1d>d9Z2d:ed;ed e ejfzde4fd<Z5xZ6S)ARunwayMLVideoConfiga& Configuration class for RunwayML video generation. RunwayML uses a task-based API where: 1. POST /v1/text_to_video, /v1/image_to_video, or /v1/video_to_video creates a task 2. The task returns immediately with a task ID 3. Client must poll or wait for task completion c"t|yr>)super__init__)self __class__s r$rlzRunwayMLVideoConfig.__init__us r#modelr<c gdS)a& Get the list of supported OpenAI parameters for video generation. Maps OpenAI params to RunwayML equivalents: - prompt -> promptText - input_reference -> promptImage - size -> ratio (e.g., "1280x720" -> "1280:720") - seconds -> duration )ropromptinput_referencerYsizeuser extra_headersr")rmros r$get_supported_openai_paramsz/RunwayMLVideoConfig.get_supported_openai_paramsxs   r#video_create_optional_params drop_paramsci}d|vr |d}||d<d|vr.|d}t|trd|vr|jdd|d<d|vr:|d}|3 t|trtt |n t||d<|j|}|jD]\} } | |vs | || <|S#t t f$rYFwxYw) a  Map OpenAI parameters to RunwayML format. Mappings: - prompt -> promptText - input_reference -> promptImage - size -> ratio (convert "WIDTHxHEIGHT" to "WIDTH:HEIGHT") - seconds -> duration (convert to integer) rr promptImagersxrIrGrYduration) rQr0replacer9r2r[ TypeErrorrvitems) rmrwrorx mapped_paramsrrrsrYsupported_openai_paramskeyvalues r$map_openai_paramsz%RunwayMLVideoConfig.map_openai_paramss 35   < <%ABS%TO+:M- ( 1 16v>D$$)-c3)? g& 4 49)DG"GQRY[^G_E'N0CehipeqM*- *.)I)I%)P6<<> +JC11%* c" +#I.s2B66CCNheadersapi_keylitellm_paramsc|r|jr|xs |j}|xs*tjxstdxs td}| td|j d|t dd|S)z Validate environment and set up authentication headers. RunwayML uses Bearer token authentication via RUNWAYML_API_SECRET. RUNWAYML_API_SECRETRUNWAYML_API_KEYzeRunwayML API key is required. Set RUNWAYML_API_SECRET environment variable or pass api_key parameter.zBearer zapplication/json) AuthorizationzX-Runway-Versionz Content-Type)rlitellmrr[updater)rmrrorrs r$validate_environmentz(RunwayMLVideoConfig.validate_environments n447!7!7G  uw u.9N*O uSabtSu  ?w  #*7)!4$@ 2  r#api_basec,|d}|jdS)z Get the base URL for RunwayML API. The specific endpoint path will be added in the transform methods. zhttps://api.dev.runwayml.com/v1/)rstrip)rmrorrs r$get_complete_urlz$RunwayMLVideoConfig.get_complete_urls  8Hs##r#rq$video_create_optional_request_paramsct||d|}|j|}|jD cic]\} } |dk(s| dk7s| | } } } g} | | |d|fScc} } w)a Transform the video creation request for RunwayML API. RunwayML has three generation endpoints discriminated by which input is present, and each request body rejects unknown fields: - /text_to_video: promptText only (rejects promptImage) - /image_to_video: promptImage (+ optional promptText) - /video_to_video: promptVideo or videoUri (rejects promptImage) )ro promptTextimage_to_videorzr)r_select_generation_endpointr) rmrorqrrrr merged_paramsendpointrr request_data files_lists r$transform_video_create_requestz2RunwayMLVideoConfig.transform_video_create_requests$ 0$ 7  ::=I*7)<)<)>2 %3(N^B^beivbvCJ2 2 +- ZH:Qxj)AAA 2 s AArcn|jd|jdy|jdyy)N promptVideovideoUrivideo_to_videorzr text_to_video)r_)rmrs r$rz/RunwayMLVideoConfig._select_generation_endpoints=   M * 6,:J:J::V:b#   M * 6#r#r; logging_objcustom_llm_providerct|}|jddd|j|jdd|j|jdd}d|vr(|dr#t |dt r|dd n|d|d <d |vr#|j|jd |d <d |vsd|vr(|jd d|jddd|d<|rSd|vr|d|d<d|vr.|d}t |t rd|vr|jdd|d<d|vrt |d|d<tdi|} |r(| jrt| j||| _ dt| jfdt|r|jdndfdt|ffD cic] \} } | | |  c} } | _| Scc} } w)ah Transform the RunwayML video creation response. RunwayML returns a task object that looks like: { "id": "task_123...", "status": "PENDING" | "RUNNING" | "SUCCEEDED" | "FAILED", "output": ["https://...video.mp4"] (when succeeded) } We map this to OpenAI VideoObject format. r'r6r(pendingr)r'r7r(r8r+r output_urlr* completed_atr,r-unknownVideo generation failedcodemessageerrorrorGrIr{rsr|rYduration_secondsvideo_resolutionNprovider_reported_cost_usdr")rAr__map_runway_status_parse_runway_timestamprQlistr0r}rr'rr\rYrXrausage) rmror;rrrr@ video_datarG video_objrrs r$transform_video_create_responsez3RunwayMLVideoConfig.transform_video_create_responses( ;<H  ##D"---m.?.?).TU66}7H7H7UV /  } $x)@/9x9PRV.W h'*]jks]t | $ M ))-)E)EmFWFWXeFf)gJ~ & M )Y--G%))-C(,,Y8QR#Jw  ,&&27&; 7#,&+G4eS)cUl).sC)@Jv&\)(+L,D(E 9%&44 9<<8GZ\abIL $%6y7H7H%IJ#%9Wc,:J:J7:Sim%no-/B=/QR U  J  s3G  runway_statuscTddddddd}|j|jdS)z Map RunwayML status to OpenAI status format. RunwayML statuses: PENDING, RUNNING, SUCCEEDED, FAILED, CANCELLED OpenAI statuses: queued, in_progress, completed, failed queued in_progress completedfailed)PENDINGRUNNING SUCCEEDEDFAILED CANCELLED THROTTLED)r_upper)rmr status_maps r$rz&RunwayMLVideoConfig._map_runway_statusas9 $$!!  ~~m113X>>r# timestamp_strc|sy tj|jdd}t|j S#t t f$rYywxYw)z Convert RunwayML ISO 8601 timestamp to Unix timestamp. RunwayML returns timestamps like: "2025-11-11T21:48:50.448Z" We need to convert to Unix timestamp (seconds since epoch). rZz+00:00)r fromisoformatr}r9 timestampr[AttributeError)rmrdts r$rz+RunwayMLVideoConfig._parse_runway_timestamprsT  ..}/D/DS(/STBr||~& &N+  s=AAAvideo_idvariantcLt|}t|d}|d|}i} || fS)a Transform the video content request for RunwayML API. RunwayML doesn't have a separate content download endpoint. The video URL is returned in the task output field. We'll retrieve the task and extract the video URL. r field_name/tasks/rr) rmrrrrroriginal_video_idencoded_video_idurlparamss r$transform_video_content_requestz3RunwayMLVideoConfig.transform_video_content_requestsA$=X#F"9:KXb"c!z)9(:;(*F{r#r@cd}d|vr!|dr|d}t|tr|dn|}|sU|jdd}|dvrtd|d|d k(r |jd d }td |td |S)z~ Helper method to extract video URL from RunwayML response. Shared between sync and async transforms. Nr+rr(UNKNOWN)rrrz#Video is still processing (status: z). Please wait and try again.rr-z Unknown errorzVideo generation failed: z