zrb 1.11.0__tar.gz → 1.12.0__tar.gz

{zrb-1.11.0 → zrb-1.12.0}/PKG-INFO

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

{zrb-1.11.0 → zrb-1.12.0}/pyproject.toml

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

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/builtin/llm/chat_session.py

@@ -25,6 +25,7 @@ async def read_user_prompt(ctx: AnyContext) -> str:
     is_tty = ctx.is_tty
     reader = await _setup_input_reader(is_tty)
     multiline_mode = False
+    current_modes = ctx.input.modes
     user_inputs = []
     while True:
         await asyncio.sleep(0.01)
@@ -38,7 +39,9 @@ async def read_user_prompt(ctx: AnyContext) -> str:
         if user_input.strip().lower() in ("/bye", "/quit", "/q", "/exit"):
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
-            result = await _trigger_ask_and_wait_for_result(ctx, user_prompt)
+            result = await _trigger_ask_and_wait_for_result(
+                ctx, user_prompt, current_modes
+            )
             if result is not None:
                 final_result = result
             break
@@ -49,9 +52,18 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             multiline_mode = False
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
-            result = await _trigger_ask_and_wait_for_result(ctx, user_prompt)
+            result = await _trigger_ask_and_wait_for_result(
+                ctx, user_prompt, current_modes
+            )
             if result is not None:
                 final_result = result
+        elif user_input.strip().lower().startswith("/mode"):
+            mode_parts = user_input.split(" ", maxsplit=2)
+            if len(mode_parts) > 1:
+                current_modes = mode_parts[1]
+            ctx.print(f"Current mode: {current_modes}", plain=True)
+            ctx.print("", plain=True)
+            continue
         elif user_input.strip().lower() in ("/help", "/info"):
             _show_info(ctx)
             continue
@@ -61,7 +73,9 @@ async def read_user_prompt(ctx: AnyContext) -> str:
                 continue
             user_prompt = "\n".join(user_inputs)
             user_inputs = []
-            result = await _trigger_ask_and_wait_for_result(ctx, user_prompt)
+            result = await _trigger_ask_and_wait_for_result(
+                ctx, user_prompt, current_modes
+            )
             if result is not None:
                 final_result = result
     return final_result
@@ -74,16 +88,26 @@ def _show_info(ctx: AnyContext):
         ctx: The context object for the task.
     """
     ctx.print(
-        (
-            f"  {stylize_bold_yellow('/bye')}   {stylize_faint('Quit from chat session')}\n"
-            f"  {stylize_bold_yellow('/multi')} {stylize_faint('Start multiline input')}\n"
-            f"  {stylize_bold_yellow('/end')}   {stylize_faint('End multiline input')}\n"
-            f"  {stylize_bold_yellow('/help')}  {stylize_faint('Show this message')}\n"
+        "\n".join(
+            [
+                _format_info_line("/bye", "Quit from chat session"),
+                _format_info_line("/multi", "Start multiline input"),
+                _format_info_line("/end", "End multiline input"),
+                _format_info_line("/modes", "Show current modes"),
+                _format_info_line("/modes <mode1,mode2,..>", "Set current modes"),
+                _format_info_line("/help", "Show this message"),
+            ]
         ),
         plain=True,
     )
 
 
+def _format_info_line(command: str, description: str) -> str:
+    styled_command = stylize_bold_yellow(command.ljust(25))
+    styled_description = stylize_faint(description)
+    return f"  {styled_command} {styled_description}"
+
+
 async def _handle_initial_message(ctx: AnyContext) -> str:
     """Processes the initial message from the command line."""
     if not ctx.input.message or ctx.input.message.strip() == "":
@@ -94,6 +118,7 @@ async def _handle_initial_message(ctx: AnyContext) -> str:
     result = await _trigger_ask_and_wait_for_result(
         ctx,
         user_prompt=ctx.input.message,
+        modes=ctx.input.modes,
         previous_session_name=ctx.input.previous_session,
         start_new=ctx.input.start_new,
     )
@@ -131,6 +156,7 @@ async def _read_next_line(is_interactive: bool, reader, ctx: AnyContext) -> str:
 async def _trigger_ask_and_wait_for_result(
     ctx: AnyContext,
     user_prompt: str,
+    modes: str,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ) -> str | None:
@@ -148,7 +174,7 @@ async def _trigger_ask_and_wait_for_result(
     """
     if user_prompt.strip() == "":
         return None
