banks 2.1.1__py3-none-any.whl → 2.1.2__py3-none-any.whl

banks/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2023-present Massimiliano Pippi <mpippi@gmail.com>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "2.1.1"
+__version__ = "2.1.2"

banks/extensions/chat.py

@@ -1,7 +1,6 @@
 # SPDX-FileCopyrightText: 2023-present Massimiliano Pippi <mpippi@gmail.com>
 #
 # SPDX-License-Identifier: MIT
-import re
 
 from jinja2 import TemplateSyntaxError, nodes
 from jinja2.ext import Extension
@@ -9,7 +8,6 @@ from jinja2.ext import Extension
 from banks.types import chat_message_from_text
 
 SUPPORTED_TYPES = ("system", "user", "assistant")
-CONTENT_BLOCK_REGEX = re.compile(r"<content_block>((?s:.)*)<\/content_block>")
 
 
 class ChatExtension(Extension):
@@ -52,8 +50,8 @@ class ChatExtension(Extension):
             raise TemplateSyntaxError(error_msg, lineno)
 
         if attr_value.value not in SUPPORTED_TYPES:
-            types = ",".join(SUPPORTED_TYPES)
-            msg = f"Unknown role type '{attr_value}', use one of ({types})"
+            types = ", ".join(SUPPORTED_TYPES)
+            msg = f"Unknown role type '{attr_value.value}', use one of ({types})"
             raise TemplateSyntaxError(msg, lineno)
 
         # Pass the role name to the CallBlock node

banks/types.py

@@ -16,7 +16,7 @@ from typing_extensions import Self
 from .utils import parse_params_from_docstring, python_type_to_jsonschema
 
 # pylint: disable=invalid-name
-CONTENT_BLOCK_REGEX = re.compile(r"(<content_block>\{.*?\}<\/content_block>)|([^<](?:(?!<content_block>).)*)")
+CONTENT_BLOCK_REGEX = re.compile(r"(<content_block>\{.*?\}<\/content_block>)|([^<](?:(?!<content_block>)[\s\S])*)")
 
 
 class ContentBlockType(str, Enum):

{banks-2.1.1.dist-info → banks-2.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banks
-Version: 2.1.1
+Version: 2.1.2
 Summary: A prompt programming language
 Project-URL: Documentation, https://github.com/masci/banks#readme
 Project-URL: Issues, https://github.com/masci/banks/issues
@@ -123,8 +123,14 @@ print(p.chat_messages({"persona": "helpful assistant"}))
 
 # Output:
 # [
-#   ChatMessage(role='system', content='You are a helpful assistant.\n'),
-#   ChatMessage(role='user', content='Hello, how are you?\n')
+#   ChatMessage(role='system', content=[
+#      ContentBlock(type=<ContentBlockType.text: 'text'>, cache_control=None, text='You are a helpful assistant.', 
+#                  image_url=None, input_audio=None)
+#   ], tool_call_id=None, name=None), 
+#   ChatMessage(role='user', content=[
+#      ContentBlock(type=<ContentBlockType.text: 'text'>, cache_control=None, text='Hello, how are you?', 
+#                  image_url=None, input_audio=None)
+#   ], tool_call_id=None, name=None)
 # ]
 ```
 

{banks-2.1.1.dist-info → banks-2.1.2.dist-info}/RECORD

@@ -1,14 +1,14 @@
-banks/__about__.py,sha256=S57kL24mDlNkjLlC1XN2yFEPPClI55FfNyJjlqC8_Fw,132
+banks/__about__.py,sha256=FgQqeZsldfp1gHFQOfAIpO9mYK5Yl83xo-6r31g2O-0,132
 banks/__init__.py,sha256=4IBopxXstFZliCvSjOuTurSQb32Vy26EXOPhmNZ4Hus,334
 banks/cache.py,sha256=uUGAu82-mfrscc2q24x19ZMZBkoQzf3hh7_V300J-Ik,1069
 banks/config.py,sha256=c6B1cXUZ-NN0XmJvfezXeHPXHP7knk8TfbmcZL7gCzk,1082
 banks/env.py,sha256=PBSz-iH_E3_KYofAfsQZLrYMZXVLNJLxgSFD7GB1Xd0,1233
 banks/errors.py,sha256=I5cgsa7wtolRVKBSq_aH5xs27yVcErBlMyUswCnM-es,580
 banks/prompt.py,sha256=RhPq3wpE-AiCfCftZpPFj2HXGdazwYD502Pr1e-j7FY,8162
-banks/types.py,sha256=R53mLqmTm9HoDJDfRP4mjfpORthkKJbn-6kxudsi_bw,5515
+banks/types.py,sha256=03x7E7FPVfuN39xY--c0fKumnyVUVzNrq9pgG5R-pAU,5520
 banks/utils.py,sha256=ZetGG3qhXMYOitDZQCWbE33wHEqR0ih2ZEg_dIW8OeI,1827
 banks/extensions/__init__.py,sha256=Lx4UrOzywYQY7a8qvIqvc3ql54nwK0lNP7x3jYdbREY,110
-banks/extensions/chat.py,sha256=3eGsvtBxcaXwjCNiaKVgVc7HN8p1p9CIAGgqlt9F_-U,2560
+banks/extensions/chat.py,sha256=VV6UV1wQZcJ0KbIFHSFmDeptWtww4o2IXF5pXB6TpTM,2478
 banks/extensions/completion.py,sha256=kF55PiNxjqpslUTAd46H4jOy0eFiLLm5hEcwxS4_oxs,7356
 banks/extensions/docs.py,sha256=vWOZvu2JoS4LwUG-BR3jPqThirYvu3Fdba331UxooYM,1098
 banks/filters/__init__.py,sha256=LV4mtTYkmLPo0OST8PMhJJmyC8azZJgU-ZKtpAVU1_0,325
@@ -21,7 +21,7 @@ banks/registries/__init__.py,sha256=iRK-8420cKBckOTd5KcIFQyV66EsF0Mc7UHCkzf8qZU,
 banks/registries/directory.py,sha256=WXVL8gIrH2OqDYcC89TuTW-iXSwT9WqJq_OWe_kYC2Q,5803
 banks/registries/file.py,sha256=8ayvFrcM8Tk0DWgGXmKD2DRBfGXr5CmgtdQaQ5cXhow,4054
 banks/registries/redis.py,sha256=eBL92URJa-NegOxRLS4b2xrDRDxz6iiaz_7Ddi32Rtc,2756
-banks-2.1.1.dist-info/METADATA,sha256=cMavPBlQKZ02YwJ77whxc8UvBw0u7pDSP723SkF1y30,11749
-banks-2.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-banks-2.1.1.dist-info/licenses/LICENSE.txt,sha256=NZJne_JTwMFwq_g-kq-sm4PuaeVOgu1l3NUGOgBHX-g,1102
-banks-2.1.1.dist-info/RECORD,,
+banks-2.1.2.dist-info/METADATA,sha256=o-tu_uFkiR7T3pkakUGoIZq4iOWIy7bkTay_9vX36lw,12098
+banks-2.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+banks-2.1.2.dist-info/licenses/LICENSE.txt,sha256=NZJne_JTwMFwq_g-kq-sm4PuaeVOgu1l3NUGOgBHX-g,1102
+banks-2.1.2.dist-info/RECORD,,