From f452f4522ef312b3a49cd40b7eb07970a000ae3f Mon Sep 17 00:00:00 2001
From: HwVanICI <zhenan.fan1@huawei.com>
Date: Wed, 24 Dec 2025 14:48:03 -0500
Subject: [PATCH] Fixed: Camel example with wrong and missing agent arguments

---
 examples/camel/config.yaml |  1 +
 examples/camel/train.py    | 10 ++++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/examples/camel/config.yaml b/examples/camel/config.yaml
index 11e4fb8ee..c1eacea89 100644
--- a/examples/camel/config.yaml
+++ b/examples/camel/config.yaml
@@ -27,6 +27,7 @@ rollout:
   consumer_batch_size: ${train_dataset.batch_size}
   max_head_offpolicyness: 2
   enable_rollout_tracing: false
+  request_timeout: 3600
 
 gconfig:
   n_samples: 1
diff --git a/examples/camel/train.py b/examples/camel/train.py
index d047cd9fb..a17ed74aa 100644
--- a/examples/camel/train.py
+++ b/examples/camel/train.py
@@ -55,12 +55,18 @@ def __init__(
         self.async_reward_fn = AsyncRewardWrapper(gsm8k_reward_fn)
 
     async def run_agent(self, data, client: ArealOpenAI):
+        model_config_dict = {"max_tokens": self.max_total_tokens}
+        rollout_engine_request_timeout = client.engine.config.request_timeout
+
         messages = data["messages"].copy()
         agent = ChatAgent(
             model=AReaLOpenAICompatibleModel(
-                openai_client=client, tokenizer=self.tokenizer, model_type="areal"
+                openai_client=client,
+                tokenizer=self.tokenizer,
+                model_type="areal",
+                model_config_dict=model_config_dict,
             ),
-            token_limit=self.max_total_tokens,
+            step_timeout=rollout_engine_request_timeout,
         )
         response = await agent.astep(messages[-1]["content"])
         content = response.msg.content