Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
9823509
initial, minimal working version of WebSocketCopilotTarget
paulinek13 Dec 22, 2025
62fc335
add useful error message for "server rejected WebSocket connection: H…
paulinek13 Dec 23, 2025
107b715
improve error handling and logging
paulinek13 Dec 23, 2025
41013a2
enhance WebSocket URL validation
paulinek13 Dec 23, 2025
c8e7e83
small fix
paulinek13 Dec 23, 2025
af636b2
fix
paulinek13 Dec 23, 2025
c9ebcee
improve `_parse_message`
paulinek13 Dec 23, 2025
99040d0
useful links
paulinek13 Dec 23, 2025
1f21ebf
add timeouts for responses and connection
paulinek13 Dec 24, 2025
1806e79
start with tests
paulinek13 Dec 25, 2025
3962bbb
require `wss://` only
paulinek13 Dec 25, 2025
7588e8d
add configurable response timeout
paulinek13 Dec 25, 2025
b98f675
fix
paulinek13 Dec 25, 2025
0dab7bb
replace Enum with IntEnum and actually use it
paulinek13 Dec 25, 2025
c2df619
test_dict_to_websocket_static_method
paulinek13 Dec 25, 2025
18fd238
fix
paulinek13 Dec 25, 2025
73b07d0
Refactor WebSocket message parser to handle multiple frames per message
paulinek13 Dec 25, 2025
9a8a878
rename message types in the enum
paulinek13 Dec 25, 2025
4d3c15d
add raw WebSocket messages for testing
paulinek13 Dec 25, 2025
b095d74
remove emojis
paulinek13 Dec 25, 2025
38e6868
simpler way to get the final result
paulinek13 Dec 25, 2025
2430dbe
log full raw message when no parseable content found
paulinek13 Dec 25, 2025
5b2c54a
_value2member_map_
paulinek13 Dec 25, 2025
4a7a7b8
TestParseRawMessage
paulinek13 Dec 25, 2025
acb0a6d
test fix
paulinek13 Dec 25, 2025
276290f
TODO: use msal for auth
paulinek13 Dec 25, 2025
ded56c6
add device code flow authentication method
paulinek13 Dec 26, 2025
558f48f
Revert "TODO: use msal for auth" -- as we need browser automation any…
paulinek13 Dec 26, 2025
02e3a4e
Revert "add device code flow authentication method"
paulinek13 Dec 26, 2025
0a9ee34
add Playwright-based way of getting sydney access token
paulinek13 Dec 27, 2025
cbc06f0
use `msal-extensions` for encrypted token persistence
paulinek13 Dec 28, 2025
4299673
add CopilotAuthenticator to WebSocketCopilotTarget for automated auth…
paulinek13 Dec 28, 2025
c65205b
WORKING multi prompt example
paulinek13 Dec 29, 2025
fd745ba
unit tests update for `WebSocketCopilotTarget`
paulinek13 Dec 29, 2025
fd8bc17
fixes
paulinek13 Dec 29, 2025
7266d7e
AUTH FLOW improvements
paulinek13 Dec 29, 2025
5241160
`CopilotAuthenticator` tests
paulinek13 Dec 30, 2025
dfa0ace
various fixes
paulinek13 Dec 30, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,7 @@ cython_debug/

# PyRIT secrets file
.env
.pyrit_cache/

# Cache for generating docs
doc/generate_docs/cache/*
Expand Down
2 changes: 2 additions & 0 deletions doc/api.rst
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ API Reference
Authenticator
AzureAuth
AzureStorageAuth
CopilotAuthenticator

:py:mod:`pyrit.auxiliary_attacks`
=================================
Expand Down Expand Up @@ -500,6 +501,7 @@ API Reference
PromptTarget
RealtimeTarget
TextTarget
WebSocketCopilotTarget

:py:mod:`pyrit.score`
=====================
Expand Down
2 changes: 2 additions & 0 deletions pyrit/auth/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,13 @@
get_default_azure_scope,
)
from pyrit.auth.azure_storage_auth import AzureStorageAuth
from pyrit.auth.copilot_authenticator import CopilotAuthenticator

__all__ = [
"Authenticator",
"AzureAuth",
"AzureStorageAuth",
"CopilotAuthenticator",
"TokenProviderCredential",
"get_azure_token_provider",
"get_azure_async_token_provider",
Expand Down
Loading