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 tool call parsing based on vllm impl and an openai server endpoint

Browse source
This commit is contained in:
dmahan93 2026-03-02 23:17:13 -06:00
parent 887a94374c
commit add42a2afb
11 changed files with 3370 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

27
atroposlib/tests/conftest.py
View file

@ -5,19 +5,32 @@ def pytest_addoption(parser):
parser.addoption(
"--runproviders", action="store_true", default=False, help="run provider tests"
)
parser.addoption(
"--run-gpu",
action="store_true",
default=False,
help="run GPU integration tests",
)
def pytest_configure(config):
config.addinivalue_line(
"markers", "providers: mark test as requires providers api keys to run"
)
config.addinivalue_line(
"markers", "gpu: mark test as requiring GPU (skipped unless --run-gpu)"
)
def pytest_collection_modifyitems(config, items):
if config.getoption("--runproviders"):
# --runproviders given in cli: do not skip slow tests
return
skip_providers = pytest.mark.skip(reason="need --runproviders option to run")
for item in items:
if "providers" in item.keywords:
item.add_marker(skip_providers)
if not config.getoption("--runproviders"):
skip_providers = pytest.mark.skip(reason="need --runproviders option to run")
for item in items:
if "providers" in item.keywords:
item.add_marker(skip_providers)
if not config.getoption("--run-gpu"):
skip_gpu = pytest.mark.skip(reason="need --run-gpu option to run")
for item in items:
if "gpu" in item.keywords:
item.add_marker(skip_gpu)