formatting

This commit is contained in:
Dalton Flanagan 2024-08-14 14:22:25 -04:00
parent 94dfa293a6
commit b6ccaf1778
33 changed files with 110 additions and 97 deletions

View file

@ -246,7 +246,6 @@ class AgentInstance(ShieldRunnerMixin):
await self.run_shields(messages, shields)
except SafetyException as e:
yield AgenticSystemTurnResponseStreamChunk(
event=AgenticSystemTurnResponseEvent(
payload=AgenticSystemTurnResponseStepCompletePayload(

View file

@ -23,7 +23,6 @@ class SafetyException(Exception): # noqa: N818
class ShieldRunnerMixin:
def __init__(
self,
safety_api: Safety,

View file

@ -11,7 +11,6 @@ from llama_toolchain.inference.api import Message
class BaseTool(ABC):
@abstractmethod
def get_name(self) -> str:
raise NotImplementedError

View file

@ -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",