diff --git a/shiny/ui/_chat_normalize.py b/shiny/ui/_chat_normalize.py index 7bec8102c..55bde9569 100644 --- a/shiny/ui/_chat_normalize.py +++ b/shiny/ui/_chat_normalize.py @@ -277,7 +277,7 @@ def normalize_message(message: Any) -> ChatMessage: return strategy.normalize(message) raise ValueError( f"Could not find a normalizer for message of type {type(message)}: {message}. " - "Consider registering a custom normalizer via shiny.ui._chat_types.registry.register()" + "Consider registering a custom normalizer via shiny.ui._chat_normalize.message_normalizer_registry.register()" ) @@ -288,5 +288,5 @@ def normalize_message_chunk(chunk: Any) -> ChatMessage: return strategy.normalize_chunk(chunk) raise ValueError( f"Could not find a normalizer for message chunk of type {type(chunk)}: {chunk}. " - "Consider registering a custom normalizer via shiny.ui._chat_types.registry.register()" + "Consider registering a custom normalizer via shiny.ui._chat_normalize.message_normalizer_registry.register()" )