banks 2.1.3__py3-none-any.whl → 2.2.0__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.3"
+__version__ = "2.2.0"

banks/env.py

@@ -4,7 +4,7 @@
 from jinja2 import Environment, select_autoescape
 
 from .config import config
-from .filters import audio, cache_control, image, lemmatize, tool
+from .filters import audio, cache_control, image, lemmatize, tool, xml
 
 
 def _add_extensions(_env):
@@ -38,5 +38,6 @@ env.filters["image"] = image
 env.filters["lemmatize"] = lemmatize
 env.filters["tool"] = tool
 env.filters["audio"] = audio
+env.filters["to_xml"] = xml
 
 _add_extensions(env)

banks/filters/__init__.py

@@ -6,5 +6,6 @@ from .cache_control import cache_control
 from .image import image
 from .lemmatize import lemmatize
 from .tool import tool
+from .xml import xml
 
-__all__ = ("cache_control", "image", "lemmatize", "tool", "audio")
+__all__ = ("cache_control", "image", "lemmatize", "tool", "audio", "xml")

banks/filters/image.py

@@ -1,15 +1,25 @@
 # SPDX-FileCopyrightText: 2023-present Massimiliano Pippi <mpippi@gmail.com>
 #
 # SPDX-License-Identifier: MIT
+import re
 from pathlib import Path
 from urllib.parse import urlparse
 
 from banks.types import ContentBlock, ImageUrl
 
+BASE64_PATH_REGEX = re.compile(r"image\/.*;base64,.*")
+
 
 def _is_url(string: str) -> bool:
     result = urlparse(string)
-    return all([result.scheme, result.netloc])
+    if not result.scheme:
+        return False
+
+    if not result.netloc:
+        # The only valid format when netloc is empty is base64 data urls
+        return all([result.scheme == "data", BASE64_PATH_REGEX.match(result.path)])
+
+    return True
 
 
 def image(value: str) -> str:

banks/filters/xml.py

@@ -0,0 +1,62 @@
+import json
+import xml.etree.ElementTree as ET
+from typing import Any, Optional, Union
+from xml.dom.minidom import parseString
+
+from pydantic import BaseModel
+
+
+def _deserialize(string: str) -> Optional[dict]:
+    try:
+        return json.loads(string)
+    except json.JSONDecodeError:
+        return None
+
+
+def _prepare_dictionary(value: Union[str, BaseModel, dict[str, Any]]):
+    root_tag = "input"
+    if isinstance(value, str):
+        model: Optional[dict[str, Any]] = _deserialize(value)
+        if model is None:
+            msg = f"{value} is not deserializable"
+            raise ValueError(msg)
+    elif isinstance(value, BaseModel):
+        model = value.model_dump()
+        root_tag = value.__class__.__name__.lower()
+    elif isinstance(value, dict):
+        model = value.copy()
+        for k in value.keys():
+            if not isinstance(k, str):
+                key = str(k)
+                if isinstance(k, (int, float)):
+                    key = "_" + key
+                v = model.pop(k)
+                model[key.lower()] = v
+    else:
+        msg = f"Input can only be of type BaseModel, dictionary or deserializable string. Got {type(value)}"
+        raise ValueError(msg)
+    return model, root_tag
+
+
+def xml(value: Union[str, BaseModel, dict[str, Any]]) -> str:
+    """
+    Convert a Pydantic model, a deserializable string or a dictionary into an XML string.
+
+    Example:
+        ```jinja
+        {{'{"username": "user", "email": "example@email.com"}' | to_xml}}
+        "
+        <input>
+            <username>user</username>
+            <email>example@email.com</email>
+        </input>
+        "
+        ```
+    """
+    model, root_tag = _prepare_dictionary(value)
+    xml_model = ET.Element(root_tag)
+    for k, v in model.items():
+        sub = ET.SubElement(xml_model, k)
+        sub.text = str(v)
+    xml_str = ET.tostring(xml_model, encoding="unicode")
+    return parseString(xml_str).toprettyxml().replace('<?xml version="1.0" ?>\n', "")  # noqa: S318