-    await _trigger_ask(ctx, user_prompt, previous_session_name, start_new)
+    await _trigger_ask(ctx, user_prompt, modes, previous_session_name, start_new)
     result = await _wait_ask_result(ctx)
     md_result = _render_markdown(result) if result is not None else ""
     ctx.print("\n🤖 >>", plain=True)
@@ -193,12 +219,14 @@ def get_llm_ask_input_mapping(callback_ctx: AnyContext):
         "start-new": data.get("start_new"),
         "previous-session": data.get("previous_session_name"),
         "message": data.get("message"),
+        "modes": data.get("modes"),
     }
 
 
 async def _trigger_ask(
     ctx: AnyContext,
     user_prompt: str,
+    modes: str,
     previous_session_name: str | None = None,
     start_new: bool = False,
 ):
@@ -218,6 +246,7 @@ async def _trigger_ask(
             "previous_session_name": previous_session_name,
             "start_new": start_new,
             "message": user_prompt,
+            "modes": modes,
         }
     )
 

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/builtin/llm/llm_ask.py

@@ -65,6 +65,14 @@ _llm_ask_inputs = [
         allow_positional_parsing=False,
         always_prompt=False,
     ),
+    TextInput(
+        "modes",
+        description="Modes",
+        prompt="Modes",
+        default="coding",
+        allow_positional_parsing=False,
+        always_prompt=False,
+    ),
     BoolInput(
         "start-new",
         description="Start new conversation (LLM will forget everything)",
@@ -101,6 +109,9 @@ llm_ask: LLMTask = llm_group.add_task(
         system_prompt=lambda ctx: (
             None if ctx.input.system_prompt.strip() == "" else ctx.input.system_prompt
         ),
+        modes=lambda ctx: (
+            None if ctx.input.modes.strip() == "" else ctx.input.modes.split(",")
+        ),
         message="{ctx.input.message}",
         retries=0,
     ),

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/config/config.py

@@ -287,6 +287,10 @@ class Config:
         """Number of seconds to sleep when throttling is required."""
         return float(os.getenv("ZRB_LLM_THROTTLE_SLEEP", "1.0"))
 
+    @property
+    def LLM_YOLO_MODE(self) -> bool:
+        return to_boolean(os.getenv("ZRB_LLM_YOLO_MODE", "false"))
+
     @property
     def LLM_SUMMARIZE_HISTORY(self) -> bool:
         return to_boolean(os.getenv("ZRB_LLM_SUMMARIZE_HISTORY", "true"))

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/config/llm_config.py

@@ -22,11 +22,9 @@ class LLMConfig:
         default_interactive_system_prompt: str | None = None,
         default_special_instruction_prompt: str | None = None,
         default_summarization_prompt: str | None = None,
-        default_context_enrichment_prompt: str | None = None,
         default_summarize_history: bool | None = None,
         default_history_summarization_token_threshold: int | None = None,
-        default_enrich_context: bool | None = None,
-        default_context_enrichment_token_threshold: int | None = None,
+        default_modes: list[str] | None = None,
         default_model: "Model | None" = None,
         default_model_settings: "ModelSettings | None" = None,
         default_model_provider: "Provider | None" = None,
@@ -40,18 +38,14 @@ class LLMConfig:
         self._default_interactive_system_prompt = default_interactive_system_prompt
         self._default_special_instruction_prompt = default_special_instruction_prompt
         self._default_summarization_prompt = default_summarization_prompt
-        self._default_context_enrichment_prompt = default_context_enrichment_prompt
         self._default_summarize_history = default_summarize_history
         self._default_history_summarization_token_threshold = (
             default_history_summarization_token_threshold
         )
-        self._default_enrich_context = default_enrich_context
-        self._default_context_enrichment_token_threshold = (
-            default_context_enrichment_token_threshold
-        )
+        self._default_modes = default_modes
+        self._default_model = default_model
         self._default_model_settings = default_model_settings
         self._default_model_provider = default_model_provider
-        self._default_model = default_model
 
     def _get_internal_default_prompt(self, name: str) -> str:
         if name not in self.__internal_default_prompt:
@@ -130,32 +124,18 @@ class LLMConfig:
             lambda: self._get_internal_default_prompt("persona"),
         )
 
