hX<dZddlZddlmZmZddlZGddZy)zPCloudZero Resource Names (CZRN) generation and validation for LiteLLM resources.N)AnycastceZdZdZej dZdZdee e fde fdZ de de d e d e d e d e de fd Z de de fdZde dee e e e e e ffdZde de fdZdde de de fdZy) CZRNGeneratorz@Generate CloudZero Resource Names (CZRNs) for LiteLLM resources.zO^czrn:([a-z0-9-]+):([a-zA-Z0-9-]+):([a-z0-9-]+):([a-z0-9-]+):([a-z0-9-]+):(.+)$cy)zInitialize CZRN generator.N)selfs l/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/integrations/cloudzero/cz_resource_names.py__init__zCZRNGenerator.__init__s rowreturncd}|j|jdd}d}|jdd}|j|}d}|jdd}|} |j|||||| S) aaCreate a CZRN from LiteLLM daily spend data. CZRN format: czrn:::::: For LiteLLM resources, we map: - service-type: 'litellm' (the service managing the LLM calls) - provider: The custom_llm_provider (e.g., 'openai', 'anthropic', 'azure') - region: 'cross-region' (LiteLLM operates across regions) - owner-account-id: The team_id or user_id (entity_id) - resource-type: 'llm-usage' (represents LLM usage/inference) - cloud-local-id: model litellmcustom_llm_providerunknownz cross-region entity_idz llm-usagemodel) service_typeproviderregionowner_account_id resource_typecloud_local_id)_normalize_providerget_normalize_componentcreate_from_components) r r rrrrrrrrs r create_from_litellm_dataz&CZRNGenerator.create_from_litellm_data"s! ++CGG4I9,UVGGK3 44Y?# +**%-') +  r rrrrrrc |j|d}|j|}|j|}|j|}|j|}d|d|d|d|d|d| }|j|std||S)z)Create a CZRN from individual components.T)allow_uppercasezczrn::zGenerated CZRN is invalid: )ris_valid ValueError)r rrrrrrczrns r rz$CZRNGenerator.create_from_componentsGs00t0T ,,X6**62445EF11-@ |nAhZq:J9K1]O[\]k\lm}}T":4&AB B r r%cJt|jj|S)z3Validate a CZRN string against the standard format.)bool CZRN_REGEXmatch)r r%s r r#zCZRNGenerator.is_valid`sDOO))$/00r c |jj|}|std|ttt t t t t t f|j S)zExtract all components from a CZRN. Returns: (service_type, provider, region, owner_account_id, resource_type, cloud_local_id) zInvalid CZRN format: )r(r)r$rtuplestrgroups)r r%r)s r extract_componentsz CZRNGenerator.extract_componentsdsS %%d+4TF;< <E#sCc367HHr cFtjjjdtjjjdtjj jdtjj jdtjjjdtjjjdtjjjdtjjjdtjjjdtjjjd i }|jjd d }||vr|S|j||S) z1Normalize provider names to standard CZRN format.azure anthropicawsgcpgooglecohere huggingface replicatez together-ai_-)r LlmProvidersAZUREvalueAZURE_AI ANTHROPICBEDROCK VERTEX_AIGEMINICOHERE HUGGINGFACE REPLICATE TOGETHER_AIlowerreplacer)r r provider_map normalizeds r rz!CZRNGenerator._normalize_provideros/  & & , ,g  ) ) / /  * * 0 0+  ( ( . .  * * 0 0%  ' ' - -x  ' ' - -x  , , 2 2M  * * 0 0+  , , 2 2M  ^^%--c37  \ )  J77r componentr!c|sy|s|j}tjdd|}tjdd|}|jd}|xsdS)z7Normalize a CZRN component to meet format requirements.rz [^a-zA-Z0-9-]r9z-+)rFresubstrip)r rJr!s r rz"CZRNGenerator._normalize_components]!)IFF+S)< FF5#y1 OOC( %I%r N)F)__name__ __module__ __qualname____doc__rLcompiler(r dictr,rrrr'r#r+r.rrrr r rrsJnoJ # DcN# s# J      21S1T1 Is IuS#sCc5Q/R I8C8C8.&c&D&UX&r r)rRrLtypingrrrrrr r rVs "W &&r