uPj>ddlZddlmZmZddlmZmZddlZddlmZddl m Z m Z ddl m Z mZerddlmZdd lmZeZeZneZeZGd d eZy) N)ABCabstractmethod) TYPE_CHECKINGAny)Headers)AllMessageValuesCreateBatchRequest) LiteLLMBatch LlmProviders)Logging)BaseLLMExceptionceZdZdZdZeedefdZe dZ e dde de d e ed e d e d e dzd e dzde fdZed e dzd e dzde d e d e dede fdZede ded e d e dee ze e efzf dZede dzdej,ded e def dZede d e d e dee ze e efzfdZede dzdej,ded e def dZede dede ezddfdZy)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 q/home/oem/CyberBrain26-unified/.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%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,r-r.r/r0r1r2s rvalidate_environmentz&BaseBatchesConfig.validate_environment>rrdatacy)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)rr2r1r-r/r0r5s rget_complete_batch_urlz(BaseBatchesConfig.get_complete_batch_urlYrrcreate_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)rr-r8r/r0s rtransform_create_batch_requestz0BaseBatchesConfig.transform_create_batch_requestrrr 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 Nrrr-r;r<r0s rtransform_create_batch_responsez1BaseBatchesConfig.transform_create_batch_responserrbatch_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)rr@r/r0s r transform_retrieve_batch_requestz2BaseBatchesConfig.transform_retrieve_batch_requestrrcy)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 Nrr>s r!transform_retrieve_batch_responsez3BaseBatchesConfig.transform_retrieve_batch_responserr 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)rrErFr,s rget_error_classz!BaseBatchesConfig.get_error_classrr)NN)__name__ __module__ __qualname____doc__rpropertyrr rr&r+dictstrlistrr4r r7bytesrr:httpxResponseLiteLLMLoggingObjr r?rBrDintrrHrrrrrs C\CC  &##   '(      t *    4 * t       !    0  .     tCH~ %   ( Tz nn '       (     tCH~ %   $ Tz nn '       (  S  s  TT[^  `r    rr)r#abcrrtypingrrrRrlitellm.types.llms.openairr litellm.types.utilsr r *litellm.litellm_core_utils.litellm_loggingr _LiteLLMLoggingObjchat.transformationr_BaseLLMExceptionrTrrrrr^sM #% ;XK*(v v r