Skip to content

Commit

Permalink
feat(Bump release):
Browse files Browse the repository at this point in the history
  • Loading branch information
msoedov committed Sep 2, 2024
1 parent 9de2c55 commit 8b77239
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 8 deletions.
1 change: 0 additions & 1 deletion Readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@
- LLM API integration and stress testing 🛠️
- Wide range of fuzzing and attack techniques 🌀


Note: Please be aware that Agentic Security is designed as a safety scanner tool and not a foolproof solution. It cannot guarantee complete protection against all possible threats.

## 📦 Installation
Expand Down
12 changes: 6 additions & 6 deletions agentic_security/probe_actor/fuzzer.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
import os
import asyncio
from typing import List, Dict, AsyncGenerator
from typing import AsyncGenerator

import httpx
import numpy as np
import pandas as pd
from agentic_security.probe_actor.refusal import refusal_heuristic
from agentic_security.probe_data.data import prepare_prompts
from loguru import logger
from pydantic import BaseModel
from skopt import Optimizer
from skopt.space import Real

from agentic_security.probe_actor.refusal import refusal_heuristic
from agentic_security.probe_data.data import prepare_prompts

IS_VERCEL = os.getenv("IS_VERCEL", "f") == "t"


Expand All @@ -35,7 +35,7 @@ def status_msg(cls, msg: str) -> str:
).model_dump_json()


async def prompt_iter(prompts: List[str] | AsyncGenerator) -> AsyncGenerator[str, None]:
async def prompt_iter(prompts: list[str] | AsyncGenerator) -> AsyncGenerator[str, None]:
if isinstance(prompts, list):
for p in prompts:
yield p
Expand All @@ -47,7 +47,7 @@ async def prompt_iter(prompts: List[str] | AsyncGenerator) -> AsyncGenerator[str
async def perform_scan(
request_factory,
max_budget: int,
datasets: List[Dict[str, str]] = [],
datasets: list[dict[str, str]] = [],
tools_inbox=None,
optimize=False,
) -> AsyncGenerator[str, None]:
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "agentic_security"
version = "0.1.12"
version = "0.2.0"
description = "Agentic LLM vulnerability scanner"
authors = ["Alexander Miasoiedov <[email protected]>"]
maintainers = ["Alexander Miasoiedov <[email protected]>"]
Expand Down

0 comments on commit 8b77239

Please sign in to comment.