mirror of
https://github.com/langgenius/dify.git
synced 2026-07-15 01:17:04 +08:00
1b3e8e9943
Signed-off-by: yyh <yuanyouhuilyz@gmail.com> Co-authored-by: 盐粒 Yanli <mail@yanli.one> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: zyssyz123 <916125788@qq.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: 盐粒 Yanli <yanli@dify.ai> Co-authored-by: 林玮 (Jade Lin) <linw1995@icloud.com>
110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
from __future__ import annotations
|
|
|
|
import tomllib
|
|
from pathlib import Path
|
|
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[2]
|
|
|
|
CLIENT_SHARED_DTO_DEPENDENCIES = {
|
|
"httpx==0.28.1",
|
|
"pydantic>=2.12.5,<2.13",
|
|
"pydantic-ai-slim>=1.102.0,<2.0.0",
|
|
"typer>=0.16.1,<0.17",
|
|
"typing-extensions>=4.12.2,<5.0.0",
|
|
}
|
|
|
|
SERVER_RUNTIME_DEPENDENCIES = {
|
|
"fastapi==0.136.0",
|
|
"graphon==0.5.2",
|
|
"jsonschema>=4.23.0,<5.0.0",
|
|
"jwcrypto>=1.5.6,<2",
|
|
"logfire[fastapi,httpx,redis]>=4.37.0,<5.0.0",
|
|
"pydantic-ai-slim[anthropic,google,openai]>=1.85.1,<2.0.0",
|
|
"pydantic-settings>=2.12.0,<3.0.0",
|
|
"redis>=7.4.0,<8.0.0",
|
|
"shell-session-manager==2.3.1",
|
|
"uvicorn[standard]==0.46.0",
|
|
}
|
|
|
|
GRPC_RUNTIME_DEPENDENCIES = {
|
|
"grpclib[protobuf]>=0.4.9,<0.5.0",
|
|
"protobuf>=6.33.5,<7.0.0",
|
|
}
|
|
|
|
DEV_DEPENDENCIES = {
|
|
"basedpyright>=1.39.3",
|
|
"coverage[toml]>=7.10.7",
|
|
"grpcio-tools>=1.81.0,<2.0.0",
|
|
"pytest>=9.0.3",
|
|
"pytest-examples>=0.0.18",
|
|
"pytest-mock>=3.14.0",
|
|
"ruff>=0.15.11",
|
|
}
|
|
|
|
|
|
def _read_pyproject():
|
|
return tomllib.loads((PROJECT_ROOT / "pyproject.toml").read_text(encoding="utf-8"))
|
|
|
|
|
|
def test_project_dependencies_split_client_and_server_requirements() -> None:
|
|
pyproject = _read_pyproject()
|
|
project = pyproject["project"]
|
|
|
|
assert set(project["dependencies"]) == CLIENT_SHARED_DTO_DEPENDENCIES
|
|
assert set(project["optional-dependencies"]["grpc"]) == GRPC_RUNTIME_DEPENDENCIES
|
|
assert set(project["optional-dependencies"]["server"]) == SERVER_RUNTIME_DEPENDENCIES
|
|
assert set(pyproject["dependency-groups"]["dev"]) == DEV_DEPENDENCIES
|
|
|
|
|
|
def test_default_package_discovery_excludes_example_packages() -> None:
|
|
pyproject = _read_pyproject()
|
|
find_config = pyproject["tool"]["setuptools"]["packages"]["find"]
|
|
|
|
assert find_config["where"] == ["src"]
|
|
assert "agenton_examples*" not in find_config["include"]
|
|
assert "dify_agent_examples*" not in find_config["include"]
|
|
|
|
|
|
def test_project_declares_console_scripts() -> None:
|
|
pyproject = _read_pyproject()
|
|
scripts = pyproject["project"]["scripts"]
|
|
|
|
assert scripts["dify-agent"] == "dify_agent.agent_stub.cli.main:main"
|
|
assert scripts["dify-agent-stub-server"] == "dify_agent.agent_stub.server.cli:main"
|
|
|
|
|
|
def test_local_sandbox_dockerfile_installs_stub_client_and_shellctl() -> None:
|
|
dockerfile = (PROJECT_ROOT / "docker" / "local-sandbox" / "Dockerfile").read_text(encoding="utf-8")
|
|
|
|
assert "ARG NODE_VERSION=22.22.1" in dockerfile
|
|
assert "ARG PNPM_VERSION=11.9.0" in dockerfile
|
|
assert "ARG UV_VERSION=0.8.9" in dockerfile
|
|
assert "ARG DIFY_AGENT_TOOL_SPEC=.[grpc]" in dockerfile
|
|
assert "ARG SHELL_SESSION_MANAGER_TOOL_SPEC=shell-session-manager" in dockerfile
|
|
assert "UV_TOOL_DIR=/opt/dify-agent-tools/envs" in dockerfile
|
|
assert "UV_TOOL_BIN_DIR=/opt/dify-agent-tools/bin" in dockerfile
|
|
assert "git" in dockerfile
|
|
assert "jq" in dockerfile
|
|
assert "openssh-client" in dockerfile
|
|
assert "ripgrep" in dockerfile
|
|
assert 'node_dist="node-v${NODE_VERSION}-linux-${node_arch}"' in dockerfile
|
|
assert 'curl -fsSLO "https://nodejs.org/dist/v${NODE_VERSION}/SHASUMS256.txt"' in dockerfile
|
|
assert 'tar -xJf "${node_dist}.tar.xz" -C /usr/local --strip-components=1' in dockerfile
|
|
assert 'npm install --global "pnpm@${PNPM_VERSION}"' in dockerfile
|
|
assert "uv export --frozen --no-dev --all-extras --no-emit-project --no-hashes" in dockerfile
|
|
assert "> /tmp/dify-agent-constraints.txt" in dockerfile
|
|
assert '--constraints /tmp/dify-agent-constraints.txt --link-mode=copy "${DIFY_AGENT_TOOL_SPEC}"' in dockerfile
|
|
assert (
|
|
'--constraints /tmp/dify-agent-constraints.txt --link-mode=copy "${SHELL_SESSION_MANAGER_TOOL_SPEC}"'
|
|
in dockerfile
|
|
)
|
|
assert "DIFY_AGENT_STUB_DRIVE_BASE=/mnt/drive" in dockerfile
|
|
assert "COPY --from=tools ${UV_TOOL_DIR} ${UV_TOOL_DIR}" in dockerfile
|
|
assert "COPY --from=tools ${UV_TOOL_BIN_DIR} ${UV_TOOL_BIN_DIR}" in dockerfile
|
|
assert "VIRTUAL_ENV" not in dockerfile
|
|
assert "uv sync" not in dockerfile
|
|
assert "mkdir -p /mnt/drive" in dockerfile
|
|
assert "chown dify:dify /home" in dockerfile
|
|
assert '["shellctl", "serve", "--listen", "0.0.0.0:5004"]' in dockerfile
|