+    @property
+    def default_modes(self) -> list[str]:
+        return self._get_property(
+            self._default_modes, CFG.LLM_MODES, lambda: ["coding"]
+        )
+
     @property
     def default_special_instruction_prompt(self) -> str:
         return self._get_property(
             self._default_special_instruction_prompt,
             CFG.LLM_SPECIAL_INSTRUCTION_PROMPT,
-            lambda: self._get_workflow_prompt(CFG.LLM_MODES),
-        )
-
-    def _get_workflow_prompt(self, modes: list[str]) -> str:
-        workflows = llm_context_config.get_workflows()
-        dir_path = os.path.dirname(__file__)
-        default_workflow_names = ("code", "content", "research")
-        for workflow_name in default_workflow_names:
-            if workflow_name in workflows:
-                continue
-            workflow_file_path = os.path.join(
-                dir_path, "default_workflow", f"{workflow_name}.md"
-            )
-            with open(workflow_file_path, "r") as f:
-                workflows[workflow_name] = f.read()
-        return "\n".join(
-            [
-                make_prompt_section(header, content)
-                for header, content in workflows.items()
-                if header.lower() in modes
-            ]
+            lambda: "",
         )
 
     @property
@@ -206,6 +186,19 @@ class LLMConfig:
     def set_default_special_instruction_prompt(self, special_instruction_prompt: str):
         self._default_special_instruction_prompt = special_instruction_prompt
 
+    def set_default_modes(self, modes: list[str]):
+        self._default_modes = modes
+
+    def add_default_mode(self, mode: str):
+        if self._default_modes is None:
+            self._default_modes = []
+        self._default_modes.append(mode)
+
+    def remove_default_mode(self, mode: str):
+        if self._default_modes is None:
+            self._default_modes = []
+        self._default_modes.remove(mode)
+
     def set_default_summarization_prompt(self, summarization_prompt: str):
         self._default_summarization_prompt = summarization_prompt
 

zrb-1.11.0/src/zrb/config/default_workflow/code.md → zrb-1.12.0/src/zrb/task/llm/default_workflow/coding.md

@@ -1,5 +1,3 @@
-# Special Instructions for Software Engineering
-
 When the user's request involves writing or modifying code, you MUST follow these domain-specific rules in addition to your core workflow.
 
 ## 1. Critical Prohibitions

zrb-1.11.0/src/zrb/config/default_workflow/content.md → zrb-1.12.0/src/zrb/task/llm/default_workflow/copywriting.md

@@ -1,5 +1,3 @@
-# Special Instructions for Content Creation & Management
-
 When the user's request involves creating, refining, or organizing textual content, you MUST follow these domain-specific rules in addition to your core workflow.
 
 ## 1. Core Principles

zrb-1.11.0/src/zrb/config/default_workflow/research.md → zrb-1.12.0/src/zrb/task/llm/default_workflow/researching.md

