-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
42 lines (40 loc) · 1.37 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
from setuptools import setup
with open('promptwizard/doc/doc.md', 'r', encoding='utf-8') as f:
long_description = f.read()
setup(
name='promptwizard',
version='0.0.24',
packages=['promptwizard', 'promptwizard.evals', 'promptwizard.cost', 'promptwizard.approximate_cost', 'promptwizard.validation_yaml', 'promptwizard.prompt_generation', 'promptwizard.openai_calls'],
python_requires='>=3.8',
long_description=long_description,
long_description_content_type='text/markdown',
install_requires=[
'openai==1.2.2',
'PyYAML>=5.4',
'matplotlib',
'numpy',
'marshmallow>=3.20.1',
'tiktoken>=0.4.0',
'tenacity>=8.2.3',
'zipp',
'six',
'aiohttp',
'urllib3',
'certifi',
'tqdm',
'python-dotenv',
'js2py',
'scikit-learn',
'cohere==4.34',
'httpx'
],
entry_points={
'console_scripts': [
'promptwizard=promptwizard.main:main',
],
},
package_data={'': ['doc/doc.md']},
include_package_data=True,
description='Prompt Wizard is a package for evaluating custom prompts using various evaluation methods. It allows you to provide your own prompts or generate them automatically and then obtain the results in a JSON file. You can also use the library for your Python projects.',
license='MIT',
)