MAINT Breaking: Message Normalizer Refactor (microsoft#1296)

Author: rlundeen2

.gitignore

@@ -174,6 +174,9 @@ cython_debug/
 doc/generate_docs/cache/*
 !doc/generate_docs/cache/.gitkeep
 
+# nbstrip output
+*_nbqa_ipynb.py
+
 # Jupyterbook build files
 doc/_build/
 doc/_autosummary/

doc/_toc.yml

@@ -75,6 +75,7 @@ chapters:
       - file: code/targets/6_rate_limiting
       - file: code/targets/7_http_target
       - file: code/targets/8_openai_responses_target
+      - file: code/targets/9_message_normalizer
       - file: code/targets/open_ai_completions
       - file: code/targets/playwright_target
       - file: code/targets/playwright_target_copilot
@@ -117,7 +118,6 @@ chapters:
       - file: code/memory/9_exporting_data
       - file: code/memory/10_schema_diagram.md
       - file: code/memory/embeddings
-      - file: code/memory/chat_message
     - file: code/setup/0_setup
       sections:
       - file: code/setup/1_configuration

doc/api.rst

@@ -43,23 +43,6 @@ API Reference
     :toctree: _autosummary/
 
 
-:py:mod:`pyrit.chat_message_normalizer`
-=======================================
-
-.. automodule:: pyrit.chat_message_normalizer
-    :no-members:
-    :no-inherited-members:
-
-.. autosummary::
-    :nosignatures:
-    :toctree: _autosummary/
-
-    ChatMessageNormalizer
-    ChatMessageNop
-    GenericSystemSquash
-    ChatMessageNormalizerChatML
-    ChatMessageNormalizerTokenizerTemplate
-
 
 :py:mod:`pyrit.cli`
 =======================================
@@ -105,7 +88,6 @@ API Reference
     get_required_value
     is_in_ipython_session
     make_request_and_raise_if_error_async
-    print_chat_messages_with_color
     reset_default_values
     set_default_value
     Singleton
@@ -298,6 +280,24 @@ API Reference
     SeedEntry
     SQLiteMemory
 
+:py:mod:`pyrit.message_normalizer`
+==================================
+
+.. automodule:: pyrit.message_normalizer
+    :no-members:
+    :no-inherited-members:
+
+.. autosummary::
+    :nosignatures:
+    :toctree: _autosummary/
+
+    MessageListNormalizer
+    MessageStringNormalizer
+    GenericSystemSquashNormalizer
+    TokenizerTemplateNormalizer
+    ConversationContextNormalizer
+    ChatMessageNormalizer
+
 :py:mod:`pyrit.models`
 ======================