From 8b7723966626a3b36a0a3e1cc7b88ca83350efae Mon Sep 17 00:00:00 2001
From: Alexander Myasoedov <msoedov@gmail.com>
Date: Mon, 2 Sep 2024 17:25:54 +0300
Subject: [PATCH] feat(Bump release):

---
 Readme.md                              |  1 -
 agentic_security/probe_actor/fuzzer.py | 12 ++++++------
 pyproject.toml                         |  2 +-
 3 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/Readme.md b/Readme.md
index 31b30d4..7c461ee 100644
--- a/Readme.md
+++ b/Readme.md
@@ -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
diff --git a/agentic_security/probe_actor/fuzzer.py b/agentic_security/probe_actor/fuzzer.py
index 1695b26..8fcc2de 100644
--- a/agentic_security/probe_actor/fuzzer.py
+++ b/agentic_security/probe_actor/fuzzer.py
@@ -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"
 
 
@@ -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
@@ -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]:
diff --git a/pyproject.toml b/pyproject.toml
index 6f8e014..37037f4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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 <msoedov@gmail.com>"]
 maintainers = ["Alexander Miasoiedov <msoedov@gmail.com>"]