uPj-TUdZddlZddlZddlmZmZmZdZeed<dede dzfdZ dd e d e de ed zfd Z d e de e fdZ d e de dzfdZd e de dzfdZd e de dzfdZd e de dzfdZ d de de de e de de dedzde fdZde de fdZde de dzfdZd e dedzfdZy)!z Utility functions for managed resources. This module provides common utility functions that can be used across different managed resource types (files, vector stores, etc.). N)AnyFinalLiteral)azureazure_ai&PASSTHROUGH_MANAGED_ID_AZURE_PROVIDERScustom_llm_providerreturnctt|d|xsdj}|sy|tvs|j dry|dk(s|j dryy)a-Map a pass-through ``custom_llm_provider`` to the provider scope that namespaces passthrough managed object IDs, or ``None`` when the route is not an OpenAI/Azure pass-through and managed IDs must not apply. Scoping is keyed on the explicit provider that the pass-through route forwards (``openai``, ``azure``, ``azure_ai``), not on the upstream URL, so a third-party OpenAI-compatible endpoint never triggers managed-ID minting. ``azure`` and ``azure_ai`` deliberately collapse to one ``"azure"`` scope: they expose the same Azure OpenAI files/batches surface, so an ID minted while routing as one must still resolve while routing as the other. Splitting them would make a managed ID minted on ``azure`` fail to resolve when replayed on ``azure_ai`` and vice versa. valueN)z.azurez .azure_airopenaiz.openai)strgetattrlowerrendswith)r providers r/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/base_llm/managed_resources/utils.py'resolve_passthrough_managed_id_providerrsc"'"5w@STZXZ[aacH 99X=N=NOf=g8x00;  resource_idprefixFct|tsy|dt| dzzz} tj|j }|j |r|Sy#t$rYywxYw)z Check if a resource ID is a base64 encoded unified ID. Args: resource_id: The resource ID to check prefix: The expected prefix for unified IDs Returns: Decoded string if valid unified ID, False otherwise F=) isinstancerlenbase64urlsafe_b64decodedecode startswith Exception)rrpaddeddecodeds ris_base64_encoded_unified_idr%*su k3 ' ##k*:):Q)>"??F11&9@@B   f %N s5A A+*A+ unified_idc t|tsgSt|}|r|}tjd|}|r<|j dj dDcgc]}|jc}SgScc}w#t$rgcYSwxYw)aZ Extract target model names from a unified resource ID. Args: unified_id: The unified resource ID (decoded or encoded) Returns: List of target model names Example: unified_id = "litellm_proxy:vector_store;unified_id,uuid;target_model_names,gpt-4,gemini-2.0" returns: ["gpt-4", "gemini-2.0"] ztarget_model_names,([^;]+),) rrr%researchgroupsplitstripr")r& decoded_idmatchmodels r*extract_target_model_names_from_unified_idr2Js *c*I9D #Jyy!> K /4{{1~/C/CC/HIeEKKMI I J  s/A>A A>A94A>7A>9A>> B  B c t|tsyt|}|r|}tjd|}|r|j dj Sy#t$rYywxYw)a. Extract resource type from a unified resource ID. Args: unified_id: The unified resource ID (decoded or encoded) Returns: Resource type string or None Example: unified_id = "litellm_proxy:vector_store;unified_id,uuid;..." returns: "vector_store" Nzlitellm_proxy:([^;]+)r(rrr%r*r+r,r.r"r&r/r0s r%extract_resource_type_from_unified_idr6osj *c*9D #Jyy!9:F ;;q>'') ) AAA A&%A&c t|tsyt|}|r|}tjd|}|r|j dj Sy#t$rYywxYw)a Extract the UUID from a unified resource ID. Args: unified_id: The unified resource ID (decoded or encoded) Returns: UUID string or None Example: unified_id = "litellm_proxy:vector_store;unified_id,abc-123;..." returns: "abc-123" Nzunified_id,([^;]+)r(r4r5s r$extract_unified_uuid_from_unified_idr9sj *c*9D #Jyy!6 C ;;q>'') ) r7c t|tsyt|}|r|}tjd|}|r|j dj Sy#t$rYywxYw)a2 Extract model ID from a unified resource ID. Args: unified_id: The unified resource ID (decoded or encoded) Returns: Model ID string or None Example: unified_id = "litellm_proxy:vector_store;...;model_id,gpt-4-model-id;..." returns: "gpt-4-model-id" Nz(?:^|;)model_id,([^;]+)r(r4r5s r extract_model_id_from_unified_idr;sj *c*9D #Jyy!;ZH ;;q>'') ) r7c t|tsyt|}|r|}gd}|D];}tj||}|s|j dj cSy#t$rYywxYw)aC Extract provider resource ID from a unified resource ID. Args: unified_id: The unified resource ID (decoded or encoded) Returns: Provider resource ID string or None Example: unified_id = "litellm_proxy:vector_store;...;resource_id,vs_abc123;..." returns: "vs_abc123" N)zresource_id,([^;]+)zvector_store_id,([^;]+)zfile_id,([^;]+)r(r4)r&r/patternspatternr0s r,extract_provider_resource_id_from_unified_idr?s *c*9D #J    .GIIgz2E{{1~++-- .  s"A'0A' A'%A'' A32A3 resource_type unified_uuidtarget_model_namesprovider_resource_idmodel_idadditional_fieldscd|d|ddj|d|d|g}|r.|jD]\}}|j|d|dj|S)a Generate a unified ID string (before base64 encoding). Args: resource_type: Type of resource (e.g., "vector_store", "file") unified_uuid: UUID for this unified resource target_model_names: List of target model names provider_resource_id: Resource ID from the provider model_id: Model ID from the router additional_fields: Additional fields to include in the ID Returns: Unified ID string (not yet base64 encoded) Example: generate_unified_id_string( resource_type="vector_store", unified_uuid="abc-123", target_model_names=["gpt-4", "gemini"], provider_resource_id="vs_xyz", model_id="model-id-123", ) returns: "litellm_proxy:vector_store;unified_id,abc-123;target_model_names,gpt-4,gemini;resource_id,vs_xyz;model_id,model-id-123" litellm_proxy:z unified_id,ztarget_model_names,r)z resource_id,z model_id,;)joinitemsappend) r@rArBrCrDrEpartskeyr s rgenerate_unified_id_stringrN sD ( l^$ chh'9:;< +,- H: E+113 +JC LLC5%) * + 88E?runified_id_stringctj|jjj dS)z Encode a unified ID string to base64. Args: unified_id_string: The unified ID string to encode Returns: Base64 encoded unified ID (URL-safe, padding stripped) r)rurlsafe_b64encodeencoder rstrip)rOs rencode_unified_idrT=s4  # #$5$<$<$> ? F F H O OPS TTrencoded_unified_idc |dt| dzzz}tj|j}|j dr|Sy#t $rYywxYw)z Decode a base64 encoded unified ID. Args: encoded_unified_id: The base64 encoded unified ID Returns: Decoded unified ID string or None if invalid rrrGN)rrrr r!r")rUr#r$s rdecode_unified_idrWJsj *SS9K5L4Lq4P-QQ 11&9@@B   . /N sA A AAc t|}|s|jdr|}nyt|t|t |t |t |dS#t$rYywxYw)a Parse a unified ID into its components. Args: unified_id: The unified ID (encoded or decoded) Returns: Dictionary with parsed components or None if invalid Example: { "resource_type": "vector_store", "unified_uuid": "abc-123", "target_model_names": ["gpt-4", "gemini"], "provider_resource_id": "vs_xyz", "model_id": "model-id-123" } rGN)r@rArBrCrD)rWr!r6r9r2r?r;r")r&r/s rparse_unified_idrYdsu*&z2 $$%56' C:N@L"LZ"X$PQ[$\8D    s!A4A A%$A%)rG)N)__doc__rr*typingrrrr__annotations__rrr%listr2r6r9r;r?dictrNrTrWrYrrr`s &&0E&E4Z:#  75>@"" #Y"J!!4Z!H!!4Z!H''4Z'T((4Z(b&* ...S . .  . d{ . .b U U U##*4'' D['r