Refactor code formatting in lightrag_api_openai_compatible_demo.py
This commit is contained in:
31
setup.py
31
setup.py
@@ -1,6 +1,7 @@
|
||||
import setuptools
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
# Reading the long description from README.md
|
||||
def read_long_description():
|
||||
try:
|
||||
@@ -8,6 +9,7 @@ def read_long_description():
|
||||
except FileNotFoundError:
|
||||
return "A description of LightRAG is currently unavailable."
|
||||
|
||||
|
||||
# Retrieving metadata from __init__.py
|
||||
def retrieve_metadata():
|
||||
vars2find = ["__author__", "__version__", "__url__"]
|
||||
@@ -17,18 +19,26 @@ def retrieve_metadata():
|
||||
for line in f.readlines():
|
||||
for v in vars2find:
|
||||
if line.startswith(v):
|
||||
line = line.replace(" ", "").replace('"', "").replace("'", "").strip()
|
||||
line = (
|
||||
line.replace(" ", "")
|
||||
.replace('"', "")
|
||||
.replace("'", "")
|
||||
.strip()
|
||||
)
|
||||
vars2readme[v] = line.split("=")[1]
|
||||
except FileNotFoundError:
|
||||
raise FileNotFoundError("Metadata file './lightrag/__init__.py' not found.")
|
||||
|
||||
|
||||
# Checking if all required variables are found
|
||||
missing_vars = [v for v in vars2find if v not in vars2readme]
|
||||
if missing_vars:
|
||||
raise ValueError(f"Missing required metadata variables in __init__.py: {missing_vars}")
|
||||
|
||||
raise ValueError(
|
||||
f"Missing required metadata variables in __init__.py: {missing_vars}"
|
||||
)
|
||||
|
||||
return vars2readme
|
||||
|
||||
|
||||
# Reading dependencies from requirements.txt
|
||||
def read_requirements():
|
||||
deps = []
|
||||
@@ -36,9 +46,12 @@ def read_requirements():
|
||||
with open("./requirements.txt") as f:
|
||||
deps = [line.strip() for line in f if line.strip()]
|
||||
except FileNotFoundError:
|
||||
print("Warning: 'requirements.txt' not found. No dependencies will be installed.")
|
||||
print(
|
||||
"Warning: 'requirements.txt' not found. No dependencies will be installed."
|
||||
)
|
||||
return deps
|
||||
|
||||
|
||||
metadata = retrieve_metadata()
|
||||
long_description = read_long_description()
|
||||
requirements = read_requirements()
|
||||
@@ -51,7 +64,9 @@ setuptools.setup(
|
||||
description="LightRAG: Simple and Fast Retrieval-Augmented Generation",
|
||||
long_description=long_description,
|
||||
long_description_content_type="text/markdown",
|
||||
packages=setuptools.find_packages(exclude=("tests*", "docs*")), # Automatically find packages
|
||||
packages=setuptools.find_packages(
|
||||
exclude=("tests*", "docs*")
|
||||
), # Automatically find packages
|
||||
classifiers=[
|
||||
"Development Status :: 4 - Beta",
|
||||
"Programming Language :: Python :: 3",
|
||||
@@ -66,6 +81,8 @@ setuptools.setup(
|
||||
project_urls={ # Additional project metadata
|
||||
"Documentation": metadata.get("__url__", ""),
|
||||
"Source": metadata.get("__url__", ""),
|
||||
"Tracker": f"{metadata.get('__url__', '')}/issues" if metadata.get("__url__") else ""
|
||||
"Tracker": f"{metadata.get('__url__', '')}/issues"
|
||||
if metadata.get("__url__")
|
||||
else "",
|
||||
},
|
||||
)
|
||||
|
Reference in New Issue
Block a user