phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-31 16:01:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

minor updates

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-07 22:40:15 -08:00
parent 04a2965967
commit 39f0c5f544
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/agents/meta_reference/safety.py
View file

@ -37,7 +37,7 @@ class ShieldRunnerMixin:
responses = await asyncio.gather( responses = await asyncio.gather(
*[ *[
self.safety_api.run_shield( self.safety_api.run_shield(
identifier=identifier, shield_id=identifier,
messages=messages, messages=messages,
) )
for identifier in identifiers for identifier in identifiers

2
llama_stack/providers/inline/agents/meta_reference/tests/test_chat_agent.py
View file

@ -80,7 +80,7 @@ class MockInferenceAPI:
class MockSafetyAPI: class MockSafetyAPI:
async def run_shield( async def run_shield(
self, shield_identifier: str, messages: List[Message] self, shield_id: str, messages: List[Message]
) -> RunShieldResponse: ) -> RunShieldResponse:
return RunShieldResponse(violation=None) return RunShieldResponse(violation=None)