FilesExpand file tree

 main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

113 lines (101 loc) · 3.73 KB

 main

/

setup.py

Top

File metadata and controls

• Code
• Blame

113 lines (101 loc) · 3.73 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

import setuptools

from pathlib import Path

# Reading the long description from README.md

def read_long_description():

try:

return Path("README.md").read_text(encoding="utf-8")

except FileNotFoundError:

return "A description of RAGAnything is currently unavailable."

# Retrieving metadata from __init__.py

def retrieve_metadata():

vars2find = ["__author__", "__version__", "__url__"]

vars2readme = {}

try:

with open("./raganything/__init__.py") as f:

for line in f.readlines():

for v in vars2find:

if line.startswith(v):

line = (

line.replace(" ", "")

.replace('"', "")

.replace("'", "")

.strip()

)

vars2readme[v] = line.split("=")[1]

except FileNotFoundError:

raise FileNotFoundError("Metadata file './raganything/__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}"

)

return vars2readme

# Reading dependencies from requirements.txt

def read_requirements():

deps = []

try:

with open("./requirements.txt") as f:

deps = [

line.strip() for line in f if line.strip() and not line.startswith("#")

]

except FileNotFoundError:

print(

"Warning: 'requirements.txt' not found. No dependencies will be installed."

)

return deps

metadata = retrieve_metadata()

long_description = read_long_description()

requirements = read_requirements()

# Define extras_require for optional features

extras_require = {

"image": ["Pillow>=10.0.0"], # For image format conversion (BMP, TIFF, GIF, WebP)

"text": ["reportlab>=4.0.0"], # For text file to PDF conversion (TXT, MD)

"office": [], # Office document processing requires LibreOffice (external program)

"paddleocr": ["paddleocr>=2.7.0", "pypdfium2>=4.25.0"], # PaddleOCR parser for scanned PDFs

"all": [

"Pillow>=10.0.0",

"reportlab>=4.0.0",

"paddleocr>=2.7.0",

"pypdfium2>=4.25.0",

"markdown>=3.4.0",

"weasyprint>=60.0",

"pygments>=2.10.0",

], # All optional features

"markdown": [

"markdown>=3.4.0",

"weasyprint>=60.0",

"pygments>=2.10.0",

], # Enhanced markdown conversion

}

setuptools.setup(

name="raganything",

url=metadata["__url__"],

version=metadata["__version__"],

author=metadata["__author__"],

description="RAGAnything: All-in-One RAG System",

long_description=long_description,

long_description_content_type="text/markdown",

packages=setuptools.find_packages(

exclude=("tests*", "docs*")

), # Automatically find packages

classifiers=[

"Development Status :: 4 - Beta",

"Programming Language :: Python :: 3",

"License :: OSI Approved :: MIT License",

"Operating System :: OS Independent",

"Intended Audience :: Developers",

"Topic :: Software Development :: Libraries :: Python Modules",

],

python_requires=">=3.9",

install_requires=requirements,

extras_require=extras_require,

include_package_data=True, # Includes non-code files from MANIFEST.in

project_urls={ # Additional project metadata

"Documentation": metadata.get("__url__", ""),

"Source": metadata.get("__url__", ""),

"Tracker": f"{metadata.get('__url__', '')}/issues"

if metadata.get("__url__")

else "",

},

)