giving up on the impossible and adjusting to have multiple targets

Author: cagostino

setup.py

@@ -1,17 +1,8 @@
-# Force lite installation if environment variable is set
-import os
-
-import sys
-
-if os.environ.get("NPCSH_LITE_INSTALL", "").lower() == "true":
-
-    sys.argv.append("--config-settings")
-    sys.argv.append("install.lite=true")
-
 from setuptools import setup, find_packages
 import site
 import platform
 from pathlib import Path
+import os
 
 
 def package_files(directory):
@@ -58,47 +49,35 @@ def get_setup_message():
 To configure your API keys and preferences.
 ==============================================
 """
-    return ""  # Return empty string for non-Windows platforms
+    return ""
 
 
-# Define core (lite) requirements
-core_requirements = [
-    "redis",
-    "flask_sse",
-    "anthropic",
-    "beautifulsoup4",
-    "google-generativeai",
-    "google-genai",
-    "duckduckgo-search",
-    "openai",
+# Base requirements (no LLM packages)
+base_requirements = [
     "jinja2",
+    "scipy",
+    "numpy",
     "requests",
     "markdown",
     "PyYAML",
-    "langchain",
-    "langchain_community",
     "pygments",
     "termcolor",
     "colorama",
     "python-dotenv",
-    "pytest",
-    "googlesearch-python",
+    "pandas",
+    "beautifulsoup4",
+    "duckduckgo-search",
     "flask",
     "flask_cors",
-    "librosa",
-    "pandas",
-    "matplotlib",
-    "IPython",
-    "pyautogui",
-    "nltk",
-    "thefuzz",
-    "pypdf",
-    "PyMuPDF",
-    "screeninfo",
+    "redis",
+    "flask_sse",
 ]
 
-# Define additional requirements for full installation
-extra_requirements = [
+# API integration requirements
+api_requirements = ["anthropic", "openai", "google-generativeai", "google-genai"]
+
+# Local ML/AI requirements
+local_requirements = [
     "sentence_transformers",
     "opencv-python",
     "ollama",
@@ -107,8 +86,8 @@ def get_setup_message():
     "diffusers",
 ]
 
-# Define audio requirements
-audio_requirements = [
+# Voice/Audio requirements
+voice_requirements = [
     "openai-whisper",
     "pyaudio",
     "gtts",
@@ -118,21 +97,17 @@ def get_setup_message():
 
 extra_files = package_files("npcsh/npc_team/")
 
-
-def get_requirements():
-    # Check if lite installation was requested via sys.argv
-    if any("install.lite=true" in arg for arg in sys.argv):
-        return core_requirements
-    elif os.environ.get("NPCSH_AUDIO_INSTALL", "").lower() == "true":
-        return core_requirements + extra_requirements + audio_requirements
-    return core_requirements + extra_requirements
-
-
 setup(
     name="npcsh",
     version="0.3.27.1",
     packages=find_packages(exclude=["tests*"]),
-    install_requires=get_requirements(),
+    install_requires=base_requirements,  # Only install base requirements by default
+    extras_require={
+        "lite": api_requirements,  # Just API integrations
+        "local": local_requirements,  # Local AI/ML features
+        "whisper": voice_requirements,  # Voice/Audio features
+        "all": api_requirements + local_requirements + voice_requirements,  # Everything
+    },
     entry_points={
         "console_scripts": [
             "npcsh=npcsh.shell:main",
@@ -154,6 +129,5 @@ def get_requirements():
     python_requires=">=3.10",
 )
 
-# Print setup message only on Windows
 if platform.system() == "Windows":
     print(get_setup_message())