uPj7(ddlmZmZddlmZddlmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZdd lmZmZd d lmZerddlZdd lmZdd l m!Z"ddlm#Z#eZ$eZ%e"Z!neZ$eZ%eZ!GddeZ&GddeZ'GddeZ(y))ABCabstractmethod)Iterator) TYPE_CHECKINGAnyUnionN) FileDeleted)UserAPIKeyAuth)TwoStepFileUploadConfig)AllMessageValuesCreateFileRequestFileContentRequest FileListPage%OpenAICreateFileRequestOptionalParamsOpenAIFileObject) LlmProviders ModelResponse) BaseConfig)Logging)Router)HttpxBinaryResponseContentc,eZdZdZedeefdZy)BaseFileUploadStreamaRe-iterable request body that yields an upload's bytes lazily. A provider returns one of these (inside the upload config from ``transform_create_file_request``) when the upload body can be produced incrementally; the HTTP handler then sends it in bounded chunks instead of buffering the whole payload, which is what exhausts memory on large uploads. ``iter_bytes`` must return a fresh iterator each call so the body can be replayed if the upload is retried. returncyNselfs o/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/files/transformation.py iter_byteszBaseFileUploadStream.iter_bytes2s-0N)__name__ __module__ __qualname____doc__rrbytesr"rr#r!rr&s  0HUO00r#rc eZdZeedefdZedefdZedede e fdZ dedzdedzded e d e d e f d Zeded e d e d e dee eedff dZededzdej&ded e def dZeded e d e deee ffdZedej&ded e defdZeded e d e deee ffdZedej&ded e ddfdZededzd e d e deee ffdZedej&ded e de efdZeddd e d e deee ffdZedej&ded e ddfd Zded!e e d e d e d"e de f d#Z! d*dedej&d$e"ded%e d!e e d e d e d&d'dedzd(e#dzde"fd)Z$y)+BaseFilesConfigrcyrrrs r!custom_llm_providerz#BaseFilesConfig.custom_llm_provider7s r#cy)a HTTP method to use for file uploads. Override this in provider configs if they need different methods. Default is POST (used by most providers like OpenAI, Anthropic). S3-based providers like Bedrock should return "PUT". POSTrrs r!file_upload_http_methodz'BaseFilesConfig.file_upload_http_method<sr#modelcyrr)r r0s r!get_supported_openai_paramsz+BaseFilesConfig.get_supported_openai_paramsFs r#api_baseNapi_keyoptional_paramslitellm_paramsdatac.|j|||||S)N)r3r4r0r5r6)get_complete_url)r r3r4r0r5r6r7s r!get_complete_file_urlz%BaseFilesConfig.get_complete_file_urlJs+$$+) %  r#create_file_datar cy)aE Transform OpenAI-style file creation request into provider-specific format. Returns: - dict: For pre-signed single-step uploads (e.g., Bedrock S3) - str/bytes: For traditional file uploads - TwoStepFileUploadConfig: For two-step upload process (e.g., Manus, GCS) Nr)r r0r;r5r6s r!transform_create_file_requestz-BaseFilesConfig.transform_create_file_request[r# raw_response logging_objcyrr)r r0r?r@r6s r!transform_create_file_responsez.BaseFilesConfig.transform_create_file_responsels r#file_idcy)z>Transform file retrieve request into provider-specific format.Nrr rCr5r6s r!transform_retrieve_file_requestz/BaseFilesConfig.transform_retrieve_file_requestvr>r#cy)z4Transform file retrieve response into OpenAI format.Nrr r?r@r6s r! transform_retrieve_file_responsez0BaseFilesConfig.transform_retrieve_file_responser>r#cy)zr#r cy)z2Transform file delete response into OpenAI format.NrrHs r!transform_delete_file_responsez.BaseFilesConfig.transform_delete_file_responser>r#purposecy)z:Transform file list request into provider-specific format.Nr)r rNr5r6s r!transform_list_files_requestz,BaseFilesConfig.transform_list_files_requestr>r#cy)z0Transform file list response into OpenAI format.NrrHs r!transform_list_files_responsez-BaseFilesConfig.transform_list_files_responser>r#file_content_requestrcy)z=Transform file content request into provider-specific format.Nr)r rSr5r6s r!transform_file_content_requestz.BaseFilesConfig.transform_file_content_requestr>r#rcy)z3Transform file content response into OpenAI format.NrrHs r!transform_file_content_responsez/BaseFilesConfig.transform_file_content_responser>r#messagesheadersctd)Nz^AudioTranscriptionConfig does not need a request transformation for audio transcription modelsNotImplementedError)r r0rXr5r6rYs r!transform_requestz!BaseFilesConfig.transform_requests" l  r#model_response request_dataencodingztiktoken.Encoding | None json_modec td)Nz_AudioTranscriptionConfig does not need a response transformation for audio transcription modelsr[) r r0r?r^r@r_rXr5r6r`r4ras r!transform_responsez"BaseFilesConfig.transform_responses" m  r#NN)%r$r%r&propertyrrr,strr/listrr2dictr r:rr(r=httpxResponseLiteLLMLoggingObjrrBtuplerFrIrKrMrPrRrUrWr r]rboolrcrr#r!r*r*6s^  \    >c9d   * t         "  ,     tS%!:: ;    Tz nn '       MMM M sDy  MMCnnC'C C  CCKKK K sDy  KKAnnA'A A  AAItII I sDy  II?nn?'? ?   ??L2LL L sDy  LLBnnB'B B & BB    '(           .#!%  nn &  '    '(   - t $;   r#r*ceZdZedededeedede de f dZ e dd ededzdedzde fd Z e dd edzdedzde d e dzd edzdedefdZed ededzdedede f dZed ededzdededdf dZy)BaseFileEndpointscreate_file_request llm_routertarget_model_names_listlitellm_parent_otel_spanuser_api_key_dictrc Kywrr)r rprqrrrsrts r! acreate_filezBaseFileEndpoints.acreate_files  NrCc Kywrr)r rCrsrqs r!afile_retrievez BaseFileEndpoints.afile_retrieves  rwrNlimitafterr7c Kywrr)r rNrsrtrzr{r7s r! afile_listzBaseFileEndpoints.afile_lists  rwc Kywrrr rCrsrqr7s r! afile_deletezBaseFileEndpoints.afile_delete  rwrc Kywrrrs r! afile_contentzBaseFileEndpoints.afile_contentrrwrrd)r$r%r&rr rrgrfSpanr rrvryintrhrr}rrrr#r!roros .  "&c  #'  *      %)   #'+ TM      !  t  #'+  *  Tz  Tz           #'+          #'+     &   r#ro))abcrrcollections.abcrtypingrrrriopenai.types.file_deletedr litellm.proxy._typesr litellm.types.filesr litellm.types.llms.openair r rrrrlitellm.types.utilsrrchat.transformationrtiktoken*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjlitellm.routerr_Routerrrkrrr*rorr#r!rs#$,, 1/7<,X0D* D F D F 13 1 d jd N3 3 r#