fix(prompt): invalidate dynamic wildcard cache without seed (#895)

This commit is contained in:
Will Miao
2026-04-15 20:43:21 +08:00
parent b910388d54
commit 2640258902
5 changed files with 65 additions and 3 deletions

View File

@@ -3,7 +3,11 @@ from __future__ import annotations
from typing import Any
import inspect
from ..services.wildcard_service import get_wildcard_service, is_trigger_words_input
from ..services.wildcard_service import (
contains_dynamic_syntax,
get_wildcard_service,
is_trigger_words_input,
)
class _PromptOptionalInputs:
@@ -90,6 +94,19 @@ class PromptLM:
)
FUNCTION = "encode"
@classmethod
def IS_CHANGED(
cls,
text: str,
clip: Any | None = None,
seed: int | None = None,
**kwargs: Any,
):
del clip, kwargs
if contains_dynamic_syntax(text) and seed is None:
return float("NaN")
return False
def encode(
self,
text: str,

View File

@@ -1,6 +1,6 @@
from __future__ import annotations
from ..services.wildcard_service import get_wildcard_service
from ..services.wildcard_service import contains_dynamic_syntax, get_wildcard_service
class TextLM:
@@ -41,5 +41,11 @@ class TextLM:
OUTPUT_TOOLTIPS = ("The text output.",)
FUNCTION = "process"
@classmethod
def IS_CHANGED(cls, text: str, seed: int | None = None):
if contains_dynamic_syntax(text) and seed is None:
return float("NaN")
return False
def process(self, text: str, seed: int | None = None):
return (get_wildcard_service().expand_text(text, seed=seed),)

View File

@@ -31,6 +31,14 @@ def _is_numeric_string(value: str) -> bool:
return bool(_NUMERIC_PATTERN.match(value))
def contains_dynamic_syntax(text: str) -> bool:
"""Return True when text contains supported wildcard or option syntax."""
return isinstance(text, str) and bool(
_WILDCARD_PATTERN.search(text) or _OPTION_PATTERN.search(text)
)
def get_wildcards_dir(create: bool = False) -> str:
"""Return the managed wildcard directory inside the settings folder."""
@@ -397,6 +405,7 @@ def get_wildcard_service() -> WildcardService:
__all__ = [
"WildcardService",
"contains_dynamic_syntax",
"get_wildcard_service",
"get_wildcards_dir",
"is_trigger_words_input",

View File

@@ -59,3 +59,26 @@ def test_text_lm_input_types_expose_input_only_seed():
assert seed_type == "INT"
assert seed_options["forceInput"] is True
assert "wildcard generation" in seed_options["tooltip"]
def test_text_lm_is_changed_forces_rerun_without_seed_when_text_is_dynamic():
result = TextLM.IS_CHANGED("__flower__", seed=None)
assert result != result
def test_text_lm_is_changed_keeps_cache_for_seeded_or_static_text():
assert TextLM.IS_CHANGED("__flower__", seed=7) is False
assert TextLM.IS_CHANGED("plain text", seed=None) is False
assert TextLM.IS_CHANGED("{red|blue}", seed=7) is False
def test_prompt_lm_is_changed_forces_rerun_without_seed_when_text_is_dynamic():
result = PromptLM.IS_CHANGED("{red|blue}", clip="clip", seed=None)
assert result != result
def test_prompt_lm_is_changed_keeps_cache_for_seeded_or_static_text():
assert PromptLM.IS_CHANGED("__flower__", clip="clip", seed=11) is False
assert PromptLM.IS_CHANGED("plain text", clip="clip", seed=None) is False

View File

@@ -2,7 +2,7 @@ from __future__ import annotations
import json
from py.services.wildcard_service import WildcardService
from py.services.wildcard_service import WildcardService, contains_dynamic_syntax
def _make_service(monkeypatch, tmp_path):
@@ -121,3 +121,10 @@ def test_expand_text_leaves_unresolved_reference_visible(monkeypatch, tmp_path):
wildcards_dir.mkdir()
assert service.expand_text("__missing__", seed=1) == "__missing__"
def test_contains_dynamic_syntax_detects_wildcards_and_options():
assert contains_dynamic_syntax("plain text") is False
assert contains_dynamic_syntax("__flower__") is True
assert contains_dynamic_syntax("{red|blue}") is True
assert contains_dynamic_syntax("{2$$, $$red|blue|green}") is True