Skip to content

Commit 4da7cfa

Browse files
authored
Create setup.py
1 parent bac9994 commit 4da7cfa

1 file changed

Lines changed: 69 additions & 0 deletions

File tree

setup.py

Lines changed: 69 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
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

Comments
 (0)