pyproject.toml

[build-system]
requires = ["setuptools>=68.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "net-gpt"
version = "1.0.0"
description = "Net-GPT: A LLM-Empowered Man-in-the-Middle Chatbot for Unmanned Aerial Vehicle"
readme = "README.md"
license = {text = "MIT"}
requires-python = ">=3.10"
authors = [
    {name = "Brett Piggott", email = "bpiggott@oakland.edu"},
    {name = "Siddhant Patil"},
    {name = "Guohuan Feng"},
    {name = "Ibrahim Odat", email = "ibrahimodat@oakland.edu"},
    {name = "Rajdeep Mukherjee"},
    {name = "Balakrishnan Dharmalingam"},
    {name = "Anyi Liu", email = "anyiliu@oakland.edu"},
]
keywords = [
    "LLM", "cybersecurity", "UAV", "man-in-the-middle",
    "QLoRA", "fine-tuning", "edge-computing", "network-security",
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Security",
]

dependencies = [
    "torch>=2.0.0",
    "transformers>=4.31.0",
    "datasets>=2.14.0",
    "accelerate>=0.21.0",
    "peft>=0.4.0",
    "bitsandbytes>=0.40.0",
    "trl>=0.5.0",
    "scapy>=2.5.0",
    "pandas>=2.0.0",
    "numpy>=1.24.0",
    "scikit-learn>=1.3.0",
    "pyyaml>=6.0",
    "tqdm>=4.65.0",
    "rich>=13.0.0",
    "tabulate>=0.9.0",
]

[project.optional-dependencies]
viz = ["matplotlib>=3.7.0", "seaborn>=0.12.0"]
dev = ["flake8", "pytest", "black", "isort"]

[project.urls]
Homepage = "https://github.com/OdatSec/Net-GPT"
Paper = "https://doi.org/10.1145/3583740.3626809"
Issues = "https://github.com/OdatSec/Net-GPT/issues"

[tool.setuptools.packages.find]
where = ["."]
include = ["src*"]

[tool.black]
line-length = 100
target-version = ["py310"]

[tool.isort]
profile = "black"
line_length = 100