Skip to content

Commit d90c0bf

Browse files
authored
Fix NormalizerRegistry.register() (#1619)
1 parent 419db53 commit d90c0bf

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

shiny/ui/_chat_normalize.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -256,11 +256,13 @@ def __init__(self) -> None:
256256
def register(
257257
self, provider: str, strategy: BaseMessageNormalizer, force: bool = False
258258
) -> None:
259-
if provider in self._strategies and not force:
260-
raise ValueError(f"Provider {provider} already exists in registry")
259+
if provider in self._strategies:
260+
if force:
261+
del self._strategies[provider]
262+
else:
263+
raise ValueError(f"Provider {provider} already exists in registry")
261264
# Update the strategies dict such that the new strategy is the first to be considered
262-
strategies = {provider: strategy}
263-
strategies.update(self._strategies)
265+
self._strategies = {provider: strategy, **self._strategies}
264266

265267

266268
message_normalizer_registry = NormalizerRegistry()

0 commit comments

Comments
 (0)