From 9540454b0a9b5d00a2d2a86cb48d197e8250840c Mon Sep 17 00:00:00 2001
From: Meshal Nayim <92199237+MeshalNayim@users.noreply.github.com>
Date: Mon, 8 Dec 2025 23:59:25 -0800
Subject: [PATCH] Fix demo agent (PromptAgent) reset(): add vm_ip and kwargs
 for compatibility with lib_run_single.py (#388)

---
 mm_agents/agent.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/mm_agents/agent.py b/mm_agents/agent.py
index fd5c355..b2c3dc9 100644
--- a/mm_agents/agent.py
+++ b/mm_agents/agent.py
@@ -1134,10 +1134,12 @@ class PromptAgent:
 
             return actions
 
-    def reset(self, _logger=None):
+    def reset(self, _logger=None, vm_ip=None, **kwargs):
         global logger
         logger = _logger if _logger is not None else logging.getLogger("desktopenv.agent")
 
+        self.vm_ip = vm_ip
+
         self.thoughts = []
         self.actions = []
-        self.observations = []
+        self.observations = []
\ No newline at end of file