zrb 1.15.26__py3-none-any.whl → 1.15.27__py3-none-any.whl

zrb/builtin/llm/chat_session.py

@@ -88,6 +88,8 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             )
             if result is not None:
                 final_result = result
+                if ctx.is_web_mode:
+                    return final_result
     return final_result
 
 

zrb/builtin/llm/llm_ask.py

@@ -162,7 +162,7 @@ _llm_ask_inputs = [
     ),
 ]
 
-llm_ask = llm_group.add_task(
+llm_ask: LLMTask = llm_group.add_task(
     LLMTask(
         name="llm-ask",
         input=_llm_ask_inputs,

zrb/builtin/llm/tool/code.py

@@ -69,6 +69,9 @@ async def analyze_repo(
     clear and specific goal. Vague goals will result in a vague analysis and
     may cause the tool to run for a long time.
 
+    Make sure to skim the repository or directory first so that you can put
+    effective parameters.
+
     Use this tool for:
     - Understanding a large or unfamiliar codebase.
     - Generating high-level summaries of a project's architecture.

zrb/config/llm_context/config.py

@@ -115,7 +115,12 @@ class LLMContextConfig:
         if context_path is None:
             context_path = cwd
 
-        config_file = os.path.join(cwd, CFG.LLM_CONTEXT_FILE)
+        config_files = self._find_config_files(cwd)
+        if config_files:
+            config_file = config_files[0]
+        else:
+            home_dir = os.path.expanduser("~")
+            config_file = os.path.join(home_dir, CFG.LLM_CONTEXT_FILE)
 
         sections = {}
         if os.path.exists(config_file):

zrb/group/any_group.py

@@ -1,9 +1,13 @@
 from abc import ABC, abstractmethod
+from typing import Generic, TypeVar
 
 from zrb.task.any_task import AnyTask
 
+GroupType = TypeVar("GroupType", bound="AnyGroup")
+TaskType = TypeVar("TaskType", bound=AnyTask)
 
-class AnyGroup(ABC):
+
+class AnyGroup(ABC, Generic[GroupType, TaskType]):
     @property
     @abstractmethod
     def name(self) -> str:
@@ -35,11 +39,11 @@ class AnyGroup(ABC):
         pass
 
     @abstractmethod
-    def add_group(self, group: "AnyGroup | str") -> "AnyGroup":
+    def add_group(self, group: GroupType, alias: str | None) -> GroupType:
         pass
 
     @abstractmethod
-    def add_task(self, task: AnyTask, alias: str | None = None) -> AnyTask:
+    def add_task(self, task: TaskType, alias: str | None = None) -> TaskType:
         pass
 
     @abstractmethod
@@ -55,5 +59,5 @@ class AnyGroup(ABC):
         pass
 
     @abstractmethod
-    def get_group_by_alias(self, name: str) -> "AnyGroup | None":
+    def get_group_by_alias(self, alias: str) -> "AnyGroup | None":
         pass

zrb/group/group.py

@@ -1,8 +1,10 @@
-from zrb.group.any_group import AnyGroup
+from typing import Generic, TypeVar
+
+from zrb.group.any_group import AnyGroup, GroupType, TaskType
 from zrb.task.any_task import AnyTask
 
 
-class Group(AnyGroup):
+class Group(AnyGroup, Generic[GroupType, TaskType]):
     def __init__(
         self, name: str, description: str | None = None, banner: str | None = None
     ):
@@ -41,13 +43,13 @@ class Group(AnyGroup):
         alias.sort()
         return {name: self._tasks.get(name) for name in alias}
 
-    def add_group(self, group: AnyGroup | str, alias: str | None = None) -> AnyGroup:
-        real_group = Group(group) if isinstance(group, str) else group
+    def add_group(self, group: GroupType, alias: str | None = None) -> GroupType:
+        real_group: GroupType = Group(group) if isinstance(group, str) else group
         alias = alias if alias is not None else real_group.name
         self._groups[alias] = real_group
         return real_group
 
-    def add_task(self, task: AnyTask, alias: str | None = None) -> AnyTask:
+    def add_task(self, task: TaskType, alias: str | None = None) -> TaskType:
         alias = alias if alias is not None else task.name
         self._tasks[alias] = task
         return task

{zrb-1.15.26.dist-info → zrb-1.15.27.dist-info}/METADATA

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

{zrb-1.15.26.dist-info → zrb-1.15.27.dist-info}/RECORD

@@ -9,15 +9,15 @@ zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,35
 zrb/builtin/group.py,sha256=zYC5uw0VE97TXiLCr464kFJ-CJIJyeQ2RXjnVRY5ovs,2577
 zrb/builtin/http.py,sha256=L6RE73c65wWwG5iHFN-tpOhyh56KsrgVskDd3c3YXtk,4246
 zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
-zrb/builtin/llm/chat_session.py,sha256=p0giSVYuQMQ5H1hbQzl7cMb49XZqWG0SF7X-ixB5EBU,10203
+zrb/builtin/llm/chat_session.py,sha256=LRBrIA4XaZTL8v4pjFrc47O-eHOKwuLOD4aWG0BeAkE,10279
 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=b3eS31NA55OUyeZN3O-sD971kz1LeX4QWzQQECpMaXo,6579
+zrb/builtin/llm/llm_ask.py,sha256=U-0QKI15bio_SoXxR9p18jCmIEdFNLCCyA2MohtHvyY,6588
 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=T8NGhBe59sQiu8LfdPOIBmsTNMXWFEKaPPSY9bolsQ8,2401
 zrb/builtin/llm/tool/cli.py,sha256=GCGB8GMFjvVcH0Ac-bD44VG6Bj3mQSuIcNHAwJbx4Ts,1210
-zrb/builtin/llm/tool/code.py,sha256=fr9FbmtfwizQTyTztvuvwAb9MD_auRZhPZfoJVBlKT4,8777
+zrb/builtin/llm/tool/code.py,sha256=2GBwDRocFwuIKIDEvHvekbxo1-NmEp1hJjd-labjAtw,8880
 zrb/builtin/llm/tool/file.py,sha256=FPPvKUZY-w1XEa7EN6D6X4VQoQNQ9ggGWLj-xJt-Ysc,23524
 zrb/builtin/llm/tool/rag.py,sha256=n4ATdr-2gCzPb7LnaBSD_TuAG4TUXKhE9ElKrSDHvFc,9763
 zrb/builtin/llm/tool/sub_agent.py,sha256=qJTJ2GSH-2Cma2QyHEJm8l_VuDHMHwhAWGls217YA6A,5078
@@ -226,7 +226,7 @@ zrb/config/default_prompt/repo_summarizer_system_prompt.md,sha256=RNy37Wg7ibXj3D
 zrb/config/default_prompt/summarization_prompt.md,sha256=qGSZJHQ0KVaSoeDVeAmvxaOHUGeLPaq813jepGqnkbk,1564
 zrb/config/default_prompt/system_prompt.md,sha256=gEb6N-cFg6VvOV-7ZffNwVt39DavAGesMqn9u0epbRc,2282
 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.py,sha256=PrF_aiZwdW8zhTu5__52IdOEd22XrdScFGsnZK3qCHQ,6067
 zrb/config/llm_context/config_parser.py,sha256=h95FbOjvVobhrsfGtG_BY3hxS-OLzQj-9F5vGZuehkY,1473
 zrb/config/llm_rate_limitter.py,sha256=_iQRv3d6kUPeRvmUYZX_iwCE7iDSEK1oKa4bQ9GROho,5261
 zrb/config/web_auth_config.py,sha256=_PXatQTYh2mX9H3HSYSQKp13zm1RlLyVIoeIr6KYMQ8,6279
@@ -246,8 +246,8 @@ zrb/env/env.py,sha256=zT-xj0l5G_lp_ginV_InuBr5qTqZMcGaMKE_p5FoqlU,1061
 zrb/env/env_file.py,sha256=dccsR_cjXp1Q_qz692YPaZSny3Xa049V5NAZ-WLDDEQ,746
 zrb/env/env_map.py,sha256=CKMmXUbuvf6h2rgFWj-WvW_xLZNgG3TtWdfrqHDe4qk,1253
 zrb/group/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zrb/group/any_group.py,sha256=sdUk82YRuJBMwDhD_b-vJ3v-Yxz8K7LFW52TM19k3HI,1242
-zrb/group/group.py,sha256=t5JnXlrLfthFfqmkVdRLKCIyms4JqsthcTODVncSmPk,4312
+zrb/group/any_group.py,sha256=BstkxtRQTGjMKdr1ErEHhB0UopXtYvwbCEOJnX5LjXA,1420
+zrb/group/group.py,sha256=JBiigohvtLdR1t3irXJzfUPeYqiNgqUWhlKaB79stpE,4409
 zrb/input/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/input/any_input.py,sha256=3EQgg2Qk_2t2Ege_4pHV2G95tV3XAIThLLuFZ6uV6oM,1051
 zrb/input/base_input.py,sha256=HCTxBZwnK-gP-F7XQGDbyyZHvwYR0fwNICzqVUw5Cg8,4169
@@ -410,7 +410,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.26.dist-info/METADATA,sha256=Eu_JTWqERNjfPIA4s8ChoWYZ7Da_PsT4p4OwB5BtxIw,9892
-zrb-1.15.26.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zrb-1.15.26.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.15.26.dist-info/RECORD,,
+zrb-1.15.27.dist-info/METADATA,sha256=qktgoudjcOq_BhgqBxuetkbMx-a4PfK7bTTAOlCFgog,9892
+zrb-1.15.27.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zrb-1.15.27.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.15.27.dist-info/RECORD,,