fix(pc):

Author: msoedov

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())

agentic_security/integrations/__init__.py

@@ -5,6 +5,8 @@
 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:
+        ...

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,

agentic_security/probe_data/modules/rl_model.py

@@ -121,7 +121,8 @@ def update_rewards(
         current_prompt: str,
         reward: float,
         passed_guard: bool,
-    ) -> None: ...
+    ) -> None:
+        ...
 
 
 class QLearningPromptSelector(PromptSelectionInterface):