fix: comments

Author: k11kirky

posthog/ai/gemini/gemini.py

@@ -128,6 +128,9 @@ def _merge_posthog_params(
         if call_properties:
             properties.update(call_properties)
 
+        if call_trace_id is None:
+            call_trace_id = str(uuid.uuid4())
+
         return distinct_id, call_trace_id, properties, privacy_mode, groups
 
     def generate_content(
@@ -162,9 +165,6 @@ def generate_content(
             posthog_distinct_id, posthog_trace_id, posthog_properties, posthog_privacy_mode, posthog_groups
         )
 
-        if trace_id is None:
-            trace_id = str(uuid.uuid4())
-
         kwargs_with_contents = {"model": model, "contents": contents, **kwargs}
 
         return call_llm_and_track_usage(
@@ -319,9 +319,6 @@ def generate_content_stream(
             posthog_distinct_id, posthog_trace_id, posthog_properties, posthog_privacy_mode, posthog_groups
         )
 
-        if trace_id is None:
-            trace_id = str(uuid.uuid4())
-
         return self._generate_content_streaming(
             model,
             contents,