# -*- coding: utf-8 -*-

from setuptools import setup, find_packages

setup(
    name='ai-0x02',
    version='0.1',
    # 你要安装的包，通过 setuptools.find_packages 找到当前目录下有哪些包
    packages=find_packages(),
    install_requires=[
        'transformers==4.36.2',
        'torch==2.1.2',
        'sentencepiece==0.1.99',
        'pymilvus==2.3.4',
        'pymysql==1.0.2',
        'sentence_transformers==2.2.2',
        'tabulate==0.9.0',  # 控制台格式化工具（可选）
        'streamlit==1.30.0',
        'plotly==5.18.0',
    ],
    entry_points={
        'console_scripts': [
            'ai-0x02 = startup:main',  # 定义命令行工具入口点，将调用 startup.py 中的 main 函数
        ],
    },
    author='Mao Si yu',
    author_email='mao_siyu@qq.com',
    description='有事儿请留言',
    url='http://www.dev-share.top',

)