agentx-python 0.8.4__tar.gz → 0.8.6__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of agentx-python might be problematic. Click here for more details.
- {agentx_python-0.8.4/agentx_python.egg-info → agentx_python-0.8.6}/PKG-INFO +1 -1
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/client.py +24 -1
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/datasets.py +39 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/models.py +8 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/runner.py +145 -34
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/client.py +56 -9
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/judge_scorers.py +23 -5
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/online_evaluators.py +7 -3
- agentx_python-0.8.6/agentx/monitor/review_queue.py +93 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/testing.py +3 -1
- agentx_python-0.8.6/agentx/version.py +1 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6/agentx_python.egg-info}/PKG-INFO +1 -1
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/SOURCES.txt +3 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_integrations.py +14 -0
- agentx_python-0.8.6/tests/test_review_queue.py +73 -0
- agentx_python-0.8.6/tests/test_runner_features.py +146 -0
- agentx_python-0.8.4/agentx/version.py +0 -1
- {agentx_python-0.8.4 → agentx_python-0.8.6}/LICENSE +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/README.md +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/agentx.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/cli.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/_term.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/adapters/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/adapters/http_endpoint.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/adapters/precomputed.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/adapters/raw.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/evaluation_settings.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/prompts.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/reporting.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/results.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/tool_schemas.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/evaluations/tracing.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/exceptions.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/export.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/feedback.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/_traced_call.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/anthropic.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/autogen.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/crewai.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/databricks.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/google_adk.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/google_genai.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/langchain.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/litellm.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/llamaindex.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/moveworks.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/openai.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/integrations/openai_agents.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/agents.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/models.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/patterns.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/profile.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/scorers.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/sessions.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/monitor/signals.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/outcomes.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/projects.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/py.typed +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/resources/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/resources/agent.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/resources/conversation.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/resources/workforce.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/traces.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/tracing/__init__.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/tracing/ci_types.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/tracing/eval_scope.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/tracing/ingest_client.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/tracing/tracer.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx/util.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/dependency_links.txt +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/entry_points.txt +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/not-zip-safe +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/requires.txt +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/agentx_python.egg-info/top_level.txt +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/setup.cfg +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/setup.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_deep_dive_fixes.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_docs_match_sdk.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_eval_scope.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_integration.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_judge_scorers.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_pairwise.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_selfhost_analysis_fallback.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_span_tree.py +0 -0
- {agentx_python-0.8.4 → agentx_python-0.8.6}/tests/test_testing.py +0 -0
|
@@ -60,6 +60,13 @@ class AgentXValidationError(AgentXEvaluationsError):
|
|
|
60
60
|
pass
|
|
61
61
|
|
|
62
62
|
|
|
63
|
+
class EvaluationSubmissionError(AgentXEvaluationsError):
|
|
64
|
+
"""A result batch could not be submitted (after one retry). The run is left unfinalized;
|
|
65
|
+
re-running execute() on the same context resumes past already-submitted cases."""
|
|
66
|
+
|
|
67
|
+
pass
|
|
68
|
+
|
|
69
|
+
|
|
63
70
|
def _resolve_scorer_id(scorer_id: Optional[str], evaluation_settings_id: Optional[str]) -> Optional[str]:
|
|
64
71
|
"""One grader, two spellings: ``scorer_id`` is the post-consolidation name for what the wire
|
|
65
72
|
still calls ``evaluationSettingsId`` (the ids are identical by design). Both kwargs are
|
|
@@ -220,6 +227,12 @@ class EvaluationsClient:
|
|
|
220
227
|
data = self._request("POST", "/datasets", json=self._with_workspace(payload))
|
|
221
228
|
return Dataset(**data)
|
|
222
229
|
|
|
230
|
+
def delete_dataset(self, dataset_id: str) -> None:
|
|
231
|
+
"""Deletes the dataset, its grading config, and both version histories. Past runs are
|
|
232
|
+
kept (their dataset reference degrades to a bare id). The engine refuses (409) when the
|
|
233
|
+
dataset's config is attached to a live scorer."""
|
|
234
|
+
self._request("DELETE", f"/datasets/{dataset_id}")
|
|
235
|
+
|
|
223
236
|
def list_datasets(self) -> List[Dataset]:
|
|
224
237
|
data = self._request("GET", "/datasets", params=self._workspace_params())
|
|
225
238
|
return [
|
|
@@ -297,10 +310,11 @@ class EvaluationsClient:
|
|
|
297
310
|
python_version: Optional[str] = None,
|
|
298
311
|
scorer_id: Optional[str] = None,
|
|
299
312
|
evaluation_settings_id: Optional[str] = None,
|
|
313
|
+
split: Optional[str] = None,
|
|
300
314
|
) -> EvaluationRun:
|
|
301
315
|
"""``scorer_id`` names the LLM Judge Scorer grading this run (its id doubles as the
|
|
302
316
|
wire's ``evaluationSettingsId``). ``evaluation_settings_id`` is the pre-consolidation
|
|
303
|
-
alias and keeps working."""
|
|
317
|
+
alias and keeps working. ``split`` records the named case subset this run covers."""
|
|
304
318
|
from agentx.version import VERSION
|
|
305
319
|
|
|
306
320
|
grader_id = _resolve_scorer_id(scorer_id, evaluation_settings_id)
|
|
@@ -318,6 +332,8 @@ class EvaluationsClient:
|
|
|
318
332
|
}
|
|
319
333
|
if grader_id:
|
|
320
334
|
payload["evaluationSettingsId"] = grader_id
|
|
335
|
+
if split:
|
|
336
|
+
payload["split"] = split
|
|
321
337
|
data = self._request("POST", "/runs", json=self._with_workspace(payload))
|
|
322
338
|
return EvaluationRun(**data)
|
|
323
339
|
|
|
@@ -436,6 +452,13 @@ class EvaluationsClient:
|
|
|
436
452
|
data = self._request("GET", f"/runs/{run_id}/missing-results")
|
|
437
453
|
return data if isinstance(data, list) else data.get("missing", [])
|
|
438
454
|
|
|
455
|
+
def get_submitted_keys(self, run_id: str) -> List[str]:
|
|
456
|
+
"""Idempotency keys this run has already accepted - what execute() uses to resume a
|
|
457
|
+
crashed or interrupted run without re-running (and re-paying for) finished cases."""
|
|
458
|
+
data = self._request("GET", f"/runs/{run_id}/missing-results")
|
|
459
|
+
keys = data.get("submittedKeys", []) if isinstance(data, dict) else []
|
|
460
|
+
return [k for k in keys if isinstance(k, str)]
|
|
461
|
+
|
|
439
462
|
# ------------------------------------------------------------------
|
|
440
463
|
# Self-host analysis fallback
|
|
441
464
|
#
|
|
@@ -105,9 +105,15 @@ class DatasetBuilder:
|
|
|
105
105
|
expected_tools: Optional[List[str]] = None,
|
|
106
106
|
trajectory_match_mode: str = "strict",
|
|
107
107
|
expected_retrieval_context: Optional[Union[str, List[str]]] = None,
|
|
108
|
+
splits: Optional[List[str]] = None,
|
|
108
109
|
) -> "DatasetBuilder":
|
|
109
110
|
"""Add a case. `judge_guideline` is optional grading guidance specific to this question.
|
|
110
111
|
|
|
112
|
+
`splits` tags this case with named subsets (e.g. ``["smoke"]``): a run started with
|
|
113
|
+
``client.evaluations.run(dataset_id, subject, split="smoke")`` executes only the tagged
|
|
114
|
+
cases (original case indexes are preserved, so per-case comparisons still line up with
|
|
115
|
+
full runs). An untagged case belongs to no split and only runs in full runs.
|
|
116
|
+
|
|
111
117
|
`expected_tools` declares the tool calls a correct run of this case should make. When a
|
|
112
118
|
result links its trace (return `{"output": ..., "trace_id": span.trace_id}` from the
|
|
113
119
|
agent function), the engine matches the trace's actual tool-call sequence against it and
|
|
@@ -149,6 +155,8 @@ class DatasetBuilder:
|
|
|
149
155
|
main["expectedTrajectory"] = {"tools": expected_tools, "mode": trajectory_match_mode}
|
|
150
156
|
if expected_retrieval_context:
|
|
151
157
|
main["expectedRetrievalContext"] = expected_retrieval_context
|
|
158
|
+
if splits:
|
|
159
|
+
main["splits"] = splits
|
|
152
160
|
self._payload["questions"].append(
|
|
153
161
|
{
|
|
154
162
|
"main_question": main,
|
|
@@ -325,6 +333,37 @@ class DatasetClient:
|
|
|
325
333
|
def list(self) -> List[Dataset]:
|
|
326
334
|
return self._client.list_datasets()
|
|
327
335
|
|
|
336
|
+
def delete(self, dataset_id: str) -> None:
|
|
337
|
+
"""Delete a dataset (and its grading config + version histories; past runs are kept)."""
|
|
338
|
+
self._client.delete_dataset(dataset_id)
|
|
339
|
+
|
|
340
|
+
def import_dataset(self, source: Any, name: Optional[str] = None) -> Dataset:
|
|
341
|
+
"""Create a NEW dataset from an exported/fetched one (a ``Dataset`` from ``get()``, or
|
|
342
|
+
the engine's wire/NDJSON-export dict). Always a copy with a fresh id - never a
|
|
343
|
+
restore-in-place. ``name`` optionally renames the copy."""
|
|
344
|
+
wire: Dict[str, Any] = (
|
|
345
|
+
source.model_dump(by_alias=True) if hasattr(source, "model_dump") else dict(source)
|
|
346
|
+
)
|
|
347
|
+
payload: Dict[str, Any] = {
|
|
348
|
+
"name": name or wire.get("name") or "Imported dataset",
|
|
349
|
+
"questions": wire.get("questions") or [],
|
|
350
|
+
}
|
|
351
|
+
for key in (
|
|
352
|
+
"description",
|
|
353
|
+
"numberOfRequests",
|
|
354
|
+
"acceptanceCriteria",
|
|
355
|
+
"rejectionCriteria",
|
|
356
|
+
"evaluationCriteria",
|
|
357
|
+
"vectorSimilarity",
|
|
358
|
+
"jaccardSimilarity",
|
|
359
|
+
"bleuScore",
|
|
360
|
+
"rougeScore",
|
|
361
|
+
"codeScorers",
|
|
362
|
+
):
|
|
363
|
+
if wire.get(key) is not None:
|
|
364
|
+
payload[key] = wire[key]
|
|
365
|
+
return self._client.create_dataset(payload)
|
|
366
|
+
|
|
328
367
|
|
|
329
368
|
# ---------------------------------------------------------------------------
|
|
330
369
|
# Helpers
|
|
@@ -52,6 +52,9 @@ class TestCase(BaseModel):
|
|
|
52
52
|
expected_delegations: Optional[List[str]] = Field(default=None, alias="expectedDelegations")
|
|
53
53
|
judge_guideline: Optional[str] = Field(default=None, alias="judgeGuideline")
|
|
54
54
|
smoke_test: Optional[SmokeTestSettings] = Field(default=None, alias="smokeTest")
|
|
55
|
+
# Named subsets this case belongs to (e.g. ["smoke"], ["full", "regression"]).
|
|
56
|
+
# ``run(dataset_id, split="smoke")`` runs only cases tagged with that split.
|
|
57
|
+
splits: Optional[List[str]] = None
|
|
55
58
|
|
|
56
59
|
class Config:
|
|
57
60
|
populate_by_name = True
|
|
@@ -412,6 +415,11 @@ class RunResultRow(BaseModel):
|
|
|
412
415
|
|
|
413
416
|
rating: Optional[float] = None
|
|
414
417
|
justification: Optional[str] = None
|
|
418
|
+
# scored | skipped (the judge could not score this row) | failed (the submitted result
|
|
419
|
+
# carried an error). A skipped row has rating None - a fact about the judge, not a 0.
|
|
420
|
+
status: Optional[str] = None
|
|
421
|
+
question_index: Optional[int] = Field(default=None, alias="questionIndex")
|
|
422
|
+
run_number: Optional[int] = Field(default=None, alias="runNumber")
|
|
415
423
|
question_text: Optional[str] = Field(default=None, alias="questionText")
|
|
416
424
|
response: Optional[str] = None
|
|
417
425
|
trace_id: Optional[str] = Field(default=None, alias="traceId")
|
|
@@ -4,12 +4,12 @@ import logging
|
|
|
4
4
|
import os
|
|
5
5
|
import time
|
|
6
6
|
import uuid
|
|
7
|
-
from typing import Any, Callable, Dict, List, Optional, Set, Union
|
|
7
|
+
from typing import Any, Callable, Dict, Iterator, List, Optional, Set, Union
|
|
8
8
|
|
|
9
9
|
from agentx.evaluations.adapters.raw import RawCallableAdapter
|
|
10
10
|
from agentx.evaluations.adapters.precomputed import PrecomputedAdapter
|
|
11
11
|
from agentx.evaluations.adapters.http_endpoint import HttpEndpointAdapter
|
|
12
|
-
from agentx.evaluations.client import EvaluationsClient
|
|
12
|
+
from agentx.evaluations.client import EvaluationsClient, EvaluationSubmissionError
|
|
13
13
|
from agentx.evaluations.models import (
|
|
14
14
|
AnalysisStatus,
|
|
15
15
|
Dataset,
|
|
@@ -96,11 +96,14 @@ class EvaluationRunContext:
|
|
|
96
96
|
run: EvaluationRun,
|
|
97
97
|
subject: EvaluationSubject,
|
|
98
98
|
evaluation_settings: Optional[EvaluationSettings] = None,
|
|
99
|
+
split: Optional[str] = None,
|
|
99
100
|
):
|
|
100
101
|
self._client = client
|
|
101
102
|
self._dataset = dataset
|
|
102
103
|
self._run = run
|
|
103
104
|
self._subject = subject
|
|
105
|
+
# Named case subset for this run - only cases tagged with it are built/executed.
|
|
106
|
+
self._split = split
|
|
104
107
|
# When set, this run was started with an independently chosen grading
|
|
105
108
|
# config (evaluation_settings_id) - its fields take precedence over the
|
|
106
109
|
# dataset's own for anything execution-time reads (see _build_cases).
|
|
@@ -118,25 +121,63 @@ class EvaluationRunContext:
|
|
|
118
121
|
# Step 1: execute
|
|
119
122
|
# ------------------------------------------------------------------
|
|
120
123
|
|
|
121
|
-
def execute(
|
|
124
|
+
def execute(
|
|
125
|
+
self,
|
|
126
|
+
adapter: AdapterLike,
|
|
127
|
+
concurrency: int = 1,
|
|
128
|
+
reuse_outputs_from: Optional[str] = None,
|
|
129
|
+
) -> "EvaluationRunContext":
|
|
122
130
|
"""Run all cases locally and submit batches to AgentX.
|
|
123
131
|
|
|
124
132
|
The whole loop runs inside the eval-run scope (tracing/eval_scope.py): any trace the
|
|
125
133
|
agent function creates is stamped source="eval-run" + monitor=False automatically, so
|
|
126
134
|
eval traffic never skews production monitoring and no one has to remember a flag.
|
|
135
|
+
|
|
136
|
+
``concurrency`` > 1 runs the agent callable across a thread pool (results are still
|
|
137
|
+
submitted in case order, and the eval-run scope is propagated into the workers).
|
|
138
|
+
``reuse_outputs_from`` replays a previous run's recorded outputs for cases whose query
|
|
139
|
+
text is unchanged instead of re-running (and re-paying for) the agent - the judge still
|
|
140
|
+
re-scores them, which makes iterating on scorers cheap. Changed or new cases run
|
|
141
|
+
normally.
|
|
127
142
|
"""
|
|
128
143
|
from agentx.tracing.eval_scope import enter_eval_run, exit_eval_run
|
|
129
144
|
|
|
130
145
|
scope_token = enter_eval_run(self._run.run_id)
|
|
131
146
|
try:
|
|
132
|
-
return self._execute_inner(adapter)
|
|
147
|
+
return self._execute_inner(adapter, concurrency=concurrency, reuse_outputs_from=reuse_outputs_from)
|
|
133
148
|
finally:
|
|
134
149
|
exit_eval_run(scope_token)
|
|
135
150
|
|
|
136
|
-
def
|
|
151
|
+
def _fetch_reusable_outputs(self, run_id: str) -> Dict[tuple, str]:
|
|
152
|
+
"""(query, run_number, is_smoke_variant) -> output text, from a previous run's rows.
|
|
153
|
+
Keyed on the query TEXT so a reworded case never silently reuses a stale answer."""
|
|
154
|
+
try:
|
|
155
|
+
prior = self._client.get_run(run_id)
|
|
156
|
+
except Exception as exc:
|
|
157
|
+
logger.warning("reuse_outputs_from: could not load run %s (%s) - running everything", run_id, exc)
|
|
158
|
+
return {}
|
|
159
|
+
reusable: Dict[tuple, str] = {}
|
|
160
|
+
for row in prior.get("results", []) or []:
|
|
161
|
+
if row.get("error") or row.get("status") == "failed":
|
|
162
|
+
continue
|
|
163
|
+
output = (row.get("output") or {}).get("text")
|
|
164
|
+
query = (row.get("input") or {}).get("query")
|
|
165
|
+
if not output or not query:
|
|
166
|
+
continue
|
|
167
|
+
key = (query, row.get("runNumber") or 1, bool(row.get("isSmokeTestVariant")))
|
|
168
|
+
reusable[key] = output
|
|
169
|
+
return reusable
|
|
170
|
+
|
|
171
|
+
def _execute_inner(
|
|
172
|
+
self,
|
|
173
|
+
adapter: AdapterLike,
|
|
174
|
+
concurrency: int = 1,
|
|
175
|
+
reuse_outputs_from: Optional[str] = None,
|
|
176
|
+
) -> "EvaluationRunContext":
|
|
137
177
|
normalized = _wrap_adapter(adapter)
|
|
138
|
-
cases = _build_cases(self._dataset, self._run, self._evaluation_settings)
|
|
178
|
+
cases = _build_cases(self._dataset, self._run, self._evaluation_settings, split=self._split)
|
|
139
179
|
max_batch = self._run.limits.max_batch_size
|
|
180
|
+
reusable = self._fetch_reusable_outputs(reuse_outputs_from) if reuse_outputs_from else {}
|
|
140
181
|
|
|
141
182
|
# Banner
|
|
142
183
|
sep = "─" * 60
|
|
@@ -144,7 +185,7 @@ class EvaluationRunContext:
|
|
|
144
185
|
framework = self._subject.framework or "custom"
|
|
145
186
|
runtime = self._subject.runtime or "local"
|
|
146
187
|
display = self._subject.display_name or ""
|
|
147
|
-
n_q = len(
|
|
188
|
+
n_q = len({c.question_index for c in cases if not c.is_smoke_test_variant})
|
|
148
189
|
n_r = (
|
|
149
190
|
self._evaluation_settings.number_of_requests
|
|
150
191
|
if self._evaluation_settings
|
|
@@ -171,6 +212,45 @@ class EvaluationRunContext:
|
|
|
171
212
|
batch: List[EvaluationResult] = []
|
|
172
213
|
total = len(cases)
|
|
173
214
|
|
|
215
|
+
def produce(case: EvaluationCase) -> EvaluationResult:
|
|
216
|
+
# Cached replay: same query text at the same repetition reuses the recorded output
|
|
217
|
+
# (the server still re-scores it with THIS run's grading config).
|
|
218
|
+
cached = reusable.get((case.query, case.run_number, case.is_smoke_test_variant))
|
|
219
|
+
if cached is not None:
|
|
220
|
+
return normalize_result(
|
|
221
|
+
case, {"output": cached, "metadata": {"reusedFromRun": reuse_outputs_from}}
|
|
222
|
+
)
|
|
223
|
+
return normalized(case)
|
|
224
|
+
|
|
225
|
+
if concurrency > 1:
|
|
226
|
+
import concurrent.futures
|
|
227
|
+
import contextvars
|
|
228
|
+
|
|
229
|
+
def in_scope(case: EvaluationCase) -> EvaluationResult:
|
|
230
|
+
# ContextVars (the eval-run scope) do not cross thread boundaries on their own -
|
|
231
|
+
# each worker task runs inside a copy of the submitting thread's context so the
|
|
232
|
+
# agent's traces still get stamped source="eval-run".
|
|
233
|
+
return contextvars.copy_context().run(produce, case)
|
|
234
|
+
|
|
235
|
+
pending = [
|
|
236
|
+
case
|
|
237
|
+
for case in cases
|
|
238
|
+
if _idem_key(self._run.run_id, case.case_id, case.run_number) not in already_done
|
|
239
|
+
]
|
|
240
|
+
executor = concurrent.futures.ThreadPoolExecutor(max_workers=concurrency)
|
|
241
|
+
# map() yields in submission order, so batching/submission below stays deterministic.
|
|
242
|
+
mapped = executor.map(in_scope, pending)
|
|
243
|
+
|
|
244
|
+
def ordered() -> "Iterator[EvaluationResult]":
|
|
245
|
+
try:
|
|
246
|
+
yield from mapped
|
|
247
|
+
finally:
|
|
248
|
+
executor.shutdown(wait=True)
|
|
249
|
+
|
|
250
|
+
results_iter = ordered()
|
|
251
|
+
else:
|
|
252
|
+
results_iter = None # sequential path below produces inline
|
|
253
|
+
|
|
174
254
|
for idx, case in enumerate(cases, start=1):
|
|
175
255
|
idem_key = _idem_key(self._run.run_id, case.case_id, case.run_number)
|
|
176
256
|
|
|
@@ -179,7 +259,7 @@ class EvaluationRunContext:
|
|
|
179
259
|
_print_progress(idx, total, case, skipped=True)
|
|
180
260
|
continue
|
|
181
261
|
|
|
182
|
-
result =
|
|
262
|
+
result = next(results_iter) if results_iter is not None else produce(case)
|
|
183
263
|
result.idempotency_key = idem_key
|
|
184
264
|
# Tag the result with the case's model so the server can group it into
|
|
185
265
|
# the Sovereignty & Portability matrix (the callable may also set it).
|
|
@@ -207,31 +287,46 @@ class EvaluationRunContext:
|
|
|
207
287
|
batch_id = str(uuid.uuid4())
|
|
208
288
|
n = len(batch)
|
|
209
289
|
with Spinner(f"Scoring - AI is rating {n} result{'s' if n != 1 else ''}"):
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
self.
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
290
|
+
last_exc: Optional[Exception] = None
|
|
291
|
+
for attempt in (1, 2):
|
|
292
|
+
try:
|
|
293
|
+
resp = self._client.append_results(self._run.run_id, batch_id, batch)
|
|
294
|
+
if resp.live_statistics is not None:
|
|
295
|
+
self._live_stats = resp.live_statistics
|
|
296
|
+
_say(
|
|
297
|
+
f" {green('✓')} Scored {resp.accepted} result{'s' if resp.accepted != 1 else ''}"
|
|
298
|
+
)
|
|
299
|
+
logger.info(
|
|
300
|
+
"Batch %s: accepted=%d duplicates=%d failed=%d",
|
|
301
|
+
batch_id[:8],
|
|
302
|
+
resp.accepted,
|
|
303
|
+
resp.duplicates,
|
|
304
|
+
resp.failed_validation,
|
|
305
|
+
)
|
|
306
|
+
return
|
|
307
|
+
except Exception as exc:
|
|
308
|
+
last_exc = exc
|
|
309
|
+
if attempt == 1:
|
|
310
|
+
logger.warning("Batch %s submission failed, retrying once: %s", batch_id[:8], exc)
|
|
311
|
+
# A batch that cannot be submitted must FAIL the run, not print a red line and carry on:
|
|
312
|
+
# execute() used to finish "successfully" having uploaded nothing. Failing fast also
|
|
313
|
+
# stops paying for agent calls whose results would hit the same broken engine; a
|
|
314
|
+
# re-execute() of this context resumes past everything already accepted (idempotency
|
|
315
|
+
# keys are deterministic and the engine returns the submitted set).
|
|
316
|
+
_say(f" {red('✗')} Scoring failed: {dim(str(last_exc))}")
|
|
317
|
+
logger.error("Failed to submit batch %s after retry: %s", batch_id[:8], last_exc)
|
|
318
|
+
raise EvaluationSubmissionError(
|
|
319
|
+
f"Failed to submit a batch of {n} result(s) to the engine after a retry: {last_exc}. "
|
|
320
|
+
"The run was left unfinalized; re-running execute() resumes past already-submitted cases."
|
|
321
|
+
) from last_exc
|
|
227
322
|
|
|
228
323
|
def _fetch_submitted_keys(self) -> Set[str]:
|
|
324
|
+
"""Keys already accepted by this run - the engine's /missing-results route returns them
|
|
325
|
+
so a re-execute() after a crash skips (and never re-pays for) finished cases."""
|
|
229
326
|
try:
|
|
230
|
-
|
|
231
|
-
# missing-results returns cases NOT yet submitted - we want the inverse
|
|
232
|
-
# but if the endpoint isn't live yet, just return empty set
|
|
233
|
-
return set()
|
|
327
|
+
return set(self._client.get_submitted_keys(self._run.run_id))
|
|
234
328
|
except Exception:
|
|
329
|
+
# Older engines without the route: no resume, identical to the historical behavior.
|
|
235
330
|
return set()
|
|
236
331
|
|
|
237
332
|
# ------------------------------------------------------------------
|
|
@@ -517,11 +612,16 @@ class EvaluationsRunner:
|
|
|
517
612
|
subject: Union[Dict[str, Any], EvaluationSubject],
|
|
518
613
|
scorer_id: Optional[str] = None,
|
|
519
614
|
evaluation_settings_id: Optional[str] = None,
|
|
615
|
+
split: Optional[str] = None,
|
|
520
616
|
) -> EvaluationRunContext:
|
|
521
617
|
"""Start a run of ``dataset_id`` against ``subject``. Pass ``scorer_id`` (an LLM Judge
|
|
522
618
|
Scorer's id, e.g. from ``client.monitor.judge_scorers``) to grade with a specific
|
|
523
619
|
scorer instead of the dataset's default. ``evaluation_settings_id`` is the
|
|
524
|
-
pre-consolidation alias for the same id and keeps working.
|
|
620
|
+
pre-consolidation alias for the same id and keeps working.
|
|
621
|
+
|
|
622
|
+
``split`` runs only the cases tagged with that named subset (``add_case(...,
|
|
623
|
+
splits=["smoke"])``) - the cheap-PR-run vs nightly-full-run workflow. Original case
|
|
624
|
+
indexes are preserved so per-case comparisons line up with full runs."""
|
|
525
625
|
from agentx.evaluations.client import _resolve_scorer_id
|
|
526
626
|
|
|
527
627
|
if isinstance(subject, dict):
|
|
@@ -532,18 +632,24 @@ class EvaluationsRunner:
|
|
|
532
632
|
evaluation_settings = (
|
|
533
633
|
self._client.get_evaluation_settings(grader_id) if grader_id else None
|
|
534
634
|
)
|
|
535
|
-
run = self._client.init_run(dataset_id, subject, scorer_id=grader_id)
|
|
635
|
+
run = self._client.init_run(dataset_id, subject, scorer_id=grader_id, split=split)
|
|
636
|
+
case_count = (
|
|
637
|
+
sum(1 for q in dataset.questions if split in (q.main_question.splits or []))
|
|
638
|
+
if split
|
|
639
|
+
else len(dataset.questions)
|
|
640
|
+
)
|
|
536
641
|
logger.info(
|
|
537
|
-
"Started evaluation run %s on dataset %s (%d case(s), %d repetition(s))",
|
|
642
|
+
"Started evaluation run %s on dataset %s (%d case(s)%s, %d repetition(s))",
|
|
538
643
|
run.run_id,
|
|
539
644
|
dataset_id,
|
|
540
|
-
|
|
645
|
+
case_count,
|
|
646
|
+
f' in split "{split}"' if split else "",
|
|
541
647
|
evaluation_settings.number_of_requests
|
|
542
648
|
if evaluation_settings
|
|
543
649
|
else dataset.number_of_requests,
|
|
544
650
|
)
|
|
545
651
|
return EvaluationRunContext(
|
|
546
|
-
self._client, dataset, run, subject, evaluation_settings=evaluation_settings
|
|
652
|
+
self._client, dataset, run, subject, evaluation_settings=evaluation_settings, split=split
|
|
547
653
|
)
|
|
548
654
|
|
|
549
655
|
|
|
@@ -568,6 +674,7 @@ def _build_cases(
|
|
|
568
674
|
dataset: Dataset,
|
|
569
675
|
run: EvaluationRun,
|
|
570
676
|
evaluation_settings: Optional[EvaluationSettings] = None,
|
|
677
|
+
split: Optional[str] = None,
|
|
571
678
|
) -> List[EvaluationCase]:
|
|
572
679
|
cases: List[EvaluationCase] = []
|
|
573
680
|
# When an independent evaluation_settings was chosen (evaluation_settings_id
|
|
@@ -596,6 +703,10 @@ def _build_cases(
|
|
|
596
703
|
}
|
|
597
704
|
for q_idx, question in enumerate(dataset.questions):
|
|
598
705
|
mq = question.main_question
|
|
706
|
+
# Split filtering preserves q_idx: a "smoke" run's case 7 is the same case 7 a full run
|
|
707
|
+
# scores, so per-case comparisons line up across the two.
|
|
708
|
+
if split and split not in (mq.splits or []):
|
|
709
|
+
continue
|
|
599
710
|
for run_num in range(1, n_runs + 1):
|
|
600
711
|
for model in models:
|
|
601
712
|
suffix = f"::{model}" if model else ""
|
|
@@ -39,6 +39,35 @@ class AgentXValidationError(AgentXMonitorError):
|
|
|
39
39
|
pass
|
|
40
40
|
|
|
41
41
|
|
|
42
|
+
class CalibrationSummary(dict):
|
|
43
|
+
"""Judge Calibration numbers (dict subclass, so existing key access keeps working).
|
|
44
|
+
Properties mirror the wire's exact camelCase keys."""
|
|
45
|
+
|
|
46
|
+
@property
|
|
47
|
+
def compared_count(self) -> int:
|
|
48
|
+
return int(self.get("comparedCount") or 0)
|
|
49
|
+
|
|
50
|
+
@property
|
|
51
|
+
def agreement_rate(self):
|
|
52
|
+
return self.get("agreementRate")
|
|
53
|
+
|
|
54
|
+
@property
|
|
55
|
+
def false_positive_rate(self):
|
|
56
|
+
return self.get("falsePositiveRate")
|
|
57
|
+
|
|
58
|
+
@property
|
|
59
|
+
def false_negative_rate(self):
|
|
60
|
+
return self.get("falseNegativeRate")
|
|
61
|
+
|
|
62
|
+
@property
|
|
63
|
+
def reported_count(self) -> int:
|
|
64
|
+
return int(self.get("reportedCount") or 0)
|
|
65
|
+
|
|
66
|
+
@property
|
|
67
|
+
def review_label_count(self) -> int:
|
|
68
|
+
return int(self.get("reviewLabelCount") or 0)
|
|
69
|
+
|
|
70
|
+
|
|
42
71
|
class MonitorClient:
|
|
43
72
|
"""Low-level HTTP client for the Monitor API (``/monitor``). Accessed via
|
|
44
73
|
``client.monitor`` on the top-level :class:`agentx.AgentX` instance; most callers
|
|
@@ -82,6 +111,11 @@ class MonitorClient:
|
|
|
82
111
|
|
|
83
112
|
self.patterns = MonitorPatternClient(self)
|
|
84
113
|
self.signals = MonitorSignalClient(self)
|
|
114
|
+
from agentx.monitor.review_queue import ReviewQueueClient
|
|
115
|
+
|
|
116
|
+
# The human-review queue (list / queue / label / dismiss) - what makes the
|
|
117
|
+
# label-and-calibrate loop scriptable instead of dashboard-only.
|
|
118
|
+
self.review_queue = ReviewQueueClient(self)
|
|
85
119
|
from agentx.monitor.scorers import ScorersClient
|
|
86
120
|
# Scorers-catalog administration as code: template enable/disable, code/external scorer
|
|
87
121
|
# CRUD and dry runs - full parity with the dashboard's Scorers page (P1.3).
|
|
@@ -230,15 +264,19 @@ class MonitorClient:
|
|
|
230
264
|
plus deltas vs the prior window and the run-outcome breakdown."""
|
|
231
265
|
return self._request("GET", "/kpis", params={"window": window})
|
|
232
266
|
|
|
233
|
-
def calibration(self, window: str = "7d") ->
|
|
267
|
+
def calibration(self, window: str = "7d") -> "CalibrationSummary":
|
|
234
268
|
"""Project-level judge calibration over a window ("24h", "7d", or "30d"): how often
|
|
235
269
|
AgentX's own verdicts agreed with real-world ground truth reported later (ops outcomes
|
|
236
|
-
via ``client.outcomes
|
|
237
|
-
Calibration numbers
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
270
|
+
via ``client.outcomes``, end-user downvotes, and human review labels). Returns the
|
|
271
|
+
dashboard's Judge Calibration numbers with these exact keys: ``comparedCount``,
|
|
272
|
+
``agreementRate``, ``falsePositiveRate``, ``falseNegativeRate`` (plus
|
|
273
|
+
``reportedCount``/``reviewLabelCount``/``noVerdictCount``). Per-scorer calibration
|
|
274
|
+
lives on ``client.monitor.judge_scorers.calibration(scorer_id)``."""
|
|
275
|
+
return CalibrationSummary(
|
|
276
|
+
self._request(
|
|
277
|
+
"GET", "/agent-monitoring/calibration",
|
|
278
|
+
base=self._api_root(), params={"window": window},
|
|
279
|
+
)
|
|
242
280
|
)
|
|
243
281
|
|
|
244
282
|
# ------------------------------------------------------------------
|
|
@@ -352,10 +390,19 @@ class MonitorClient:
|
|
|
352
390
|
base=self._api_root(), json={**criteria, "window": window}, timeout=600,
|
|
353
391
|
)
|
|
354
392
|
|
|
355
|
-
def publish_online_evaluator_tuning(
|
|
393
|
+
def publish_online_evaluator_tuning(
|
|
394
|
+
self, evaluator_id: str, criteria: dict, *, validation: Optional[dict] = None, force: bool = False
|
|
395
|
+
) -> dict:
|
|
396
|
+
# The engine gates publish on validation provenance (and refuses a measured regression)
|
|
397
|
+
# unless forced - see judge_scorers.publish_tuning for the full story.
|
|
398
|
+
payload = dict(criteria)
|
|
399
|
+
if validation is not None:
|
|
400
|
+
payload["validation"] = validation
|
|
401
|
+
if force:
|
|
402
|
+
payload["force"] = True
|
|
356
403
|
return self._request(
|
|
357
404
|
"POST", f"/agent-monitoring/online-evaluators/{evaluator_id}/tune/publish",
|
|
358
|
-
base=self._api_root(), json=
|
|
405
|
+
base=self._api_root(), json=payload, timeout=60,
|
|
359
406
|
)
|
|
360
407
|
|
|
361
408
|
def update_profile(self, agent_id: str, payload: dict) -> MonitorProfile:
|
|
@@ -280,12 +280,30 @@ class JudgeScorersClient:
|
|
|
280
280
|
timeout=600,
|
|
281
281
|
)
|
|
282
282
|
|
|
283
|
-
def publish_tuning(
|
|
283
|
+
def publish_tuning(
|
|
284
|
+
self,
|
|
285
|
+
scorer_id: str,
|
|
286
|
+
criteria: Dict[str, Any],
|
|
287
|
+
*,
|
|
288
|
+
validation: Optional[Dict[str, Any]] = None,
|
|
289
|
+
force: bool = False,
|
|
290
|
+
) -> dict:
|
|
284
291
|
"""Write tuned criteria onto the scorer's rubric - it applies everywhere the scorer is
|
|
285
|
-
used: online scoring, offline dataset runs, and the playground.
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
)
|
|
292
|
+
used: online scoring, offline dataset runs, and the playground.
|
|
293
|
+
|
|
294
|
+
The engine gates publish on provenance: pass ``validation`` (the dict returned by
|
|
295
|
+
``validate_tuning``, or at least its ``verdict``/``netAgreementGain``) so the version
|
|
296
|
+
history records what the change measurably did; a ``regressed`` verdict is refused.
|
|
297
|
+
``force=True`` publishes without (or despite) validation - deliberate escape hatch."""
|
|
298
|
+
payload = dict(criteria)
|
|
299
|
+
if validation is not None:
|
|
300
|
+
payload["validation"] = {
|
|
301
|
+
"verdict": validation.get("verdict"),
|
|
302
|
+
"netAgreementGain": validation.get("netAgreementGain"),
|
|
303
|
+
}
|
|
304
|
+
if force:
|
|
305
|
+
payload["force"] = True
|
|
306
|
+
return self._request("POST", f"/online-evaluators/{self._profile_id(scorer_id)}/tune/publish", json=payload)
|
|
289
307
|
|
|
290
308
|
def ratings(self, scorer_id: str, window: str = "7d") -> "List[OnlineEvaluatorRatingPoint]":
|
|
291
309
|
"""Bucketed average-rating-over-time for this scorer's live checks - same typed points
|
|
@@ -157,9 +157,13 @@ class MonitorOnlineEvaluatorClient:
|
|
|
157
157
|
is {acceptanceCriteria, rejectionCriteria, evaluationCriteria} from tune()."""
|
|
158
158
|
return self._client.validate_online_evaluator_tuning(evaluator_id, criteria, window)
|
|
159
159
|
|
|
160
|
-
def publish_tuning(
|
|
161
|
-
|
|
162
|
-
|
|
160
|
+
def publish_tuning(
|
|
161
|
+
self, evaluator_id: str, criteria: dict, *, validation: Optional[dict] = None, force: bool = False
|
|
162
|
+
) -> dict:
|
|
163
|
+
"""Publish tuned criteria onto the evaluator's config (the human-approval step).
|
|
164
|
+
Pass ``validation`` (the ``validate_tuning`` result) - the engine refuses an unvalidated
|
|
165
|
+
publish, and a ``regressed`` verdict, unless ``force=True``."""
|
|
166
|
+
return self._client.publish_online_evaluator_tuning(evaluator_id, criteria, validation=validation, force=force)
|
|
163
167
|
|
|
164
168
|
def ratings(self, evaluator_id: str, window: str = "7d") -> List[OnlineEvaluatorRatingPoint]:
|
|
165
169
|
"""Bucketed average-rating-over-time for this evaluator. ``window`` is one of
|