i <ddlmZddlmZddlmZGddZy))BatchRequestCounts)Metadata LiteLLMBatchc2eZdZdZe ddededdfdZy) BedrockBatchesHandlerz Handler for Bedrock Batches. Specific providers/models needed some special handling. E.g. Twelve Labs Embedding Async Invoke Nbatch_idaws_region_namereturnrc  ddl ddlmfd ddl} fd}|jj 5}|j |}|jcdddS#1swYyxYw)av Handle async invoke status check for AWS Bedrock. This is for Twelve Labs Embedding Async Invoke. Args: batch_id: The async invoke ARN aws_region_name: AWS region name **kwargs: Additional parameters Returns: dict: Status information including status, output_file_id (S3 URL), etc. rN)BedrockEmbeddingcK}|jd dd{}ddlm}|ddd|jdxsd|d d }||d d |d |d|d|jd|d dk(r|jdndt d|d dk(rdnd|d dk(rdnd|ddd }|S7w)N)invocation_arnr logging_objrroutputDataConfigs3OutputDataConfigs3UrifailureMessagemodelArn)output_file_idfailure_message model_arn invocationArnbatchstatus submitTimelastModifiedTimeendTimefailed completed)totalr"r 24hz/v1/embeddings) idobjectr created_atin_progress_at completed_at failed_atrequest_countsmetadatacompletion_windowendpoint input_file_id)_get_async_invoke_statuslitellm.types.utilsrgetr) embedding_handlerstatus_responseropenai_batch_metadataresultr r r kwargsrs u/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/bedrock/batches/handler.py_async_get_statuszLBedrockBatchesHandler._handle_async_invoke_status.._async_get_status%s0 0 2 %O$5$N$N%' /'% %O 9#22D"E(##$3#6#67G#H#NB,Z8 : !""?3&x0*<8./AB,00;"8,8*--i81#28#< #KaQR / 9X E11  /"') %F*MOs!CB>BCcj}j| |j|jS#|jwxYwN)new_event_loopset_event_looprun_until_completeclose)new_loopr:asyncios r9 run_in_threadzHBedrockBatchesHandler._handle_async_invoke_status..run_in_threadWsK--/H  " "8 , !223D3FG  s A A)rB$litellm.llms.bedrock.embed.embeddingr concurrent.futuresfuturesThreadPoolExecutorsubmitr7) r r rr8 concurrentrCexecutorfuturer r:rBs ```` @@@r9_handle_async_invoke_statusz1BedrockBatchesHandler._handle_async_invoke_statussY" I, , ` " !   2 2 4__]3F==?5 4 4s !A**A3r<)__name__ __module__ __qualname____doc__ staticmethodstrrLr0r9rrs;9=P#P#(+P# P#P#rSrN)openai.types.batchrrOpenAIBatchMetadatar2rrr0rSr9rVs1>,Z#Z#rS