change extraction

lavague-ai · Aug 26, 2024 · 7ecd7e0 · 7ecd7e0
1 parent 4ece722
commit 7ecd7e0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lavague-core/lavague/core/python_engine.py b/lavague-core/lavague/core/python_engine.py
@@ -56,7 +56,7 @@ def __init__(
         display: bool = False,
         batch_size: int = 5,
         confidence_threshold: float = 0.85,
-        fallback_threshold: float = 0.65,
+        fallback_threshold: float = 0.85,
         temp_screenshots_path="./tmp_screenshots",
         n_search_attemps=10,
     ):
@@ -81,7 +81,8 @@ def from_context(cls, context: Context, driver: BaseDriver):
         return cls(llm=context.llm, embedding=context.embedding, driver=driver)
 
     def extract_json(self, output: str) -> Optional[dict]:
-        clean = JsonFromMarkdownExtractor.extract(output)
+        extractor = JsonFromMarkdownExtractor()
+        clean = extractor.extract(markdown_text=output)
         try:
             output_dict = json.loads(clean)
         except json.JSONDecodeError as e: