|
| 1 | +from setuptools import setup, find_packages |
| 2 | +import os |
| 3 | + |
| 4 | +# Read README |
| 5 | +with open("README.md", "r", encoding="utf-8") as fh: |
| 6 | + long_description = fh.read() |
| 7 | + |
| 8 | +# Read requirements |
| 9 | +with open("requirements.txt", "r", encoding="utf-8") as fh: |
| 10 | + requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")] |
| 11 | + |
| 12 | +setup( |
| 13 | + name="truthprobe", |
| 14 | + version="4.0.0", |
| 15 | + author="Dafydd Napier", |
| 16 | + author_email="napiersnotes@github.com", |
| 17 | + description="A lightweight, model-agnostic deception detector for LLMs", |
| 18 | + long_description=long_description, |
| 19 | + long_description_content_type="text/markdown", |
| 20 | + url="https://github.com/Napiersnotes/TruthProbe", |
| 21 | + project_urls={ |
| 22 | + "Bug Tracker": "https://github.com/Napiersnotes/TruthProbe/issues", |
| 23 | + "Documentation": "https://github.com/Napiersnotes/TruthProbe#readme", |
| 24 | + }, |
| 25 | + classifiers=[ |
| 26 | + "Development Status :: 4 - Beta", |
| 27 | + "Intended Audience :: Science/Research", |
| 28 | + "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| 29 | + "License :: OSI Approved :: MIT License", |
| 30 | + "Programming Language :: Python :: 3", |
| 31 | + "Programming Language :: Python :: 3.9", |
| 32 | + "Programming Language :: Python :: 3.10", |
| 33 | + "Programming Language :: Python :: 3.11", |
| 34 | + "Operating System :: OS Independent", |
| 35 | + ], |
| 36 | + packages=find_packages(include=['src', 'src.*']), |
| 37 | + python_requires=">=3.9", |
| 38 | + install_requires=requirements, |
| 39 | + extras_require={ |
| 40 | + "dev": [ |
| 41 | + "pytest>=7.0.0", |
| 42 | + "pytest-cov>=4.0.0", |
| 43 | + "black>=23.0.0", |
| 44 | + "flake8>=6.0.0", |
| 45 | + "mypy>=1.0.0", |
| 46 | + ], |
| 47 | + "dashboard": [ |
| 48 | + "dash>=2.9.0", |
| 49 | + "dash-bootstrap-components>=1.4.0", |
| 50 | + "plotly>=5.14.0", |
| 51 | + ], |
| 52 | + "full": [ |
| 53 | + "torch>=2.0.0", |
| 54 | + "transformers>=4.30.0", |
| 55 | + "sentence-transformers>=2.2.0", |
| 56 | + "spacy>=3.5.0", |
| 57 | + "scikit-learn>=1.3.0", |
| 58 | + ] |
| 59 | + }, |
| 60 | + entry_points={ |
| 61 | + "console_scripts": [ |
| 62 | + "truthprobe=src.truthprobe_v3:main", |
| 63 | + ], |
| 64 | + }, |
| 65 | + include_package_data=True, |
| 66 | + package_data={ |
| 67 | + "": ["*.json", "*.txt", "*.md"], |
| 68 | + }, |
| 69 | +) |
0 commit comments