Upload alpha_factory/infra/llm_client.py

alpha_factory/infra/llm_client.py

@@ -268,13 +268,14 @@ class LLMClient:
         Parse JSON from LLM response, handling common issues:
         - Markdown code blocks
         - Leading/trailing text
-        - Thinking tags
+        - Thinking tags (<think>...</think>, <thinking>...</thinking>)
         """
         text = content.strip()
 
         # Remove thinking tags (Qwen/DeepSeek R1 style)
-        text = re.sub(r'\s*\s*', '', text, flags=re.DOTALL).strip()
-        text = re.sub(r'\s*<thinking>.*?</thinking>\s*', '', text, flags=re.DOTALL).strip()
+        # Match <think>...</think> or <thinking>...</thinking>
+        text = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL).strip()
+        text = re.sub(r'<thinking>.*?</thinking>', '', text, flags=re.DOTALL).strip()
 
         # Remove markdown code blocks
         if "```json" in text: