[release] v0.128.7

api/oss/src/core/applications/service.py

@@ -161,9 +161,16 @@ async def edit_application(
         #
         application_edit: ApplicationEdit,
     ) -> Optional[Application]:
+        # Only a folder_id the caller sent may reach the workflow edit, where a null
+        # moves the application to the top level.
         workflow_edit = WorkflowEdit(
             **application_edit.model_dump(
                 mode="json",
+                exclude=(
+                    None
+                    if "folder_id" in application_edit.model_fields_set
+                    else {"folder_id"}
+                ),
             ),
         )
 

api/oss/src/core/evaluations/runtime/sources.py

@@ -70,6 +70,33 @@ def _extract_ag_data(trace: Any) -> Dict[str, Any]:
     return data if isinstance(data, dict) else {}
 
 
+def _trace_source_item(
+    *,
+    trace: Any,
+    step_key: str,
+    trace_id: Optional[str] = None,
+) -> ResolvedSourceItem:
+    """A trace source item with the root span id, inputs and outputs read from the trace.
+
+    The evaluator steps read the outputs from these fields, and the span id becomes the
+    evaluator's `invocation` link. Without the link the evaluator trace is typed
+    `invocation` and a live query on invocation traces reads it back.
+    """
+    ag_data = _extract_ag_data(trace) if trace is not None else {}
+    root_span = _extract_root_span(trace) if trace is not None else None
+    return ResolvedSourceItem(
+        kind="trace",
+        step_key=step_key,
+        #
+        trace_id=trace_id or getattr(trace, "trace_id", None),
+        span_id=_extract_span_id(root_span),
+        trace=trace,
+        #
+        inputs=ag_data.get("inputs"),
+        outputs=ag_data.get("outputs"),
+    )
+
+
 class SourceResolutionError(Exception):
     """An input step does not carry exactly one recognized source reference."""
 
@@ -397,23 +424,14 @@ async def resolve_direct_source_items(
             )
             for testcase_id in testcase_ids
         )
