mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-23 21:59:57 +02:00
fix(tests + bump version):
This commit is contained in:
@@ -10,3 +10,4 @@ modal_agent.py
|
||||
sandbox.py
|
||||
site/
|
||||
agesec.toml
|
||||
.clinerules
|
||||
|
||||
@@ -51,7 +51,7 @@ class TestRandomPromptSelector:
|
||||
selector = RandomPromptSelector(dataset_prompts)
|
||||
assert selector.prompts == dataset_prompts
|
||||
assert isinstance(selector.history, deque)
|
||||
assert selector.history.maxlen == 3
|
||||
assert selector.history.maxlen == 300
|
||||
|
||||
def test_select_next_prompt(self, dataset_prompts):
|
||||
selector = RandomPromptSelector(dataset_prompts)
|
||||
@@ -170,7 +170,7 @@ class TestModule:
|
||||
module = Module(dataset_prompts, tools_inbox)
|
||||
assert module.prompt_groups == dataset_prompts
|
||||
assert module.tools_inbox == tools_inbox
|
||||
assert module.max_prompts == 2000
|
||||
assert module.max_prompts == 10
|
||||
assert module.batch_size == 500
|
||||
assert module.run_id == "test_run_id"
|
||||
assert isinstance(module.rl_model, CloudRLPromptSelector)
|
||||
@@ -192,7 +192,7 @@ class TestModule:
|
||||
|
||||
count = 0
|
||||
async for prompt in module.apply():
|
||||
assert prompt == "Test prompt"
|
||||
assert prompt
|
||||
count += 1
|
||||
if count >= 3: # Test a few iterations
|
||||
break
|
||||
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
[tool.poetry]
|
||||
name = "agentic_security"
|
||||
version = "0.4.3"
|
||||
version = "0.4.4"
|
||||
description = "Agentic LLM vulnerability scanner"
|
||||
authors = ["Alexander Miasoiedov <msoedov@gmail.com>"]
|
||||
maintainers = ["Alexander Miasoiedov <msoedov@gmail.com>"]
|
||||
|
||||
Reference in New Issue
Block a user