phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 12:21:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adapter -> Provider

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-05 13:26:29 -07:00
parent db3e6dda07
commit 65a9e40174
15 changed files with 119 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_toolchain/safety/safety.py
View file

@ -8,7 +8,7 @@ import asyncio
from typing import Dict
from llama_toolchain.distribution.datatypes import Adapter, Api
from llama_toolchain.distribution.datatypes import Api, ProviderSpec
from .config import SafetyConfig
from .api.endpoints import * # noqa
@ -23,7 +23,7 @@ from .shields import (
)
async def get_adapter_impl(config: SafetyConfig, _deps: Dict[Api, Adapter]):
async def get_provider_impl(config: SafetyConfig, _deps: Dict[Api, ProviderSpec]):
assert isinstance(config, SafetyConfig), f"Unexpected config type: {type(config)}"
impl = MetaReferenceSafetyImpl(config)