unique_toolkit 1.29.1__py3-none-any.whl → 1.29.3__py3-none-any.whl

unique_toolkit/_common/utils/jinja/helpers.py

@@ -0,0 +1,10 @@
+from pathlib import Path
+
+
+def get_parent_dir(file_path: str) -> Path:
+    return Path(file_path).parent
+
+
+def load_template(parent_dir: Path, template_name: str) -> str:
+    with open(parent_dir / template_name, "r") as file:
+        return file.read().strip()

unique_toolkit/_common/utils/jinja/render.py

@@ -0,0 +1,12 @@
+from typing import Any
+
+from jinja2 import Template
+
+from unique_toolkit._common.utils.jinja.schema import Jinja2PromptParams
+
+
+def render_template(template: str, params: Jinja2PromptParams | dict[str, Any]) -> str:
+    if isinstance(params, Jinja2PromptParams):
+        params = params.model_dump(exclude_none=True, mode="json")
+
+    return Template(template, lstrip_blocks=True).render(**params)

unique_toolkit/_common/utils/jinja/schema.py

@@ -0,0 +1,65 @@
+from datetime import date, datetime
+from typing import Annotated, Any
+
+from jinja2 import Template
+from pydantic import (
+    BaseModel,
+    ConfigDict,
+    Field,
+    SerializerFunctionWrapHandler,
+    WrapSerializer,
+)
+
+from unique_toolkit.agentic.tools.tool import Tool
+
+
+class Jinja2PromptParams(BaseModel):
+    model_config = ConfigDict(str_strip_whitespace=True)
+
+    def render_template(self, template: str) -> str:
+        params = self.model_dump(exclude_none=True, mode="json")
+
+        return Template(template, lstrip_blocks=True).render(**params)
+
+
+class ToolPromptParams(Jinja2PromptParams):
+    name: str
+    tool_description_for_system_prompt: str = ""
+    tool_format_information_for_system_prompt: str = ""
+    tool_format_reminder_for_user_prompt: str = ""
+
+    @classmethod
+    def from_tool(cls, tool: Tool) -> "ToolPromptParams":
+        return cls(
+            name=tool.name,
+            tool_description_for_system_prompt=tool.tool_description_for_system_prompt(),
+            tool_format_information_for_system_prompt=tool.tool_format_information_for_system_prompt(),
+            tool_format_reminder_for_user_prompt=tool.tool_format_reminder_for_user_prompt(),
+        )
+
+
+def serialize_iso8601_date(v: Any, handler: SerializerFunctionWrapHandler) -> str:
+    if isinstance(v, date):
+        return v.isoformat()
+    return handler(v)
+
+
+ISO8601Date = Annotated[
+    date,
+    WrapSerializer(serialize_iso8601_date, return_type=str),
+]
+
+
+class AgentSystemPromptParams(Jinja2PromptParams):
+    info_cutoff_at: ISO8601Date | None
+    current_date: ISO8601Date = Field(default_factory=lambda: datetime.now().date())
+    tools: list[ToolPromptParams]
+    used_tools: list[ToolPromptParams]
+    add_citation_appendix: bool = True
+    max_tools_per_iteration: int
+    max_loop_iterations: int
+    current_iteration: int
+
+
+class AgentUserPromptParams(Jinja2PromptParams):
+    user_prompt: str

unique_toolkit/_common/utils/jinja/utils.py

@@ -0,0 +1,80 @@
+from jinja2 import Environment
+from jinja2.nodes import Const, Getattr, Getitem, Name
+from pydantic import BaseModel
+
+
+class TemplateValidationResult(BaseModel):
+    is_valid: bool
+    missing_placeholders: list[str]
+    optional_placeholders: list[str]
+    unexpected_placeholders: list[str]
+
+
+def _get_nested_variables(node):
+    """Recursively extract all variable references from a Jinja2 AST node."""
+    variables = set()
+
+    if isinstance(node, Name):
+        variables.add(node.name)
+    elif isinstance(node, (Getattr, Getitem)):
+        # For nested attributes like example.category
+        if isinstance(node.node, Name):
+            if isinstance(node, Getattr):
+                variables.add(f"{node.node.name}.{node.attr}")
+            else:  # Getitem
+                if isinstance(node.arg, Const):
+                    variables.add(f"{node.node.name}.{node.arg.value}")
+                else:
+                    # For dynamic indices, just use the base variable
+                    variables.add(node.node.name)
+        # Recursively process nested nodes
+        variables.update(_get_nested_variables(node.node))
+
+    # Process child nodes
+    for child in node.iter_child_nodes():
+        variables.update(_get_nested_variables(child))
+
+    return variables
+
+
+def validate_template_placeholders(
+    template_content: str,
+    required_placeholders: set[str],
+    optional_placeholders: set[str],
+) -> TemplateValidationResult:
+    """
+    Validates that all required placeholders in the template are present.
+    Handles both top-level and nested variables (e.g. example.category).
+
+    Args:
+        template_content (str): The content of the Jinja template
+        required_placeholders (set[str]): Set of required placeholder names
+        optional_placeholders (set[str]): Set of optional placeholder names
+
+    Returns:
+        TemplateValidationResult: A result object containing validation information
+    """
+    # Create a Jinja environment
+    env = Environment()
+
+    # Parse the template and get all variables including nested ones
+    ast = env.parse(template_content)
+    template_vars = _get_nested_variables(ast)
+
+    # Check for missing required placeholders
+    missing_placeholders = required_placeholders - template_vars
+
+    # Check for optional placeholders present
+    present_optional = optional_placeholders & template_vars
+
+    # Check for any unexpected placeholders
+    unexpected_placeholders = template_vars - (
+        required_placeholders | optional_placeholders
+    )
+
+    return TemplateValidationResult(
+        is_valid=len(missing_placeholders) == 0,
+        missing_placeholders=sorted(list(missing_placeholders)),
+        optional_placeholders=sorted(list(present_optional)),
+        unexpected_placeholders=sorted(list(unexpected_placeholders)),
+    )

