small bug fix

gui_agents/s3/utils/common_utils.py

@@ -72,9 +72,11 @@ def call_llm_formatted(generator, format_checkers, **kwargs):
     max_retries = 3  # Set the maximum number of retries
     attempt = 0
     response = ""
-    messages = (
-        generator.messages.copy()
-    )  # Copy messages to avoid modifying the original
+    if kwargs.get("messages") is None:
+        messages = generator.messages.copy()  # Copy messages to avoid modifying the original
+    else:
+        messages = kwargs["messages"]
+        del kwargs["messages"]  # Remove messages from kwargs to avoid passing it twice
     while attempt < max_retries:
         response = call_llm_safe(generator, messages=messages, **kwargs)
 

osworld_setup/s3/bbon/generate_facts.py

@@ -45,7 +45,11 @@ async def generate_single_fact_caption(
 
     # Generate fact caption using behavior narrator
     result = await asyncio.to_thread(
-        judge.judge, before_bytes, after_bytes, pyautogui_action
+        judge.judge,
+        screenshot_num=i + 1,
+        before_img_bytes=before_bytes,
+        after_img_bytes=after_bytes,
+        pyautogui_action=pyautogui_action,
     )
     result["screenshot_num"] = i + 1