zrb 1.8.10__tar.gz → 1.8.11__tar.gz

{zrb-1.8.10 → zrb-1.8.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.8.10
+Version: 1.8.11
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.8.10 → zrb-1.8.11}/pyproject.toml

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

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/builtin/llm/tool/code.py

@@ -86,8 +86,7 @@ async def analyze_repo(
     summarization_token_limit: int = 30000,
 ) -> str:
     """
-    Extract and summarize information from a directory that probably
-    contains a large resources.
+    Extract and summarize information from any directory.
     You should state the goal specifically so that the tool can return relevant informations.
     Use this tool for:
     - summarization

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/builtin/llm/tool/web.py

@@ -42,7 +42,7 @@ async def open_web_page(url: str) -> str:
                     return {"content": content, "links_on_page": links}
                 finally:
                     await browser.close()
-        except ImportError:
+        except BaseException:
             import requests
 
             response = requests.get(url, headers={"User-Agent": user_agent})

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/llm_config.py

@@ -69,14 +69,8 @@ DEFAULT_SPECIAL_INSTRUCTION_PROMPT = (
     "is clean.\n"
     "2.  **Halt if Dirty:** If the directory is not clean, STOP. Inform the "
     "user and wait for their instructions.\n"
-    "3.  **Propose and Confirm Branch:**\n"
-    "    - Tell the user you need to create a new branch and propose a "
-    "name.\n"
-    "    - Example: 'I will create a branch named `feature/add-user-login`. "
-    "Is this okay?'\n"
-    "    - **Wait for the user to say 'yes' or approve.**\n"
-    "4.  **Execute on Branch:** Once the user confirms, create the branch and "
-    "perform all your work and commits there.\n\n"
+    "3.  **Create a New Branch:** Create a new branch and inform the user.\n"
+    "4.  **Execute on Branch:** Perform all your work and commits there.\n\n"
     "**4. Debugging Protocol**\n"
     "1.  **Hypothesize:** State the most likely cause of the bug in one "
     "sentence.\n"

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/llm_rate_limitter.py

@@ -88,6 +88,7 @@ class LLMRateLimiter:
             prompt_parts = prompt.split(" ")
             last_part_index = len(prompt_parts) - 2
             clipped_prompt = " ".join(prompt_parts[:last_part_index])
+            clipped_prompt += "(Content clipped...)"
             token_count = self.count_token(clipped_prompt)
             if token_count < limit:
                 return clipped_prompt

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/task/llm_task.py

@@ -357,7 +357,7 @@ class LLMTask(BaseTask):
                 if xcom_usage_key not in ctx.xcom:
                     ctx.xcom[xcom_usage_key] = Xcom([])
                 usage = agent_run.result.usage()
-                ctx.xcom.get(xcom_usage_key).push(usage)
+                ctx.xcom[xcom_usage_key].push(usage)
                 ctx.print(stylize_faint(f"[Token Usage] {usage}"), plain=True)
                 return agent_run.result.output
             else:

{zrb-1.8.10 → zrb-1.8.11}/src/zrb/builtin/llm/llm_ask.py

@@ -127,13 +127,13 @@ llm_group.add_task(
 
 if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
     llm_ask.append_tool(
+        analyze_repo,
+        analyze_file,
+        search_files,
         list_files,
         read_from_file,
-        write_to_file,
-        search_files,
         apply_diff,
-        analyze_file,
-        analyze_repo,
+        write_to_file,
     )
 
 if CFG.LLM_ALLOW_ACCESS_SHELL: