unique_toolkit 1.42.9__py3-none-any.whl → 1.43.1__py3-none-any.whl

unique_toolkit/_common/experimental/write_up_agent/services/template_handler/utils.py

@@ -0,0 +1,182 @@
+"""Template utilities."""
+
+from jinja2 import Environment
+from jinja2.nodes import For, Getattr, Name
+from pydantic import BaseModel
+
+
+class TemplateStructureInfo(BaseModel):
+    """Information about the structure expected by a Jinja template.
+
+    Attributes:
+        grouping_columns: List of column names detected from {{ group.column }} patterns
+        row_columns: List of column names detected from {{ row.column }} patterns
+        expects_groups: True if template iterates over 'groups' variable
+        expects_rows: True if template iterates over 'rows' variable
+    """
+
+    grouping_columns: list[str]
+    row_columns: list[str]
+    expects_groups: bool
+    expects_rows: bool
+
+
+# TODO [UN-16142]: Simplify template logic
+def parse_template(template_str: str) -> TemplateStructureInfo:
+    """
+    Parse a Jinja template to extract structure information.
+
+    The parser detects:
+    - {% for X in groups %} → expects_groups = True, X becomes the group variable
+    - {{ X.column_name }} → grouping_columns contains 'column_name' (excluding 'rows', 'instructions')
+    - {% for Y in rows %} or {% for Y in X.rows %} → expects_rows = True
+    - {{ row.column_name }} → row_columns contains 'column_name'
+
+    Args:
+        template_str: Jinja template string to parse
+
+    Returns:
+        TemplateStructureInfo with detected structure
+
+    Raises:
+        Exception: If template parsing fails
+
+    Example:
+        >>> template = '''
+        ... {% for g in groups %}
+        ... Region: {{ g.region }}
+        ... {% for row in g.rows %}
+        ... - {{ row.product }}: ${{ row.price }}
+        ... {% endfor %}
+        ... {% endfor %}
+        ... '''
+        >>> info = parse_template(template)
+        >>> info.grouping_columns
+        ['region']
+        >>> info.row_columns
+        ['product', 'price']
+        >>> info.expects_groups
+        True
+    """
+    env = Environment()
+
+    try:
+        ast = env.parse(template_str)
+    except Exception as e:
+        raise ValueError(f"Failed to parse Jinja template: {e}") from e
+
+    # First, find what variable name is used for groups iteration
+    # e.g., {% for g in groups %} -> group_var = 'g'
+    group_var = None
+    for node in ast.find_all(For):
+        if isinstance(node.iter, Name) and node.iter.name == "groups":
+            if isinstance(node.target, Name):
+                group_var = node.target.name
+                break
+
+    # Detect if template expects 'groups' and 'rows' loops
+    expects_groups = _check_for_loop_variable(ast, "groups")
+    expects_rows = _check_for_loop_variable(ast, "rows")
+
+    # Extract column references
+    # For grouping columns, use the group variable name (e.g., 'g', 'group', etc.)
+    if group_var:
+        grouping_columns_raw = _extract_attribute_references(ast, group_var)
+        # Filter out special attributes that are not DataFrame grouping columns:
+        # - 'rows': structural template variable for row data
+        # - 'llm_response': reserved for LLM-generated summaries
+        # - 'instructions': structural template variable for group-specific instructions
+        # - anything starting with '_': internal/computed variables
+        grouping_columns = sorted(
+            [
+                col
+                for col in grouping_columns_raw
+                if col not in ["rows", "llm_response", "instructions"]
+                and not col.startswith("_")
+            ]
+        )
+    else:
+        # Fallback to 'group' if no explicit group var found
+        grouping_columns = sorted(list(_extract_attribute_references(ast, "group")))
+        grouping_columns = sorted(
+            [
+                col
+                for col in grouping_columns
+                if col not in ["rows", "llm_response", "instructions"]
+                and not col.startswith("_")
+            ]
+        )
+
+    row_columns = sorted(list(_extract_attribute_references(ast, "row")))
+
+    return TemplateStructureInfo(
+        grouping_columns=grouping_columns,
+        row_columns=row_columns,
+        expects_groups=expects_groups,
+        expects_rows=expects_rows,
+    )
+
+
+def _extract_attribute_references(node, target_var: str) -> set[str]:
+    """
+    Recursively extract attribute references for a specific variable.
+
+    For example, if target_var='group', this extracts:
+    - 'region' from {{ group.region }}
+    - 'region' from {{ group.group.region }} (nested access)
+
+    Args:
+        node: Jinja2 AST node to traverse
+        target_var: Variable name to look for (e.g., 'group', 'row')
+
+    Returns:
+        Set of attribute names referenced on the target variable
+    """
+    attributes = set()
+
+    if isinstance(node, Getattr):
+        # Check if this is an attribute access on our target variable
+        if isinstance(node.node, Name) and node.node.name == target_var:
+            attributes.add(node.attr)
+        # Handle nested attributes like group.group.section
+        # If node.node is also Getattr, check if it eventually resolves to target_var
+        elif isinstance(node.node, Getattr):
+            # Recursively extract from nested getattr
+            nested_attrs = _extract_attribute_references(node.node, target_var)
+            if nested_attrs:
+                # If the nested part references our target, add this attr too
+                attributes.add(node.attr)
+        # Recursively check the node being accessed
+        attributes.update(_extract_attribute_references(node.node, target_var))
+
+    # Process all child nodes
+    for child in node.iter_child_nodes():
+        attributes.update(_extract_attribute_references(child, target_var))
+
+    return attributes
+
+
+def _check_for_loop_variable(node, loop_var: str) -> bool:
+    """
+    Check if a for-loop iterates over a specific variable.
+
+    For example, checks if template contains {% for X in groups %}.
+
+    Args:
+        node: Jinja2 AST node to traverse
+        loop_var: Variable name to look for (e.g., 'groups', 'rows')
+
+    Returns:
+        True if a for-loop over the variable is found
+    """
+    if isinstance(node, For):
+        # Check if this for-loop iterates over our target variable
+        if isinstance(node.iter, Name) and node.iter.name == loop_var:
+            return True
+
+    # Recursively check child nodes
+    for child in node.iter_child_nodes():
+        if _check_for_loop_variable(child, loop_var):
+            return True
+
+    return False

