Skip to content

Commit 8e21942

Browse files
committed
Fix lint on vcr_replay.py — pyupgrade/black/isort/flake8 clean
1 parent 91dbd7f commit 8e21942

1 file changed

Lines changed: 5 additions & 8 deletions

File tree

opencontractserver/utils/vcr_replay.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,14 @@
3030
differ across runs. We therefore strip volatile fields from the request
3131
body before matching. See ``_match_llm_body`` below.
3232
"""
33+
3334
from __future__ import annotations
3435

3536
import logging
3637
import os
3738
import re
39+
from collections.abc import Iterator
3840
from contextlib import contextmanager
39-
from typing import Iterator, Optional
4041

4142
logger = logging.getLogger(__name__)
4243

@@ -60,9 +61,7 @@
6061
re.compile(rb'"id"\s*:\s*"call_[A-Za-z0-9]+"'), # OpenAI tool-call IDs
6162
re.compile(rb'"tool_call_id"\s*:\s*"call_[A-Za-z0-9]+"'),
6263
# UUIDs that occasionally appear in tool returns (annotation IDs etc.)
63-
re.compile(
64-
rb"\b[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}\b"
65-
),
64+
re.compile(rb"\b[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}\b"),
6665
]
6766

6867

@@ -135,7 +134,7 @@ def _match_llm_body(r1, r2) -> None:
135134

136135

137136
@contextmanager
138-
def maybe_vcr_cassette() -> Iterator[Optional[object]]:
137+
def maybe_vcr_cassette() -> Iterator[object | None]:
139138
"""
140139
Yield either an active VCR cassette context (when env says so) or
141140
None (the no-op path used in production).
@@ -210,9 +209,7 @@ def maybe_vcr_cassette() -> Iterator[Optional[object]]:
210209
def _ignore_request(request) -> bool:
211210
return request.host not in _LLM_HOSTS
212211

213-
my_vcr.before_record_request = (
214-
lambda req: req if not _ignore_request(req) else None
215-
)
212+
my_vcr.before_record_request = lambda req: req if not _ignore_request(req) else None
216213
# ``record_mode=none`` (replay) plus our ignore filter means: for
217214
# non-LLM hosts the request passes through to the real network; for
218215
# LLM hosts a missing cassette entry raises

0 commit comments

Comments
 (0)