feat(Bump release):

This commit is contained in:
Alexander Myasoedov committed 2024-09-02 17:25:54 +03:00
1 parent 9de2c55474
commit 8b77239666
3 files changed
+7 -8

No files matched your search

-1
View File
@@ -26,7 +26,6 @@
- LLM API integration and stress testing 🛠️ - LLM API integration and stress testing 🛠️
- Wide range of fuzzing and attack techniques 🌀 - 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. 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 ## 📦 Installation
+6 -6
View File
@@ -1,17 +1,17 @@
import os import os
import asyncio from typing import AsyncGenerator
from typing import List, Dict, AsyncGenerator
import httpx import httpx
import numpy as np import numpy as np
import pandas as pd 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 loguru import logger
from pydantic import BaseModel from pydantic import BaseModel
from skopt import Optimizer from skopt import Optimizer
from skopt.space import Real 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" IS_VERCEL = os.getenv("IS_VERCEL", "f") == "t"
@@ -35,7 +35,7 @@ class ScanResult(BaseModel):
).model_dump_json() ).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): if isinstance(prompts, list):
for p in prompts: for p in prompts:
yield p yield p
@@ -47,7 +47,7 @@ async def prompt_iter(prompts: List[str] | AsyncGenerator) -> AsyncGenerator[str
async def perform_scan( async def perform_scan(
request_factory, request_factory,
max_budget: int, max_budget: int,
datasets: List[Dict[str, str]] = [], datasets: list[dict[str, str]] = [],
tools_inbox=None, tools_inbox=None,
optimize=False, optimize=False,
) -> AsyncGenerator[str, None]: ) -> AsyncGenerator[str, None]:
+1 -1
View File
@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "agentic_security" name = "agentic_security"
version = "0.1.12" version = "0.2.0"
description = "Agentic LLM vulnerability scanner" description = "Agentic LLM vulnerability scanner"
authors = ["Alexander Miasoiedov <msoedov@gmail.com>"] authors = ["Alexander Miasoiedov <msoedov@gmail.com>"]
maintainers = ["Alexander Miasoiedov <msoedov@gmail.com>"] maintainers = ["Alexander Miasoiedov <msoedov@gmail.com>"]