Skip to content

[FEAT]: LLM-based Prompt Driver - #33

Merged
Bashir Partovi (bashirpartovi) merged 10 commits into
microsoft:mainfrom
bashirpartovi:dev/bashirpartovi/llm-driver-2nd-try
Apr 28, 2026
Merged

[FEAT]: LLM-based Prompt Driver#33
Bashir Partovi (bashirpartovi) merged 10 commits into
microsoft:mainfrom
bashirpartovi:dev/bashirpartovi/llm-driver-2nd-try

Conversation

@bashirpartovi

Copy link
Copy Markdown
Contributor

Summary

Adds LLMDriver, a new PromptDriver implementation that uses a driving LLM to generate the next prompt on each turn based on the agent's prior response and the evaluator's feedback. Complements the existing StaticDriver (fixed prompt sequence) for tests that need adaptive behaviour, multi-turn jailbreak attempts, exploratory probes, or attacks that adjust strategy based on what the agent does.

Built on PyRIT's PromptChatTarget and PromptNormalizer.


Also moved _pyrit, which was originally a hidden boundary for everything PyRIT-related, to an actual pyrit_bridge package. This is to be more open about what RAMPART is built upon — we will always depend on PyRIT.

@bashirpartovi
Bashir Partovi (bashirpartovi) requested a review from a team April 26, 2026 18:01
Comment thread rampart/drivers/llm.py
Comment thread rampart/drivers/llm.py
Comment thread rampart/drivers/llm.py
Comment thread rampart/core/execution.py Outdated
Comment thread rampart/drivers/llm.py Outdated
Comment thread .github/workflows/coverage.yml
Comment thread .github/workflows/coverage.yml
Comment thread rampart/drivers/llm.py
Comment thread pyproject.toml Outdated
Comment thread pyproject.toml
Comment thread rampart/drivers/llm.py Outdated
Comment thread rampart/drivers/llm.py Outdated
Comment thread rampart/pyrit_bridge/__init__.py
Comment thread rampart/core/execution.py Outdated

@nina-msft Nina Chikanov (nina-msft) left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Great work Bashir!

@bashirpartovi
Bashir Partovi (bashirpartovi) merged commit bc0cd78 into microsoft:main Apr 28, 2026
11 checks passed
@bashirpartovi
Bashir Partovi (bashirpartovi) deleted the dev/bashirpartovi/llm-driver-2nd-try branch April 28, 2026 00:11
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants