Auto-fix: formatting, bug fixes, import sorting, and type check improvements

Signed-off-by: ikhanganin <ismailmac39@gmail.com>
This commit is contained in:
Ismail mach
2025-03-18 15:11:59 +00:00
committed by ikhanganin
parent b7fef85750
commit d3ccea76b6
4 changed files with 8 additions and 11 deletions
+3 -3
View File
@@ -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())
+2 -4
View File
@@ -5,8 +5,6 @@ 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: ...
+2 -2
View File
@@ -1,5 +1,5 @@
from agentic_security.primitives.models import ( # noqa
CompletionRequest,
from agentic_security.primitives.models import (
CompletionRequest, # noqa
FileProbeResponse,
LLMInfo,
Message,
@@ -121,8 +121,7 @@ class CloudRLPromptSelector(PromptSelectionInterface):
current_prompt: str,
reward: float,
passed_guard: bool,
) -> None:
...
) -> None: ...
class QLearningPromptSelector(PromptSelectionInterface):