mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
add env using the tool api stuff
This commit is contained in:
parent
c8eb63f33d
commit
12d61d197f
15 changed files with 2632 additions and 21 deletions
|
|
@ -7,7 +7,12 @@ import json
|
|||
|
||||
import pytest
|
||||
|
||||
from atroposlib.envs.server_handling.tool_call_translator import ToolCallTranslator
|
||||
from atroposlib.envs.server_handling.tool_call_translator import (
|
||||
VLLM_AVAILABLE,
|
||||
ToolCallTranslator,
|
||||
)
|
||||
|
||||
pytestmark = pytest.mark.skipif(not VLLM_AVAILABLE, reason="vLLM not installed")
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Mock tokenizer (same one from test_managed_server.py)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue