add authorized paths to gradio

lavague-ai · Aug 26, 2024 · bf37b41 · bf37b41
1 parent fe3eeb0
commit bf37b41
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lavague-core/lavague/core/navigation.py b/lavague-core/lavague/core/navigation.py
@@ -248,8 +248,11 @@ def execute_instruction_gradio(self, instruction: str, action_engine: Any):
                 except:
                     pass
             start = time.time()
+            authorized_xpaths = extract_xpaths_from_html(llm_context)
             prompt = self.prompt_template.format(
-                context_str=llm_context, query_str=instruction
+                context_str=llm_context,
+                query_str=instruction,
+                authorized_xpaths=authorized_xpaths,
             )
             response = self.llm.complete(prompt).text
             end = time.time()