zrb 1.15.3__py3-none-any.whl → 2.0.0a4__py3-none-any.whl

zrb/util/yaml.py

@@ -0,0 +1,204 @@
+from typing import Any
+
+
+def yaml_dump(obj: Any, key: str = "") -> str:
+    """
+    Convert any Python object to a YAML string representation.
+
+    Args:
+        obj: Any Python object to convert to YAML
+
+    Returns:
+        str: YAML string representation of the object
+
+    Rules:
+    - Any non-first level multiline string should be rendered as block (using `|`)
+    - None values are rendered correctly (not omitted)
+    - Non-primitive/list/dict/set objects are ignored
+    """
+    import yaml
+
+    # Process the object
+    processed_obj = _sanitize_obj(obj)
+    if key:
+        key_parts = _parse_key(key)
+        obj_to_dump = _get_obj_value(processed_obj, key_parts)
+    else:
+        obj_to_dump = processed_obj
+    # Add custom representer for multiline strings
+    yaml.add_representer(str, _multiline_string_presenter)
+    # Generate YAML
+    yaml_str = yaml.dump(
+        obj_to_dump,
+        default_flow_style=False,
+        allow_unicode=True,
+        sort_keys=False,
+        explicit_end=False,
+        width=float("inf"),
+    )
+    if not isinstance(obj_to_dump, (dict, list)):
+        # PyYAML appends '...\n' (document-end) for top-level scalars.
+        # So, we remove it.
+        if yaml_str.endswith("...\n"):
+            yaml_str = yaml_str[:-4]
+    return yaml_str
+
+
+def edit_obj(obj: Any, key: str, val: str) -> Any:
+    """
+    Edit a property or subproperty of an object using YAML syntax.
+
+    Args:
+        obj: The object to edit
+        key: The key to edit, can be nested with '.' as separator
+        val: The string value to set, will be parsed as YAML
+
+    Returns:
+        Any: The modified object
+
+    Example:
+        edit({"a": {"b": 1}}, "a.b", "2") -> {"a": {"b": 2}}
+        edit({"flag": False}, "flag", "true") -> {"flag": True}
+        edit({"a": 1}, "", "2") -> 2  # Replace entire object with scalar
+        edit({"a": 1}, "", "b: 2") -> {"a": 1, "b": 2}  # Patch dict if obj is dict
+    """
+    # Parse the value using YAML rules
+    parsed_value = _load_yaml(val)
+
+    # Handle empty key - replace entire object
+    if not key:
+        if isinstance(obj, dict) and isinstance(parsed_value, dict):
+            # Patch/merge the dict values
+            return {**obj, **parsed_value}
+        # Replace entire object with parsed value
+        return parsed_value
+
+    # Split the key by dots
+    key_parts = _parse_key(key)
+    # Set the nested value
+    return _set_obj_value(obj, key_parts, parsed_value)
+
+
+def _sanitize_obj(obj: Any) -> Any:
+    """Process a value for YAML conversion."""
+    if obj is None:
+        return None
+    elif isinstance(obj, (int, float, bool, str)):
+        return obj
+    elif isinstance(obj, (list, tuple)):
+        return [_sanitize_obj(item) for item in obj if not _is_complex_obj(item)]
+    elif isinstance(obj, dict):
+        return {k: _sanitize_obj(v) for k, v in obj.items() if not _is_complex_obj(v)}
+    elif isinstance(obj, set):
+        return [
+            _sanitize_obj(item) for item in sorted(obj) if not _is_complex_obj(item)
+        ]
+    else:
+        # Ignore non-primitive/list/dict/set objects
+        return None
+
+
+def _is_complex_obj(obj: Any) -> bool:
+    return obj is not None and not isinstance(
+        obj, (int, float, bool, str, list, tuple, dict, set)
+    )
+
+
+def _multiline_string_presenter(dumper, data):
+    """Custom representer for multiline strings."""
+    if "\n" in data:
+        # Clean up the string for block style
+        lines = [line.rstrip() for line in data.splitlines()]
+        clean_data = "\n".join(lines)
+        return dumper.represent_scalar("tag:yaml.org,2002:str", clean_data, style="|")
+    return dumper.represent_scalar("tag:yaml.org,2002:str", data)
+
+
+def _parse_key(key: str) -> list[str]:
+    return key.split(".")
+
+
+def _load_yaml(value_str: str) -> Any:
+    """Parse a string value using YAML rules."""
+    import yaml
+
+    # Handle empty string explicitly
+    if value_str == "":
+        return ""
+    try:
+        # Use yaml.safe_load to parse the value
+        parsed = yaml.safe_load(value_str)
+        return parsed
+    except yaml.YAMLError:
+        # If YAML parsing fails, treat as string
+        return value_str
+
+
+def _set_obj_value(obj: Any, keys: list[str], value: Any) -> Any:
+    """Set a value in a nested structure."""
+    if not keys:
+        return value
+    current_key = keys[0]
+    remaining_keys = keys[1:]
+    if isinstance(obj, dict):
+        # Handle dictionary
+        if remaining_keys:
+            # There are more keys to traverse
+            if current_key not in obj:
+                obj[current_key] = {}
+            obj[current_key] = _set_obj_value(obj[current_key], remaining_keys, value)
+        else:
+            # This is the final key
+            obj[current_key] = value
+        return obj
+    elif isinstance(obj, list):
+        # Handle list - convert key to index
+        try:
+            index = int(current_key)
+            if 0 <= index < len(obj):
+                if remaining_keys:
+                    obj[index] = _set_obj_value(obj[index], remaining_keys, value)
+                else:
+                    obj[index] = value
+            else:
+                raise IndexError(
+                    f"Index {index} out of range for list of length {len(obj)}"
+                )
+        except ValueError:
+            raise KeyError(f"Cannot use non-integer key '{current_key}' with list")
+        return obj
+    else:
+        # Handle other types by converting to dict
+        if remaining_keys:
+            # Create nested structure
+            new_obj = {current_key: _set_obj_value({}, remaining_keys, value)}
+            return new_obj
+        else:
+            # Replace the entire object
+            return {current_key: value}
+
+
+def _get_obj_value(obj: Any, keys: list[str]) -> Any:
+    """
+    Get a value from a nested structure using a list of keys.
+    Returns None if the key path does not exist.
+    """
+    current_val = obj
+    for key in keys:
+        if isinstance(current_val, dict):
+            if key in current_val:
+                current_val = current_val[key]
+            else:
+                return None
+        elif isinstance(current_val, list):
+            try:
+                index = int(key)
+                if 0 <= index < len(current_val):
+                    current_val = current_val[index]
+                else:
+                    return None
+            except (ValueError, TypeError):
+                return None
+        else:
+            return None
+    return current_val

