NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

make preserve thinking optional

Browse source
This commit is contained in:
dmahan93 2026-03-04 15:44:12 -06:00
parent 12d61d197f
commit f4875c5dc6
3 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
environments/t1_tool_planning/t1_core.py
View file

@ -201,7 +201,9 @@ async def collect_multistep_trajectory(
f"collect_multistep_trajectory: {len(conversation)} turns, temp={temperature}"
)
async with server.managed_server(tokenizer=tokenizer) as managed:
async with server.managed_server(
tokenizer=tokenizer, preserve_think_blocks=True
) as managed:
for i, turn in enumerate(conversation):
role = turn["Role"].strip().lower()