iddlZddlmZmZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddl mZmZddlmZmZerddlmZdd lmZeZeZneZeZGd d eZy) N)ABCabstractmethod) TYPE_CHECKINGAnyDictListOptionalUnion)Headers)AllMessageValuesCreateBatchRequest) LiteLLMBatch LlmProviders)Logging)BaseLLMExceptionceZdZdZdZeedefdZe dZ e dde de d e ed e d e d ee d ee de fdZed ee d ee de d ed edede fdZede ded e d e deee ee efff dZedee dej2ded e def dZede d e d e deee ee efffdZedee dej2ded e def dZede dedeee fddfdZ!y)BaseBatchesConfiga Abstract base class for batch processing configurations across different LLM providers. This class defines the interface that all provider-specific batch configurations must implement to work with LiteLLM's unified batch processing system. cy)Nselfs }/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/batches/transformation.py__init__zBaseBatchesConfig.__init__"s returncy)z4Return the LLM provider type for this configuration.Nrrs rcustom_llm_providerz%BaseBatchesConfig.custom_llm_provider%s rc |jjDcic]`\}}|jdsJ|jds9t|tj tj ttfs|||bc}}Scc}}w)z,Get configuration dictionary for this class.___abc) __dict__items startswith isinstancetypes FunctionTypeBuiltinFunctionType classmethod staticmethod)clskvs r get_configzBaseBatchesConfig.get_config+s  **, ,1<<%LL(&&--   qD,   sA%BNheadersmodelmessagesoptional_paramslitellm_paramsapi_keyapi_basecy)a Validate and prepare environment-specific headers and parameters. Args: headers: HTTP headers dictionary model: Model name messages: List of messages optional_params: Optional parameters litellm_params: LiteLLM parameters api_key: API key api_base: API base URL Returns: Updated headers dictionary Nr)rr/r0r1r2r3r4r5s rvalidate_environmentz&BaseBatchesConfig.validate_environment?s4 rdatacy)a~ Get the complete URL for batch creation request. Args: api_base: Base API URL api_key: API key model: Model name optional_params: Optional parameters litellm_params: LiteLLM parameters data: Batch creation request data Returns: Complete URL for the batch request Nr)rr5r4r0r2r3r8s rget_complete_batch_urlz(BaseBatchesConfig.get_complete_batch_url[s0 rcreate_batch_datacy)aR Transform the batch creation request to provider-specific format. Args: model: Model name create_batch_data: Batch creation request data optional_params: Optional parameters litellm_params: LiteLLM parameters Returns: Transformed request data Nr)rr0r;r2r3s rtransform_create_batch_requestz0BaseBatchesConfig.transform_create_batch_requestu( r raw_response logging_objcy)a2 Transform provider-specific batch response to LiteLLM format. Args: model: Model name raw_response: Raw HTTP response logging_obj: Logging object litellm_params: LiteLLM parameters Returns: LiteLLM batch object Nrrr0r?r@r3s rtransform_create_batch_responsez1BaseBatchesConfig.transform_create_batch_responser>rbatch_idcy)a% Transform the batch retrieval request to provider-specific format. Args: batch_id: Batch ID to retrieve optional_params: Optional parameters litellm_params: LiteLLM parameters Returns: Transformed request data Nr)rrDr2r3s r transform_retrieve_batch_requestz2BaseBatchesConfig.transform_retrieve_batch_requests$ rcy)a< Transform provider-specific batch retrieval response to LiteLLM format. Args: model: Model name raw_response: Raw HTTP response logging_obj: Logging object litellm_params: LiteLLM parameters Returns: LiteLLM batch object NrrBs r!transform_retrieve_batch_responsez3BaseBatchesConfig.transform_retrieve_batch_responser>r error_message status_codercy)a  Get the appropriate error class for this provider. Args: error_message: Error message status_code: HTTP status code headers: Response headers Returns: Provider-specific exception class Nr)rrIrJr/s rget_error_classz!BaseBatchesConfig.get_error_classs r)NN)"__name__ __module__ __qualname____doc__rpropertyrrrr)r.dictstrrr r r7rr r:r bytesrr=httpxResponseLiteLLMLoggingObjrrCrFrHintr rLrrrrrs  \    &"&"&   '(      # 3-    6 3- #       !    2  .     uc4S>) *   * } nn '       *     uc4S>) *   & } nn '       *   /2 =B4==Q   rr)r&abcrrtypingrrrrr r rUr litellm.types.llms.openair r litellm.types.utilsrr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjchat.transformationr_BaseLLMExceptionrWrrrrrasP #BB ;XK*(@ @ r