Skip to content

Commit

Permalink
Fix NormalizerRegistry.register() (#1619)
Browse files Browse the repository at this point in the history
  • Loading branch information
cpsievert authored Aug 20, 2024
1 parent 419db53 commit d90c0bf
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions shiny/ui/_chat_normalize.py
Original file line number Diff line number Diff line change
Expand Up @@ -256,11 +256,13 @@ def __init__(self) -> None:
def register(
self, provider: str, strategy: BaseMessageNormalizer, force: bool = False
) -> None:
if provider in self._strategies and not force:
raise ValueError(f"Provider {provider} already exists in registry")
if provider in self._strategies:
if force:
del self._strategies[provider]
else:
raise ValueError(f"Provider {provider} already exists in registry")
# Update the strategies dict such that the new strategy is the first to be considered
strategies = {provider: strategy}
strategies.update(self._strategies)
self._strategies = {provider: strategy, **self._strategies}


message_normalizer_registry = NormalizerRegistry()
Expand Down

0 comments on commit d90c0bf

Please sign in to comment.