fix(tests + bump version):

This commit is contained in:
Alexander Myasoedov committed 2025-02-08 10:41:43 +02:00
1 parent eb27f7bbaa
commit 693c5743c0
3 files changed
+5 -4

No files matched your search

+1
View File
@@ -10,3 +10,4 @@ modal_agent.py
sandbox.py sandbox.py
site/ site/
agesec.toml agesec.toml
.clinerules
@@ -51,7 +51,7 @@ class TestRandomPromptSelector:
selector = RandomPromptSelector(dataset_prompts) selector = RandomPromptSelector(dataset_prompts)
assert selector.prompts == dataset_prompts assert selector.prompts == dataset_prompts
assert isinstance(selector.history, deque) assert isinstance(selector.history, deque)
assert selector.history.maxlen == 3 assert selector.history.maxlen == 300
def test_select_next_prompt(self, dataset_prompts): def test_select_next_prompt(self, dataset_prompts):
selector = RandomPromptSelector(dataset_prompts) selector = RandomPromptSelector(dataset_prompts)
@@ -170,7 +170,7 @@ class TestModule:
module = Module(dataset_prompts, tools_inbox) module = Module(dataset_prompts, tools_inbox)
assert module.prompt_groups == dataset_prompts assert module.prompt_groups == dataset_prompts
assert module.tools_inbox == tools_inbox assert module.tools_inbox == tools_inbox
assert module.max_prompts == 2000 assert module.max_prompts == 10
assert module.batch_size == 500 assert module.batch_size == 500
assert module.run_id == "test_run_id" assert module.run_id == "test_run_id"
assert isinstance(module.rl_model, CloudRLPromptSelector) assert isinstance(module.rl_model, CloudRLPromptSelector)
@@ -192,7 +192,7 @@ class TestModule:
count = 0 count = 0
async for prompt in module.apply(): async for prompt in module.apply():
assert prompt == "Test prompt" assert prompt
count += 1 count += 1
if count >= 3: # Test a few iterations if count >= 3: # Test a few iterations
break break
+1 -1
View File
@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "agentic_security" name = "agentic_security"
version = "0.4.3" version = "0.4.4"
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>"]