remove qwen agent producer

Author: YeAnbang

applications/ColossalChat/coati/distributed/agent/qwen_math_agentic_producer.py

@@ -5,7 +5,6 @@
 
 import ray
 from coati.distributed.agent.agentic_producer import AgenticProducer
-from coati.distributed.agent.qwen_math_agentic_producer import QwenMathAgenticProducer
 from coati.distributed.agent.tool_worker import ToolWorker
 
 from .consumer import SimpleConsumer
@@ -21,7 +20,6 @@
 }
 Producer_MAP = {"Simple": SimpleProducer, "Async": AsyncSimpleProducer}
 AGENTIC_PRODUCER_MAP = {
-    "QwenMathAgent": QwenMathAgenticProducer,
     "Agentic": AgenticProducer,
 }  # supported agentic producers
 

applications/ColossalChat/coati/distributed/agent/qwen_math_agentic_utils.py

@@ -165,7 +165,7 @@
         "--agentic-type",
         type=str,
         default="Agentic",
-        choices=["Agentic", "QwenMathAgent"],
+        choices=["Agentic"],
         help="Agentic model type for agentic training.",
     )
     parser.add_argument(
@@ -418,19 +418,7 @@
     if "agentic" in args.backend:
         assert "vllm" in args.backend, "Agentic backend only supports async-agentic-vllm backends."
         generate_config["n"] = 1  # agentic producer use AsyncProducer which processes one request a time
-        if args.agentic_type == "QwenMathAgent":
-            agentic_config = {
-                "agentic_producer": "QwenMathAgent",
-                "model": args.model,
-                "model_type": "transformers",
-                "generate_cfg": {
-                    "max_input_tokens": args.max_new_tokens + args.max_prompt_tokens,
-                },
-            }
-            agentic_config["generate_cfg"].update(
-                {k: v for k, v in generate_config.items() if k in ["top_k", "top_p", "temperature"]}
-            )
-        elif args.agentic_type == "Agentic":
+        if args.agentic_type == "Agentic":
             generate_config["stop"] = ["<|im_end|>"]
             generate_config["prompt_logprobs"] = 0
             agentic_config = {