Azure / PyRIT

The Python Risk Identification Tool for generative AI (PyRIT) is an open access automation framework to empower security professionals and machine learning engineers to proactively find risks in their generative AI systems.
MIT License
1.9k stars 366 forks source link

FEAT Improving HTTPTarget Configuration #553

Open AlexRRR opened 6 days ago

AlexRRR commented 6 days ago

Currently, the HTTPTarget is very limited in its configuration. In my use case, I'm testing a RAG application which quickly runs into the httpx.AsyncClient default timeout (5 seconds). I believe there might be other use cases, such as using cookies for authentication or disabling verify

Describe the solution you'd like

Make the timeout configurable.

Describe alternatives you've considered, if relevant

Injecting the HTTP client into the target.

Cheers!

romanlutz commented 6 days ago

Makes sense. Thanks for contributing the PR.