zrb/xcom/xcom.py

@@ -34,6 +34,16 @@ class Xcom(deque):
         else:
             raise IndexError("Xcom is empty")
 
+    def get(self, default_value: Any = None) -> Any:
+        if len(self) > 0:
+            return self[0]
+        return default_value
+
+    def set(self, new_value: Any):
+        self.push(new_value)
+        while len(self) > 1:
+            self.pop()
+
     def add_push_callback(self, callback: Callable[[], Any]):
         if not hasattr(self, "push_callbacks"):
             self.push_callbacks: list[Callable[[], Any]] = []

{zrb-1.15.3.dist-info → zrb-2.0.0a4.dist-info}/METADATA

@@ -1,38 +1,58 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: zrb
-Version: 1.15.3
+Version: 2.0.0a4
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code
 Author: Go Frendi Gunawan
 Author-email: gofrendiasgard@gmail.com
-Requires-Python: >=3.10.0,<4.0.0
+Requires-Python: >=3.11.0,<4.0.0
 Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: all
+Provides-Extra: anthropic
+Provides-Extra: bedrock
+Provides-Extra: cohere
+Provides-Extra: google
+Provides-Extra: groq
+Provides-Extra: huggingface
+Provides-Extra: mistral
 Provides-Extra: playwright
 Provides-Extra: rag
