phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-02 20:40:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove the safety adapter for Together; we can just use "meta-reference" (#387)

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-06 17:36:57 -08:00 • committed by GitHub
parent 8fc2d212a2
commit 064d2a5287
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 4 additions and 179 deletions
Download patch file Download diff file Expand all files Collapse all files

20
llama_stack/providers/tests/safety/fixtures.py
View file

@ -12,12 +12,10 @@ from llama_stack.providers.inline.meta_reference.safety import (
LlamaGuardShieldConfig,
SafetyConfig,
)
from llama_stack.providers.remote.safety.together import TogetherSafetyConfig
from llama_stack.providers.tests.resolver import resolve_impls_for_test_v2
from ..conftest import ProviderFixture, remote_stack_fixture
from ..env import get_env_or_fail
@pytest.fixture(scope="session")
@ -49,23 +47,7 @@ def safety_meta_reference(safety_model) -> ProviderFixture:
)
@pytest.fixture(scope="session")
def safety_together() -> ProviderFixture:
return ProviderFixture(
providers=[
Provider(
provider_id="together",
provider_type="remote::together",
config=TogetherSafetyConfig().model_dump(),
)
],
provider_data=dict(
together_api_key=get_env_or_fail("TOGETHER_API_KEY"),
),
)
SAFETY_FIXTURES = ["meta_reference", "together", "remote"]
SAFETY_FIXTURES = ["meta_reference", "remote"]
@pytest_asyncio.fixture(scope="session")