ai-security-toolkit 1.0.0__tar.gz → 1.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (24) hide show
  1. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/PKG-INFO +6 -1
  2. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/README.md +1 -0
  3. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/run.py +11 -2
  4. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/PKG-INFO +6 -1
  5. ai_security_toolkit-1.1.1/ai_security_toolkit.egg-info/requires.txt +11 -0
  6. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/pyproject.toml +10 -1
  7. ai_security_toolkit-1.0.0/ai_security_toolkit.egg-info/requires.txt +0 -5
  8. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/__init__.py +0 -0
  9. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/__init__.py +0 -0
  10. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/backdoor_trigger_attack.py +0 -0
  11. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/fgsm_mobilenet.py +0 -0
  12. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/label_flip_attack.py +0 -0
  13. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/membership_inference_attack.py +0 -0
  14. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/simulate_inversion.py +0 -0
  15. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/steal_model.py +0 -0
  16. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/modules/train_mnist_model.py +0 -0
  17. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/shared/__init__.py +0 -0
  18. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit/shared/log_utils.py +0 -0
  19. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/SOURCES.txt +0 -0
  20. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/dependency_links.txt +0 -0
  21. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/entry_points.txt +0 -0
  22. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/ai_security_toolkit.egg-info/top_level.txt +0 -0
  23. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/setup.cfg +0 -0
  24. {ai_security_toolkit-1.0.0 → ai_security_toolkit-1.1.1}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ai-security-toolkit
3
- Version: 1.0.0
3
+ Version: 1.1.1
4
4
  Summary: A red-team AI security framework with adversarial attack modules
5
5
  Author: Rishit Goel
6
6
  License: MIT
@@ -11,6 +11,10 @@ Requires-Dist: numpy
11
11
  Requires-Dist: matplotlib
12
12
  Requires-Dist: pandas
13
13
  Requires-Dist: cleverhans
14
+ Provides-Extra: tensorflow
15
+ Requires-Dist: tensorflow; extra == "tensorflow"
16
+ Provides-Extra: cleverhans
17
+ Requires-Dist: cleverhans; extra == "cleverhans"
14
18
  Dynamic: requires-python
15
19
 
16
20
  # 🛡️ AI Security Toolkit
