Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .jules/bolt.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
## 2024-06-25 - Context Lexicon Regex Compilation Bottleneck
**Learning:** Evaluated span assertions in the openmed NLP pipeline heavily depend on repeated deterministic cue compilations via `_compiled_context_lexicon`. This function constructs and formats numerous regexes for every evaluated axis (negation, temporality, certainty, etc). Doing this on every decision axis per text/span iteration caused massive redundant regex compilations and became a significant CPU bottleneck on the main thread.
**Action:** Always memoize (e.g. `functools.lru_cache`) deterministic regex lexicons and configuration functions where inputs are purely hashable parameters like `language` strings.
4 changes: 4 additions & 0 deletions openmed/openmed/clinical/context.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@
from collections.abc import Iterable, Iterator, Mapping, Sequence
from dataclasses import dataclass, replace
from datetime import date
from functools import lru_cache
from typing import Any, Literal

from openmed.clinical.lexicons import (
Expand Down Expand Up @@ -154,6 +155,9 @@ class _CompiledContextLexicon:
backward_context_cues: frozenset[str]


# ⚡ Bolt: Cache deterministic context lexicons to avoid compiling the same
# regex patterns thousands of times during span/text evaluations.
@lru_cache(maxsize=32)
def _compiled_context_lexicon(language: str | None = None) -> _CompiledContextLexicon:
lexicon = get_clinical_cue_lexicon(language)
token_boundaries = lexicon.token_boundaries
Expand Down