From ce65d06ba67ca4fad85ba3b3f71a1eec359bc6e5 Mon Sep 17 00:00:00 2001 From: Jer Miller Date: Thu, 23 Jul 2026 13:58:34 -0600 Subject: [PATCH] test: update classifier caller and stats baseline for endpoint-budget contract Full make ci surfaced two mechanical consequences of the endpoint-budget contract change: update the _classify_byo_generate_error test caller in test_describe_capacity_retry.py for its new (exc, endpoint) signature, and refresh the stats API baseline to reflect the segment_summary talent's new max_output_tokens: 1024. No product change. Co-Authored-By: Claude Opus 4.8 (1M context) --- tests/baselines/api/stats/stats.json | 1 + tests/test_describe_capacity_retry.py | 9 ++++++++- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/tests/baselines/api/stats/stats.json b/tests/baselines/api/stats/stats.json index de5499f37..e69454805 100644 --- a/tests/baselines/api/stats/stats.json +++ b/tests/baselines/api/stats/stats.json @@ -430,6 +430,7 @@ "talents": false, "transcripts": false }, + "max_output_tokens": 1024, "mtime": 0, "output": "json", "path": "/solstone/apps/timeline/talent/segment_summary.md", diff --git a/tests/test_describe_capacity_retry.py b/tests/test_describe_capacity_retry.py index 1942c2235..72e493b80 100644 --- a/tests/test_describe_capacity_retry.py +++ b/tests/test_describe_capacity_retry.py @@ -65,11 +65,18 @@ def _assert_no_describe_temp(directory: Path) -> None: def _capacity_error(): from solstone.think.providers import local as local_provider + from solstone.think.providers.local_endpoint import LocalEndpoint inner = type("ReadTimeout", (Exception,), {})("capacity wait timed out") outer = RuntimeError("outer") outer.__cause__ = inner - return local_provider._classify_byo_generate_error(outer) + endpoint = LocalEndpoint( + base_url="http://endpoint.test", + served_model_id="m", + credential=None, + is_bundled=False, + ) + return local_provider._classify_byo_generate_error(outer, endpoint) class RetryBatch: -- 2.51.2