diff --git a/pyproject.toml b/pyproject.toml index 50dd7a4..38d393e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ dependencies = [ "python-dotenv", "pytest-asyncio", "pytest", - "attackmate @ git+https://github.com/thorinaboenke/attackmate.git@_api_refactor", + "attackmate @ git+https://github.com/ait-testbed/attackmate.git@development", "pytest-asyncio", "mypy>=1.19.1", "pydantic-settings>=2.13.1", diff --git a/src/attackmate_api_server/routers/playbooks.py b/src/attackmate_api_server/routers/playbooks.py index a825384..90d8757 100644 --- a/src/attackmate_api_server/routers/playbooks.py +++ b/src/attackmate_api_server/routers/playbooks.py @@ -10,7 +10,7 @@ from attackmate_api_server.schemas import PlaybookResponseModel from attackmate_api_server.utils import varstore_to_state_model from attackmate_api_server.log_utils import instance_logging -from attackmate_api_server.state import attackmate_config +from attackmate_api_server.state import get_attackmate_config from attackmate_api_server.config import settings router = APIRouter(prefix='/playbooks', tags=['Playbooks']) @@ -71,7 +71,9 @@ async def execute_playbook_from_yaml( raise ValueError('Received empty or invalid playbook YAML content.') playbook = Playbook.model_validate(playbook_dict) logger.info(f'Creating transient AttackMate instance, ID: {instance_id}') - am_instance = AttackMate(playbook=playbook, config=attackmate_config, varstore=None) + am_config = get_attackmate_config() + am_instance = AttackMate(playbook=playbook, config=am_config, + varstore=None, is_api_instance=True) return_code = await am_instance.main() final_state = varstore_to_state_model(am_instance.varstore) logger.info(f'Transient playbook execution finished. Return code: {return_code}')