unique_toolkit/_common/experimental/write_up_agent/utils.py

@@ -0,0 +1,24 @@
+from pathlib import Path
+
+
+def template_loader(parent_dir: Path, template_name: str) -> str:
+    """
+    Load a Jinja2 template file from the filesystem.
+
+    Args:
+        parent_dir: Path object pointing to the directory containing the template
+        template_name: Name of the template file to load (e.g., 'template.j2')
+
+    Returns:
+        Template content as a string
+
+    Raises:
+        FileNotFoundError: If the template file does not exist
+        IOError: If there's an error reading the template file
+
+    Example:
+        >>> from pathlib import Path
+        >>> template = template_loader(Path(__file__).parent, "my_template.j2")
+    """
+    template_path = parent_dir / template_name
+    return template_path.read_text()

{unique_toolkit-1.42.9.dist-info → unique_toolkit-1.43.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.42.9
+Version: 1.43.1
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -124,6 +124,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.43.1] - 2026-01-12
+- Remove accidental example report.md from repo
+
+## [1.43.0] - 2026-01-11
+- Add `WriteUpAgent` as an experimental service
+
 ## [1.42.9] - 2026-01-11
 - Include feature flag to have message logs compatible with new ChatUI
 

{unique_toolkit-1.42.9.dist-info → unique_toolkit-1.43.1.dist-info}/RECORD

@@ -20,6 +20,31 @@ unique_toolkit/_common/exception.py,sha256=ho0uBcPeZXU2w15IrSBhO5w7KUgxp1HcKAQrf
 unique_toolkit/_common/execution.py,sha256=ocPGGfUwa851207HNTLYiBJ1pNzJp4VhMZ49OPP33gU,8022
 unique_toolkit/_common/experimental/endpoint_builder.py,sha256=pEDwgeDzt67qbyaM98u8X7UAy29mQIw9Qufjz2bxgEA,11410
 unique_toolkit/_common/experimental/endpoint_requestor.py,sha256=YnDr8wASAEjZjLAeBmOWuFn4wUIZslTHBN_aApWeJBA,16079
