NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +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

5
atroposlib/tests/test_managed_server_proxy.py
View file

@ -13,6 +13,9 @@ from fastapi.testclient import TestClient
from atroposlib.envs.server_handling.managed_server_proxy import create_app
from atroposlib.envs.server_handling.server_harness import ServerHarness
from atroposlib.envs.server_handling.server_manager import ServerManager
from atroposlib.envs.server_handling.tool_call_translator import VLLM_AVAILABLE
skip_no_vllm = pytest.mark.skipif(not VLLM_AVAILABLE, reason="vLLM not installed")
# ---------------------------------------------------------------------------
# Mock tokenizer (same as test_managed_server.py / test_tool_call_translator.py)
@ -304,6 +307,7 @@ class TestChatCompletions:
# ---------------------------------------------------------------------------
@skip_no_vllm
class TestToolCalls:
def test_tool_call_outbound(self, client_and_backend):
"""Model generates <tool_call> tags → response has structured tool_calls."""
@ -535,6 +539,7 @@ class TestNodes:
# ---------------------------------------------------------------------------
@skip_no_vllm
class TestMultiStepNodeHandling:
"""Test that multi-step conversations with tool calls produce exactly 1 node.