tPjgtUdZddlmZddlZddlmZddlmZddlZ ddl m Z dZ d e d <dd ZGd d Zy)zFocus export data transformer.) annotationsN) timedelta)Final)FOCUS_NORMALIZED_SCHEMA) team_id team_aliasorganization_idorganization_aliasuser_id user_email api_key_aliasmodel model_groupcustom_llm_providerr _TAG_KEYScdd}tj|j|tjj dS)aRBuild a Polars expression that produces a JSON Tags string per row. Uses ``pl.struct`` + ``map_elements`` to avoid materialising the entire DataFrame to a list of Python dicts. The JSON serialisation callback still runs in Python (GIL-bound), but struct-packing and loop dispatch are handled by Polars' Rust engine. c|jDcic]\}}| |t|}}}|rtj|SdScc}}w)N{})itemsstrjsondumps)rowkvtagss k/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/focus/transformer.py_struct_to_jsonz)_build_tags_expr.._struct_to_json$sG-0YY[JTQAMq#a&yJJ#'tzz$1T1Ks A A ) return_dtypeTags)rdictreturnr)plstruct map_elementsStringalias)available_keysrs r_build_tags_exprr*s;2 99^ $ 1 1/PRPYPY 1 Z ` `ag hhceZdZdZeZddZy)FocusTransformerz8Transforms LiteLLM DB rows into Focus-compatible schema.c|jr tj|jStDcgc]}||j vs|}}|r|j t|}n3|j tjdjd}|j tjdjtjjjtjddd jd }|j tjd jd tjd t!d zjd}d}tj"ddfd}tjdtj}tjdtj"dd}|j$g|tjdj'djdtjdjtj(jdtjdjtj(jdtjdjdtjdjd|tjdjd |tjd jd!tjd"jd#|jd$tjd%jtj(jd&tjd'jd(|tjdjd|tjd jd |tjd)jtj*jtj,j'djd*tjd+jd,|tjdj'djd-|jd.|tjdj'djd/tjd0jtj(jd1|jd2|tjdj'djd3|jd4|jd5tjdjd6|jd7|tjd)jtj*jtj,j'djd8|jd9|tjdj'djd:|jd;tjd+jd<tjd0jtj(jd=tjd0jtj(jd>|jd?|jd@tjd%jtj(jdAtjd%jtj(jdBtjd%jtj(jdCtjdDjdEtjdFjdGtjdHjtj(jdItjdJjtj(jdKtjdLjtj(jdM|jdNtjdjtj(jdScc}w)Oz=Return a normalized frame expected by downstream serializers.)schemarr!dateus) time_unitz%Y-%m-%dF)formatstrict usage_dateChargePeriodStartr)daysChargePeriodEndc8|jjdS)Nz%Y-%m-%dT%H:%M:%SZ)dtstrftime)cols rfmtz'FocusTransformer.transform..fmtHs66??#78 8r+c&|jS)N)cast)r<DECs rdecz'FocusTransformer.transform..decMs88C= r+N)dtypespendg BilledCostapi_keyBillingAccountIdrBillingAccountNamezAPI KeyBillingAccountTypeUSDBillingCurrencyBillingPeriodEndBillingPeriodStartUsageChargeCategory ChargeClassrChargeDescriptionz Usage-BasedChargeFrequency api_requestsConsumedQuantityRequests ConsumedUnitContractedCostContractedUnitPrice EffectiveCostrInvoiceIssuerName InvoiceIdListCost ListUnitPriceAvailabilityZonePricingCurrencyPricingCategoryPricingQuantity"PricingCurrencyContractedUnitPricePricingCurrencyEffectiveCostPricingCurrencyListUnitPrice PricingUnit ProviderName PublisherNameRegionId RegionName ResourceId ResourceName ResourceTypezAI and Machine LearningServiceCategoryz Generative AIServiceSubcategoryr ServiceNamer SubAccountIdr SubAccountNameSubAccountType)is_emptyr$ DataFramer/rcolumns with_columnsr*litr(r<rAUtf8rstrptimeDatetimerDecimalselect fill_nullr'Int64Float64) selfframerr)r=rCnone_strnone_decrBs @r transformzFocusTransformer.transform0s >> <?u||- w))#. / 5 5l C- FF9  " "299 - 3 34F G-  FF? # ( ( 3 9 9:N O-  FF9  # #$8 9 - FF5M   1 2 - () * 0 01C D -  *+ , 2 23G H-  FF7O ! !"2 3-  NN= )-  FF7O  + 1 12E F-  FF= ! ' '(9 :-  () * 0 01B C-  *+ , 2 23F G-  ~&++BHH5::2::FPPQTU V \ \]o p-  FF:  $ $^ 4- w))#. / 5 56F G!- " NN0 1#- $ w))#. / 5 5o F%- & FF( ) . .ryy 9 ? ?@S T'- ( NN; ')- * w))#. / 5 5j A+- , NN? +-- . NN- ./- 0 FF5M   1 21- 2 NN, -3- 4 ~&++BHH5::2::FPPQTU V \ \]n o5- 6 NN? @7- 8 w))#. / 5 56T U9- : NN9 :;- < FF:  $ $] 3=- > FF( ) . .ryy 9 ? ? O?- @ FF( ) . .ryy 9 ? ? PA- B NN: &C- D NN< (E- F FF7O  + 1 1, ?G- H FF7O  + 1 1. AI- J FF7O  + 1 1. AK- L FF, - 3 34E FM- N FF? # ) )*> ?O- P FF= ! & &ryy 1 7 7 FQ- R FF9  " "299 - 3 3N CS- T FF< % %bii 0 6 67G HU- V NN+ ,W- X FF6N   * 0 0 8Y- - ;!Ms g<g<N)r pl.DataFramer#r)__name__ __module__ __qualname____doc__rr/rr+rr-r-+sB $FP r+r-)r)z list[str]r#zpl.Expr)r __future__rrdatetimertypingrpolarsr$r/rr__annotations__r*r-rr+rrs:$" +  5  i U U r+