mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-24 06:09:55 +02:00
Auto-fix: formatting, bug fixes, import sorting, and type check improvements
Signed-off-by: ikhanganin <ismailmac39@gmail.com>
This commit is contained in:
@@ -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())
|
||||
|
||||
@@ -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: ...
|
||||
|
||||
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user