update get-action-from-context

lavague-ai · Aug 26, 2024 · 24350a0 · 24350a0
1 parent e6a6588
commit 24350a0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lavague-core/lavague/core/navigation.py b/lavague-core/lavague/core/navigation.py
@@ -166,7 +166,14 @@ def get_action_from_context(self, context: str, query: str) -> str:
         """
         Generate the code from a query and a context
         """
-        prompt = self.prompt_template.format(context_str=context, query_str=query)
+        authorized_xpaths = extract_xpaths_from_html(context)
+
+        prompt = self.prompt_template.format(
+            context_str=context,
+            query_str=query,
+            authorized_xpaths=authorized_xpaths,
+        )
+
         response = self.llm.complete(prompt).text
         code = self.extractor.extract(response)
         return code