46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import os
|
|
from setuptools import find_packages, setup
|
|
|
|
extras = {
|
|
"llama-cpp": ["llama-cpp-python"],
|
|
"transformers": ["torch", "transformers"],
|
|
}
|
|
|
|
def package_files(directory):
|
|
paths = []
|
|
for path, directories, filenames in os.walk(directory):
|
|
for filename in filenames:
|
|
paths.append(os.path.join("..", path, filename))
|
|
return paths
|
|
|
|
next_static_files = package_files("sqlcoder/static")
|
|
|
|
setup(
|
|
name="sqlcoder",
|
|
packages=find_packages(),
|
|
package_data={"sqlcoder": next_static_files},
|
|
version="0.2.1",
|
|
description="SQLCoder is a large language model for converting text questions in SQL queries.",
|
|
author="Defog, Inc",
|
|
license="Apache-2",
|
|
install_requires=[
|
|
"psycopg2-binary>=2.9.5",
|
|
"fastapi",
|
|
"uvicorn",
|
|
"defog",
|
|
"pandas",
|
|
"sqlparse",
|
|
"argparse",
|
|
"huggingface-hub",
|
|
],
|
|
entry_points={
|
|
"console_scripts": [
|
|
"sqlcoder=sqlcoder.cli:main",
|
|
],
|
|
},
|
|
author_email="founders@defog.ai",
|
|
url="https://github.com/defog-ai/sqlcoder",
|
|
long_description="SQLCoder is a large language model for converting text questions in SQL queries.",
|
|
long_description_content_type="text/markdown",
|
|
extras_require=extras,
|
|
) |