Add mustache rendering for ManageEngine and BrowserUse

Author: whoisarpit

patchwork/steps/BrowserUse/BrowserUse.py

@@ -3,6 +3,7 @@
 import os
 from datetime import datetime
 
+from patchwork.common.utils.utils import mustache_render
 from patchwork.step import Step
 from patchwork.steps import SimplifiedLLMOnce
 from patchwork.steps.BrowserUse.typed import BrowserUseInputs, BrowserUseOutputs
@@ -178,7 +179,7 @@ def run(self) -> dict:
         agent = Agent(
             browser=browser,
             controller=controller,
-            task=self.inputs["task"],
+            task=mustache_render(self.inputs["task"], self.inputs["task_value"]),
             llm=self.llm,
             generate_gif=self.generate_gif,
             validate_output=True,

patchwork/steps/BrowserUse/typed.py

@@ -1,21 +1,19 @@
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Annotated, Any, Dict, Optional, TypedDict
 
 from patchwork.common.utils.step_typing import StepTypeConfig
 
 
-class BrowserUseInputs(TypedDict, total=False):
+class __BrowserUseInputsRequired(TypedDict):
     task: str
-    example_json: str
-    openai_api_key: Annotated[
-        str,
-        StepTypeConfig(is_config=True, or_op=["google_api_key", "anthropic_api_key"]),
-    ]
-    anthropic_api_key: Annotated[str, StepTypeConfig(is_config=True, or_op=["google_api_key", "openai_api_key"])]
-    google_api_key: Annotated[
-        str,
-        StepTypeConfig(is_config=True, or_op=["openai_api_key", "anthropic_api_key"]),
-    ]
-    generate_gif: Annotated[bool, StepTypeConfig(is_config=True)]
+    task_value: Dict[str, Any]
+
+
+class BrowserUseInputs(__BrowserUseInputsRequired, total=False):
+    example_json: Optional[str]
+    openai_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "anthropic_api_key"])]
+    anthropic_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "openai_api_key"])]
+    google_api_key: Annotated[str, StepTypeConfig(or_op=["openai_api_key", "anthropic_api_key"])]
+    generate_gif: Optional[bool]
 
 
 class BrowserUseOutputs(TypedDict):

patchwork/steps/ManageEngineAgent/ManageEngineAgent.py

@@ -4,14 +4,13 @@
     AgenticStrategyV2,
 )
 from patchwork.common.tools.api_tool import APIRequestTool
+from patchwork.common.utils.utils import mustache_render
 from patchwork.step import Step
 
 from .typed import ManageEngineAgentInputs, ManageEngineAgentOutputs
 
 
-class ManageEngineAgent(
-    Step, input_class=ManageEngineAgentInputs, output_class=ManageEngineAgentOutputs
-):
+class ManageEngineAgent(Step, input_class=ManageEngineAgentInputs, output_class=ManageEngineAgentOutputs):
     def __init__(self, inputs: dict):
         super().__init__(inputs)
 
@@ -43,7 +42,7 @@ def __init__(self, inputs: dict):
             llm_client=llm_client,
             system_prompt_template=system_prompt,
             template_data={},
-            user_prompt_template=inputs.get("user_prompt"),
+            user_prompt_template=mustache_render(inputs.get("user_prompt"), inputs.get("prompt_value")),
             agent_configs=[
                 AgentConfig(
                     name="ManageEngine Assistant",

patchwork/steps/ManageEngineAgent/typed.py

@@ -1,24 +1,22 @@
-from typing_extensions import Annotated, Dict, List, Optional, TypedDict
+from typing_extensions import Annotated, Any, Dict, List, Optional, TypedDict
 
 from patchwork.common.utils.step_typing import StepTypeConfig
 
 
-class ManageEngineAgentInputs(TypedDict, total=False):
+class __ManageEngineAgentInputsRequired(TypedDict):
     me_access_token: str
+    user_prompt: str
+    prompt_value: Dict[str, Any]
+
+
+class ManageEngineAgentInputs(__ManageEngineAgentInputsRequired, total=False):
     max_agent_calls: int
-    openai_api_key: Annotated[
-        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "anthropic_api_key"])
-    ]
-    anthropic_api_key: Annotated[
-        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "openai_api_key"])
-    ]
-    google_api_key: Annotated[
-        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "openai_api_key", "anthropic_api_key"])
-    ]
+    openai_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "anthropic_api_key"])]
+    anthropic_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "openai_api_key"])]
+    google_api_key: Annotated[str, StepTypeConfig(or_op=["openai_api_key", "anthropic_api_key"])]
 
     # Prompt and strategy configuration
     system_prompt: Optional[str]
-    user_prompt: str
     example_json: Optional[Dict]