diff --git a/agentic_security/agents/operator_crew.py b/agentic_security/agents/operator_crew.py index e836569..72b706a 100644 --- a/agentic_security/agents/operator_crew.py +++ b/agentic_security/agents/operator_crew.py @@ -246,9 +246,9 @@ async def run_crew(): os.environ["OPENAI_API_KEY"] = os.environ.get( "DEEPSEEK_API_KEY", "" ) # CrewAI uses OPENAI_API_KEY -os.environ[ - "OPENAI_MODEL_NAME" -] = "deepseek:chat" # Specify DeepSeek model (adjust if needed) +os.environ["OPENAI_MODEL_NAME"] = ( + "deepseek:chat" # Specify DeepSeek model (adjust if needed) +) if __name__ == "__main__": asyncio.run(run_crew()) diff --git a/agentic_security/integrations/__init__.py b/agentic_security/integrations/__init__.py index 4b2a360..dc138c1 100644 --- a/agentic_security/integrations/__init__.py +++ b/agentic_security/integrations/__init__.py @@ -5,8 +5,6 @@ class IntegrationProto(Protocol): def __init__( self, prompt_groups: list, tools_inbox: asyncio.Queue, opts: dict = {} - ): - ... + ): ... - async def apply(self) -> list: - ... + async def apply(self) -> list: ... diff --git a/agentic_security/primitives/__init__.py b/agentic_security/primitives/__init__.py index 1098fa1..e2e9424 100644 --- a/agentic_security/primitives/__init__.py +++ b/agentic_security/primitives/__init__.py @@ -1,5 +1,5 @@ -from agentic_security.primitives.models import ( # noqa - CompletionRequest, +from agentic_security.primitives.models import ( + CompletionRequest, # noqa FileProbeResponse, LLMInfo, Message, diff --git a/agentic_security/probe_data/modules/rl_model.py b/agentic_security/probe_data/modules/rl_model.py index 1b5befd..002158c 100644 --- a/agentic_security/probe_data/modules/rl_model.py +++ b/agentic_security/probe_data/modules/rl_model.py @@ -121,8 +121,7 @@ def update_rewards( current_prompt: str, reward: float, passed_guard: bool, - ) -> None: - ... + ) -> None: ... class QLearningPromptSelector(PromptSelectionInterface):