zrb 1.15.18__py3-none-any.whl → 1.15.20__py3-none-any.whl

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/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/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/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/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"]
 
 

{zrb-1.15.18.dist-info → zrb-1.15.20.dist-info}/METADATA

@@ -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.dist-info → zrb-1.15.20.dist-info}/RECORD

@@ -12,7 +12,7 @@ zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
 zrb/builtin/llm/chat_session.py,sha256=p0giSVYuQMQ5H1hbQzl7cMb49XZqWG0SF7X-ixB5EBU,10203
 zrb/builtin/llm/history.py,sha256=LDOrL0p7r_AHLa5L8Dp7bHNsOALugmJd7OguXRWGnm4,3087
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
-zrb/builtin/llm/llm_ask.py,sha256=BuZdQKw4oXCdZyFYr-pKKMKGMG3XDQKa9aaCGOoSsHM,6015
+zrb/builtin/llm/llm_ask.py,sha256=XtnSZoBvwHqnBUi8R0rt8VDfnBmWgwFlDuuo1WA1W_w,6209
 zrb/builtin/llm/previous-session.js,sha256=xMKZvJoAbrwiyHS0OoPrWuaKxWYLoyR5sguePIoCjTY,816
 zrb/builtin/llm/tool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/builtin/llm/tool/api.py,sha256=vMEiZhhTZ3o2jRBxWcJ62b0M85wd_w4W0X4Hx23NXto,2380
@@ -217,7 +217,7 @@ zrb/callback/callback.py,sha256=PFhCqzfxdk6IAthmXcZ13DokT62xtBzJr_ciLw6I8Zg,4030
 zrb/cmd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/cmd/cmd_result.py,sha256=L8bQJzWCpcYexIxHBNsXj2pT3BtLmWex0iJSMkvimOA,597
 zrb/cmd/cmd_val.py,sha256=7Doowyg6BK3ISSGBLt-PmlhzaEkBjWWm51cED6fAUOQ,1014
-zrb/config/config.py,sha256=AtJ61i_ZTi95vRV5h4EKtMWhPdU2YVEdv08mU1DePEk,14161
+zrb/config/config.py,sha256=afsH-a7yTPT5hhvnmxGarAiU5yRUcUbua3Y_hRgltEg,14326
 zrb/config/default_prompt/file_extractor_system_prompt.md,sha256=42Co-uNiL1Wn_KtH-7vjvNA0nja-xM0Sh2HpZF9OFU8,3880
 zrb/config/default_prompt/interactive_system_prompt.md,sha256=XvXI51dMpQmuuYah_LEWUwJzvWHxNTz8XqbdMsF6qdI,2555
 zrb/config/default_prompt/persona.md,sha256=GfUJ4-Mlf_Bm1YTzxFNkPkdVbAi06ZDVYh-iIma3NOs,253
@@ -225,7 +225,7 @@ zrb/config/default_prompt/repo_extractor_system_prompt.md,sha256=EGZ-zj78RlMEg2j
 zrb/config/default_prompt/repo_summarizer_system_prompt.md,sha256=RNy37Wg7ibXj3DlsFKaYvgMpMS-lyXlM1LZlc59_4ic,2009
 zrb/config/default_prompt/summarization_prompt.md,sha256=vbxzJGeH7uCdgnfbVt3GYOjY-BygmeRfJ6tnzXmFdoY,1698
 zrb/config/default_prompt/system_prompt.md,sha256=gEb6N-cFg6VvOV-7ZffNwVt39DavAGesMqn9u0epbRc,2282
-zrb/config/llm_config.py,sha256=-XmSCkFk858NlKHSbAWo4e2bMzE9-mMriIoR5_RaXtE,8865
+zrb/config/llm_config.py,sha256=9I_msAibvTY-ADG6lXCBXQ0EshaA-GQzJym9EZKsetw,8901
 zrb/config/llm_context/config.py,sha256=TPQX_kU772r0AHmVFeo1WGLDAidacT-qDyuMWxY_avg,5878
 zrb/config/llm_context/config_parser.py,sha256=h95FbOjvVobhrsfGtG_BY3hxS-OLzQj-9F5vGZuehkY,1473
 zrb/config/llm_rate_limitter.py,sha256=_iQRv3d6kUPeRvmUYZX_iwCE7iDSEK1oKa4bQ9GROho,5261
@@ -347,7 +347,7 @@ zrb/task/cmd_task.py,sha256=myM8WZm6NrUD-Wv0Vb5sTOrutrAVZLt5LVsSBKwX6SM,10860
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
 zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=uI0X_injDrxq-IhYuZsAMdoq9AXy-gnipozGuZ72QIs,8990
-zrb/task/llm/config.py,sha256=UrRdjgCpTllJaH6LIptHOj69h7upnYd0q9iKHhszErk,4007
+zrb/task/llm/config.py,sha256=zOPf4NpdWPuBc_R8d-kljYcOKfUAypDxiSjRDrxV66M,4059
 zrb/task/llm/conversation_history.py,sha256=oMdKUV2__mBZ4znnA-prl-gfyoleKC8Nj5KNpmLQJ4o,6764
 zrb/task/llm/conversation_history_model.py,sha256=kk-7niTl29Rm2EUIhTHzPXgZ5tp4IThMnIB3dS-1OdU,3062
 zrb/task/llm/default_workflow/coding.md,sha256=2uythvPsnBpYfIhiIH1cCinQXX0i0yUqsL474Zpemw0,2484
@@ -401,7 +401,7 @@ zrb/util/llm/prompt.py,sha256=HMpKby27DE8lJWpytYKylp7Iw9ENwsYQI0nMMKCCi54,2190
 zrb/util/load.py,sha256=DK0KYSlu48HCoGPqnW1IxnE3pHrZSPCstfz8Fjyqqv8,2140
 zrb/util/run.py,sha256=vu-mcSWDP_WuuvIKqM_--Gk3WkABO1oTXiHmBRTvVQk,546
 zrb/util/string/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/util/string/conversion.py,sha256=XEvN1hyziEwP9mEC_LdNaisoLLZF3l4DE40DgNEo6is,6257
+zrb/util/string/conversion.py,sha256=ovSQtogMCt2hQcyh1neSp_XdbFx4WX-Mw5xoUlyOnRQ,6271
 zrb/util/string/format.py,sha256=MwWGAwSdtOgR_2uz-JCXlg_q-uRYUUI-G8CGkfdgqik,1198
 zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
@@ -409,7 +409,7 @@ zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/util/truncate.py,sha256=eSzmjBpc1Qod3lM3M73snNbDOcARHukW_tq36dWdPvc,921
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.15.18.dist-info/METADATA,sha256=cGOrmv5M8MXtOP4PyxO1Wfxln5G8EVsL6R4gAaN5ANo,9892
-zrb-1.15.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.18.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.18.dist-info/RECORD,,
+zrb-1.15.20.dist-info/METADATA,sha256=xEpmCD_LUxH-rR-OqQcBtkOQHbvYhkEsw2fu12gfUtI,9892
+zrb-1.15.20.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.20.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.20.dist-info/RECORD,,