feat: sync latest ARR implementation

tests/test_agent_client.py

@@ -52,8 +52,11 @@ class AgentClientTests(unittest.TestCase):
         )
 
     def test_stream_parser_preserves_event_id_retry_and_multiline_json(self):
+        seen = {}
+
         def handler(request: httpx.Request) -> httpx.Response:
             self.assertEqual(request.headers.get("accept"), "text/event-stream")
+            seen["include_trace"] = request.url.params.get("include_trace")
             return httpx.Response(
                 200,
                 headers={"content-type": "text/event-stream"},
@@ -75,7 +78,13 @@ class AgentClientTests(unittest.TestCase):
                 api_key="df_open_test",
                 http_client=http_client,
             )
-            events = list(client.stream_message("open_sess_123", "你好"))
+            events = list(
+                client.stream_message(
+                    "open_sess_123",
+                    "你好",
+                    include_trace=True,
+                )
+            )
 
         self.assertEqual(
             events,
@@ -89,6 +98,7 @@ class AgentClientTests(unittest.TestCase):
                 OpenAgentEvent(event="run.completed", data={"status": "completed"}),
             ],
         )
+        self.assertEqual(seen["include_trace"], "true")
 
     def test_non_stream_routes_support_x_api_key_auth(self):
         calls = []
@@ -150,4 +160,3 @@ class AgentClientTests(unittest.TestCase):
 
 if __name__ == "__main__":
     unittest.main(verbosity=2)
-