uPj7 ddlmZmZddlmZddlmZeddGddZdeee fd e efd Z deee fd e efd Z deee fd e fd Zdeee fd e fd Zdeeee fd efdZy))MappingSequence) dataclass)FinalT)frozenslotsc:eZdZUedzed<edzed<defdZy)GroundingRequestsNweb_search_requestsgoogle_maps_grounding_requestsreturncHt|jxs |jSN)boolr r )selfs u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/vertex_ai/gemini/grounding_requests.pyhas_billable_groundingz(GroundingRequests.has_billable_grounding sD,,S0S0STT)__name__ __module__ __qualname__int__annotations__rrrrr r s$t#$'$J.UUrr itemr c||jd}t|ts tStd|DS)NgroundingChunksc3PK|]}t|ts|D]}| ywr isinstancer).0chunkkinds r z_chunk_kinds..s'^eE71KX]^PTT^T^s&&getr list frozenset)rchunkss r _chunk_kindsr*s4HH./F fd #{ ^v^ ^^rc||jd}t|ts tStd|DS)NwebSearchQueriesc3HK|]}t|ts|s|ywr)r str)r!querys rr$z_queries..sTu:eS3IeUTs """r%)rqueriess r_queriesr1s4XX01G gt ${ TT TTrcTdt|vxst|jdS)NmapsgoogleMapsWidgetContextToken)r*rr&rs r _is_maps_itemr6s& \$' ' Y49W0X+YYrc6t|xr dt|vS)Nweb)r6r*r5s r_attributes_queries_to_mapsr9!s   B5 T0B#BBrgrounding_metadatactd|D}td|D}td|D}td|D}tt |xsd|rt t |dSdS)aBillable grounding requests across candidates, counting each distinct query once. Duplicate queries within and across grounding metadata items collapse to the distinct-query count (#36377), and empty strings are ignored. Maps grounding is floored at one request whenever a candidate carries maps chunks or a widget token, since per-prompt billing charges the prompt even when no query is reported. c3BK|]}t|ts|ywrrr!rs rr$z/calculate_grounding_requests..-sZ$ 4QX@YZsc3XK|]"}t|rt|D]}|$ywrr9r1r!rr/s rr$z/calculate_grounding_requests...s6#'B4'HV^_cVd#MR# #**c3XK|]"}t|st|D]}|$ywrr?r@s rr$z/calculate_grounding_requests..1s6$#>t#DRZ[_R`$IN$ $rAc32K|]}t|ywr)r6r=s rr$z/calculate_grounding_requests..4s@$--@sN)r r )tupler(anyr lenmax)r:items web_queries maps_querieshas_mapss rcalculate_grounding_requestsrM%sZ*<ZZE"##K$$$L@%@@H  ,4DLs3|+rRs-! $d#UU$U_wsF{+_ #_U73;'UIcNUZV ,ZZCgc6k&:CtCXgc6k>R5SXir