ps-fuzz/setup.py at 7f1ed916a2e5764385ebc68124472f823fabb4a8 · prompt-security/ps-fuzz · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import os
from setuptools import setup, find_packages

with open('README.md', 'r', encoding='utf-8') as fh:
    long_description = fh.read()

setup(
    name="prompt-security-fuzzer",
    version=os.getenv('PKG_VERSION', '0.0.1'),
    author="Prompt Security",
    author_email="support@prompt.security",
    description="LLM and System Prompt vulnerability scanner tool",
    long_description=long_description,
    long_description_content_type='text/markdown',
    url="https://github.com/prompt-security/ps-fuzz",
    packages=find_packages(),
    package_data={
        'ps_fuzz': ['attack_data/*'],
    },
    classifiers=[
        "Development Status :: 5 - Production/Stable",
        "Intended Audience :: Developers",
        "Topic :: Software Development :: Quality Assurance",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11"
    ],
    python_requires='>=3.9',
    install_requires=[
        "httpx>=0.24.0,<0.25.0",
        "openai==1.6.1",
        "langchain>=0.3.0,<0.4.0",
        "langchain-community>=0.3.0,<0.4.0",
        "langchain-core>=0.3.81,<0.4.0",
        "argparse==1.4.0",
        "python-dotenv==1.0.0",
        "tqdm>=4.66.3",
        "colorama==0.4.6",
        "prettytable==3.10.0",
        "pandas==2.2.2",
        "inquirer==3.2.4",
        "prompt-toolkit==3.0.43",
        "fastparquet==2024.2.0",
        "chromadb>=0.4.0",
        "tiktoken>=0.11.0"
    ],
    extras_require={
        "dev": ["pytest==7.4.4"]
    },
    entry_points={
        'console_scripts': [
            'prompt-security-fuzzer=ps_fuzz.cli:main',
        ],
    },
    license="MIT",
)