add setup

Author: shell-nlp

gpt_server/model_worker/base/__init__.py

@@ -0,0 +1,27 @@
+from typing import Tuple
+
+__version__ = "0.3.2"
+short_version = __version__
+
+
+def parse_version_info(version_str: str) -> Tuple:
+    """Parse version from a string.
+
+    Args:
+        version_str (str): A string represents a version info.
+
+    Returns:
+        tuple: A sequence of integer and string represents version.
+    """
+    _version_info = []
+    for x in version_str.split("."):
+        if x.isdigit():
+            _version_info.append(int(x))
+        elif x.find("rc") != -1:
+            patch_version = x.split("rc")
+            _version_info.append(int(patch_version[0]))
+            _version_info.append(f"rc{patch_version[1]}")
+    return tuple(_version_info)
+
+
+version_info = parse_version_info(__version__)

gpt_server/openai_api_protocol/__init__.py

@@ -0,0 +1,58 @@
+import os
+import sys
+import subprocess
+from setuptools import setup, find_packages
+from setuptools.command.install import install
+
+
+pwd = os.path.dirname(__file__)
+version_file = "gpt_server/version.py"
+
+
+# 自定义安装类
+class CustomInstallCommand(install):
+    """自定义安装命令类，用于在安装过程中执行额外的脚本"""
+
+    def run(self):
+        # 调用父类的 run 方法
+        install.run(self)
+
+        # 运行 Bash 脚本
+        script_path = os.path.join(os.path.dirname(__file__), "install.sh")
+        if os.path.exists(script_path):
+            print("Running install_script.sh...")
+            try:
+                subprocess.check_call(["/bin/bash", script_path])
+            except subprocess.CalledProcessError as e:
+                print(f"Error executing script {script_path}: {e}")
+            sys.exit(1)
+        else:
+            print(f"Script {script_path} not found!")
+
+
+def readme():
+    with open(os.path.join(pwd, "README.md"), encoding="utf-8") as f:
+        content = f.read()
+    return content
+
+
+def get_version():
+    with open(os.path.join(pwd, version_file), "r") as f:
+        exec(compile(f.read(), version_file, "exec"))
+    return locals()["__version__"]
+
+
+setup(
+    name="gpt_server",
+    version=get_version(),
+    description="gpt_server是一个用于生产级部署LLMs或Embedding的开源框架。",
+    long_description=readme(),
+    long_description_content_type="text/markdown",
+    author="Yu Liu",
+    author_email="[email protected]",
+    packages=find_packages(exclude=()),
+    # ... 其他 setup 参数 ...
+    cmdclass={
+        "install": CustomInstallCommand,  # 关联自定义安装类
+    },
+)