forked from phoenix-oss/llama-stack-mirror
formatting
This commit is contained in:
parent
94dfa293a6
commit
b6ccaf1778
33 changed files with 110 additions and 97 deletions
|
@ -246,7 +246,6 @@ class AgentInstance(ShieldRunnerMixin):
|
|||
await self.run_shields(messages, shields)
|
||||
|
||||
except SafetyException as e:
|
||||
|
||||
yield AgenticSystemTurnResponseStreamChunk(
|
||||
event=AgenticSystemTurnResponseEvent(
|
||||
payload=AgenticSystemTurnResponseStepCompletePayload(
|
||||
|
|
|
@ -23,7 +23,6 @@ class SafetyException(Exception): # noqa: N818
|
|||
|
||||
|
||||
class ShieldRunnerMixin:
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
safety_api: Safety,
|
||||
|
|
|
@ -11,7 +11,6 @@ from llama_toolchain.inference.api import Message
|
|||
|
||||
|
||||
class BaseTool(ABC):
|
||||
|
||||
@abstractmethod
|
||||
def get_name(self) -> str:
|
||||
raise NotImplementedError
|
||||
|
|
|
@ -66,7 +66,6 @@ class SingleMessageBuiltinTool(BaseTool):
|
|||
|
||||
|
||||
class PhotogenTool(SingleMessageBuiltinTool):
|
||||
|
||||
def __init__(self, dump_dir: str) -> None:
|
||||
self.dump_dir = dump_dir
|
||||
|
||||
|
@ -87,7 +86,6 @@ class PhotogenTool(SingleMessageBuiltinTool):
|
|||
|
||||
|
||||
class BraveSearchTool(SingleMessageBuiltinTool):
|
||||
|
||||
def __init__(self, api_key: str) -> None:
|
||||
self.api_key = api_key
|
||||
|
||||
|
@ -204,7 +202,6 @@ class BraveSearchTool(SingleMessageBuiltinTool):
|
|||
|
||||
|
||||
class WolframAlphaTool(SingleMessageBuiltinTool):
|
||||
|
||||
def __init__(self, api_key: str) -> None:
|
||||
self.api_key = api_key
|
||||
self.url = "https://api.wolframalpha.com/v2/query"
|
||||
|
@ -287,7 +284,6 @@ class WolframAlphaTool(SingleMessageBuiltinTool):
|
|||
|
||||
|
||||
class CodeInterpreterTool(BaseTool):
|
||||
|
||||
def __init__(self) -> None:
|
||||
ctx = CodeExecutionContext(
|
||||
matplotlib_dump_dir=f"/tmp/{os.environ['USER']}_matplotlib_dump",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue