pyproject.toml

[tool.poetry]
name = "metAIsploit-assistant"
version = "0.1.0"
description = "Use LLMs and open source models to generate metaploit modules from published CVEs"
authors = ["Atilla"]
license = "BSD 2-Clause License"
readme = "README.md"
packages = [{include = "metAIsploit_assistant"}]

[tool.poetry.dependencies]
python = "^3.11"
gymnasium = "^0.29.1"
poetry-exec-plugin = "^1.0.1"
pydantic = "^2.7.1"
pyyaml = "^6.0.1"
huggingface-hub = "^0.23.0"
langchain = "^0.3.26"
gpt4all = "^2.8.2"
beautifulsoup4 = "^4.12.3"
transformers = "^4.41.2"
tokenizers = "^0.19.1"
accelerate = "^0.30.1"
torch = "^2.3.1"
pymetasploit3 = "^1.0.5"
langchain-community = "^0.3.27"

[tool.poetry.group.dev.dependencies]
black = "^24.4.2"


[tool.poetry-exec-plugin.commands]
lint = "black ."

[tool.poetry.scripts]
init = "metAIsploit_assistant.utilities.models:get_models_inventory"
chat = "metAIsploit_assistant.actions.chat:perform_chat"
demo = "examples.explicit_poc:main"
prompt-demo = "examples.explicit_poc:poetry_run_prompt_demo"
generate-prompts = "metAIsploit_assistant.utilities.gatherers:get_full_prompt_list_for_msf"
metai = "metAIsploit_assistant.utilities.recon_pipeline:main"

[build-system]
requires = ["poetry-core>=1.9.0,<2.0.0"]
build-backend = "poetry.core.masonry.api"