minions/setup.py at main · HazyResearch/minions · GitHub

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
from setuptools import setup, find_packages

setup(
    name="minions",
    version="0.1.0",
    packages=find_packages(),
    install_requires=[
        "ollama",  # for local LLM
        "streamlit==1.42.2",  # for the UI
        "openai",  # for OpenAI client
        "anthropic",  # for Anthropic client
        "together",  # for Together client
        "groq",  # for Groq client
        "mistralai",  # for Mistral AI client
        "requests",  # for API calls
        "tiktoken",  # for token counting
        "pymupdf",  # for PDF processing
        "st-theme",
        "mcp",
        "spacy",  # for PII extraction, worked on python 3.11 and not 3.13
        "rank_bm25",  # for smart retrieval
        "PyMuPDF",  # for PDF handling
        "firecrawl-py",  # for scraping urls
        "google-genai",  # for Gemini client
        "serpapi",  # for web search
        "google_search_results",  # for web search
        "psutil",
        "flask",  # for the worker server
        "cryptography",  # for crypto utils
        "orjson",
        "twilio",
        "pyjwt",  # for JWT utilities
        "torch",
        "cerebras-cloud-sdk",  # for Cerebras client
        "nv-attestation-sdk",
        "nv-local-gpu-verifier",
        "azure-security-attestation",
        "azure-identity",
	"gitingest",
    ],
    extras_require={
        "mlx": ["mlx-lm"],
        "csm-mlx": ["csm-mlx @ git+https://github.com/senstella/csm-mlx.git"],
        "embeddings": [
            "faiss-cpu",  # for embedding search
            "sentence-transformers",  # for pretrained embedding models
            "torch",  # for running embedding models on CUDA
            "chromadb",  # for vector database
        ],
        "secure": [
            "flask",  # for the worker server
            "cryptography",  # for crypto utils
            "orjson",
            "twilio",
            "pyjwt",  # for JWT utilities
            "nv-attestation-sdk",
            "nv-local-gpu-verifier",
            "azure-security-attestation",
            "azure-identity",
            "PyJWT[crypto]", # TODO: check if this conflicts with the pyjwt installed above
        ],
    },
    author="Sabri, Avanika, and Dan",
    description="A package for running minion protocols with local and remote LLMs",
    python_requires=">=3.8",
    entry_points={
        "console_scripts": [
            "minions=minions_cli:main",
        ],
    },
)