Skip to content

Conversation

@jul-sh
Copy link

@jul-sh jul-sh commented Dec 4, 2025

Summary

Adds native Google Gemini API support using pydantic-ai's built-in google-gla: provider prefix.

Changes

  • Added google to pydantic-ai-slim optional dependencies
  • Updated .env.example with GEMINI_API_KEY

No custom provider needed - pydantic-ai's native google-gla: prefix handles everything!

Usage

export GEMINI_API_KEY="your-key"

# Run with Gemini as target
uv run prompt-siren run attack +dataset=agentdojo-workspace +attack=mini-goat agent.config.model=google-gla:gemini-3-pro-preview

# Can also set attacker model
attack.config.attacker_model=google-gla:gemini-3-pro-preview

Features Supported

  • Native Gemini API with full thinking/reasoning support
  • Thought signatures for multi-turn conversations
  • All Gemini models (gemini-3-pro-preview, gemini-2.5-flash, etc.)

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 4, 2025
@evtimovi
Copy link
Contributor

evtimovi commented Dec 6, 2025

Thank you so much @jul-sh ! This looks very reasonable to me! I'll wait on the CI signals just in case but I think we are okay to merge this next week.

@evtimovi evtimovi self-assigned this Dec 6, 2025
Copy link
Contributor

@evtimovi evtimovi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me! Thank you for your contribution!

@evtimovi
Copy link
Contributor

evtimovi commented Dec 6, 2025

Actually, @jul-sh can I ask you to also resolve the conflicts in toml and regenerate the uv.lock in a new commit? I merged Edoardo's pull request making dependencies optional and I don't think we can manually resolve the uv lock differences.

Copy link
Contributor

@evtimovi evtimovi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry - updating this with one more: can you put the Llama API back in and also address the comment about the conflicts?

@jul-sh
Copy link
Author

jul-sh commented Dec 6, 2025

resolved the conflicts. I'm happy to add the llama API key; but would note that this PR isn't removing it (main branch doesn't have it). But happy to add it.

@jul-sh jul-sh requested a review from evtimovi December 6, 2025 01:40
@evtimovi
Copy link
Contributor

@jul-sh Thank you so much! I am sorry - maybe I am misreading the UI but I was referring to this commit? 4736263

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants