o ưi**@sddlmZmZmZmZmZmZddlZddlZddl m Z ddl m Z ddl mZmZmZmZmZddlmZddlmZerQdd lmZdd lmZeZeZneZeZGd d d eZdS) ) TYPE_CHECKINGAnyDictOptionalTupleUnionN)StandardBuiltInToolCostTracking)get_secret_str)$ContainerCreateOptionalRequestParamsContainerFileListResponseContainerListResponseContainerObjectDeleteContainerResult)GenericLiteLLMParams)BaseContainerConfig)LoggingBaseLLMExceptioncspeZdZdZfddZdefddZdedede fd d Z d:d e d e e de fddZde e de de fddZde de ded e de f ddZdejdedefddZ d;de ded e de e de ede e de e e efdee e ffd d!Zdejdedefd"d#Zd$e de ded e dee e ff d%d&Zdejdedefd'd(Zd$e de ded e dee e ff d)d*Zdejdede fd+d,Z! d;d$e de ded e de e de ede e de e e efdee e ffd-d.Z"dejdede#fd/d0Z$d$e d1e de ded e dee e ff d2d3Z%dejdede&fd4d5Z'd6e d7ed e(e ej)fde*fd8d9Z+Z,S)<OpenAIContainerConfigz2Configuration class for OpenAI container API. cstdSN)super__init__self __class__d/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/openai/containers/transformation.pyr%szOpenAIContainerConfig.__init__returncCsgdS)zGGet the list of supported OpenAI parameters for container API. )nameZ expires_afterZfile_ids extra_headersrrrrrget_supported_openai_params(sz1OpenAIContainerConfig.get_supported_openai_params container_create_optional_params drop_paramscCst|S)z:No mapping applied since inputs are in OpenAI spec already)dict)rr#r$rrrmap_openai_params2sz'OpenAIContainerConfig.map_openai_paramsNheadersapi_keycCs0|p tjp tjp td}|dd|i|S)NZOPENAI_API_KEY AuthorizationzBearer )litellmr(Z openai_keyr update)rr'r(rrrvalidate_environment:s z*OpenAIContainerConfig.validate_environmentapi_baselitellm_paramscCs.|ptjptdptdpd}|ddS)z7Get the complete URL for OpenAI container API. ZOPENAI_BASE_URLZOPENAI_API_BASEzhttps://api.openai.com/v1/z /containers)r*r-r rstrip)rr-r.rrrget_complete_urlLsz&OpenAIContainerConfig.get_complete_urlr (container_create_optional_request_paramscCs"dd|D}d|i|}|S)zATransform the container creation request for OpenAI API. cSsi|] \}}|dvr||qS))r!r).0kvrrr gs zLOpenAIContainerConfig.transform_container_create_request..r )items)rr r2r.r'Z request_dictrrr"transform_container_create_request]s z8OpenAIContainerConfig.transform_container_create_request raw_response logging_objcCsd|}tdi|}tjddd}t|dr|jduri|_d|jvr)i|jd<||jdd<|S) z:Transform the OpenAI container creation response. Zopenai)sessionsprovider_hidden_paramsNZadditional_headersz$llm_provider-x-litellm-response-costr)jsonr rZget_cost_for_code_interpreterhasattrr>)rr9r: response_data container_objZcontainer_costrrr#transform_container_create_responsets  z9OpenAIContainerConfig.transform_container_create_responseafterlimitorder extra_queryc CsR|}i} |dur || d<|durt|| d<|dur|| d<|r%| ||| fS)zTransform the container list request for OpenAI API. OpenAI API expects the following request: - GET /v1/containers NrDrErF)strr+) rr-r.r'rDrErFrGurlparamsrrr transform_container_list_requests  z6OpenAIContainerConfig.transform_container_list_requestcC|}tdi|}|S)z6Transform the OpenAI container list response. Nr)r?r )rr9r:rAZcontainer_listrrr!transform_container_list_responsez7OpenAIContainerConfig.transform_container_list_response container_idcC |dd|}i}||fS)z9Transform the OpenAI container retrieve request. r/r0rrOr-r.r'rIdatarrr$transform_container_retrieve_requests z:OpenAIContainerConfig.transform_container_retrieve_requestcCrL)z:Transform the OpenAI container retrieve response. Nr)r?r )rr9r:rArBrrr%transform_container_retrieve_responsesz;OpenAIContainerConfig.transform_container_retrieve_responsecCrP)zTransform the container delete request for OpenAI API. OpenAI API expects the following request: - DELETE /v1/containers/{container_id} r/rQrRrrr"transform_container_delete_requests z8OpenAIContainerConfig.transform_container_delete_requestcCrL)z8Transform the OpenAI container delete response. Nr)r?r)rr9r:rAZ delete_resultrrr#transform_container_delete_responserNz9OpenAIContainerConfig.transform_container_delete_responsec Csd|dd|d} i} |dur|| d<|durt|| d<|dur'|| d<|r.| || | fS)zTransform the container file list request for OpenAI API. OpenAI API expects the following request: - GET /v1/containers/{container_id}/files r/z/filesNrDrErF)r0rHr+) rrOr-r.r'rDrErFrGrIrJrrr%transform_container_file_list_requests  z;OpenAIContainerConfig.transform_container_file_list_requestcCrL)z;Transform the OpenAI container file list response. Nr)r?r )rr9r:rA file_listrrr&transform_container_file_list_response!rNzOpenAIContainerConfig.transform_container_file_content_requestcCs|jS)z{Transform the OpenAI container file content response. Returns the raw binary content of the file. )content)rr9r:rrr)transform_container_file_content_responseDs z?OpenAIContainerConfig.transform_container_file_content_response error_message status_codecCsddlm}||||d)Nrr)r`messager')base_llm.chat.transformationr)rr_r`r'rrrrget_error_classOs z%OpenAIContainerConfig.get_error_classr)NNNN)-__name__ __module__ __qualname____doc__rlistr"r boolrr&r%rrHr,r1rr8httpxResponseLiteLLMLoggingObjr rCintrrrKr rMrTrUrVrrWrXr rZr\bytesr^rZHeadersrrc __classcell__rrrrr!sN        ! !          "      r)typingrrrrrrrjr*Z@litellm.litellm_core_utils.llm_cost_calc.tool_call_cost_trackingrZlitellm.secret_managers.mainr Zlitellm.types.containers.mainr r r r rZlitellm.types.routerrZ"base_llm.containers.transformationrZ*litellm.litellm_core_utils.litellm_loggingrZ_LiteLLMLoggingObjrbrZ_BaseLLMExceptionrlrrrrrs