uPjBddlZddlZddlmZmZmZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZddlmZmZer ddlZdd lmZeZneZGd d e Zy) N) TYPE_CHECKINGAnyFinal)verbose_logger)RUNWAYML_DEFAULT_API_VERSIONRUNWAYML_POLLING_TIMEOUT)BaseImageGenerationConfig)get_secret_str)AllMessageValues#OpenAIImageGenerationOptionalParams) ImageObject ImageResponse)LoggingceZdZUdZdZeed<dZeed< d*dedzdedzd ed ed ed e dzd efdZ d+ded ede e d ed ededzdedzd efdZ edeeefded efdZededed dfdZedeeefd efdZ d,dededeeefded ej,f dZ d,dededeeefded ej,f dZ d+d edej,dededed ed edd dedzd!e dzd efd"Z d+d edej,dededed ed edd dedzd!e dzd efd#Zd ed e efd$Zd%ed ed ed&e d ef d'Zd ed(ed ed eded ef d)Zy)-RunwayMLImageGenerationConfigz= Configuration for RunwayML image generation models. zhttps://api.dev.runwayml.comDEFAULT_BASE_URLzv1/text_to_imageIMAGE_GENERATION_ENDPOINTNapi_baseapi_keymodeloptional_paramslitellm_paramsstreamreturnc|xstdxs |j}|jd}|jr|d|j}|S)zi Get the complete url for the request Some providers need `model` in `api_base` RUNWAYML_API_BASE/)r rrstripr)selfrrrrrr complete_urls z/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/runwayml/image_generation/transformation.pyget_complete_urlz.RunwayMLImageGenerationConfig.get_complete_url(sU%d7J(KdtOdOd #**3/  ) )*^1T-K-K,LMLheadersmessagescz|xstdxs td}|s tdd||d<t|d<|S)NRUNWAYML_API_SECRETRUNWAYML_API_KEYz2RUNWAYML_API_SECRET or RUNWAYML_API_KEY is not setzBearer AuthorizationX-Runway-Version)r ValueErrorr) rr$rr%rrrr final_api_keys r!validate_environmentz2RunwayMLImageGenerationConfig.validate_environment=sT  b~&;< bOa@b QR R%,]O#< &B"#r# response_datamodel_responsec |jsg|_|jdg}t|tr|D]}t|tr'|jj t |d:t|tsK|jj t |jdd|jdd|S)a Transform RunwayML response format to OpenAI ImageResponse format. RunwayML response format (after polling): { "id": "task_123...", "status": "SUCCEEDED", "output": ["https://cloudfront.net/.../image.png"], "completedAt": "2025-11-13T..." } OpenAI ImageResponse format: { "data": [ { "url": "https://cloudfront.net/.../image.png", "b64_json": null } ] } Args: response_data: JSON response from RunwayML (after polling completes) model_response: ImageResponse object to populate Returns: Populated ImageResponse in OpenAI format outputN)urlb64_jsonr2r3)dataget isinstanceliststrappendr dict)r.r/r1 image_items r!&_transform_runwayml_response_to_openaizDRunwayMLImageGenerationConfig._transform_runwayml_response_to_openaiQsB"""$N &))(B7 fd #$  j#."''..# *%)  D1"''..# *ud ;%/^^J%E $r# start_time timeout_secscVtj|z |kDrtd|dy)z Check if operation has timed out. Args: start_time: Start time of the operation timeout_secs: Timeout duration in seconds Raises: TimeoutError: If operation has exceeded timeout z&RunwayML task polling timed out after z secondsN)time TimeoutErrorr=r>s r!_check_timeoutz,RunwayMLImageGenerationConfig._check_timeouts2 99; #l 2!G ~U]^_ _ 3r#c6|jddj}tjd||dk(ry|dk(r6|jdd}|jd d }t d |d |d |dk(r t d|dvryt d|)a Check RunwayML task status from response. RunwayML statuses: PENDING, RUNNING, SUCCEEDED, FAILED, CANCELLED, THROTTLED Args: response_data: JSON response from RunwayML task endpoint Returns: Normalized status string: "running", "succeeded", or raises on failure Raises: ValueError: If task failed or status is unknown statuszRunwayML task status: %s SUCCEEDED succeededFAILEDfailurez Unknown error failureCodeunknownz"RunwayML image generation failed: z (code: ) CANCELLEDz'RunwayML image generation was cancelled)PENDINGRUNNING THROTTLEDrunningzUnknown RunwayML task status: )r5upperrdebugr+)r.rEfailure_reason failure_codes r!_check_task_statusz0RunwayMLImageGenerationConfig._check_task_statuss &))(B7==?7@ [  x $1$5$5i$QN"/"3"3M9"MLA.AQQYZfYgghij j { "FG G : :=fXFG Gr#task_idcddlm}|}tj}|jd}|d|}t j d| |j |||j||} | j| j} |j| } | d k(r| S| d k(rtjd z) a Poll RunwayML task until completion (sync). RunwayML POST returns immediately with a task that has status PENDING/RUNNING. We need to poll GET /v1/tasks/{task_id} until status is SUCCEEDED or FAILED. Args: task_id: The task ID to poll api_base: Base URL for RunwayML API headers: Request headers (including auth) timeout_secs: Total timeout in seconds (default: 600s = 10 minutes) Returns: Final response with completed task r)_get_httpx_clientr /v1/tasks/zPolling RunwayML task: %srBr2r$r.rHrR) &litellm.llms.custom_httpx.http_handlerrZr@rrrTrCr5raise_for_statusjsonrWsleep) rrXrr$r>rZclientr=task_urlresponser.rEs r!_poll_task_syncz-RunwayMLImageGenerationConfig._poll_task_syncs, M)+ IIK ??3'%Jj :8(C   :L  Qzzhz@H  % % '$MMOM,,=,IF$9$ 1 !r#cKddl}ddlm}||jj}t j }|j d}|d|} tjd|  |j|||j| | d{} | j| j} |j| } | d k(r| S| d k(rtjd d{7`7w)am Poll RunwayML task until completion (async). Args: task_id: The task ID to poll api_base: Base URL for RunwayML API headers: Request headers (including auth) timeout_secs: Total timeout in seconds (default: 600s = 10 minutes) Returns: Final response with completed task rN)get_async_httpx_client) llm_providerrr[z!Polling RunwayML task (async): %srBr\r]rHrRr^)litellmr_rh LlmProvidersRUNWAYMLr@rrrTrCr5r`rarWasynciorb) rrXrr$r>rjrhrcr=rdrer.rEs r!_poll_task_asyncz.RunwayMLImageGenerationConfig._poll_task_asyncs& Q.Gz8RunwayML polling complete, transforming to OpenAI formatr.r/)ra Exceptionget_error_classrxr$rrTr5r+requestrrfrrr<rrror/rprqrrrrrrsr.erX poll_headerss r!#transform_image_generation_responsezARunwayMLImageGenerationConfig.transform_image_generation_response"s:D (--/M ;<'**40@A A*1199==orR , 4 4 < < @ @ASUq r ++** 1 , %))+ WX::');  E && NqcR(44$,,'  sC11 D*:+D%%D*c vK |j} t j d| jd} | s td|jjjdd|jjjd td }|j| |j|t d{}|j} t j d |j| | S#t$r0} |jd| |j|jd} ~ wwxYw7xw)a Async transform the image generation response to the litellm image response. RunwayML returns a task immediately with status PENDING/RUNNING. We need to poll the task until it completes (status SUCCEEDED) using async polling. rurvNz$RunwayML starting polling (async)...ryrzr)rFr*r{r|z@RunwayML polling complete (async), transforming to OpenAI formatr})rar~rrxr$rrTr5r+rrrnrrr<rs r!)async_transform_image_generation_responsezGRunwayMLImageGenerationConfig.async_transform_image_generation_responsemsG& (--/M CD'**40@A A*1199==orR , 4 4 < < @ @ASUq r "22** 1 3  %))+ _`::');  E && NqcR(44$,,'  * s4D9C;B*D9>D7?OA& $$QC'CE7Jefvewxno r#promptc|xsd|d}d|vr |d|d<nd|d<|jD]\}}||vs |dvs|||<|S)a. Transform the image generation request to the RunwayML image generation request body RunwayML expects: - model: The model to use (e.g., 'gen4_image') - promptText: The text prompt - ratio: The aspect ratio (e.g., '1920:1080', '1080:1920', '1024:1024') gen4_image)r promptTextrr)r)items) rrrrrr$runwayml_request_bodyrvs r!"transform_image_generation_requestz@RunwayMLImageGenerationConfig.transform_image_generation_requests"*l (  o %-a9b ! !! !  !  !F!%!%!% !%  !%  !% !%r#r)rmr@typingrrrrlitellm._loggingrlitellm.constantsrr5litellm.llms.base_llm.image_generation.transformationr litellm.secret_managers.mainr litellm.types.llms.openair r litellm.types.utilsr rtiktoken*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrrrr#r!rsT ,, +8;X*U%$=U%r#