forked from phoenix-oss/llama-stack-mirror
# What does this PR do? Includes SambaNova safety adaptor to use the sambanova cloud served Meta-Llama-Guard-3-8B minor updates in sambanova docs ## Test Plan pytest -s -v tests/integration/safety/test_safety.py --stack-config=sambanova --safety-shield=sambanova/Meta-Llama-Guard-3-8B
18 lines
480 B
Python
18 lines
480 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
|
|
from typing import Any
|
|
|
|
from .config import SambaNovaSafetyConfig
|
|
|
|
|
|
async def get_adapter_impl(config: SambaNovaSafetyConfig, _deps) -> Any:
|
|
from .sambanova import SambaNovaSafetyAdapter
|
|
|
|
impl = SambaNovaSafetyAdapter(config)
|
|
await impl.initialize()
|
|
return impl
|