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

Attachment / add TTL api

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-26 13:10:27 -07:00
parent 3230af4910
commit 986a865e62
3 changed files with 5 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

17
llama_toolchain/safety/meta_reference/shields/base.py
View file

@ -5,9 +5,9 @@
# the root directory of this source tree.
from abc import ABC, abstractmethod
from typing import List, Union
from typing import List
from llama_models.llama3.api.datatypes import Attachment, Message
from llama_models.llama3.api.datatypes import interleaved_text_media_as_str, Message
from llama_toolchain.safety.api.datatypes import * # noqa: F403
CANNED_RESPONSE_TEXT = "I can't answer that. Can I help with something else?"
@ -30,18 +30,7 @@ class ShieldBase(ABC):
def message_content_as_str(message: Message) -> str:
def _to_str(content: Union[str, Attachment]) -> str:
if isinstance(content, str):
return content
elif isinstance(content, Attachment):
return f"File: {str(content.url)}"
else:
raise
if isinstance(message.content, list) or isinstance(message.content, tuple):
return "\n".join([_to_str(c) for c in message.content])
else:
return _to_str(message.content)
return interleaved_text_media_as_str(message.content)
# For shields that operate on simple strings