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.91k
stars
368
forks
source link
FEAT: Functionality to update PromptMemoryEntries #531
The following functions: update_entries, update_prompt_entries_by_conversation_id, update_labels_by_conversation_id, and update_prompt_metadata_by_conversation_id now exist and can be used for both DuckDBMemory and AzureSQLMemory. This allows for easier updating of entries in memory without adjusting them in PromptRequestPiece prior.
update_entries is now an abstract method in memory_interface following the format of other methods like query_entries and insert_entries.
Updated a deprecated function in PAIR orchestrator class
Testing:
Used dev Azure SQL DB to ensure that entries were updated correctly when the appropriate functions were called and wrote additional unit tests for both DuckDBMemory and AzureSQLMemory
Description:
update_entries,update_prompt_entries_by_conversation_id,update_labels_by_conversation_id, andupdate_prompt_metadata_by_conversation_idnow exist and can be used for both DuckDBMemory and AzureSQLMemory. This allows for easier updating of entries in memory without adjusting them inPromptRequestPieceprior.update_entriesis now an abstract method inmemory_interfacefollowing the format of other methods likequery_entriesandinsert_entries.Testing: