new: implemented --cot-tags argument to pass a custom set of CoT tags

src/agent/mod.rs

@@ -106,6 +106,7 @@ pub struct Config {
     pub force_strategy: bool,
     pub user_only: bool,
     pub max_iterations: usize,
+    pub cot_tags: Vec<String>,
 }
 
 pub struct TaskSpecs {
@@ -340,7 +341,7 @@ impl Agent {
     }
 
     async fn on_invalid_action(&self, invocation: Invocation, error: Option<String>) {
-        if invocation.action == "thinking" || invocation.action == "think" {
+        if self.config.cot_tags.contains(&invocation.action) {
             self.on_event(Event::new(EventType::Thinking(invocation.payload.unwrap())))
                 .unwrap();
             return;

src/cli/cli.rs

@@ -33,6 +33,9 @@ pub struct Args {
     /// Force specified serialization format even if the model supports native tools calling.
     #[arg(long)]
     pub force_format: bool,
+    /// Chain-of-Thought xml tags.
+    #[arg(long, value_parser, num_args = 1.., value_delimiter = ' ', default_values_t = ["think".to_string(), "thinking".to_string(), "reasoning".to_string()])]
+    pub cot_tags: Vec<String>,
     /// Tasklet file.
     #[arg(short = 'T', long)]
     pub tasklet: Option<String>,

src/cli/setup.rs

@@ -176,6 +176,7 @@ pub async fn setup_agent_for_task(
         force_strategy: args.force_format,
         user_only: args.user_only,
         max_iterations: args.max_iterations,
+        cot_tags: args.cot_tags.clone(),
     };
 
     // create the agent