diff --git a/llama_stack/providers/remote/safety/nvidia/nvidia.py b/llama_stack/providers/remote/safety/nvidia/nvidia.py index ce9b2953b..7988d83c6 100644 --- a/llama_stack/providers/remote/safety/nvidia/nvidia.py +++ b/llama_stack/providers/remote/safety/nvidia/nvidia.py @@ -10,11 +10,9 @@ from typing import Any, Dict, List import requests from llama_stack.apis.inference import Message -from llama_stack.apis.safety import (RunShieldResponse, Safety, - SafetyViolation, ViolationLevel) +from llama_stack.apis.safety import RunShieldResponse, Safety, SafetyViolation, ViolationLevel from llama_stack.apis.shields import Shield -from llama_stack.distribution.library_client import \ - convert_pydantic_to_json_value +from llama_stack.distribution.library_client import convert_pydantic_to_json_value from llama_stack.providers.datatypes import ShieldsProtocolPrivate from .config import NVIDIASafetyConfig