+unique_toolkit/_common/experimental/write_up_agent/README.md,sha256=ygmSpks3to5yye_XddLTpBXTiW61bW8w0hg2T33UP7U,28701
+unique_toolkit/_common/experimental/write_up_agent/__init__.py,sha256=ka0eMrZiIAZVaoAbM-SPIT7heMd7rsuqVHDNIaNOQBQ,534
+unique_toolkit/_common/experimental/write_up_agent/agent.py,sha256=bn-ftCx2GBRK2-yw9YspAob8zzHFX-X8Bk9Nx95Uczw,5835
+unique_toolkit/_common/experimental/write_up_agent/config.py,sha256=5nGku-_feFWhM6RBI-Y4QRv9CG4Bgi7RKGOk65LsybY,1640
+unique_toolkit/_common/experimental/write_up_agent/examples/data.csv,sha256=JHaYFBvmsSYOP7Ay1IKdMjBApsU-AJ902IDWp_85Lho,1569
+unique_toolkit/_common/experimental/write_up_agent/examples/example_usage.py,sha256=--4rnjQ9_MY9FzHHl6LBAaqF6HM1TeqoZbouxssZxXY,2484
+unique_toolkit/_common/experimental/write_up_agent/schemas.py,sha256=DUZ_RrjtqrtTUkqcecs-lW8eCOEAwB1Qxw_hrOXm3HE,861
+unique_toolkit/_common/experimental/write_up_agent/services/__init__.py,sha256=w2SR53rQl1yM9dYKxP9dJWF6qiovN2oyyGis0-aCslc,341
+unique_toolkit/_common/experimental/write_up_agent/services/dataframe_handler/__init__.py,sha256=hd2IktiTyCiwE_JJTC43zmu-67LL2vDhaLoi1tgZH_U,539
+unique_toolkit/_common/experimental/write_up_agent/services/dataframe_handler/exceptions.py,sha256=6F9VmWD5j3GI7JbHzakoYUM83y8rFWTTMVjTD6YQni8,855
+unique_toolkit/_common/experimental/write_up_agent/services/dataframe_handler/service.py,sha256=gaXXjIpTBNMPD_8KD94i6eSl4q1Em7iqIcmLFLhwHAM,5981
+unique_toolkit/_common/experimental/write_up_agent/services/dataframe_handler/utils.py,sha256=E-Yupm2d-Xo6wXha669L4B7g4cKa3qL03iJe7nIF5Ts,3382
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/__init__.py,sha256=u5Z2-XhEXbmMoH2X2PmonkMe5rr1VGc-XYp3gdCXOig,750
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/config.py,sha256=944Wn47t3MrC2f85zRA7vzrtuVKV22Um9olHLsv81B4,1927
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/exceptions.py,sha256=6hbP4ZbbvCGQVOVzUav2sfIkU-sVbst1K2grNCvo2Rk,1996
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/prompts/config.py,sha256=ep8NJNyBEbKIN7FudkeAfFaHzl1JygJxfNR2Ab0MHU0,1153
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/prompts/system_prompt.j2,sha256=U66WhKnA-_AHM-APCzHVRn2bu5rZMqRwyZ2kvAmF2xA,674
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/prompts/user_prompt.j2,sha256=9OMngS80URT4g96c0pYWjIcZn2Jn-C2aSFUdE7uIfCs,709
+unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/service.py,sha256=aIUc7HUbqBkBZGxXhwdhwQPYEcXoLy1NLCfAKOMav9E,12825
+unique_toolkit/_common/experimental/write_up_agent/services/template_handler/__init__.py,sha256=dyUAoV0uL-eEtSQQQ7XV8YMnNvwcyowiquZrgbhzNII,800
+unique_toolkit/_common/experimental/write_up_agent/services/template_handler/default_template.j2,sha256=mL2l4rvrQFEHvjZ8uf2uQFujoKsnT1CoYHbEC45zHDY,1036
+unique_toolkit/_common/experimental/write_up_agent/services/template_handler/exceptions.py,sha256=FGkJsbEeLwGDtFROH0TMxMFeXo4zfF40pXbLk6peYAo,1256
+unique_toolkit/_common/experimental/write_up_agent/services/template_handler/service.py,sha256=lg9rIbQKTzPhQTSWMt66z1X3hHnu8v7uIylV7Jn9vyY,6494
+unique_toolkit/_common/experimental/write_up_agent/services/template_handler/utils.py,sha256=FeJhTAU7AUlSFCfFwmEzzI22PtAdFJluzsGeiEPK49Q,6393
+unique_toolkit/_common/experimental/write_up_agent/utils.py,sha256=I_OCDXCxY6uZk_fNQg_bIXHwjnN7ND_ngcisThnGGIc,739
 unique_toolkit/_common/feature_flags/schema.py,sha256=X32VqH4VMK7bhEfSd8Wbddl8FVs7Gh7ucuIEbmqc4Kw,268
 unique_toolkit/_common/pydantic/rjsf_tags.py,sha256=T3AZIF8wny3fFov66s258nEl1GqfKevFouTtG6k9PqU,31219
 unique_toolkit/_common/pydantic_helpers.py,sha256=Yg1CHD603wVrqvinHiyh3stjIK3MjuexUe9aQQUfmXs,5406
@@ -217,7 +242,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.42.9.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.42.9.dist-info/METADATA,sha256=UlLw100oSlKFi6ycPSH6swuqJZwbzMQebOdbfbookk0,47453
-unique_toolkit-1.42.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.42.9.dist-info/RECORD,,
+unique_toolkit-1.43.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.43.1.dist-info/METADATA,sha256=esEOCD00HzfGppkkGkf53KU-JWbQBcKJbOq1sw8uDRA,47602
+unique_toolkit-1.43.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.43.1.dist-info/RECORD,,