mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-23 01:55:06 +02:00
Sreekara YachamaneniandGitHub
1211af45b1
- Accept agent_id and environment args for `lanngraph deploy` - Validate both arguments present or none - If agent arguments present, make sure deployment_id and name are not present
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
import json
|
|
from unittest.mock import Mock
|
|
|
|
import httpx
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
import langgraph_cli.deploy as deploy
|
|
from langgraph_cli.cli import cli
|
|
from langgraph_cli.host_backend import HostBackendClient
|
|
|
|
|
|
@pytest.fixture
|
|
def deployment_api(monkeypatch, tmp_path):
|
|
monkeypatch.chdir(tmp_path)
|
|
monkeypatch.delenv("LANGSMITH_DEPLOYMENT_NAME", raising=False)
|
|
monkeypatch.setattr(deploy, "_emitter", None)
|
|
monkeypatch.setattr(deploy, "_no_input", False)
|
|
(tmp_path / "langgraph.json").write_text(
|
|
json.dumps({"dependencies": ["."], "graphs": {"agent": "./agent.py:graph"}})
|
|
)
|
|
(tmp_path / ".env").write_text("LANGSMITH_DEPLOYMENT_NAME=legacy\n")
|
|
requests = []
|
|
state = {"enabled": True, "resources": []}
|
|
|
|
def handler(request):
|
|
requests.append(request)
|
|
assert request.url.path == "/v2/deployments"
|
|
if request.method == "GET":
|
|
if not state["enabled"] and (
|
|
"agent_id" in request.url.params
|
|
or "agent_environment" in request.url.params
|
|
):
|
|
return httpx.Response(
|
|
400, text="Agent filters are not available for this tenant."
|
|
)
|
|
return httpx.Response(200, json={"resources": state["resources"]})
|
|
assert request.method == "POST"
|
|
return httpx.Response(200, json={"id": "runtime-id", "name": "server-name"})
|
|
|
|
client = HostBackendClient("https://api.example.com", "test-key")
|
|
client._client.close()
|
|
client._client = httpx.Client(
|
|
base_url="https://api.example.com",
|
|
transport=httpx.MockTransport(handler),
|
|
headers={"X-Api-Key": "test-key"},
|
|
)
|
|
monkeypatch.setattr(deploy, "_create_host_backend_client", lambda *a, **kw: client)
|
|
monkeypatch.setattr(deploy, "find_tracked_packages", lambda *a: [])
|
|
remote_build = Mock(return_value=deploy.BuildResult())
|
|
monkeypatch.setattr(deploy, "_run_remote_build", remote_build)
|
|
monkeypatch.setattr(deploy, "_resolve_build_mode", lambda flag, **kw: (flag, None))
|
|
yield state, requests, remote_build
|
|
client._client.close()
|
|
|
|
|
|
AGENT_ARGS = [
|
|
"deploy",
|
|
"--agent-id",
|
|
"customer-support",
|
|
"--environment",
|
|
"staging",
|
|
"--remote",
|
|
"--no-wait",
|
|
"--no-input",
|
|
]
|
|
|
|
|
|
def test_agent_create(deployment_api, tmp_path, monkeypatch):
|
|
monkeypatch.setenv("LANGSMITH_DEPLOYMENT_NAME", "legacy")
|
|
_, requests, build = deployment_api
|
|
result = CliRunner().invoke(cli, AGENT_ARGS)
|
|
assert result.exit_code == 0, result.output
|
|
assert dict(requests[0].url.params) == {
|
|
"name_contains": "",
|
|
"agent_id": "customer-support",
|
|
"agent_environment": "staging",
|
|
}
|
|
payload = json.loads(requests[1].content)
|
|
assert payload["agent"] == {
|
|
"agent_id": "customer-support",
|
|
"environment": "staging",
|
|
}
|
|
assert "name" not in payload
|
|
assert build.call_args.kwargs["deployment_id"] == "runtime-id"
|
|
assert "server-name" in result.output
|
|
assert (tmp_path / ".env").read_text() == "LANGSMITH_DEPLOYMENT_NAME=legacy\n"
|
|
|
|
|
|
def test_agent_update(deployment_api):
|
|
state, requests, build = deployment_api
|
|
state["resources"] = [{"id": "existing-id", "is_preview": False}]
|
|
result = CliRunner().invoke(cli, AGENT_ARGS)
|
|
assert result.exit_code == 0, result.output
|
|
assert len(requests) == 1
|
|
assert build.call_args.kwargs["deployment_id"] == "existing-id"
|
|
|
|
|
|
def test_agent_rejects_explicit_name(deployment_api, monkeypatch):
|
|
monkeypatch.setenv("LANGSMITH_DEPLOYMENT_NAME", "legacy")
|
|
_, requests, _ = deployment_api
|
|
result = CliRunner().invoke(cli, [*AGENT_ARGS, "--name", "legacy"])
|
|
assert result.exit_code == 2
|
|
assert "cannot be combined" in result.output
|
|
assert not requests
|