lib/water_drop/contracts/message.rb in waterdrop-2.0.1 vs lib/water_drop/contracts/message.rb in waterdrop-2.0.2

@@ -20,9 +20,10 @@
       params do
         required(:topic).filled(:str?, format?: TOPIC_REGEXP)
         required(:payload).filled(:str?)
         optional(:key).maybe(:str?, :filled?)
         optional(:partition).filled(:int?, gteq?: -1)
+        optional(:partition_key).maybe(:str?, :filled?)
         optional(:timestamp).maybe { time? | int? }
         optional(:headers).maybe(:hash?)
       end
 
       rule(:headers) do