-Requires-Dist: beautifulsoup4 (>=4.13.3,<5.0.0)
-Requires-Dist: black (>=25.1.0,<25.2.0)
-Requires-Dist: chromadb (>=0.6.3,<0.7.0) ; extra == "rag" or extra == "all"
-Requires-Dist: fastapi[standard] (>=0.116.1,<0.117.0)
-Requires-Dist: isort (>=6.0.1,<6.1.0)
-Requires-Dist: libcst (>=1.7.0,<2.0.0)
-Requires-Dist: markdownify (>=1.2.0,<2.0.0)
-Requires-Dist: openai (>=1.99.7)
-Requires-Dist: pdfplumber (>=0.11.6,<0.12.0)
-Requires-Dist: playwright (>=1.53.0,<2.0.0) ; extra == "playwright" or extra == "all"
+Provides-Extra: vertexai
+Requires-Dist: anthropic (>=0.75.0) ; extra == "anthropic" or extra == "all"
+Requires-Dist: beautifulsoup4 (>=4.14.2,<5.0.0)
+Requires-Dist: black (>=25.11.0,<26.0.0)
+Requires-Dist: boto3 (>=1.42.14) ; extra == "bedrock"
+Requires-Dist: chromadb (>=1.3.5,<2.0.0) ; extra == "rag" or extra == "all"
+Requires-Dist: cohere (>=5.18.0) ; extra == "cohere" or extra == "all"
+Requires-Dist: fastapi[standard] (>=0.123.9,<0.124.0)
+Requires-Dist: google-auth (>=2.36.0) ; extra == "vertexai" or extra == "all"
+Requires-Dist: google-genai (>=1.56.0) ; extra == "google" or extra == "all"
+Requires-Dist: groq (>=0.25.0) ; extra == "groq" or extra == "all"
+Requires-Dist: huggingface-hub[inference] (>=0.33.5,<1.0.0) ; extra == "huggingface"
+Requires-Dist: isort (>=7.0.0,<8.0.0)
+Requires-Dist: libcst (>=1.8.6,<2.0.0)
+Requires-Dist: markdownify (>=1.2.2,<2.0.0)
+Requires-Dist: mcp (>1.18.0)
+Requires-Dist: mistralai (>=1.9.11) ; extra == "mistral"
+Requires-Dist: openai (>=2.11.0)
+Requires-Dist: pdfplumber (>=0.11.7,<0.12.0)
+Requires-Dist: playwright (>=1.56.0,<2.0.0) ; extra == "playwright" or extra == "all"
+Requires-Dist: prompt-toolkit (>=3)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,google,groq,huggingface,mcp,mistral,openai,vertexai] (>=0.7.0,<0.8.0)
+Requires-Dist: pydantic-ai-slim (>=1.42.0,<1.43.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)
-Requires-Dist: python-jose[cryptography] (>=3.4.0,<4.0.0)
-Requires-Dist: requests (>=2.32.4,<3.0.0)
-Requires-Dist: tiktoken (>=0.8.0,<0.9.0)
+Requires-Dist: python-jose[cryptography] (>=3.5.0,<4.0.0)
+Requires-Dist: pyyaml (>=6.0.3,<7.0.0)
+Requires-Dist: requests (>=2.32.5,<3.0.0)
+Requires-Dist: rich (>=13)
+Requires-Dist: tiktoken (>=0.12.0,<0.13.0)
 Requires-Dist: ulid-py (>=1.1.0,<2.0.0)
 Project-URL: Documentation, https://github.com/state-alchemists/zrb
 Project-URL: Homepage, https://github.com/state-alchemists/zrb
@@ -111,8 +131,8 @@ Add the following Python code to your `zrb_init.py`:
 
 ```python
 from zrb import cli, LLMTask, CmdTask, StrInput, Group
-from zrb.builtin.llm.tool.code import analyze_repo
-from zrb.builtin.llm.tool.file import write_to_file
+from zrb.llm.tool.code import analyze_code
+from zrb.llm.tool.file import write_file
 
 
 # Create a group for Mermaid-related tasks
@@ -136,7 +156,7 @@ make_mermaid_script = mermaid_group.add_task(
             "Write the script into `{ctx.input.dir}/{ctx.input.diagram}.mmd`"
         ),
         tools=[
-            analyze_repo, write_to_file
+            analyze_code, write_file
         ],
     )
 )
@@ -212,13 +232,7 @@ Start a chat session with an LLM to ask questions, brainstorm ideas, or get codi
 zrb llm chat
 ```
 
-### Quick Questions
 
-For a single question, use the `ask` command for a fast response.
-
-```bash
-zrb llm ask "What is the capital of Indonesia?"
-```
 
 ---