different field updates collapsed

deeppavlov · Feb 11, 2025 · 1a6efa7 · 1a6efa7
1 parent 25a0b7d
commit 1a6efa7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/chatsky/context_storages/database.py b/chatsky/context_storages/database.py
@@ -232,21 +232,21 @@ async def update_context(self, ctx_id: str, ctx_info: ContextInfo, field_info: L
         :param ctx_info: Context information (will be written to different tables at once).
         """
 
-        joined_field_info = list()
+        joined_field_info = dict()
         logger.debug(f"Updating context for {ctx_id}...")
         for field, added, deleted in field_info:
             field = self._validate_field_name(field)
             if len(added) == 0:
                 logger.debug(f"\tNo fields to add in {field}!")
             else:
-                joined_field_info += [(field, added)]
+                joined_field_info.setdefault(field, list()) += added
                 logger.debug(f"\tAdding fields for {field}: {collapse_num_list(list(k for k, _ in added))}...")
             if len(deleted) == 0:
                 logger.debug(f"\tNo fields to delete in {field}!")
             else:
-                joined_field_info += [(field, [(k, None) for k in deleted])]
+                joined_field_info.setdefault(field, list()) += [(k, None) for k in deleted]
                 logger.debug(f"\tDeleting fields for {field}: {collapse_num_list(deleted)}...")
-        await self._update_context(ctx_id, ctx_info, joined_field_info)
+        await self._update_context(ctx_id, ctx_info, list(joined_field_info.items()))
         logger.debug(f"Context updated for {ctx_id}")
 
     @abstractmethod