Merge pull request #202 from ikhanganin/main

Improvements to Code Quality and Bug Fixes

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,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: ...

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,

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