fix(api): replace unknown wildcards with empty string
This commit is contained in:
parent
c8a9dd4475
commit
fa3a36ed25
|
@ -376,11 +376,11 @@ def replace_wildcards(prompt: str, seed: int, wildcards: Dict[str, List[str]]) -
|
||||||
logger.debug("found wildcard in prompt: %s", next_match)
|
logger.debug("found wildcard in prompt: %s", next_match)
|
||||||
name = next_match.groups()
|
name = next_match.groups()
|
||||||
|
|
||||||
if name not in wildcards:
|
wildcard = ""
|
||||||
logger.warning("unknown wildcard: %s", name)
|
if name in wildcards:
|
||||||
continue
|
|
||||||
|
|
||||||
wildcard = random.choice(wildcards.get(name))
|
wildcard = random.choice(wildcards.get(name))
|
||||||
|
else:
|
||||||
|
logger.warning("unknown wildcard: %s", name)
|
||||||
|
|
||||||
remaining_prompt = (
|
remaining_prompt = (
|
||||||
remaining_prompt[: next_match.start()]
|
remaining_prompt[: next_match.start()]
|
||||||
|
|
Loading…
Reference in New Issue