from setuptools import setup, find_packages

from setuptools import setup, find_packages

setup(name='prompt-lib',
      version='0.1',
      description="Library for running few-shot inference on large language models",
      packages=find_packages(),
      install_requires=[
          'openai==0.27.2',
          'pandas',
          'tqdm==4.49.0',
          'wandb==0.13.2',
          'PyYAML'
      ],
      license='MIT',
      long_description=open('README.md').read(),
)