fix(tests + bump version):

This commit is contained in:
Alexander Myasoedov
2025-02-08 10:41:43 +02:00
parent eb27f7bbaa
commit 693c5743c0
3 changed files with 5 additions and 4 deletions
+1
View File
@@ -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
View File
@@ -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>"]