Update BrowserUse inputs

Author: whoisarpit

patchwork/steps/BrowserUse/BrowserUse.py

@@ -23,6 +23,7 @@ def __init__(self, inputs):
             raise ValueError(f'Missing required data: "{self.required_keys}"')
 
         self.browser = Browser(config=config)
+
         if "google_api_key" in self.inputs:
             self.llm = ChatGoogleGenerativeAI(
                 model="gemini-2.0-flash", google_api_key=self.inputs["google_api_key"]
@@ -53,7 +54,7 @@ def run(self) -> dict:
         loop = asyncio.new_event_loop()
         self.history = loop.run_until_complete(agent.run())
 
-        if self.inputs["json_example_schema"]:
+        if "example_json" in self.inputs:
             return self.__format_history_as_json()
 
         return {
@@ -74,7 +75,7 @@ def __format_history_as_json(self):
 {self.history.final_result()}
 </history>
 """,
-            json_schema=self.inputs["json_example_schema"],
+            json_schema=self.inputs["example_json"],
             prompt_value=dict(),
         )
 

patchwork/steps/BrowserUse/typed.py

@@ -5,7 +5,7 @@
 
 class BrowserUseInputs(TypedDict, total=False):
     task: str
-    json_example_schema: str
+    example_json: str
     openai_api_key: Annotated[
         str,
         StepTypeConfig(is_config=True, or_op=["google_api_key", "anthropic_api_key"]),