zrb 1.15.18__tar.gz → 1.15.20__tar.gz

{zrb-1.15.18 → zrb-1.15.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zrb
-Version: 1.15.18
+Version: 1.15.20
 Summary: Your Automation Powerhouse
 License: AGPL-3.0-or-later
 Keywords: Automation,Task Runner,Code Generator,Monorepo,Low Code

{zrb-1.15.18 → zrb-1.15.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.15.18"
+version = "1.15.20"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"

{zrb-1.15.18 → zrb-1.15.20}/src/zrb/builtin/llm/llm_ask.py

@@ -32,6 +32,26 @@ from zrb.task.base_trigger import BaseTrigger
 from zrb.task.llm_task import LLMTask
 from zrb.util.string.conversion import to_boolean
 
+
+def _get_default_yolo_mode(ctx: AnyContext) -> str:
+    default_value = llm_config.default_yolo_mode
+    if isinstance(default_value, list):
+        return ",".join(default_value)
+    return f"{default_value}"
+
+
+def _render_yolo_mode_input(ctx: AnyContext) -> list[str] | bool | None:
+    if ctx.input.yolo.strip() == "":
+        return None
+    elements = ctx.input.yolo.split(",")
+    if len(elements) == 0:
+        try:
+            return to_boolean(elements[0])
+        except Exception:
+            pass
+    return elements
+
+
 _llm_ask_inputs = [
     StrInput(
         "model",
@@ -88,7 +108,7 @@ _llm_ask_inputs = [
         "yolo",
         description="YOLO mode (LLM Agent will start in YOLO Mode)",
         prompt="YOLO mode (LLM Agent will start in YOLO Mode)",
-        default=lambda ctx: llm_config.default_yolo_mode,
+        default=_get_default_yolo_mode,
         allow_positional_parsing=False,
         always_prompt=False,
     ),
@@ -103,19 +123,6 @@ _llm_ask_inputs = [
     ),
 ]
 
-
-def _render_yolo_mode_input(ctx: AnyContext) -> list[str] | bool | None:
-    if ctx.input.yolo.strip() == "":
-        return None
-    elements = ctx.input.yolo.split(",")
-    if len(elements) == 0:
-        try:
-            return to_boolean(elements[0])
-        except Exception:
-            pass
-    return elements
-
-
 llm_ask: LLMTask = llm_group.add_task(
     LLMTask(
         name="llm-ask",

{zrb-1.15.18 → zrb-1.15.20}/src/zrb/config/config.py

@@ -304,8 +304,12 @@ class Config:
         return float(self._getenv("LLM_THROTTLE_SLEEP", "1.0"))
 
     @property
-    def LLM_YOLO_MODE(self) -> bool:
-        return to_boolean(self._getenv("LLM_YOLO_MODE", "false"))
+    def LLM_YOLO_MODE(self) -> bool | list[str]:
+        str_val = self._getenv("LLM_YOLO_MODE", "false")
+        try:
+            return to_boolean(str_val)
+        except Exception:
+            return [val.strip() for val in str_val.split(",") if val.strip() != ""]
 
     @property
     def LLM_SUMMARIZE_HISTORY(self) -> bool:

{zrb-1.15.18 → zrb-1.15.20}/src/zrb/config/llm_config.py

@@ -26,7 +26,7 @@ class LLMConfig:
         default_model: "Model | None" = None,
         default_model_settings: "ModelSettings | None" = None,
         default_model_provider: "Provider | None" = None,
-        default_yolo_mode: bool | None = None,
+        default_yolo_mode: bool | list[str] | None = None,
     ):
         self.__internal_default_prompt: dict[str, str] = {}
         self._default_model_name = default_model_name
@@ -175,7 +175,7 @@ class LLMConfig:
         )
 
     @property
-    def default_yolo_mode(self) -> bool:
+    def default_yolo_mode(self) -> bool | list[str]:
         return self._get_property(
             self._default_yolo_mode, CFG.LLM_YOLO_MODE, lambda: False
         )
@@ -236,7 +236,7 @@ class LLMConfig:
     def set_default_model_settings(self, model_settings: "ModelSettings"):
         self._default_model_settings = model_settings
 
-    def set_default_yolo_mode(self, yolo_mode: bool):
+    def set_default_yolo_mode(self, yolo_mode: bool | list[str]):
         self._default_yolo_mode = yolo_mode
 
 

{zrb-1.15.18 → zrb-1.15.20}/src/zrb/task/llm/config.py

@@ -16,11 +16,13 @@ def get_yolo_mode(
     yolo_mode_attr: BoolAttr | StrAttr | None = None,
     render_yolo_mode: bool = True,
 ) -> bool | list[str]:
+    if yolo_mode_attr is None:
+        return llm_config.default_yolo_mode
     try:
         return get_bool_attr(
             ctx,
             yolo_mode_attr,
-            llm_config.default_yolo_mode,
+            False,
             auto_render=render_yolo_mode,
         )
     except Exception:

{zrb-1.15.18 → zrb-1.15.20}/src/zrb/util/string/conversion.py

@@ -1,7 +1,7 @@
 import re
 
 NON_ALPHA_NUM = re.compile(r"[^a-zA-Z0-9]+")
-TRUE_STRS = ["true", "1", "yes", "y", "active", "on"]
+TRUE_STRS = ["true", "1", "yes", "y", "active", "on", "okay", "ok"]
 FALSE_STRS = ["false", "0", "no", "n", "inactive", "off"]