Added assistant support to banks for few shot learning (#31)

* added assistant support to banks for few shot learning * Update src/banks/extensions/chat.py Co-authored-by: Massimiliano Pippi <mpippi@gmail.com> --------- Co-authored-by: Massimiliano Pippi <mpippi@gmail.com>
masci · Jan 30, 2025 · ae88078 · ae88078
1 parent 0db010e
commit ae88078
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/banks/extensions/chat.py b/src/banks/extensions/chat.py
@@ -8,7 +8,7 @@
 
 from banks.types import ChatMessage, ContentBlock, ContentBlockType
 
-SUPPORTED_TYPES = ("system", "user")
+SUPPORTED_TYPES = ("system", "user", "assistant")
 CONTENT_BLOCK_REGEX = re.compile(r"<content_block>((?s:.)*)<\/content_block>")