sweatstack 0.76.2__tar.gz → 0.77.0__tar.gz

{sweatstack-0.76.2 → sweatstack-0.77.0}/CHANGELOG.md

@@ -6,6 +6,16 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 
+## [0.77.0] - 2026-05-19
+
+### Added
+- Link a trace to a test via the new `test_id` argument on `create_trace()` and `update_trace()`. A linked trace appears in the test's traces regardless of its timestamp.
+- `get_test()` accepts `trace_resolution=TraceResolution.linked` to return only traces explicitly linked to the test. Defaults to `TraceResolution.auto` (unchanged behaviour).
+
+### Changed
+- `update_trace()` replaces all fields, including `test_id`. Callers that omit `test_id` will clear any existing link — pass it back in to keep the trace linked.
+
+
 ## [0.76.2] - 2026-04-27
 
 ### Fixed

{sweatstack-0.76.2 → sweatstack-0.77.0}/CONTRIBUTING.md

@@ -6,4 +6,4 @@ This repository is open-sourced primarily to make it easier for developers and A
 
 If you've found a bug or have a question about using the library, feel free to [open an issue](https://github.com/SweatStack/sweatstack-python/issues).
 
-For general questions about SweatStack, please refer to the [developer documentation](https://developer.sweatstack.no/getting-started/).
+For general questions about SweatStack, please refer to the [developer documentation](https://docs.sweatstack.no/getting-started/).

{sweatstack-0.76.2 → sweatstack-0.77.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: sweatstack
-Version: 0.76.2
+Version: 0.77.0
 Summary: The official Python client for SweatStack
 Project-URL: Homepage, https://sweatstack.no
-Project-URL: Documentation, https://developer.sweatstack.no/getting-started/
+Project-URL: Documentation, https://docs.sweatstack.no/getting-started/
 Project-URL: Repository, https://github.com/SweatStack/sweatstack-python
 Project-URL: Issues, https://github.com/SweatStack/sweatstack-python/issues
 Project-URL: Changelog, https://github.com/SweatStack/sweatstack-python/blob/main/CHANGELOG.md
@@ -43,4 +43,4 @@ Description-Content-Type: text/markdown
 
 This is the official Python client library for SweatStack.
 
-Documentation can be found [here](https://developer.sweatstack.no/getting-started/).
+Documentation can be found [here](https://docs.sweatstack.no/getting-started/).

{sweatstack-0.76.2 → sweatstack-0.77.0}/README.md

@@ -2,4 +2,4 @@
 
 This is the official Python client library for SweatStack.
 
-Documentation can be found [here](https://developer.sweatstack.no/getting-started/).
+Documentation can be found [here](https://docs.sweatstack.no/getting-started/).

sweatstack-0.77.0/plans/003_trace_test_linking.md

@@ -0,0 +1,300 @@
+# Plan: Trace ↔ Test linking + `TraceResolution`
+
+## Summary
+
+Mirror upstream `sweatstack@93cd588` ("Adds trace to test optional foreign key
+relationship") in the Python client:
+
+- Traces gain an optional `test_id` FK that *explicitly* links them to a test
+  (independent of timestamp).
+- `GET /api/v1/tests/{test_id}` gains a `traces` query parameter
+  (`TraceResolution` enum: `auto` | `linked`) that controls how the returned
+  `traces` list is resolved.
+- `POST /api/v1/traces/` and `PUT /api/v1/traces/{trace_id}` may now return
+  `404` when a referenced `test_id` does not exist.
+
+Server-side behaviour:
+
+- `auto` (default): traces whose timestamp falls in the test's window, **plus**
+  traces explicitly linked to this test, **minus** traces explicitly linked to a
+  *different* test.
+- `linked`: only traces explicitly linked via `test_id`; the time window is
+  ignored.
+
+The `activities` list is unaffected — always time-overlap matched.
+
+## Code Quality Requirements
+
+Same bar as plan 001a:
+
+- Follow existing patterns exactly (keyword-only params, enum handling,
+  `_enums_to_strings`, error surface via `_raise_for_status`, docstring style).
+- Comprehensive type hints. No `Any`. No string-typed enums on the surface.
+- Concise Google-style docstrings (Args / Returns / Raises).
+- No dead code, no TODOs, no half-finished branches.
+
+## Step 1: Regenerate `openapi_schemas.py`
+
+The **user will run the SweatStack server locally** at
+`http://localhost:8080`. Wait for confirmation that the server is up
+before running the regen, and do not attempt to start the server.
+
+Then:
+
+```bash
+uv run generate-response-models
+```
+
+Expected diff in `src/sweatstack/openapi_schemas.py`:
+
+- `TraceDetails`: new `test_id: str | None = Field(None, title='Test Id')`.
+- `TraceCreateOrUpdate`: new `test_id: str | None = Field(None, title='Test Id')`.
+- New top-level enum:
+
+  ```python
+  class TraceResolution(Enum):
+      auto = 'auto'
+      linked = 'linked'
+  ```
+
+**Caveat — regen pulls more than this feature.** Two intervening upstream
+commits since `openapi_schemas.py` was last regenerated also touch the
+schema/API surface: `feebe56` ("Adds security.txt …") and `3f29c1f`
+("Better handling of tags=null"). The regenerated diff will not be
+confined to this plan's scope.
+
+Action: review the full diff. If unrelated changes appear, **split them
+into a separate commit** ahead of the trace-linking commit so the
+trace-linking commit stays reviewable. Do not merge regen-noise as part
+of this feature.
+
+Also verify on the running server before committing: open
+`http://localhost:8080/openapi.json`, search for `TraceResolution`,
+confirm wire values are `"auto"` and `"linked"` (lowercase). This guards
+against an upstream `BaseEnum` casing change.
+
+## Step 2: Re-export `TraceResolution`
+
+Two-step exposure (mirrors `Sport`):
+
+1. `src/sweatstack/schemas.py` — add `TraceResolution` to the import from
+   `openapi_schemas`.
+2. `src/sweatstack/client.py` — add `TraceResolution` to the import block
+   pulling enums/schemas from `.schemas` (≈ line 45, where `Sport`,
+   `TraceDetails`, etc. are imported).
+
+`__init__.py` is `from .client import *`, so step 2 is what actually makes
+`from sweatstack import TraceResolution` work. Verify with a smoke test
+in the REPL after the change. (This is the step that's easy to skip and
+breaks the public surface.)
+
+## Step 3: Update `src/sweatstack/client.py`
+
+Import `TraceResolution` from `schemas` in the existing imports block
+(≈ line 45, alongside `TraceDetails`).
+
+### 3a. `create_trace` — add `test_id` keyword arg
+
+Location: ≈ line 1708.
+
+- Add `test_id: str | None = None` as the **last** keyword-only param
+  (preserves existing call-site compatibility — every existing caller uses
+  keyword args, but appending keeps the visual ordering stable).
+- Add `"test_id": test_id` to the JSON body. Do not omit when `None`; the
+  server accepts `null` and other optional fields are already passed as
+  `None` (consistent with `lactate`, `rpe`, etc.).
+- Docstring:
+  - Add `test_id: Optional ID of a test to explicitly link this trace to.
+    If the test does not exist, the API raises 404.` to Args.
+  - Add a `Raises:` note that an `HTTPStatusError(404)` is raised when
+    `test_id` references a non-existent test (existing `_raise_for_status`
+    already surfaces this; just document it).
+
+### 3b. `update_trace` — add `test_id` keyword arg
+
+Location: ≈ line 1762.
+
+Identical treatment to `create_trace`. Same docstring additions.
+
+**Behavioural-change callout (required in both docstring and CHANGELOG):**
+`update_trace` is full-replace. Once `test_id` is in the signature with a
+default of `None`, **existing callers that don't pass `test_id` will silently
+unlink any previously linked test** the next time they call
+`update_trace`. This is consistent with how every other optional field on
+`update_trace` already behaves (e.g. `lactate=None` nulls lactate), but
+it's a real, observable change for upgraders.
+
+Docstring wording suggestion:
+
+> Note: This is a full-replace operation. If a trace was previously linked
+> to a test via `test_id`, you must pass that `test_id` again to preserve
+> the link — otherwise it is cleared.
+
+Do not try to work around this with "preserve if omitted" sentinels — that
+would diverge from how every other field on this method behaves and rot
+the contract.
+
+### 3c. `get_test` — add `trace_resolution` keyword arg
+
+Location: ≈ line 1934.
+
+Current signature:
+
+```python
+def get_test(self, test_id: str) -> TestDetails:
+```
+
+New signature:
+
+```python
+def get_test(
+    self,
+    test_id: str,
+    *,
+    trace_resolution: TraceResolution = TraceResolution.auto,
+) -> TestDetails:
+```
+
+Naming decision: **`trace_resolution`, not `traces`.** The wire name is
+`traces=` (server query param), but on the Python method
+`client.get_test("t1", traces=TraceResolution.linked)` reads as "give me
+the test and traces=X" — ambiguous. `trace_resolution=` is
+self-documenting. Note the wire mapping in the docstring so future
+maintainers know they don't match.
+
+Typing decision: **strict `TraceResolution`, not `TraceResolution | str`.**
+This diverges from the `Sport | str` pattern used elsewhere in the
+client. The deviation is deliberate: the enum has exactly two values, it
+is a brand-new surface (no historical string callers to support), and
+the brief asks for "extremely clean". Document the deviation in a code
+review note when the PR goes up; don't surprise the reviewer.
+
+Implementation:
+
+```python
+with self._http_client() as client:
+    params = {}
+    if trace_resolution is not TraceResolution.auto:
+        params["traces"] = trace_resolution.value
+    response = client.get(
+        url=f"/api/v1/tests/{test_id}",
+        params=params,
+    )
+    self._raise_for_status(response)
+    return TestDetails.model_validate(response.json())
+```
+
+Notes:
+
+- Use `.value` (lowercase `"auto"` / `"linked"`) explicitly. `httpx` does
+  not unwrap `Enum.value` on its own — it `str()`s the value, which gives
+  `"TraceResolution.linked"`. Verified pattern: `_get_tests_generator` and
+  every other enum query usage in this file calls `.value`.
+- **Omit the param when at default** — matches the convention used by
+  `_get_activities_generator`, `_get_tests_generator`, etc., which only
+  add optional query params when the caller supplied a non-default.
+  Sending `?traces=auto` would deviate from this pattern.
+- Keep `test_id` positional (existing call sites pass it positionally);
+  only `trace_resolution` is keyword-only. Strictly additive.
+- Docstring: explain `auto` vs `linked` in user terms. Copy the
+  server-side wording verbatim from `app/routers/api.py` — it is the
+  canonical description and survives doc drift.
+
+### 3d. No other method changes
+
+- `get_traces()` is unchanged. Upstream does not yet expose a `test_id`
+  filter on the list endpoint; do not add a client-side filter
+  (out of scope, would lie about API support).
+- `_get_traces_generator()` is unchanged. The new `test_id` field rides
+  along automatically in `TraceDetails`.
+
+## Step 4: Tests
+
+Add `tests/test_trace_test_linking.py`. Follow the no-network pattern
+established by the other tests in this repo.
+
+Coverage:
+
+- **Schema round-trip**: build a `TraceDetails` with `test_id="t_123"`,
+  `.model_dump()` → `.model_validate()` round-trip preserves it.
+- **TraceCreateOrUpdate accepts `test_id`**: `TraceCreateOrUpdate(timestamp=...,
+  test_id="t_123")` validates; `.model_dump()` includes the field.
+- **`TraceResolution` enum values**: assert `TraceResolution.auto.value ==
+  "auto"` and `TraceResolution.linked.value == "linked"` (guards against the
+  generator regenerating the enum with different casing on the wire).
+- **Re-export**: `from sweatstack import TraceResolution` resolves, *and*
+  `import sweatstack; sweatstack.TraceResolution` resolves (catches the
+  client-level import step easy to skip).
+
+If the repo already has an httpx-mock based test harness (it currently does
+not, judging by the existing files), add a single integration-style test for
+`get_test(test_id, traces=TraceResolution.linked)` that asserts the request
+URL carries `?traces=linked`. If not, skip — the enum-value test above
+provides the same guarantee without inventing a test infrastructure for this
+plan alone.
+
+## Step 5: Examples / docs
+
+- Quick scan of `examples/` — if there is an existing "tests" example, append
+  a short snippet showing:
+
+  ```python
+  test = client.create_test(sport=Sport.cycling, start=...)
+  client.create_trace(timestamp=..., lactate=4.0, test_id=test.id)
+  detail = client.get_test(test.id, traces=TraceResolution.linked)
+  ```
+
+  If none exists, do **not** add a new example file for this alone.
+
+- `docs/everything.rst` — `TraceDetails` and `TraceCreateOrUpdate` autoclass
+  entries pick up the new field automatically. Add a `.. autoclass::
+  sweatstack.openapi_schemas.TraceResolution` entry in the same enums block
+  where `Sport` lives (if `Sport` is documented there — check before adding).
+
+- `.claude/skills/sweatstack-python/client.md` — update the `create_trace`,
+  `update_trace`, and `get_test` signatures, and add `TraceResolution` to
+  the list of exported enums.
+
+## Step 6: Changelog + version bump
+
+- `CHANGELOG.md`: new entry under the next minor version. Must include
+  both the additive feature and the behavioural-change callout.
+  Wording suggestion:
+
+  > **Added** — Traces can be explicitly linked to a test via the new
+  > `test_id` parameter on `create_trace` and `update_trace`. `get_test`
+  > accepts a `trace_resolution` parameter (`TraceResolution.auto` or
+  > `TraceResolution.linked`) controlling how the returned traces list is
+  > resolved.
+  >
+  > **Behaviour change** — `update_trace` is full-replace. Callers that
+  > previously linked a trace to a test (e.g. via the SweatStack UI or
+  > another client) and then call `update_trace` without passing
+  > `test_id` will now unlink that trace. Pass the existing `test_id`
+  > back in to preserve the link.
+
+- `pyproject.toml`: bump `0.76.2 → 0.77.0` (minor — additive feature, no
+  breaking change). Confirm the version cadence with the maintainer if
+  unsure; this is the only judgement call in the plan.
+
+## API Reference
+
+| Method | Path | New parameter | Notes |
+|--------|------|---------------|-------|
+| POST | `/api/v1/traces/` | body `test_id?: str` | 404 if test missing |
+| PUT  | `/api/v1/traces/{id}` | body `test_id?: str` | full-replace; `null` unlinks; 404 if test missing |
+| GET  | `/api/v1/tests/{id}` | query `traces=auto\|linked` | default `auto` |
+
+### Resolution semantics (server-side, reference only)
+
+- `auto`: `(traces in window AND not claimed by another test) ∪ (traces with test_id == this test)`
+- `linked`: `traces with test_id == this test` (window ignored)
+
+## Out of Scope
+
+- Listing/filtering traces by `test_id` on `GET /traces/` (upstream does not
+  expose this).
+- Bulk attach/detach helpers (`attach_traces_to_test(...)`). Not requested by
+  upstream; would be premature abstraction.
+- A `linked_only: bool` convenience on `get_test`. Rejected in favour of the
+  enum to keep a single, faithful surface.

{sweatstack-0.76.2 → sweatstack-0.77.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sweatstack"
-version = "0.76.2"
+version = "0.77.0"
 description = "The official Python client for SweatStack"
 readme = "README.md"
 license = "MIT"
@@ -32,7 +32,7 @@ dependencies = [
 
 [project.urls]
 Homepage = "https://sweatstack.no"
-Documentation = "https://developer.sweatstack.no/getting-started/"
+Documentation = "https://docs.sweatstack.no/getting-started/"
 Repository = "https://github.com/SweatStack/sweatstack-python"
 Issues = "https://github.com/SweatStack/sweatstack-python/issues"
 Changelog = "https://github.com/SweatStack/sweatstack-python/blob/main/CHANGELOG.md"

{sweatstack-0.76.2 → sweatstack-0.77.0}/src/sweatstack/client.py

@@ -43,7 +43,7 @@ from .schemas import (
     BackfillStatus, DailyMeasure, DailyResponse,
     Marker, Metric, Scope, Sport,
     TeamResponse, TestDetails, TestResults, TestSummary, TokenResponse, TraceDetails,
-    UserInfoResponse, UserResponse, UserSummary
+    TraceResolution, UserInfoResponse, UserResponse, UserSummary
 )
 from .utils import convert_to_standard_dtypes, decode_jwt_body, make_dataframe_streamlit_compatible
 
@@ -1717,6 +1717,7 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
         heart_rate: int | None = None,
         tags: list[str] | None = None,
         sport: Sport | str | None = None,
+        test_id: str | None = None,
     ) -> TraceDetails:
         """Creates a new trace with the specified parameters.
 
@@ -1733,12 +1734,18 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
             heart_rate: Optional heart rate measurement in beats per minute.
             tags: Optional list of tags to associate with this trace.
             sport: Optional sport to associate with this trace.
+            test_id: Optional ID of a test to explicitly link this trace to.
+                The link is independent of timestamp — a linked trace appears
+                in the test's traces list regardless of whether its timestamp
+                falls inside the test window.
 
         Returns:
             TraceDetails: The created trace object with all details.
 
         Raises:
-            HTTPStatusError: If the API request fails.
+            SweatStackNotFoundError: If ``test_id`` references a test that
+                does not exist.
+            SweatStackAPIError: If the API request fails for any other reason.
         """
         sport = self._enums_to_strings([sport])[0] if sport else None
         with self._http_client() as client:
@@ -1754,6 +1761,7 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
                     "heart_rate": heart_rate,
                     "tags": tags,
                     "sport": sport,
+                    "test_id": test_id,
                 },
             )
             self._raise_for_status(response)
@@ -1772,6 +1780,7 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
         heart_rate: int | None = None,
         tags: list[str] | None = None,
         sport: Sport | str | None = None,
+        test_id: str | None = None,
     ) -> None:
         """Updates a trace by replacing all fields.
 
@@ -1779,6 +1788,10 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
         server-side. To modify a single field, first fetch the trace with
         ``get_traces()``, then pass all fields back.
 
+        In particular: if the trace was previously linked to a test via
+        ``test_id`` and you do not pass ``test_id`` here, the link is cleared.
+        Pass the existing ``test_id`` back in to preserve it.
+
         Args:
             trace_id: The unique identifier of the trace to update.
             timestamp: The date and time when the trace was recorded.
@@ -1790,9 +1803,13 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
             heart_rate: Optional heart rate measurement in beats per minute.
             tags: Optional list of tags to associate with this trace.
             sport: Optional sport to associate with this trace.
+            test_id: Optional ID of a test to explicitly link this trace to.
+                Pass ``None`` (or omit) to leave the trace unlinked.
 
         Raises:
-            HTTPStatusError: If the API request fails.
+            SweatStackNotFoundError: If ``trace_id`` does not exist, or if
+                ``test_id`` references a test that does not exist.
+            SweatStackAPIError: If the API request fails for any other reason.
         """
         sport = self._enums_to_strings([sport])[0] if sport else None
         with self._http_client() as client:
@@ -1808,6 +1825,7 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
                     "heart_rate": heart_rate,
                     "tags": tags,
                     "sport": sport,
+                    "test_id": test_id,
                 },
             )
             self._raise_for_status(response)
@@ -1931,11 +1949,29 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
         else:
             return tests
 
-    def get_test(self, test_id: str) -> TestDetails:
+    def get_test(
+        self,
+        test_id: str,
+        *,
+        trace_resolution: TraceResolution | str = TraceResolution.auto,
+    ) -> TestDetails:
         """Gets details for a specific test by ID.
 
         Args:
             test_id: The unique identifier of the test to retrieve.
+            trace_resolution: How traces are matched to this test. Affects only
+                the ``traces`` list on the response; ``activities`` is always
+                time-overlap matched. Accepts a ``TraceResolution`` enum or
+                its string value (``"auto"`` or ``"linked"``).
+
+                - ``"auto"`` (default): traces whose timestamp falls in the
+                  test's time range, plus any traces explicitly linked to
+                  this test, minus any traces explicitly linked to a
+                  different test.
+                - ``"linked"``: only traces explicitly linked to this test
+                  via ``test_id``, regardless of timestamp.
+
+                Maps to the ``traces`` query parameter on the wire.
 
         Returns:
             TestDetails: The test details including resolved traces and overlapping activities.
@@ -1943,8 +1979,10 @@ class Client(_OAuth2Mixin, _DelegationMixin, _TokenStorageMixin, _LocalCacheMixi
         Raises:
             HTTPStatusError: If the API request fails.
         """
+        resolution = self._enums_to_strings([trace_resolution])[0]
+        params = {} if resolution == TraceResolution.auto.value else {"traces": resolution}
         with self._http_client() as client:
-            response = client.get(url=f"/api/v1/tests/{test_id}")
+            response = client.get(url=f"/api/v1/tests/{test_id}", params=params)
             self._raise_for_status(response)
             return TestDetails.model_validate(response.json())
 

{sweatstack-0.76.2 → sweatstack-0.77.0}/src/sweatstack/openapi_schemas.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2026-04-27T08:28:54+00:00
+#   timestamp: 2026-05-19T10:00:53+00:00
 
 from __future__ import annotations
 
@@ -13,7 +13,6 @@ from pydantic import (
     AnyUrl,
     AwareDatetime,
     BaseModel,
-    NaiveDatetime,
     EmailStr,
     Field,
     RootModel,
@@ -26,7 +25,7 @@ from pydantic import (
 
 class Activity(BaseModel):
     id: str = Field(..., title='Id')
-    start_date_local: NaiveDatetime | None = Field(None, title='Start Date Local')
+    start_date_local: AwareDatetime | None = Field(None, title='Start Date Local')
 
 
 class ActivityUpdate(BaseModel):
@@ -1116,8 +1115,8 @@ class TestSummary(BaseModel):
     end: AwareDatetime | None = Field(None, title='End')
     results: TestResults | None = None
     app_metadata: dict[str, Any] | None = Field(None, title='App Metadata')
-    start_local: NaiveDatetime = Field(..., title='Start Local')
-    end_local: NaiveDatetime | None = Field(..., title='End Local')
+    start_local: AwareDatetime = Field(..., title='Start Local')
+    end_local: AwareDatetime | None = Field(..., title='End Local')
 
 
 class TestUpdate(BaseModel):
@@ -1150,6 +1149,7 @@ class TokenResponse(BaseModel):
 
 class TraceCreateOrUpdate(BaseModel):
     timestamp: AwareDatetime = Field(..., title='Timestamp')
+    test_id: str | None = Field(None, title='Test Id')
     lactate: float | None = Field(None, title='Lactate')
     rpe: int | None = Field(None, title='Rpe')
     notes: str | None = Field(None, title='Notes')
@@ -1161,6 +1161,11 @@ class TraceCreateOrUpdate(BaseModel):
     sport: Sport | None = None
 
 
+class TraceResolution(Enum):
+    auto = 'auto'
+    linked = 'linked'
+
+
 class UserFlow(Enum):
     session = 'session'
     signup = 'signup'
@@ -1172,7 +1177,7 @@ class UserInfoResponse(BaseModel):
     given_name: str | None = Field(None, title='Given Name')
     family_name: str | None = Field(None, title='Family Name')
     email: str | None = Field(None, title='Email')
-    registered_at: AwareDatetime | NaiveDatetime = Field(..., title='Registered At')
+    registered_at: AwareDatetime = Field(..., title='Registered At')
     name: str = Field(..., title='Name')
 
 
@@ -1181,7 +1186,7 @@ class UserResponse(BaseModel):
     first_name: str | None = Field(..., title='First Name')
     last_name: str | None = Field(..., title='Last Name')
     admin: bool = Field(..., title='Admin')
-    registered_at: AwareDatetime | NaiveDatetime = Field(..., title='Registered At')
+    registered_at: AwareDatetime = Field(..., title='Registered At')
     display_name: str = Field(..., title='Display Name')
     is_managed: bool = Field(..., title='Is Managed')
 
@@ -1294,6 +1299,12 @@ class BodyAuthorizeOauthAuthorizePost(BaseModel):
     code_challenge_method: str | None = Field(None, title='Code Challenge Method')
 
 
+class BodyConnectAddEmailPostConnectAddEmailPost(BaseModel):
+    email: EmailStr = Field(..., title='Email')
+    state: str | None = Field(None, title='State')
+    user_flow: UserFlow | None = 'session'
+
+
 class BodyCreateCheckoutSessionApiV1PaymentStripeCheckoutPost(BaseModel):
     subscription_plan: SubscriptionPlan
 
@@ -1304,12 +1315,6 @@ class BodyCreateDailyDataDailiesPost(BaseModel):
     value: float = Field(..., title='Value')
 
 
-class BodyExpressAddEmailPostExpressAddEmailPost(BaseModel):
-    email: EmailStr = Field(..., title='Email')
-    state: str | None = Field(None, title='State')
-    user_flow: UserFlow | None = 'session'
-
-
 class BodyGenerateApiKeyPartialsGenerateApiKeyPost(BaseModel):
     scopes: list[Scope] = Field(..., title='Scopes')
 
@@ -1381,8 +1386,8 @@ class Lap(BaseModel):
     start: AwareDatetime = Field(..., title='Start')
     end: AwareDatetime = Field(..., title='End')
     duration: timedelta = Field(..., title='Duration')
-    start_local: NaiveDatetime = Field(..., title='Start Local')
-    end_local: NaiveDatetime = Field(..., title='End Local')
+    start_local: AwareDatetime = Field(..., title='Start Local')
+    end_local: AwareDatetime = Field(..., title='End Local')
 
 
 class RampValueInput(BaseModel):
@@ -1563,8 +1568,8 @@ class ActivityDetails(BaseModel):
     distance: float | None = Field(None, title='Distance')
     devices: list[str] | None = Field(None, title='Devices')
     duration: timedelta = Field(..., title='Duration')
-    start_local: NaiveDatetime = Field(..., title='Start Local')
-    end_local: NaiveDatetime = Field(..., title='End Local')
+    start_local: AwareDatetime = Field(..., title='Start Local')
+    end_local: AwareDatetime = Field(..., title='End Local')
 
 
 class ActivitySummary(BaseModel):
@@ -1580,8 +1585,8 @@ class ActivitySummary(BaseModel):
     traces: list[TraceDetails] | None = Field(None, title='Traces')
     app_metadata: dict[str, Any] | None = Field(None, title='App Metadata')
     duration: timedelta = Field(..., title='Duration')
-    start_local: NaiveDatetime = Field(..., title='Start Local')
-    end_local: NaiveDatetime = Field(..., title='End Local')
+    start_local: AwareDatetime = Field(..., title='Start Local')
+    end_local: AwareDatetime = Field(..., title='End Local')
 
 
 class TestDetails(BaseModel):
@@ -1596,13 +1601,14 @@ class TestDetails(BaseModel):
     app_metadata: dict[str, Any] | None = Field(None, title='App Metadata')
     traces: list[TraceDetails] | None = Field(None, title='Traces')
     activities: list[ActivitySummary] | None = Field(None, title='Activities')
-    start_local: NaiveDatetime = Field(..., title='Start Local')
-    end_local: NaiveDatetime | None = Field(..., title='End Local')
+    start_local: AwareDatetime = Field(..., title='Start Local')
+    end_local: AwareDatetime | None = Field(..., title='End Local')
 
 
 class TraceDetails(BaseModel):
     tags: list[str] | None = Field(None, title='Tags')
     id: str = Field(..., title='Id')
+    test_id: str | None = Field(None, title='Test Id')
     timestamp: AwareDatetime = Field(..., title='Timestamp')
     lactate: float | None = Field(None, title='Lactate')
     rpe: int | None = Field(None, title='Rpe')
@@ -1615,7 +1621,7 @@ class TraceDetails(BaseModel):
     activity: ActivitySummary | None = None
     sport: Sport | None = None
     app_metadata: dict[str, Any] | None = Field(None, title='App Metadata')
-    timestamp_local: NaiveDatetime = Field(..., title='Timestamp Local')
+    timestamp_local: AwareDatetime = Field(..., title='Timestamp Local')
 
 
 RepeatInput.model_rebuild()

{sweatstack-0.76.2 → sweatstack-0.77.0}/src/sweatstack/schemas.py

@@ -18,7 +18,7 @@ from .openapi_schemas import (
     BackfillStatus, DailyMeasure, DailyResponse,
     Marker, Metric, Scope, Sport,
     TeamResponse, TestDetails, TestResults, TestSummary, TokenResponse, TraceDetails,
-    UserInfoResponse, UserResponse, UserSummary
+    TraceResolution, UserInfoResponse, UserResponse, UserSummary
 )
 
 

{sweatstack-0.76.2 → sweatstack-0.77.0}/tests/test_tests.py

@@ -39,8 +39,8 @@ def sample_test_summary(sample_results: TestResults) -> TestSummary:
         end=datetime(2026, 3, 15, 11, 0, tzinfo=timezone.utc),
         results=sample_results,
         tags=["lab", "cycling"],
-        start_local=datetime(2026, 3, 15, 10, 0),
-        end_local=datetime(2026, 3, 15, 12, 0),
+        start_local=datetime(2026, 3, 15, 10, 0, tzinfo=timezone.utc),
+        end_local=datetime(2026, 3, 15, 12, 0, tzinfo=timezone.utc),
     )
 
 
@@ -152,8 +152,8 @@ class TestDataFrameConversion:
                 end=datetime(2026, 1, i + 1, 11, 0, tzinfo=timezone.utc),
                 results=sample_results if i == 0 else None,
                 tags=[],
-                start_local=datetime(2026, 1, i + 1, 10, 0),
-                end_local=datetime(2026, 1, i + 1, 12, 0),
+                start_local=datetime(2026, 1, i + 1, 10, 0, tzinfo=timezone.utc),
+                end_local=datetime(2026, 1, i + 1, 12, 0, tzinfo=timezone.utc),
             )
             for i in range(3)
         ]

sweatstack-0.77.0/tests/test_trace_test_linking.py

@@ -0,0 +1,90 @@
+"""Tests for the trace <-> test linking surface.
+
+Covers:
+
+- ``test_id`` round-trips through ``TraceDetails`` and ``TraceCreateOrUpdate``.
+- ``TraceResolution`` enum wire values are exactly ``"auto"`` / ``"linked"``
+  (these are what the server's query parameter accepts).
+- ``TraceResolution`` is exposed on the public package surface.
+"""
+
+from datetime import datetime, timezone
+
+import sweatstack
+from sweatstack import TraceDetails, TraceResolution
+from sweatstack.openapi_schemas import TraceCreateOrUpdate
+
+
+def _trace_details(**overrides) -> TraceDetails:
+    base = dict(
+        id="trace_001",
+        timestamp=datetime(2026, 3, 15, 9, 30, tzinfo=timezone.utc),
+        timestamp_local=datetime(2026, 3, 15, 10, 30, tzinfo=timezone.utc),
+    )
+    base.update(overrides)
+    return TraceDetails(**base)
+
+
+class TestTestIdField:
+    def test_trace_details_round_trip_with_test_id(self):
+        original = _trace_details(test_id="test_123", lactate=4.2)
+        restored = TraceDetails.model_validate(original.model_dump())
+
+        assert restored.test_id == "test_123"
+        assert restored.lactate == 4.2
+
+    def test_trace_details_test_id_defaults_to_none(self):
+        trace = _trace_details()
+        assert trace.test_id is None
+
+    def test_trace_create_or_update_accepts_test_id(self):
+        payload = TraceCreateOrUpdate(
+            timestamp=datetime(2026, 3, 15, 9, 30, tzinfo=timezone.utc),
+            lactate=4.2,
+            test_id="test_123",
+        )
+
+        dumped = payload.model_dump()
+        assert dumped["test_id"] == "test_123"
+
+    def test_trace_create_or_update_test_id_defaults_to_none(self):
+        payload = TraceCreateOrUpdate(
+            timestamp=datetime(2026, 3, 15, 9, 30, tzinfo=timezone.utc),
+        )
+
+        assert payload.test_id is None
+        assert payload.model_dump()["test_id"] is None
+
+
+class TestGetTestTraceResolution:
+    """``get_test`` should accept both the enum and the bare string for
+    ``trace_resolution`` (mirroring how ``sport``, ``measure`` etc. work
+    elsewhere in the client)."""
+
+    def _build_client(self):
+        from sweatstack.client import Client
+
+        return Client.__new__(Client)
+
+    def test_enum_input(self):
+        client = self._build_client()
+        assert client._enums_to_strings([TraceResolution.linked]) == ["linked"]
+
+    def test_string_input(self):
+        client = self._build_client()
+        assert client._enums_to_strings(["linked"]) == ["linked"]
+
+
+class TestTraceResolutionEnum:
+    def test_wire_values(self):
+        """The wire values must match the server's query parameter exactly."""
+        assert TraceResolution.auto.value == "auto"
+        assert TraceResolution.linked.value == "linked"
+
+    def test_membership(self):
+        """Guards against accidental additions/removals on regen."""
+        assert {m.value for m in TraceResolution} == {"auto", "linked"}
+
+    def test_publicly_exported(self):
+        """Importable from the top-level package, like other enums (e.g. Sport)."""
+        assert sweatstack.TraceResolution is TraceResolution

{sweatstack-0.76.2 → sweatstack-0.77.0}/uv.lock

@@ -2537,7 +2537,7 @@ wheels = [
 
 [[package]]
 name = "sweatstack"
-version = "0.76.0"
+version = "0.77.0"
 source = { editable = "." }
 dependencies = [
     { name = "email-validator" },