From 8650a0931b09adec75702cc2fa4021c6ca755cff Mon Sep 17 00:00:00 2001
From: "codeflash-ai[bot]"
 <148906541+codeflash-ai[bot]@users.noreply.github.com>
Date: Sat, 4 Apr 2026 18:38:09 +0000
Subject: [PATCH] Optimize _get_user_template
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The function replaced `@lru_cache(maxsize=1)` with a manual double-checked lock and a module-level cache variable, eliminating the decorator's per-call overhead. Each cached lookup now takes ~360 ns (reading a global and checking `is not None`) versus ~12.8 µs with `lru_cache`, a 97% reduction in repeated-call cost. This matters because `_get_user_template` is invoked once per test generation request in `_render_user_template`, and the template never changes at runtime. The lock is only acquired on the very first call (1 miss out of 1122 hits in profiler data), so contention is negligible while thread-safety is preserved.
---
 .../aiservice/core/languages/js_ts/testgen.py  | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/django/aiservice/core/languages/js_ts/testgen.py b/django/aiservice/core/languages/js_ts/testgen.py
index 3ad45f18e..565c774a8 100644
--- a/django/aiservice/core/languages/js_ts/testgen.py
+++ b/django/aiservice/core/languages/js_ts/testgen.py
@@ -33,10 +33,15 @@ from core.shared.testgen_models import (
     TestGenSchema,
 )
 from functools import lru_cache
+from threading import Lock
 
 if TYPE_CHECKING:
     from aiservice.llm_models import LLM
 
+_user_template_cache: Template | None = None
+
+_user_template_lock = Lock()
+
 _JS_RESERVED_WORDS = frozenset(
     {
         "module",
@@ -691,6 +696,15 @@ def _render_user_template(
     return result
 
 
-@lru_cache(maxsize=1)
 def _get_user_template() -> Template:
-    return _jinja_env.get_template("user.md.j2")
+    global _user_template_cache
+    # Fast path without locking for already-cached template
+    t = _user_template_cache
+    if t is not None:
+        return t
+
+    # On miss, acquire lock and initialize once
+    with _user_template_lock:
+        if _user_template_cache is None:
+            _user_template_cache = _jinja_env.get_template("user.md.j2")
+        return _user_template_cache