@@ -1,5 +1,3 @@
-# Special Instructions for Research, Analysis, and Summarization
-
 When the user's request involves finding, synthesizing, or analyzing information, you MUST follow these domain-specific rules in addition to your core workflow.
 
 ## 1. Core Principles

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/task/llm/prompt.py

@@ -3,11 +3,12 @@ import platform
 import re
 from datetime import datetime, timezone
 
-from zrb.attr.type import StrAttr
+from zrb.attr.type import StrAttr, StrListAttr
 from zrb.config.llm_config import llm_config as llm_config
+from zrb.config.llm_context.config import llm_context_config
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.conversation_history_model import ConversationHistory
-from zrb.util.attr import get_attr, get_str_attr
+from zrb.util.attr import get_attr, get_str_attr, get_str_list_attr
 from zrb.util.file import read_dir, read_file_with_line_numbers
 from zrb.util.llm.prompt import make_prompt_section
 
@@ -15,13 +16,14 @@ from zrb.util.llm.prompt import make_prompt_section
 def get_persona(
     ctx: AnyContext,
     persona_attr: StrAttr | None,
+    render_persona: bool,
 ) -> str:
     """Gets the persona, prioritizing task-specific, then default."""
     persona = get_attr(
         ctx,
         persona_attr,
         None,
-        auto_render=False,
+        auto_render=render_persona,
     )
     if persona is not None:
         return persona
