phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Some lightweight cleanup and renaming for bedrock safety adapter

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-24 19:27:03 -07:00
parent a2465f3f9c
commit f45705cd10
5 changed files with 76 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/adapters/safety/bedrock/__init__.py
View file

@ -7,12 +7,12 @@
from typing import Any
from .config import BedrockShieldConfig
from .config import BedrockSafetyConfig
async def get_adapter_impl(config: BedrockShieldConfig, _deps) -> Any:
from .bedrock import BedrockShieldAdapter
async def get_adapter_impl(config: BedrockSafetyConfig, _deps) -> Any:
from .bedrock import BedrockSafetyAdapter
impl = BedrockShieldAdapter(config)
impl = BedrockSafetyAdapter(config)
await impl.initialize()
return impl
return impl