Improve demo project script (flush + batching) (#1112)

* Update demo_data_generator.py

* Fix linter

sdks/python/examples/demo_data_generator.py

@@ -31,6 +31,7 @@ def create_demo_data(base_url: str, workspace_name, comet_api_key):
         workspace=workspace_name,
         host=base_url,
         api_key=comet_api_key,
+        _use_batching=True,
     )
 
     for trace in sorted(evaluation_traces, key=lambda x: x["start_time"]):
@@ -48,6 +49,8 @@ def create_demo_data(base_url: str, workspace_name, comet_api_key):
             span["parent_span_id"] = new_parent_span_id
         client.span(**span)
 
+    client.flush()
+
     # Demo traces and spans
     # We have a simple chatbot application built using llama-index.
     # We gave it the content of Opik documentation as context, and then asked it a few questions.
@@ -57,6 +60,7 @@ def create_demo_data(base_url: str, workspace_name, comet_api_key):
         workspace=workspace_name,
         host=base_url,
         api_key=comet_api_key,
+        _use_batching=True,
     )
 
     for trace in sorted(demo_traces, key=lambda x: x["start_time"]):
@@ -152,6 +156,8 @@ def create_demo_data(base_url: str, workspace_name, comet_api_key):
 
     experiment.insert(experiment_items)
 
+    client.flush()
+
 
 if __name__ == "__main__":
     base_url = "http://localhost:5173/api"