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

add env using the tool api stuff

Browse source
This commit is contained in:
dmahan93 2026-03-03 19:51:30 -06:00
parent c8eb63f33d
commit 12d61d197f
15 changed files with 2632 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

17
atroposlib/envs/server_handling/tool_call_translator.py
View file

@ -208,11 +208,20 @@ class ToolCallTranslator:
if role == "assistant" and msg.get("tool_calls"):
# Reconstruct raw text from tool_calls
first_id = (
msg["tool_calls"][0].get("id", "") if msg["tool_calls"] else ""
)
is_exact = first_id in self.call_id_to_raw_text
raw_text = self.reconstruct_raw_text_from_tool_calls(msg["tool_calls"])
# Prepend any content that came before the tool calls
content = msg.get("content") or ""
if content:
raw_text = content + "\n" + raw_text
if not is_exact:
# Fallback reconstruction — prepend any content (e.g. <think>)
# that came before the tool calls
content = msg.get("content") or ""
if content:
raw_text = content + "\n" + raw_text
# When exact lookup succeeds, raw_text already contains
# the full model output (including any <think> blocks)
converted.append(
{