zrb 1.7.0__tar.gz → 1.7.2__tar.gz

{zrb-1.7.0 → zrb-1.7.2}/PKG-INFO

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

{zrb-1.7.0 → zrb-1.7.2}/pyproject.toml

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

{zrb-1.7.0 → zrb-1.7.2}/src/zrb/builtin/jwt.py

@@ -24,10 +24,12 @@ from zrb.task.make_task import make_task
     ],
 )
 def encode_jwt(ctx: AnyContext) -> str:
+    import json
+
     import jwt
 
     try:
-        payload = eval(ctx.input.payload)
+        payload = json.loads(ctx.input.payload)
         token = jwt.encode(
             payload=payload, key=ctx.input.secret, algorithm=ctx.input.algorithm
         )

{zrb-1.7.0 → zrb-1.7.2}/src/zrb/builtin/llm/llm_ask.py

@@ -135,13 +135,14 @@ if CFG.LLM_ALLOW_ACCESS_LOCAL_FILE:
             tool_name="analyze_file",
             tool_description="\n".join(
                 [
-                    "Analyze file using LLM capability.",
+                    "Always use this tool to analyze file using LLM capability.",
                     "This tool can do:",
                     "- summarization",
                     "- outline/structure extraction",
                     "- code review",
                     "- other tasks requiring deep understanding.",
-                    "Always use this tool to get deep understanding of file content",
+                    "This tool ONLY HAVE SINGLE ARGUMENT: query",
+                    "Make sure to include the file path you want to analyze in the query",
                 ]
             ),
             sub_agent_system_prompt="\n".join(

{zrb-1.7.0 → zrb-1.7.2}/src/zrb/task/llm/context_enrichment.py

@@ -47,6 +47,7 @@ async def enrich_context(
 
     ctx.log_info("Attempting to enrich conversation context...")
     # Prepare context and history for the enrichment prompt
+    history_summary = conversation_context.get("history_summary")
     try:
         context_json = json.dumps(conversation_context)
         history_json = json.dumps(history_list)
@@ -54,7 +55,13 @@ async def enrich_context(
         user_prompt_data = "\n".join(
             [
                 "Extract context from the following conversation info.",
-                "Maintain the relevan context and remove the unrelevant ones.",
+                "Extract only contexts that will be relevant across multiple conversations, like",  # noqa
+                "- user name",
+                "- user hobby",
+                "- user's long life goal",
+                "- standard/SOP",
+                "- etc.",
+                "Always maintain the relevant context and remove the irrelevant ones.",
                 "Restructure the context in a helpful way",
                 "Keep the context small",
                 f"Existing Context: {context_json}",
@@ -89,7 +96,9 @@ async def enrich_context(
             usage = enrichment_run.result.usage()
             ctx.print(stylize_faint(f"[Token Usage] {usage}"), plain=True)
             if response:
-                conversation_context.update(response)
+                conversation_context = response
+                # Re inject history summary
+                conversation_context["history_summary"] = history_summary
                 ctx.log_info("Context enriched based on history.")
                 ctx.log_info(
                     f"Updated conversation context: {json.dumps(conversation_context)}"