fix: extract_gemini_web_search_count

[zachary822]

extract_gemini_web_search_count errors if response.candidates is None

[greptile-apps]

_{1 file reviewed, 1 comment}

_{Edit Code Review Agent Settings | Greptile}

[greptile-apps]

[P3] Prefer parameterizing the web-search “no candidates” cases

This new test covers candidates=None, but it duplicates the structure of the existing “no web search” tests nearby; given the repo preference for parameterized tests, it’d be simpler to combine the “no candidates” variants (e.g. candidates=None and candidates=[]) into one @pytest.mark.parametrize case table so new edge cases don’t keep adding near-identical blocks.

_{Note: If this suggestion doesn't match your team's coding style, reply to this and let me know. I'll remember it for next time!}

Prompt To Fix With AI

This is a comment left during a code review.
Path: posthog/test/ai/gemini/test_gemini.py
Line: 1166:1169

Comment:
[P3] Prefer parameterizing the web-search “no candidates” cases

This new test covers `candidates=None`, but it duplicates the structure of the existing “no web search” tests nearby; given the repo preference for parameterized tests, it’d be simpler to combine the “no candidates” variants (e.g. `candidates=None` and `candidates=[]`) into one `@pytest.mark.parametrize` case table so new edge cases don’t keep adding near-identical blocks.

<sub>Note: If this suggestion doesn't match your team's coding style, reply to this and let me know. I'll remember it for next time!</sub>

How can I resolve this? If you propose a fix, please make it concise.