diff --git a/src/banks/extensions/chat.py b/src/banks/extensions/chat.py index ea2df08..4c3465a 100644 --- 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"((?s:.)*)<\/content_block>")