From da632701424c9149802ca1929acad800092018f1 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Tue, 18 Mar 2025 17:40:23 +0200 Subject: [PATCH] fix(pc): --- agentic_security/agents/operator_crew.py | 6 +++--- agentic_security/integrations/__init__.py | 6 ++++-- agentic_security/primitives/__init__.py | 5 +++-- agentic_security/probe_data/modules/rl_model.py | 3 ++- 4 files changed, 12 insertions(+), 8 deletions(-) diff --git a/agentic_security/agents/operator_crew.py b/agentic_security/agents/operator_crew.py index 72b706a..e836569 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 dc138c1..4b2a360 100644 --- a/agentic_security/integrations/__init__.py +++ b/agentic_security/integrations/__init__.py @@ -5,6 +5,8 @@ from typing import Protocol 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 e2e9424..46e0219 100644 --- a/agentic_security/primitives/__init__.py +++ b/agentic_security/primitives/__init__.py @@ -1,5 +1,6 @@ -from agentic_security.primitives.models import ( - CompletionRequest, # noqa +# noqa +from agentic_security.primitives.models import CompletionRequest # noqa +from agentic_security.primitives.models import ( # 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 002158c..1b5befd 100644 --- a/agentic_security/probe_data/modules/rl_model.py +++ b/agentic_security/probe_data/modules/rl_model.py @@ -121,7 +121,8 @@ class CloudRLPromptSelector(PromptSelectionInterface): current_prompt: str, reward: float, passed_guard: bool, - ) -> None: ... + ) -> None: + ... class QLearningPromptSelector(PromptSelectionInterface):