Removed redundant reasoning tag stripping from litellm model

Author: rolshoven

src/lighteval/models/endpoints/litellm_model.py

@@ -21,7 +21,6 @@
 # SOFTWARE.
 
 import logging
-import re
 import time
 from concurrent.futures import ThreadPoolExecutor
 from json import JSONDecodeError
@@ -233,12 +232,6 @@ def __call_api(self, prompt, return_logits, max_new_tokens, num_samples, stop_se
                     response = litellm.completion(**kwargs)
                     content = response.choices[0].message.content
 
-                if content and "<think>" in content:
-                    logger.debug(f"Removing <think> tags from response: {content}")
-                    response.choices[0].message.content = re.sub(
-                        r"<think>.*?</think>", "", content, flags=re.DOTALL
-                    ).strip()
-
                 return response
             except litellm.BadRequestError as e:
                 if "message" in e.__dict__: