Tree: Formatting

This commit is contained in:
turboderp
2026-03-18 03:29:15 +01:00
parent 53357047ef
commit 6bccc70d94
4 changed files with 24 additions and 9 deletions

View File

@@ -43,7 +43,9 @@ def _extract_think_content(text: str) -> tuple[Optional[str], Optional[str]]:
return None, text
elif model.container.reasoning_start_token in text:
start_reasoning = text.split(model.container.reasoning_start_token)[1]
reasoning_content = start_reasoning.split(model.container.reasoning_end_token)[0]
reasoning_content = start_reasoning.split(model.container.reasoning_end_token)[
0
]
content = start_reasoning.split(model.container.reasoning_end_token)[1]
return reasoning_content.strip(), content.strip()
else: