uPj ddlmZmZmZddlmZddlmZddlm Z m Z ddl m Z de de de d ee e e ffd Zed ZGd d Zy))AnyFinalTypeVar)BaseContainerConfig)ResponsesAPIRequestUtils)$ContainerCreateOptionalRequestParams"ContainerListOptionalRequestParams)GenericLiteLLMParams container_idcustom_llm_providerlitellm_paramsreturnctj|}|jd|}|jd}|r|dk(r|}|jd}|r|jds||d<|||fS)zDecode a LiteLLM-managed container ID for upstream API calls. Returns: (original_container_id, resolved_provider, updated_litellm_params) response_idr openaimodel_id)r_decode_container_idget)r r r decodedoriginal_container_iddecoded_providerdecoded_model_ids ]/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/containers/utils.py'decode_managed_container_id_for_requestr s}.BB<PG#*;;}l#K%kk*?@/8;.%kk*5 2 2: >%5z" "5~ EETc eZdZededefdZedededefdZedede fdZ e dd e d e dzd ee e fdzd ee e fdzde f d Zy)ContainerRequestUtils passed_paramsrcVt}gd}|D]}||vs||||||<|S)zLExtract only valid container creation parameters from the passed parameters.) expires_afterfile_ids extra_headers extra_body)r)r container_create_optional_params valid_paramsparams r-get_requested_container_create_optional_paramzCContainerRequestUtils.get_requested_container_create_optional_param(sR 3W2X( " OE %-*>*J:G:N07 O0/rcontainer_provider_configr%c|j}|jDcic] \}}||vs ||}}}|j|dScc}}w)z3Get the optional parameters for container creation.F)r% drop_params)get_supported_openai_paramsitemsmap_openai_params)r)r%supported_paramskvfiltered_paramss r$get_optional_params_container_createz:ContainerRequestUtils.get_optional_params_container_create<sg #<"W"W"Y4T3Y3Y3[!u41a_`dt_t!Q$!u!u(::-<;  "vs A A cVt}gd}|D]}||vs||||||<|S)zHExtract only valid container list parameters from the passed parameters.)afterlimitorderr# extra_query)r )rcontainer_list_optional_paramsr&r's r+get_requested_container_list_optional_paramzAContainerRequestUtils.get_requested_container_list_optional_paramLsR 1S0T& " ME %-*>*J8Ee8L.u5 M.-rN response_objr litellm_metadatar$c|xsi}|jdixsi}|jd}d}|d}|rzd|vrvd}|r|d}t|tr#|jddj }n:t|t r*t |dkDrt|dj }|r7|r5t|dr)tj|||j}||_ |S) a Encode container_id in response object with provider/model metadata for routing. This mirrors the responses API pattern where response IDs are encoded with routing metadata so follow-up calls can route to the correct provider. Encodes when: 1. litellm_metadata contains model_info.id (indicating router/proxy usage), OR 2. extra_body contains target_model_names (indicating model-specific routing) Direct SDK calls with explicit custom_llm_provider and no routing hints return raw IDs. Args: response_obj: Response object with an `id` attribute (ContainerObject, DeleteContainerResult, etc.) custom_llm_provider: Provider name (e.g., "azure", "openai") litellm_metadata: Optional litellm_metadata dict that may contain model_info.id extra_body: Optional extra_body dict that may contain target_model_names Returns: The same response object with encoded container_id (if routing metadata present) model_infoidFTtarget_model_names,r)r rr ) r isinstancestrsplitstriplistlenhasattrr_build_container_idr?) r;r r<r$r>r should_encode target_models encoded_ids rencode_container_id_in_responsez5ContainerRequestUtils.encode_container_id_in_responseas:,1r,<,@,@r,R,XVX >>$'    M .*< M'12F'G mS1,2237:@@BH t4]9Ka9O"=#34::rZsm&&O<6FFF)F 3) )* F0 CLwwr