remove api.endpoints imports

This commit is contained in:
Ashwin Bharambe 2024-08-26 11:18:42 -07:00
parent df489261ac
commit c1078a60e7
3 changed files with 8 additions and 8 deletions

View file

@ -9,8 +9,7 @@ from typing import List
from llama_toolchain.agentic_system.meta_reference.safety import ShieldRunnerMixin
from llama_toolchain.inference.api import Message
from llama_toolchain.safety.api.datatypes import ShieldDefinition
from llama_toolchain.safety.api.endpoints import Safety
from llama_toolchain.safety.api import Safety, ShieldDefinition
from .builtin import BaseTool