From 6a11d72df7f15b39d76570dba5f415ff2b0292b3 Mon Sep 17 00:00:00 2001
From: rishitdholakia13 <123388671+rishitdholakia13@users.noreply.github.com>
Date: Fri, 5 Jun 2026 09:51:20 -0400
Subject: [PATCH] [Reasoning][Structured Outputs] Add Command A plus tags for
 structural tags (#44588)

Signed-off-by: rishitdholakia13 <rishit+github@cohere.com>
Co-authored-by: Chauncey <chaunceyjiang@gmail.com>
---
 vllm/reasoning/cohere_command_reasoning_parser.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vllm/reasoning/cohere_command_reasoning_parser.py b/vllm/reasoning/cohere_command_reasoning_parser.py
index b28a59089e..949c9ff5d9 100644
--- a/vllm/reasoning/cohere_command_reasoning_parser.py
+++ b/vllm/reasoning/cohere_command_reasoning_parser.py
@@ -90,7 +90,7 @@ MODEL_TO_TAG_STYLE: dict[str, CohereTagStyle] = {
         tools=COMMAND_A_TOOLS_TAG,
     ),
     "Cohere2MoeForCausalLM": CohereTagStyle(
-        json_tags=(COMMAND_A_JSON_TAG,),
+        json_tags=(COMMAND_A_JSON_TAG, COMMAND_A_PLUS_JSON_TAG),
         tools=COMMAND_A_TOOLS_TAG,
     ),
 }