KiN3SSKrSSKrSSKrSSKJrJrJr SSKJr SSK J r J r Jr SSKJr "SS5r"S S 5rg!\ a \ "S5ef=f) N)AnyDictOptional)genaizTPlease install the Google Gemini SDK to use this feature: 'pip install google-genai')call_llm_and_track_usageget_model_paramswith_privacy_mode)Clientc|\rSrSrSrS S\\S\\S\\S\\\\ 4S\ S \\\\ 44 S jjr S r g) r a A drop-in replacement for genai.Client that automatically sends LLM usage events to PostHog. Usage: client = Client( api_key="your_api_key", posthog_client=posthog_client, posthog_distinct_id="default_user", # Optional defaults posthog_properties={"team": "ai"} # Optional defaults ) response = client.models.generate_content( model="gemini-2.0-flash", contents=["Hello world"], posthog_distinct_id="specific_user" # Override default ) Napi_keyposthog_clientposthog_distinct_idposthog_propertiesposthog_privacy_modeposthog_groupsc NUc [S5e[SUUUUUUS.UD6Ulg)a: Args: api_key: Google AI API key. If not provided, will use GOOGLE_API_KEY or API_KEY environment variable posthog_client: PostHog client for tracking usage posthog_distinct_id: Default distinct ID for all calls (can be overridden per call) posthog_properties: Default properties for all calls (can be overridden per call) posthog_privacy_mode: Default privacy mode for all calls (can be overridden per call) posthog_groups: Default groups for all calls (can be overridden per call) **kwargs: Additional arguments (for future compatibility) N/posthog_client is required for PostHog tracking)r rrrrr) ValueErrorModelsmodelsselfr rrrrrkwargss Z/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/posthog/ai/gemini/gemini.py__init__Client.__init__'sC(  !NO O ) 31!5)    )rNNNNFN) __name__ __module__ __qualname____firstlineno____doc__rstr PostHogClientrrboolr__static_attributes__rrrr r s&"&26-17;%*37 # !/ &c]  %T#s(^4  #  !c3h0  rr cn\rSrSr%Sr\\S'S#S\\S\\S\\S\\ \\ 4S \ S \\ \\ 44 S jjr S \\S \\S\\ \\ 4S\\ S\\ \\ 44 Sjr S$S\S\\S\\S\\ \\ 4S \\ S \\ \\ 4S\ 4SjjrS\S\\S\\S\\ \\ 4S\ S\\ \\ 4S\ 4SjrS\S\\S\\S\\ \\ 4S\ S\\ \\ 4S\ \\ 4S\ \\4S\S\4SjrS rS$S\S\\S\\S\\ \\ 4S \\ S \\ \\ 4S\ 4S!jjrS"rg)%rIzK Models interface that mimics genai.Client().models with PostHog tracking. _ph_clientNr rrrrrc \Uc [S5eX lX0lU=(d 0UlXPlX`lUcE[ RRS5=(d [ RRS5nUc [S5e[R"US9Ul SUl g)a Args: api_key: Google AI API key. If not provided, will use GOOGLE_API_KEY or API_KEY environment variable posthog_client: PostHog client for tracking usage posthog_distinct_id: Default distinct ID for all calls posthog_properties: Default properties for all calls posthog_privacy_mode: Default privacy mode for all calls posthog_groups: Default groups for all calls **kwargs: Additional arguments (for future compatibility) NrGOOGLE_API_KEYAPI_KEYz_API key must be provided either as parameter or via GOOGLE_API_KEY/API_KEY environment variable)r z)https://generativelanguage.googleapis.com) rr,_default_distinct_id_default_properties_default_privacy_mode_default_groupsosenvirongetrr _client _base_urlrs rrModels.__init__Ps(  !NO O(%8!#5#; %9"- ?jjnn%56S"**..:SG ?q ||G4 Drcall_distinct_id call_trace_idcall_propertiescall_privacy_mode call_groupscUbUO URnUbUO URnUbUO URn[UR5n U(aU R U5 Uc[ [R"55nXbXU4$)z9Merge call-level PostHog parameters with client defaults.) r0r2r3dictr1updater&uuiduuid4) rr:r;r<r=r> distinct_id privacy_modegroups propertiess r_merge_posthog_paramsModels._merge_posthog_params{s + ** !, ++  !, 7T=Q=Q$223    o .   -M:VKKrmodelposthog_trace_idrc URUUUUU5uppn XS.UEn[U URSU U U U URURR R 4 0UD6$)a  Generate content using Gemini's API while tracking usage in PostHog. This method signature exactly matches genai.Client().models.generate_content() with additional PostHog tracking parameters. Args: model: The model to use (e.g., 'gemini-2.0-flash') contents: The input content for generation posthog_distinct_id: ID to associate with the usage event (overrides client default) posthog_trace_id: Trace UUID for linking events (auto-generated if not provided) posthog_properties: Extra properties to include in the event (merged with client defaults) posthog_privacy_mode: Whether to redact sensitive information (overrides client default) posthog_groups: Group analytics properties (overrides client default) **kwargs: Arguments passed to Gemini's generate_content rJcontentsgemini)rHrr,r8r7rgenerate_content)rrJrNrrKrrrrrDtrace_idrGrErFkwargs_with_contentss rrPModels.generate_contents:  & &# "$   @ z*/OO'  OO      NN LL   0 0  #  rrDrQrGrErFc ^^^^^^^^^^ ^ ^ ^[R"5m SSS.m/m TTS.TEn TRRR"S0U D6m U UUUUUUUU UU UU4 Sjn U "5$)Nr input_tokens output_tokensrMc3h> # T Hn[US5(a@UR(a/[URSS5[URSS5S.m[US5(a,UR(aTR UR5 Uv M [ R "5nUT- nSR T5nT RT TTTT T TTTUU5 g![ R "5nUT- nSR T5nT RT TTTT T TTTUU5 f=f7f)Nusage_metadataprompt_token_countrcandidates_token_countrUtext)hasattrrYgetattrr\appendtimejoin_capture_streaming_event)chunkend_timelatencyoutputaccumulated_contentrNrDrFrrJrErGresponser start_timerQ usage_statss r generator5Models._generate_content_streaming..generators8" %Eu&677EK&" 99;"Z/!45--    99;"Z/!45--  sD2BC' AD2'AD//D2r)rar7rgenerate_content_stream)rrJrNrDrQrGrErFrkwargs_without_streamrlrhrirjrks````````` @@@@r_generate_content_streaming"Models._generate_content_streamingsiYY[ 781&M  */X P P<<&&>>WAVW% % % N{rrkrfrgc Uc[[R"55nSU[U5[ UR UUR U55[ UR UU SS./5SU RSS5U RSS5U UURS. U=(d 0En UcS U S '[UR S 5(aUR RUS U US 9 gg)NrO assistant)contentrolerVrrW) z $ai_providerz $ai_modelz$ai_model_parametersz $ai_inputz$ai_output_choicesz$ai_http_statusz$ai_input_tokensz$ai_output_tokensz $ai_latencyz $ai_trace_idz $ai_base_urlFz$process_person_profilecapturez$ai_generation)rDeventrGrF) r&rBrCrr r, _format_inputr6r8r^rw) rrJrNrDrQrGrErFrrkrfrgevent_propertiess rrcModels._capture_streaming_event s  4::<(H%$4V$<*""8, #4#[9:# # + B!,!!D"$ NN' (R) .  :? 6 7 4??I . . OO # #'&+ $  /rc[U[5(aSUS./$[U[5(a/nUH|n[U[5(aURSUS.5 M.[ US5(a URSUR S.5 M_URS[U5S.5 M~ U$S[U5S./$)z*Format input contents for PostHog trackinguser)rurtr\) isinstancer&listr`r^r\)rrN formatteditems rryModels._format_input>s h $ $#9: : $ ' 'I dC(($$f%FGT6**$$f%KL$$fT%KL ! #H >? ?rc fURUUUUU5uppn UR"UUU U U U U 40UD6$)N)rHrp)rrJrNrrKrrrrrDrQrGrErFs rrnModels.generate_content_streamOsa  & &# "$   @ z//           r)r8r7r0r3r2r1r,r )NNNNN)r!r"r#r$r%r'__annotations__rr&rrr(rrHrPrpintfloatrcryrnr)rrrrrIs"&26-17;%*37)E#)E!/)E&c] )E %T#s(^4 )E # )E!c3h0)EVL"3-L }L"$sCx.1 L $D> L d38n- LH.2*.7;/3373 3 &c] 3 #3- 3 %T#s(^4 3 'tn3 !c3h03 3 j99c] 9 3- 9 T#s(^, 99c3h(99v11c] 1 3- 1 T#s(^, 11c3h(1S#X1#s(^111f@*.2*.7;/337  &c]  #3-  %T#s(^4  'tn !c3h0   rr)r4rarBtypingrrrgoogler ImportErrorModuleNotFoundErrorposthog.ai.utilsrrr posthog.clientr r'rrrrrsd &&  31 1 he e C ^ s AA