From ae88078a3b369172f9239079330b32312edc3900 Mon Sep 17 00:00:00 2001 From: Krystof Olik <46579634+ArmykOliva@users.noreply.github.com> Date: Thu, 30 Jan 2025 11:50:31 +0100 Subject: [PATCH] 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 --------- Co-authored-by: Massimiliano Pippi --- src/banks/extensions/chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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>")