mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-02 09:40:42 +00:00
92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
from setuptools import setup, find_packages
|
|
import os
|
|
import platform
|
|
import shutil
|
|
|
|
package_name = "gpt4all"
|
|
|
|
# Define the location of your prebuilt C library files
|
|
SRC_CLIB_DIRECtORY = os.path.join("..", "..", "gpt4all-backend")
|
|
SRC_CLIB_BUILD_DIRECTORY = os.path.join("..", "..", "gpt4all-backend", "build")
|
|
|
|
LIB_NAME = "llmodel"
|
|
|
|
DEST_CLIB_DIRECTORY = os.path.join(package_name, f"{LIB_NAME}_DO_NOT_MODIFY")
|
|
DEST_CLIB_BUILD_DIRECTORY = os.path.join(DEST_CLIB_DIRECTORY, "build")
|
|
|
|
system = platform.system()
|
|
|
|
def get_c_shared_lib_extension():
|
|
|
|
if system == "Darwin":
|
|
return "dylib"
|
|
elif system == "Linux":
|
|
return "so"
|
|
elif system == "Windows":
|
|
return "dll"
|
|
else:
|
|
raise Exception("Operating System not supported")
|
|
|
|
lib_ext = get_c_shared_lib_extension()
|
|
|
|
def copy_prebuilt_C_lib(src_dir, dest_dir, dest_build_dir):
|
|
files_copied = 0
|
|
|
|
if not os.path.exists(dest_dir):
|
|
os.mkdir(dest_dir)
|
|
os.mkdir(dest_build_dir)
|
|
|
|
for dirpath, _, filenames in os.walk(src_dir):
|
|
for item in filenames:
|
|
# copy over header files to dest dir
|
|
s = os.path.join(dirpath, item)
|
|
if item.endswith(".h"):
|
|
d = os.path.join(dest_dir, item)
|
|
shutil.copy2(s, d)
|
|
files_copied += 1
|
|
if item.endswith(lib_ext) or item.endswith('.metal'):
|
|
s = os.path.join(dirpath, item)
|
|
d = os.path.join(dest_build_dir, item)
|
|
shutil.copy2(s, d)
|
|
files_copied += 1
|
|
|
|
return files_copied
|
|
|
|
|
|
# NOTE: You must provide correct path to the prebuilt llmodel C library.
|
|
# Specifically, the llmodel.h and C shared library are needed.
|
|
copy_prebuilt_C_lib(SRC_CLIB_DIRECtORY,
|
|
DEST_CLIB_DIRECTORY,
|
|
DEST_CLIB_BUILD_DIRECTORY)
|
|
|
|
setup(
|
|
name=package_name,
|
|
version="0.3.3",
|
|
description="Python bindings for GPT4All",
|
|
author="Richard Guo",
|
|
author_email="richard@nomic.ai",
|
|
url="https://pypi.org/project/gpt4all/",
|
|
classifiers = [
|
|
"Programming Language :: Python :: 3",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Operating System :: OS Independent",
|
|
],
|
|
python_requires='>=3.8',
|
|
packages=find_packages(),
|
|
install_requires=['requests', 'tqdm'],
|
|
extras_require={
|
|
'dev': [
|
|
'pytest',
|
|
'twine',
|
|
'wheel',
|
|
'setuptools',
|
|
'mkdocs-material',
|
|
'mkautodoc',
|
|
'mkdocstrings[python]',
|
|
'mkdocs-jupyter'
|
|
]
|
|
},
|
|
package_data={'llmodel': [os.path.join(DEST_CLIB_DIRECTORY, "*")]},
|
|
include_package_data=True
|
|
)
|