zrb 1.7.0__py3-none-any.whl → 1.7.2__py3-none-any.whl

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/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/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)}"

{zrb-1.7.0.dist-info → zrb-1.7.2.dist-info}/METADATA

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

@@ -8,11 +8,11 @@ zrb/builtin/git.py,sha256=8_qVE_2lVQEVXQ9vhiw8Tn4Prj1VZB78ZjEJJS5Ab3M,5461
 zrb/builtin/git_subtree.py,sha256=7BKwOkVTWDrR0DXXQ4iJyHqeR6sV5VYRt8y_rEB0EHg,3505
 zrb/builtin/group.py,sha256=t008xLM4_fgbjfZrPoi_fQAnSHIo6MOiQSCHBO4GDYU,2379
 zrb/builtin/http.py,sha256=sLqEczuSxGYXWzyJR6frGOHkPTviu4BeyroUr3-ZuAI,4322
-zrb/builtin/jwt.py,sha256=kjCf8qt7tkW9BpBDRAVTMJaEPQGzCbO1wo9xt5JoM8A,2836
+zrb/builtin/jwt.py,sha256=3M5uaQhJZbKQLjTUft1OwPz_JxtmK-xtkjxWjciOQho,2859
 zrb/builtin/llm/chat_session.py,sha256=HqFwrE1DiSlJrR-S3LRYWQBHkVsD-sfAV8_IIbnmtqY,6631
 zrb/builtin/llm/history.py,sha256=cnkOyO43uiMQ9cEvmqk-pPoCk1zCAH_fwAqSgBtsjzY,3079
 zrb/builtin/llm/input.py,sha256=Nw-26uTWp2QhUgKJcP_IMHmtk-b542CCSQ_vCOjhvhM,877
-zrb/builtin/llm/llm_ask.py,sha256=7Dx5XWf0QAbPwscGohloKOp_iqoI-JuaVojTaCJL3Jc,5332
+zrb/builtin/llm/llm_ask.py,sha256=GKjJZEVPgW0IbcRDNF9W7deHcLDPUyq3C0Vl-DqiJLc,5427
 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=yR9I0ZsI96OeQl9pgwORMASVuXsAL0a89D_iPS4C8Dc,1699
@@ -320,7 +320,7 @@ zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=6wGSsw03GdY_fj12CsJh7wxB6BnE13N8RYXaWfbiUsk,5451
 zrb/task/llm/config.py,sha256=oGxHYMdIvhASnKwNuMPwcdeJiFfS0tNskzGHakpfpQU,3458
 zrb/task/llm/context.py,sha256=U9a8lxa2ikz6my0Sd5vpO763legHrMHyvBjbrqNmv0Y,3838
-zrb/task/llm/context_enrichment.py,sha256=i2oWcs0mJGkfT8Jho3tQwNADagG9qU5oEZHbGxTTsSs,6629
+zrb/task/llm/context_enrichment.py,sha256=jB7lekwo4hTk8HIwhhq7HSMfgAzwaV2U867icBRt6Z0,7088
 zrb/task/llm/error.py,sha256=27DQXSG8SH1-XuvXFdZQKzP39wZDWmd_YnSTz6DJKKI,3690
 zrb/task/llm/history.py,sha256=3WMXoi7RquxosXQf3iv2_BCeF8iKtY1f407pR71xERs,7745
 zrb/task/llm/history_summarization.py,sha256=n3GbgwXlDIkgpJppMGfpqF_8Wpi9yAoZYh46O1pFQeU,6432
@@ -370,7 +370,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.7.0.dist-info/METADATA,sha256=CFGNJW5jVSUWPFlS6-xNemTuGvjfBrOrmymGwRc3V40,8385
-zrb-1.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.7.0.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.7.0.dist-info/RECORD,,
+zrb-1.7.2.dist-info/METADATA,sha256=JA6bk-DSmoriPSaNGzS8nmmka4IaNC7SLJQjNaCR_RE,8385
+zrb-1.7.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.7.2.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.7.2.dist-info/RECORD,,