mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 11:33:14 +00:00
889b2663d9
* fix: resolve trace URL project id by tracing project name Graphs were split into separate LangSmith tracing projects (open-swe-agent, open-swe-review) but the "View trace" link still used a single fixed project-id env var pointing at the old combined project. Resolve the project id from the tracing project name so agent and reviewer links point at their respective projects, falling back to the env var when resolution is unavailable. Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> * docs: document per-graph tracing projects for trace links View trace links now resolve project IDs from the open-swe-agent / open-swe-review project names. Document this so fresh deployments create the right projects instead of relying on a single project ID. Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> --------- Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import pytest
|
|
|
|
from agent.utils import langsmith as ls_utils
|
|
from agent.utils.tracing import AGENT_TRACING_PROJECT, REVIEW_TRACING_PROJECT
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clear_cache() -> None:
|
|
ls_utils._PROJECT_ID_CACHE.clear()
|
|
|
|
|
|
def _set_env(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("LANGSMITH_URL_PROD", "https://smith.example")
|
|
monkeypatch.setenv("LANGSMITH_TENANT_ID_PROD", "tenant-1")
|
|
monkeypatch.delenv("LANGSMITH_TRACING_PROJECT_ID_PROD", raising=False)
|
|
|
|
|
|
def test_trace_url_resolves_project_id_by_name(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
_set_env(monkeypatch)
|
|
monkeypatch.setattr(
|
|
ls_utils,
|
|
"_resolve_project_id_by_name",
|
|
lambda name: "agent-pid" if name == AGENT_TRACING_PROJECT else "review-pid",
|
|
)
|
|
|
|
agent_url = ls_utils.get_langsmith_trace_url("t1")
|
|
review_url = ls_utils.get_langsmith_trace_url("t2", project_name=REVIEW_TRACING_PROJECT)
|
|
|
|
assert agent_url == "https://smith.example/o/tenant-1/projects/p/agent-pid/t/t1"
|
|
assert review_url == "https://smith.example/o/tenant-1/projects/p/review-pid/t/t2"
|
|
|
|
|
|
def test_trace_url_falls_back_to_env_project_id(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
_set_env(monkeypatch)
|
|
monkeypatch.setenv("LANGSMITH_TRACING_PROJECT_ID_PROD", "env-pid")
|
|
monkeypatch.setattr(ls_utils, "_resolve_project_id_by_name", lambda name: None)
|
|
|
|
url = ls_utils.get_langsmith_trace_url("t3")
|
|
|
|
assert url == "https://smith.example/o/tenant-1/projects/p/env-pid/t/t3"
|
|
|
|
|
|
def test_trace_url_none_when_unresolvable(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
_set_env(monkeypatch)
|
|
monkeypatch.setattr(ls_utils, "_resolve_project_id_by_name", lambda name: None)
|
|
|
|
assert ls_utils.get_langsmith_trace_url("t4") is None
|
|
|
|
|
|
def test_resolve_project_id_caches_success(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
calls: list[str] = []
|
|
|
|
class _FakeProject:
|
|
id = "pid-123"
|
|
|
|
class _FakeClient:
|
|
def read_project(self, *, project_name: str) -> _FakeProject:
|
|
calls.append(project_name)
|
|
return _FakeProject()
|
|
|
|
monkeypatch.setattr(ls_utils, "_build_prod_langsmith_client", lambda: _FakeClient())
|
|
|
|
first = ls_utils._resolve_project_id_by_name(AGENT_TRACING_PROJECT)
|
|
second = ls_utils._resolve_project_id_by_name(AGENT_TRACING_PROJECT)
|
|
|
|
assert first == "pid-123"
|
|
assert second == "pid-123"
|
|
assert calls == [AGENT_TRACING_PROJECT]
|