{unique_toolkit-1.29.1.dist-info → unique_toolkit-1.29.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.29.1
+Version: 1.29.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: docxtpl (>=0.20.1,<0.21.0)
 Requires-Dist: jambo (>=0.1.2,<0.2.0)
+Requires-Dist: jinja2 (>=3.1.6,<4.0.0)
 Requires-Dist: markdown-it-py (>=4.0.0,<5.0.0)
 Requires-Dist: mkdocs-mermaid2-plugin (>=1.2.2,<2.0.0)
 Requires-Dist: mkdocs-multirepo-plugin (>=0.8.3,<0.9.0)
@@ -120,6 +121,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.29.3] - 2025-11-24
+- Fix jinja utility helpers import
+
+## [1.29.2] - 2025-11-21
+- Add `jinja` utility helpers to `_common`
+
 ## [1.29.1] - 2025-11-21
 - Add early return in `create_message_log_entry` if chat_service doesn't have assistant_message_id (relevant for agentic table)
 

{unique_toolkit-1.29.1.dist-info → unique_toolkit-1.29.3.dist-info}/RECORD

@@ -31,6 +31,10 @@ unique_toolkit/_common/token/token_counting.py,sha256=gM4B_aUqKqEPvmStFNcvCWNMNN
 unique_toolkit/_common/utils/__init__.py,sha256=qHrEy-3zkbFPdGFriRscPbGKuQfOuPi3O7tE5Zw5VHY,37
 unique_toolkit/_common/utils/files.py,sha256=97PkhXDUMXFEhje5HzWMMlLvZj49A6jOifqFRIrzu5M,1102
 unique_toolkit/_common/utils/image/encode.py,sha256=IJhtTcIT_azhiDsPp15oYs2LPze2l-sGlZrJASHrDVI,658
+unique_toolkit/_common/utils/jinja/helpers.py,sha256=UQWj0hFMtnuUFYDJ8NKbEblvYE8DrWX1o7fKQU3H9IQ,262
+unique_toolkit/_common/utils/jinja/render.py,sha256=n8wtslMZYU7DX9H_0HMmKjdcCvvvYk7C3qztktH5rO8,398
+unique_toolkit/_common/utils/jinja/schema.py,sha256=qCme5vI8TS8lZZVtyMleAgJtjBl9V7EfBIapcYXpJ-c,1912
+unique_toolkit/_common/utils/jinja/utils.py,sha256=0SrwN6qfLlRJrl38HG7qsMr5-DwpABNnM9LpqpMgwUA,2814
 unique_toolkit/_common/utils/structured_output/__init__.py,sha256=nm_orZrlCXL0FPLUg0Jv6Ty1flXPkCgZ9caAWaS8rz8,38
 unique_toolkit/_common/utils/structured_output/schema.py,sha256=Tp7kDYcmKtnUhcuRkH86TSYhylRff0ZZJYb2dLkISts,131
 unique_toolkit/_common/utils/write_configuration.py,sha256=fzvr4C-XBL3OSM3Od9TbqIxeeDS9_d9CLEyTq6DDknY,1409
@@ -183,7 +187,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.29.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.29.1.dist-info/METADATA,sha256=uthGA7BbiBQGrst9M84W-eUrfZ6YtHdtgzwd1vtJSiA,44224
-unique_toolkit-1.29.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.29.1.dist-info/RECORD,,
+unique_toolkit-1.29.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.29.3.dist-info/METADATA,sha256=oSEIVZwLtWw86E06C9j0RvJl0DS7GBvPLxYu0q8aam8,44393
+unique_toolkit-1.29.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.29.3.dist-info/RECORD,,