-
Notifications
You must be signed in to change notification settings - Fork 13
/
setup.py
49 lines (45 loc) · 1.38 KB
/
setup.py
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
from setuptools import setup, find_packages
with open("README.md", 'r', encoding='utf') as f:
long_description = f.read()
setup(
name='topicgpt',
version='0.0.5',
packages=find_packages(where='src'),
package_dir={'': 'src'},
install_requires=[
'gensim',
'hdbscan',
'nltk',
'numpy',
'openai>=1.0.0',
'pandas',
'plotly',
'regex',
'scikit-learn',
'seaborn',
'sentence-transformers',
'tiktoken',
'tokenizers',
'tqdm',
'umap-learn',
'umap-learn[plot]'
],
include_package_data=True,
# Additional metadata
author='Arik Reuter',
author_email='[email protected]',
description='A package for integrating LLMs like GPT-3.5 and GPT-4 into topic modelling',
long_description=long_description,
long_description_content_type="text/markdown",
license="MIT",
keywords=['Topic Modelling', 'GPT', 'LLM', 'OpenAI', 'Retrieval Augmented Generation', 'Chat-GPT', 'GPT-3', 'GPT-4'],
classifiers=[
"Development Status :: 3 - Alpha",
'Intended Audience :: Science/Research',
"Intended Audience :: Developers",
"Programming Language :: Python :: 3.11",
"Operating System :: Unix",
"Operating System :: MacOS :: MacOS X",
"Operating System :: Microsoft :: Windows",
]
)