{banks-2.1.3.dist-info → banks-2.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: banks
-Version: 2.1.3
+Version: 2.2.0
 Summary: A prompt programming language
 Project-URL: Documentation, https://github.com/masci/banks#readme
 Project-URL: Issues, https://github.com/masci/banks/issues

{banks-2.1.3.dist-info → banks-2.2.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
-banks/__about__.py,sha256=GetPXxJYXnTH5Y2Q50FaMRs7TJ2tkq34MorPa-qCNpc,132
+banks/__about__.py,sha256=77xoXWQ484-a4X30IBRswGGTfRx8t_BwItzAm7NTXwI,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/env.py,sha256=XOSz6QGNSRaqnIdKWhH5U-ci8Tfi1mDyJHit_aE27Ro,1266
 banks/errors.py,sha256=I5cgsa7wtolRVKBSq_aH5xs27yVcErBlMyUswCnM-es,580
 banks/prompt.py,sha256=RhPq3wpE-AiCfCftZpPFj2HXGdazwYD502Pr1e-j7FY,8162
 banks/types.py,sha256=03x7E7FPVfuN39xY--c0fKumnyVUVzNrq9pgG5R-pAU,5520
@@ -11,17 +11,18 @@ banks/extensions/__init__.py,sha256=Lx4UrOzywYQY7a8qvIqvc3ql54nwK0lNP7x3jYdbREY,
 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
+banks/filters/__init__.py,sha256=MMNxopwecFHW4LA76NwL2JQkdddIAGbKOaHUHG1JQs8,353
 banks/filters/audio.py,sha256=2vTPdpDo8FVQsl0WiPlXskwMCGnF8zKwWXfq1fYQzws,726
 banks/filters/cache_control.py,sha256=aOGOIzuqasV_TcuFaaXbaoGhA2W9YTFuz7wkatyjXRU,962
-banks/filters/image.py,sha256=cxp_Tlqy-DQ1y3I6IyO6TwM7KkdP8aL_xEKcSqeWX1w,1140
+banks/filters/image.py,sha256=Ls1fWCgRx0YLGIFx7hdKtR1skY575jDWlCESP0zV1Bs,1407
 banks/filters/lemmatize.py,sha256=Yvp8M4HCx6C0nrcu3UEMtjJUwsyVYI6GQDYOG4S6EEw,887
 banks/filters/tool.py,sha256=i8ukSDYw54ksShVJ2abfRQAiKzKrqUtmgBB1H04cig0,475
+banks/filters/xml.py,sha256=uQ_2zfCf8NhpdbF8F5HS7URXvDzsxfg-TEIVGufZbM0,1991
 banks/registries/__init__.py,sha256=iRK-8420cKBckOTd5KcIFQyV66EsF0Mc7UHCkzf8qZU,255
 banks/registries/directory.py,sha256=gRFO7fl9yXHt2NJ1pDA2wPSQtlORhSw1GKWxSTyFzE8,6055
 banks/registries/file.py,sha256=8ayvFrcM8Tk0DWgGXmKD2DRBfGXr5CmgtdQaQ5cXhow,4054
 banks/registries/redis.py,sha256=eBL92URJa-NegOxRLS4b2xrDRDxz6iiaz_7Ddi32Rtc,2756
-banks-2.1.3.dist-info/METADATA,sha256=TL6VlzchrWjdEduPj_aHzVKJts7peXbug1CUaogpPbo,12098
-banks-2.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-banks-2.1.3.dist-info/licenses/LICENSE.txt,sha256=NZJne_JTwMFwq_g-kq-sm4PuaeVOgu1l3NUGOgBHX-g,1102
-banks-2.1.3.dist-info/RECORD,,
+banks-2.2.0.dist-info/METADATA,sha256=m2W5swzWGUFGr0uNCFAgOqFtxv4A7fMyjsC34GDBTsk,12098
+banks-2.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+banks-2.2.0.dist-info/licenses/LICENSE.txt,sha256=NZJne_JTwMFwq_g-kq-sm4PuaeVOgu1l3NUGOgBHX-g,1102
+banks-2.2.0.dist-info/RECORD,,