update custom selection prompt (#689)

This commit is contained in:
LawyZheng
2024-08-08 14:53:08 +08:00
committed by GitHub
parent 83cbcfee28
commit 5299f9265e
3 changed files with 19 additions and 4 deletions

View File

@@ -442,8 +442,8 @@ class IncrementalScrapePage:
js_script = "() => window.globalOneTimeIncrementElements.length"
return await self.skyvern_frame.get_frame().evaluate(js_script)
def build_html_tree(self) -> str:
return "".join([json_to_html(element) for element in self.element_tree_trimmed])
def build_html_tree(self, element_tree: list[dict] = []) -> str:
return "".join([json_to_html(element) for element in (element_tree or self.element_tree_trimmed)])
def trim_element_tree(elements: list[dict]) -> list[dict]: