diff --git a/tests/gateway/test_queued_native_image_session_key.py b/tests/gateway/test_queued_native_image_session_key.py index beb0b7626d..238e0dc2d1 100644 --- a/tests/gateway/test_queued_native_image_session_key.py +++ b/tests/gateway/test_queued_native_image_session_key.py @@ -57,7 +57,9 @@ class CaptureQueuedNativeImageAgent: self.tools = [] self.tool_progress_callback = kwargs.get("tool_progress_callback") - def run_conversation(self, message, conversation_history=None, task_id=None): + def run_conversation(self, message, conversation_history=None, task_id=None, **kwargs): + # The real AIAgent.run_conversation accepts the persist_* kwargs the gateway adds for a + # message with a raw inbound id (queued follow-ups now carry theirs); tolerate them. type(self).calls.append(message) return { "final_response": f"done-{len(type(self).calls)}",