@@ -18,6 +22,7 @@ Dynamic: requires-python
18
22
  [![Made by Rishit Goel 💻](https://img.shields.io/badge/Made%20by-Rishit%20Goel-blueviolet?style=flat-square&logo=github)](https://github.com/rishit03)
19
23
  ![Python](https://img.shields.io/badge/Python-3.8+-blue?logo=python)
20
24
  ![License](https://img.shields.io/github/license/rishit03/ai-security-toolkit?style=flat)
25
+ [![PyPI](https://img.shields.io/pypi/v/ai-security-toolkit?color=brightgreen)](https://pypi.org/project/ai-security-toolkit/)
21
26
  ![GitHub Repo stars](https://img.shields.io/github/stars/rishit03/ai-security-toolkit?style=social)
22
27
  ![GitHub last commit](https://img.shields.io/github/last-commit/rishit03/ai-security-toolkit?color=green)
23
28
 
@@ -3,6 +3,7 @@
3
3
  [![Made by Rishit Goel 💻](https://img.shields.io/badge/Made%20by-Rishit%20Goel-blueviolet?style=flat-square&logo=github)](https://github.com/rishit03)
4
4
  ![Python](https://img.shields.io/badge/Python-3.8+-blue?logo=python)
5
5
  ![License](https://img.shields.io/github/license/rishit03/ai-security-toolkit?style=flat)
6
+ [![PyPI](https://img.shields.io/pypi/v/ai-security-toolkit?color=brightgreen)](https://pypi.org/project/ai-security-toolkit/)
6
7
  ![GitHub Repo stars](https://img.shields.io/github/stars/rishit03/ai-security-toolkit?style=social)
7
8
  ![GitHub last commit](https://img.shields.io/github/last-commit/rishit03/ai-security-toolkit?color=green)
8
9
 
@@ -1,6 +1,11 @@
1
1
  import importlib
2
2
  import sys
3
3
  import os
4
+ import platform
5
+
6
+ if platform.system() == "Windows":
7
+ print("ℹ️ Running on Windows. If you see DLL errors, install:")
8
+ print("👉 https://aka.ms/vs/17/release/vc_redist.x64.exe")
4
9
 
5
10
  # Add project root to PYTHONPATH
6
11
  sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '.')))
@@ -26,8 +31,9 @@ def print_menu():
26
31
  def run_selected_module(choice_idx):
27
32
  try:
28
33
  label = list(available_modules.keys())[choice_idx - 1]
29
- module_name = f"modules.{available_modules[label]}"
34
+ module_name = f"ai_security_toolkit.modules.{available_modules[label]}"
30
35
  print(f"\n🔍 Running: {label} ({module_name})...\n")
36
+
31
37
  mod = importlib.import_module(module_name)
32
38
 
33
39
  if hasattr(mod, "main"):
@@ -36,8 +42,11 @@ def run_selected_module(choice_idx):
36
42
  print("⚠️ No 'main()' found — running file as script...")
37
43
  exec(open(mod.__file__).read())
38
44
 
45
+ except ModuleNotFoundError:
46
+ print(f"❌ Could not find module: {module_name}")
47
+ print("💡 Make sure the package is installed correctly via pip.")
39
48
  except Exception as e:
40
- print(f"❌ Error: {e}")
49
+ print(f"❌ Unexpected error: {e}")
41
50
 
42
51
  def main():
43
52
  while True:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ai-security-toolkit
3
- Version: 1.0.0
3
+ Version: 1.1.1
4
4
  Summary: A red-team AI security framework with adversarial attack modules
5
5
  Author: Rishit Goel
6
6
  License: MIT
@@ -11,6 +11,10 @@ Requires-Dist: numpy
11
11
  Requires-Dist: matplotlib
12
12
  Requires-Dist: pandas
13
13
  Requires-Dist: cleverhans
14
+ Provides-Extra: tensorflow
15
+ Requires-Dist: tensorflow; extra == "tensorflow"
16
+ Provides-Extra: cleverhans
17
+ Requires-Dist: cleverhans; extra == "cleverhans"
14
18
  Dynamic: requires-python
15
19
 
16
20
  # 🛡️ AI Security Toolkit
@@ -18,6 +22,7 @@ Dynamic: requires-python
18
22
  [![Made by Rishit Goel 💻](https://img.shields.io/badge/Made%20by-Rishit%20Goel-blueviolet?style=flat-square&logo=github)](https://github.com/rishit03)
19
23
  ![Python](https://img.shields.io/badge/Python-3.8+-blue?logo=python)
20
24
  ![License](https://img.shields.io/github/license/rishit03/ai-security-toolkit?style=flat)
25
+ [![PyPI](https://img.shields.io/pypi/v/ai-security-toolkit?color=brightgreen)](https://pypi.org/project/ai-security-toolkit/)
21
26
  ![GitHub Repo stars](https://img.shields.io/github/stars/rishit03/ai-security-toolkit?style=social)
22
27
  ![GitHub last commit](https://img.shields.io/github/last-commit/rishit03/ai-security-toolkit?color=green)
23
28
 
@@ -0,0 +1,11 @@
1
+ tensorflow
2
+ numpy
3
+ matplotlib
4
+ pandas
5
+ cleverhans
6
+
7
+ [cleverhans]
8
+ cleverhans
9
+
10
+ [tensorflow]
11
+ tensorflow
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "ai-security-toolkit"
7
- version = "1.0.0"
7
+ version = "1.1.1"
8
8
  description = "A red-team AI security framework with adversarial attack modules"
9
9
  readme = "README.md"
10
10
  authors = [
@@ -23,3 +23,12 @@ dependencies = [
23
23
 
24
24
  [project.scripts]
25
25
  ai-toolkit = "ai_security_toolkit.run:main"
26
+
27
+ [project.optional-dependencies]
28
+ tensorflow = ["tensorflow"]
29
+ cleverhans = ["cleverhans"]
30
+
31
+ [tool.setuptools.packages.find]
32
+ include = ["ai_security_toolkit*"]
33
+
34
+
@@ -1,5 +0,0 @@
1
- tensorflow
2
- numpy
3
- matplotlib
4
- pandas
5
- cleverhans