Dataflow-LoopAI/setup.py at main · OpenDCAI/Dataflow-LoopAI · 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
72
73
74
from setuptools import setup, find_packages

setup(
    name="loopai",
    version="0.0.1",
    packages=find_packages(),
    install_requires=[
        "langgraph>=0.6.7",
        "langgraph-checkpoint-sqlite>=3.0.0",
        "colorlog>=6.10.0",
        "rich>=13.0.0",
        "langchain>=0.3.27",
        "langchain-community>=0.3.0",
        "langchain-openai>=0.1.0",
        "langchain-core>=0.3.0",
        "langchain-text-splitters>=0.2.0",
        "omegaconf>=2.3.0",
        "PyYAML>=6.0",
        "httpx>=0.24.0",
        "chromadb>=0.4.0",
        "ddgs",
        # Download dependencies
        "huggingface_hub>=0.20.0",
        "datasets>=2.14.0",
        "kaggle>=1.5.0",
        "kagglehub>=0.2.0",
        "playwright>=1.40.0",
        "tenacity>=8.2.0",
        "requests>=2.31.0",
        "transformers>=4.30.0",
        "pyarrow>=20.0.0",
        "mcp[cli]>=0.1.0",
        "tomlkit>=0.13.2",
        "aiosqlite>=0.21.0",
        "fastapi",
        "uvicorn",
        "pydantic",
        "tortoise-orm",
        "func-timeout",
        "psutil",
        "pyflakes>=3.4.0",
        "tree_sitter>=0.25.2",
        "tree-sitter-python>=0.25.0",
        "tree-sitter-javascript>=0.25.0",
        "tree-sitter-java>=0.23.5",
        "tree-sitter-cpp>=0.23.4",
        "tree-sitter-c>=0.24.1",
        "tree-sitter-c-sharp>=0.23.1",
        "tree-sitter-go>=0.25.0",
        "tree-sitter-rust>=0.24.0",
        "tree-sitter-ruby>=0.23.1",
        "tree-sitter-swift>=0.0.1",
        "tree-sitter-kotlin>=1.1.0",
        "tree-sitter-scala>=0.24.0",
        "tree-sitter-bash>=0.25.1",
        "tree-sitter-html>=0.23.2",
        "tree-sitter-css>=0.25.0",
        "tree-sitter-json>=0.24.8",
        "tree-sitter-yaml>=0.7.2",
        "tree-sitter-markdown>=0.5.1",
        "one-eval @ git+https://github.com/OpenDCAI/One-Eval.git@main",
        "open-dataflow==1.0.10",
        "vllm>=0.4.0"
    ],
    entry_points={
        "console_scripts": [
            "loopai-obtainercli=loopai.skills.ObtainerCLI.cli:main",
            "loopai-judger=loopai.skills.Judger.cli:main",
            "loopai-analyzer=loopai.skills.Analyzer.cli:main",
            "loopai-trainer=loopai.skills.Trainer.cli:main",
        ],
    },
    python_requires=">=3.12",
)