-        for trace_id in trace_ids:
-            trace = traces_by_id.get(trace_id)
-            ag_data = _extract_ag_data(trace) if trace is not None else {}
-            root_span = _extract_root_span(trace) if trace is not None else None
-            source_items.append(
-                ResolvedSourceItem(
-                    kind="trace",
-                    step_key="",
-                    #
-                    trace_id=trace_id,
-                    span_id=_extract_span_id(root_span),
-                    trace=trace,
-                    #
-                    inputs=ag_data.get("inputs"),
-                    outputs=ag_data.get("outputs"),
-                )
+        source_items.extend(
+            _trace_source_item(
+                trace=traces_by_id.get(trace_id),
+                trace_id=trace_id,
+                step_key="",
             )
+            for trace_id in trace_ids
+        )
 
         return source_items
 
@@ -535,13 +553,7 @@ async def resolve_query_source_items(
 
         return {
             query_step_key: [
-                ResolvedSourceItem(
-                    kind="trace",
-                    step_key=query_step_key,
-                    #
-                    trace_id=trace.trace_id,
-                    trace=trace,
-                )
+                _trace_source_item(trace=trace, step_key=query_step_key)
                 for trace in traces
                 if trace and trace.trace_id
             ]

api/oss/src/core/evaluators/service.py

@@ -158,9 +158,16 @@ async def edit_evaluator(
         #
         evaluator_edit: EvaluatorEdit,
     ) -> Optional[Evaluator]:
+        # Only a folder_id the caller sent may reach the workflow edit, where a null
+        # moves the evaluator to the top level.
         workflow_edit = WorkflowEdit(
             **evaluator_edit.model_dump(
                 mode="json",
+                exclude=(
+                    None
+                    if "folder_id" in evaluator_edit.model_fields_set
+                    else {"folder_id"}
+                ),
             ),
         )
 

api/oss/src/core/workflows/service.py

@@ -1751,6 +1751,10 @@ async def edit_workflow(
             exclude_none=True,
             exclude={"flags"},
         )
+        # exclude_none drops a null folder_id, but a null sent on purpose moves the
+        # workflow to the top level.
+        if "folder_id" in workflow_edit.model_fields_set:
+            artifact_edit_kwargs["folder_id"] = workflow_edit.folder_id
         if "flags" in workflow_edit.model_fields_set:
             artifact_flags = self._artifact_flags_from_any(workflow_edit.flags)
             artifact_edit_kwargs["flags"] = (

api/oss/tests/pytest/unit/evaluations/test_query_eval_loops.py

@@ -469,3 +469,80 @@ async def test_run_query_source_live_run_not_finalized_on_error(monkeypatch):
 
     # live runs are never finalized on error — the scheduler keeps polling.
     edit_run.assert_not_awaited()
+
+
+@pytest.mark.asyncio
+async def test_query_source_items_carry_root_span_inputs_and_outputs():
+    """A live query trace feeds the evaluator its outputs and an invocation link.
+
+    Without the outputs the code evaluator fails with "Invalid outputs". Without
+    the span id the evaluator trace has no links, so it is typed `invocation`
+    and the live query (`trace_type is invocation`) reads it back.
+    """
+    from agenta.sdk.models.tracing import SpansNode, Trace
+    from oss.src.core.evaluations.tasks.processor import _seed_context_from_source
+
+    query_revision_id = uuid4()
+    trace_id = "a" * 32
+    span_id = "b" * 16
+    trace = Trace(
+        trace_id=trace_id,
+        spans={
+            "capital": SpansNode(
+                trace_id=trace_id,
+                span_id=span_id,
+                attributes={
+                    "ag": {
+                        "data": {
+                            "inputs": {"country": "Peru"},
+                            "outputs": "Lima",
+                        }
+                    }
+                },
+            )
+        },
+    )
+    run = EvaluationRun(
+        id=uuid4(),
+        principal=any_principal(),
+        flags=EvaluationRunFlags(is_live=True, has_queries=True),
+        data=EvaluationRunData(
+            steps=[
+                EvaluationRunDataStep(
+                    key="query-live",
+                    type="input",
+                    origin="custom",
+                    references={"query_revision": Reference(id=query_revision_id)},
+                ),
+            ],
+            repeats=1,
+        ),
+    )
+    query_traces = AsyncMock(return_value=[trace])
+    sources = SourceResolution(
+        queries_service=SimpleNamespace(
+            fetch_query_revision=AsyncMock(
+                return_value=SimpleNamespace(
+                    id=query_revision_id,
+                    slug="query-live",
+                    data=SimpleNamespace(filtering=None, windowing=None),
+                )
+            )
+        ),
+        tracing_service=SimpleNamespace(query_traces=query_traces),
+    )
+
+    items = await sources.resolve_query_source_items(
+        project_id=uuid4(),
+        run=run,
+    )
+
+    [item] = items["query-live"]
+    assert item.trace_id == trace_id
+    assert item.span_id == span_id
+    assert item.inputs == {"country": "Peru"}
+    assert item.outputs == "Lima"
+
+    context = _seed_context_from_source(source_item=item, repeats=1)
+    assert context[0]["outputs"] == "Lima"
+    assert context[0]["span_id"] == span_id

api/oss/tests/pytest/unit/workflows/test_edit_workflow_folder.py

@@ -0,0 +1,104 @@
+from unittest.mock import AsyncMock
+from uuid import uuid4
+
+import pytest
+
+import oss.src.core.workflows.service as workflows_service_module
+from oss.src.core.applications.dtos import ApplicationEdit
+from oss.src.core.applications.service import ApplicationsService
+from oss.src.core.evaluators.dtos import EvaluatorEdit
+from oss.src.core.evaluators.service import EvaluatorsService
+from oss.src.apis.fastapi.workflows.models import WorkflowEditRequest
+from oss.src.core.workflows.dtos import Workflow
+from oss.src.core.workflows.service import WorkflowsService
+
+
+@pytest.fixture
+def service(monkeypatch):
+    monkeypatch.setattr(workflows_service_module, "get_cache", AsyncMock())
+    monkeypatch.setattr(workflows_service_module, "set_cache", AsyncMock())
+    monkeypatch.setattr(workflows_service_module, "invalidate_cache", AsyncMock())
+    workflows_dao = AsyncMock()
+    workflow = Workflow(id=uuid4(), slug="wf", folder_id=uuid4())
+    workflows_dao.fetch_artifact.return_value = workflow
+    workflows_dao.edit_artifact.return_value = workflow
+    return WorkflowsService(workflows_dao=workflows_dao), workflow.id
+
+
+def _artifact_edit(service):
+    return service.workflows_dao.edit_artifact.await_args.kwargs["artifact_edit"]
+
+
+def _request_body(workflow: dict):
+    """The edit as the router gets it: parsed from the JSON body of PUT /workflows/{id}."""
+    return WorkflowEditRequest.model_validate({"workflow": workflow}).workflow
+
+
+@pytest.mark.asyncio
+async def test_a_null_folder_id_moves_the_workflow_to_the_top_level(service):
+    workflows_service, workflow_id = service
+
+    await workflows_service.edit_workflow(
+        project_id=uuid4(),
+        user_id=uuid4(),
+        workflow_edit=_request_body({"id": str(workflow_id), "folder_id": None}),
+    )
+
+    artifact_edit = _artifact_edit(workflows_service)
+    assert "folder_id" in artifact_edit.model_fields_set
+    assert artifact_edit.folder_id is None
+
+
+@pytest.mark.asyncio
+async def test_a_folder_id_in_the_body_moves_the_workflow_to_that_folder(service):
+    workflows_service, workflow_id = service
+    destination = uuid4()
+
+    await workflows_service.edit_workflow(
+        project_id=uuid4(),
+        user_id=uuid4(),
+        workflow_edit=_request_body(
+            {"id": str(workflow_id), "folder_id": str(destination)}
+        ),
+    )
+
+    assert _artifact_edit(workflows_service).folder_id == destination
+
+
+@pytest.mark.asyncio
+async def test_an_edit_without_folder_id_keeps_the_folder(service):
+    workflows_service, workflow_id = service
+
+    await workflows_service.edit_workflow(
+        project_id=uuid4(),
+        user_id=uuid4(),
+        workflow_edit=_request_body({"id": str(workflow_id), "name": "Renamed"}),
+    )
+
+    assert "folder_id" not in _artifact_edit(workflows_service).model_fields_set
+
+
+@pytest.mark.asyncio
+async def test_an_application_rename_keeps_the_folder(service):
+    workflows_service, workflow_id = service
+
+    await ApplicationsService(workflows_service=workflows_service).edit_application(
+        project_id=uuid4(),
+        user_id=uuid4(),
+        application_edit=ApplicationEdit(id=workflow_id, name="Renamed"),
+    )
+
+    assert "folder_id" not in _artifact_edit(workflows_service).model_fields_set
+
+
+@pytest.mark.asyncio
+async def test_an_evaluator_rename_keeps_the_folder(service):
+    workflows_service, workflow_id = service
+
+    await EvaluatorsService(workflows_service=workflows_service).edit_evaluator(
+        project_id=uuid4(),
+        user_id=uuid4(),
+        evaluator_edit=EvaluatorEdit(id=workflow_id, name="Renamed"),
+    )
+
+    assert "folder_id" not in _artifact_edit(workflows_service).model_fields_set

api/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "api"
-version = "0.128.6"
+version = "0.128.7"
 description = "Agenta API"
 requires-python = ">=3.11,<3.14"
 authors = [

clients/python/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agenta-client"
-version = "0.128.6"
+version = "0.128.7"
 description = "Fern-generated Python client for the Agenta API."
 requires-python = ">=3.11,<3.14"
 authors = [

hosting/kubernetes/helm/Chart.yaml

@@ -2,8 +2,8 @@ apiVersion: v2
 name: agenta
 description: A Helm chart for deploying Agenta (OSS or EE) on Kubernetes
 type: application
-version: 0.128.6
-appVersion: "v0.128.6"
+version: 0.128.7
+appVersion: "v0.128.7"
 keywords:
   - agenta
   - llm

sdks/python/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agenta"
-version = "0.128.6"
+version = "0.128.7"
 description = "Agenta is the open-source workspace for your agents. Build agents through chat, improve them with feedback, and share them with your team."
 readme = "README.md"
 requires-python = ">=3.11,<3.14"