add wandb settings, remember to set WANDB_KEY

mm_agents/agent.py

@@ -15,6 +15,7 @@ import backoff
 import dashscope
 import google.generativeai as genai
 import requests
+import wandb
 from PIL import Image
 
 from mm_agents.accessibility_tree_wrap.heuristic_retrieve import find_leaf_nodes, filter_nodes, draw_bounding_boxes
@@ -441,7 +442,7 @@ class PromptAgent:
             actions = None
             self.thoughts.append("")
 
-        return actions
+        return response, actions
 
     @backoff.on_exception(
         backoff.expo,