splight-lib 5.6.3__tar.gz → 5.6.5.dev0__tar.gz

{splight_lib-5.6.3 → splight_lib-5.6.5.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: splight-lib
-Version: 5.6.3
+Version: 5.6.5.dev0
 Summary: Splight Library
 Author: Splight Dev
 Author-email: dev@splight-ae.com

{splight_lib-5.6.3 → splight_lib-5.6.5.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "splight-lib"
-version = "5.6.3"
+version = "5.6.5dev0"
 description = "Splight Library"
 authors = ["Splight Dev <dev@splight-ae.com>"]
 readme = "README.md"

{splight_lib-5.6.3 → splight_lib-5.6.5.dev0}/splight_lib/models/base.py

@@ -38,8 +38,7 @@ class SplightDatabaseBaseModel(BaseModel):
         files_dict = self._get_model_files_dict()
         saved = self._db_client.save(
             self.__class__.__name__,
-            # TODO: improve the following line
-            json.loads(self.model_dump_json(exclude_none=True)),
+            self.model_dump(exclude_none=True, mode="json"),
             files=files_dict,
         )
         if not self.id:

{splight_lib-5.6.3 → splight_lib-5.6.5.dev0}/splight_lib/models/tests/test_database_model.py

@@ -98,7 +98,7 @@ def test_save_with_id(instance_dict):
     ) as mock:
         resource = Resource.model_validate(instance_dict)
         resource.save()
-        mock.assert_called_with(Resource.__name__, instance_dict)
+        mock.assert_called_with(Resource.__name__, instance_dict, files=None)
         assert resource.model_dump() == instance_dict
 
 
@@ -114,7 +114,7 @@ def test_save_without_id(instance_dict):
         resource.id = None
         resource_dict = resource.model_dump(exclude_none=True)
         resource.save()
-        mock.assert_called_with(Resource.__name__, resource_dict)
+        mock.assert_called_with(Resource.__name__, resource_dict, files=None)
         assert resource.model_dump() == instance_dict
 
 

{splight_lib-5.6.3 → splight_lib-5.6.5.dev0}/splight_lib/client/datalake/remote_client.py

@@ -372,6 +372,7 @@ class BufferedSyncRemoteDataClient(SyncRemoteDatalakeClient):
             raise InvalidCollectionName(collection)
         buffer = self._data_buffers[collection]
         with self._lock:
+            buffer.add_documents(instances)
             if buffer.should_flush():
                 logger.debug(
                     "Flushing datalake buffer with %s elements",
@@ -379,7 +380,6 @@ class BufferedSyncRemoteDataClient(SyncRemoteDatalakeClient):
                 )
                 self._send_documents(collection, buffer.data)
                 buffer.reset()
-            buffer.add_documents(instances)
         return instances
 
     def save_dataframe(