@@ -31,13 +33,14 @@ def get_persona(
 def get_base_system_prompt(
     ctx: AnyContext,
     system_prompt_attr: StrAttr | None,
+    render_system_prompt: bool,
 ) -> str:
     """Gets the base system prompt, prioritizing task-specific, then default."""
     system_prompt = get_attr(
         ctx,
         system_prompt_attr,
         None,
-        auto_render=False,
+        auto_render=render_system_prompt,
     )
     if system_prompt is not None:
         return system_prompt
@@ -47,33 +50,95 @@ def get_base_system_prompt(
 def get_special_instruction_prompt(
     ctx: AnyContext,
     special_instruction_prompt_attr: StrAttr | None,
+    render_spcecial_instruction_prompt: bool,
 ) -> str:
     """Gets the special instruction prompt, prioritizing task-specific, then default."""
     special_instruction = get_attr(
         ctx,
         special_instruction_prompt_attr,
         None,
-        auto_render=False,
+        auto_render=render_spcecial_instruction_prompt,
     )
     if special_instruction is not None:
         return special_instruction
     return llm_config.default_special_instruction_prompt
 
 
+def get_modes(
+    ctx: AnyContext,
+    modes_attr: StrAttr | None,
+    render_modes: bool,
+) -> str:
+    """Gets the modes, prioritizing task-specific, then default."""
+    raw_modes = get_str_list_attr(
+        ctx,
+        modes_attr,
+        auto_render=render_modes,
+    )
+    modes = [mode.strip() for mode in raw_modes if mode.strip() != ""]
+    if len(modes) > 0:
+        return modes
+    return llm_config.default_modes or []
+
+
+def get_workflow_prompt(
+    ctx: AnyContext,
+    modes_attr: StrAttr | None,
+    render_modes: bool,
+) -> str:
+    modes = get_modes(ctx, modes_attr, render_modes)
+    # Get user-defined workflows
+    workflows = {
+        workflow_name: content
+        for workflow_name, content in llm_context_config.get_workflows().items()
+        if workflow_name in modes
+    }
+    # Get requested builtin-workflow names
+    requested_builtin_workflow_names = [
+        workflow_name
+        for workflow_name in ("coding", "copywriting", "researching")
+        if workflow_name in modes and workflow_name not in workflows
+    ]
+    # add builtin-workflows if requested
+    if len(requested_builtin_workflow_names) > 0:
+        dir_path = os.path.dirname(__file__)
+        for workflow_name in requested_builtin_workflow_names:
+            workflow_file_path = os.path.join(
+                dir_path, "default_workflow", f"{workflow_name}.md"
+            )
+            with open(workflow_file_path, "r") as f:
+                workflows[workflow_name] = f.read()
+    return "\n".join(
+        [
+            make_prompt_section(header.capitalize(), content)
+            for header, content in workflows.items()
+            if header.lower() in modes
+        ]
+    )
+
+
 def get_system_and_user_prompt(
     ctx: AnyContext,
     user_message: str,
     persona_attr: StrAttr | None = None,
+    render_persona: bool = False,
     system_prompt_attr: StrAttr | None = None,
+    render_system_prompt: bool = False,
     special_instruction_prompt_attr: StrAttr | None = None,
+    render_special_instruction_prompt: bool = False,
+    modes_attr: StrListAttr | None = None,
+    render_modes: bool = False,
     conversation_history: ConversationHistory | None = None,
 ) -> tuple[str, str]:
     """Combines persona, base system prompt, and special instructions."""
-    persona = get_persona(ctx, persona_attr)
-    base_system_prompt = get_base_system_prompt(ctx, system_prompt_attr)
-    special_instruction = get_special_instruction_prompt(
-        ctx, special_instruction_prompt_attr
+    persona = get_persona(ctx, persona_attr, render_persona)
+    base_system_prompt = get_base_system_prompt(
+        ctx, system_prompt_attr, render_system_prompt
     )
+    special_instruction_prompt = get_special_instruction_prompt(
+        ctx, special_instruction_prompt_attr, render_special_instruction_prompt
+    )
+    workflow_prompt = get_workflow_prompt(ctx, modes_attr, render_modes)
     if conversation_history is None:
         conversation_history = ConversationHistory()
     conversation_context, new_user_message = extract_conversation_context(user_message)
@@ -81,7 +146,8 @@ def get_system_and_user_prompt(
         [
             make_prompt_section("Persona", persona),
             make_prompt_section("System Prompt", base_system_prompt),
-            make_prompt_section("Special Instruction", special_instruction),
+            make_prompt_section("Special Instruction", special_instruction_prompt),
+            make_prompt_section("Special Workflows", workflow_prompt),
             make_prompt_section(
                 "Past Conversation",
                 "\n".join(
@@ -194,30 +260,15 @@ def get_user_message(
 def get_summarization_system_prompt(
     ctx: AnyContext,
     summarization_prompt_attr: StrAttr | None,
+    render_summarization_prompt: bool,
 ) -> str:
     """Gets the summarization prompt, rendering if configured and handling defaults."""
     summarization_prompt = get_attr(
         ctx,
         summarization_prompt_attr,
         None,
-        auto_render=False,
+        auto_render=render_summarization_prompt,
     )
     if summarization_prompt is not None:
         return summarization_prompt
     return llm_config.default_summarization_prompt
-
-
-def get_context_enrichment_prompt(
-    ctx: AnyContext,
-    context_enrichment_prompt_attr: StrAttr | None,
-) -> str:
-    """Gets the context enrichment prompt, rendering if configured and handling defaults."""
-    context_enrichment_prompt = get_attr(
-        ctx,
-        context_enrichment_prompt_attr,
-        None,
-        auto_render=False,
-    )
-    if context_enrichment_prompt is not None:
-        return context_enrichment_prompt
-    return llm_config.default_context_enrichment_prompt

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/task/llm/tool_wrapper.py

@@ -5,9 +5,12 @@ import typing
 from collections.abc import Callable
 from typing import TYPE_CHECKING
 
+from zrb.config.config import CFG
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.error import ToolExecutionError
+from zrb.util.callable import get_callable_name
 from zrb.util.run import run_async
+from zrb.util.string.conversion import to_boolean
 
 if TYPE_CHECKING:
     from pydantic_ai import Tool
@@ -71,13 +74,11 @@ def _create_wrapper(
     async def wrapper(*args, **kwargs):
         # Identify AnyContext parameter name from the original signature if needed
         any_context_param_name = None
-
         if needs_any_context_for_injection:
             for param in original_sig.parameters.values():
                 if _is_annotated_with_context(param.annotation, AnyContext):
                     any_context_param_name = param.name
                     break  # Found it, no need to continue
-
             if any_context_param_name is None:
                 # This should not happen if needs_any_context_for_injection is True,
                 # but check for safety
@@ -87,24 +88,22 @@ def _create_wrapper(
             # Inject the captured ctx into kwargs. This will overwrite if the LLM
             # somehow provided it.
             kwargs[any_context_param_name] = ctx
-
         # If the dummy argument was added for schema generation and is present in kwargs,
         # remove it before calling the original function, unless the original function
         # actually expects a parameter named '_dummy'.
         if "_dummy" in kwargs and "_dummy" not in original_sig.parameters:
             del kwargs["_dummy"]
-
         try:
-            # Call the original function.
-            # pydantic-ai is responsible for injecting RunContext if takes_ctx is True.
-            # Our wrapper injects AnyContext if needed.
-            # The arguments received by the wrapper (*args, **kwargs) are those
-            # provided by the LLM, potentially with RunContext already injected by
-            # pydantic-ai if takes_ctx is True. We just need to ensure AnyContext
-            # is injected if required by the original function.
-            # The dummy argument handling is moved to _adjust_signature's logic
-            # for schema generation, it's not needed here before calling the actual
-            # function.
+            if not CFG.LLM_YOLO_MODE and not ctx.is_web_mode and ctx.is_tty:
+                func_name = get_callable_name(func)
+                ctx.print(f"✅ >> Allow to run tool: {func_name} (Y/n)", plain=True)
+                user_confirmation_str = await _read_line()
+                user_confirmation = to_boolean(user_confirmation_str)
+                if not user_confirmation:
+                    ctx.print("❌ >> Why?", plain=True)
+                    reason = await _read_line()
+                    ctx.print("", plain=True)
+                    raise ValueError(f"User disapproval: {reason}")
             return await run_async(func(*args, **kwargs))
         except Exception as e:
             error_model = ToolExecutionError(
@@ -118,6 +117,13 @@ def _create_wrapper(
     return wrapper
 
 
+async def _read_line():
+    from prompt_toolkit import PromptSession
+
+    reader = PromptSession()
+    return await reader.prompt_async()
+
+
 def _adjust_signature(
     wrapper: Callable, original_sig: inspect.Signature, takes_no_args: bool
 ):

{zrb-1.11.0 → zrb-1.12.0}/src/zrb/task/llm_task.py

@@ -2,7 +2,7 @@ import json
 from collections.abc import Callable
 from typing import TYPE_CHECKING, Any
 
-from zrb.attr.type import BoolAttr, IntAttr, StrAttr, fstring
+from zrb.attr.type import BoolAttr, IntAttr, StrAttr, StrListAttr, fstring
 from zrb.config.llm_rate_limitter import LLMRateLimiter
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
@@ -16,14 +16,12 @@ from zrb.task.llm.config import (
     get_model_settings,
 )
 from zrb.task.llm.conversation_history import (
-    ListOfDict,
     read_conversation_history,
     write_conversation_history,
 )
 from zrb.task.llm.conversation_history_model import ConversationHistory
 from zrb.task.llm.history_summarization import maybe_summarize_history
 from zrb.task.llm.prompt import (
-    get_context_enrichment_prompt,
     get_summarization_system_prompt,
     get_system_and_user_prompt,
     get_user_message,
@@ -65,16 +63,15 @@ class LLMTask(BaseTask):
         ) = None,
         agent: "Agent | Callable[[AnySharedContext], Agent] | None" = None,
         persona: StrAttr | None = None,
+        render_persona: bool = False,
         system_prompt: StrAttr | None = None,
+        render_system_prompt: bool = False,
         special_instruction_prompt: StrAttr | None = None,
+        render_special_instruction_prompt: bool = False,
+        modes: StrListAttr | None = None,
+        render_modes: bool = True,
         message: StrAttr | None = None,
         render_message: bool = True,
-        enrich_context: BoolAttr | None = None,
-        render_enrich_context: bool = True,
-        context_enrichment_prompt: StrAttr | None = None,
-        render_context_enrichment_prompt: bool = True,
-        context_enrichment_token_threshold: IntAttr | None = None,
-        render_context_enrichment_token_threshold: bool = True,
         tools: (
             list["ToolOrCallable"]
             | Callable[[AnySharedContext], list["ToolOrCallable"]]
@@ -100,6 +97,7 @@ class LLMTask(BaseTask):
         summarize_history: BoolAttr | None = None,
         render_summarize_history: bool = True,
         summarization_prompt: StrAttr | None = None,
+        render_summarization_prompt: bool = False,
         history_summarization_token_threshold: IntAttr | None = None,
         render_history_summarization_token_threshold: bool = True,
         rate_limitter: LLMRateLimiter | None = None,
@@ -150,19 +148,17 @@ class LLMTask(BaseTask):
         self._model_settings = model_settings
         self._agent = agent
         self._persona = persona
+        self._render_persona = render_persona
         self._system_prompt = system_prompt
+        self._render_system_prompt = render_system_prompt
         self._special_instruction_prompt = special_instruction_prompt
+        self._render_special_instruction_prompt = render_special_instruction_prompt
+        self._modes = modes
+        self._render_modes = render_modes
         self._message = message
         self._render_message = render_message
         self._summarization_prompt = summarization_prompt
-        self._should_enrich_context = enrich_context
-        self._render_enrich_context = render_enrich_context
-        self._context_enrichment_prompt = context_enrichment_prompt
-        self._render_context_enrichment_prompt = render_context_enrichment_prompt
-        self._context_enrichment_token_threshold = context_enrichment_token_threshold
-        self._render_context_enrichment_token_threshold = (
-            render_context_enrichment_token_threshold
-        )
+        self._render_summarization_prompt = render_summarization_prompt
         self._tools = tools
         self._rate_limitter = rate_limitter
         self._additional_tools: list["ToolOrCallable"] = []
@@ -198,12 +194,6 @@ class LLMTask(BaseTask):
         for single_mcp_server in mcp_server:
             self._additional_mcp_servers.append(single_mcp_server)
 
-    def set_should_enrich_context(self, enrich_context: bool):
-        self._should_enrich_context = enrich_context
-
-    def set_context_enrichment_token_threshold(self, enrichment_token_threshold: int):
-        self._context_enrichment_token_threshold = enrichment_token_threshold
-
     def set_should_summarize_history(self, summarize_history: bool):
         self._should_summarize_history = summarize_history
 
@@ -227,6 +217,7 @@ class LLMTask(BaseTask):
         summarization_prompt = get_summarization_system_prompt(
             ctx=ctx,
             summarization_prompt_attr=self._summarization_prompt,
+            render_summarization_prompt=self._render_summarization_prompt,
         )
         user_message = get_user_message(ctx, self._message, self._render_message)
         # 1. Prepare initial state (read history from previous session)
@@ -243,8 +234,13 @@ class LLMTask(BaseTask):
             ctx=ctx,
             user_message=user_message,
             persona_attr=self._persona,
+            render_persona=self._render_persona,
             system_prompt_attr=self._system_prompt,
+            render_system_prompt=self._render_system_prompt,
             special_instruction_prompt_attr=self._special_instruction_prompt,
+            render_special_instruction_prompt=self._render_special_instruction_prompt,
+            modes_attr=self._modes,
+            render_modes=self._render_modes,
             conversation_history=conversation_history,
         )
         # 3. Get the agent instance