@@ -288,21 +288,25 @@ def validate_conversational_data(conversational_data: list) -> None:
288288 Args:
289289 conversational_data (list): list of dictionaries.
290290 """
291+
291292 def check_message_keys (message ):
292293 accepted_message_keys = set ([
293- "messageId" , "timestampUsec" , "content" , "user" , "align" , "canLabel" ])
294+ "messageId" , "timestampUsec" , "content" , "user" , "align" ,
295+ "canLabel"
296+ ])
294297 for key in message .keys ():
295298 if not key in accepted_message_keys :
296299 raise KeyError (
297- f"Invalid { key } key found! Accepted keys in messages list is { accepted_message_keys } " )
300+ f"Invalid { key } key found! Accepted keys in messages list is { accepted_message_keys } "
301+ )
298302
299- if conversational_data and not isinstance (conversational_data , list ):
303+ if conversational_data and not isinstance (conversational_data ,
304+ list ):
300305 raise ValueError (
301306 f"conversationalData must be a list. Found { type (conversational_data )} "
302307 )
303308
304- [check_message_keys (message )
305- for message in conversational_data ]
309+ [check_message_keys (message ) for message in conversational_data ]
306310
307311 def parse_metadata_fields (item ):
308312 metadata_fields = item .get ('metadata_fields' )
@@ -360,9 +364,10 @@ def convert_item(item):
360364 "version" : version ,
361365 "messages" : messages
362366 }
363- conversationUrl = self .client .upload_data (json .dumps (one_conversation ),
364- content_type = "application/json" ,
365- filename = "conversational_data.json" )
367+ conversationUrl = self .client .upload_data (
368+ json .dumps (one_conversation ),
369+ content_type = "application/json" ,
370+ filename = "conversational_data.json" )
366371 item ["row_data" ] = conversationUrl
367372
368373 # Convert all payload variations into the same dict format
0 commit comments