diff --git a/burr/tracking/client.py b/burr/tracking/client.py index 159e8ddd5..30b63cabe 100644 --- a/burr/tracking/client.py +++ b/burr/tracking/client.py @@ -210,7 +210,7 @@ def _log_child_relationships( child=PointerModel( app_id=app_id, sequence_id=None, - partition_key=None, # TODO -- get partition key + partition_key=fork_parent_pointer_model.partition_key, # TODO -- get partition key ), event_time=datetime.datetime.now(), event_type="fork", @@ -228,7 +228,7 @@ def _log_child_relationships( child=PointerModel( app_id=app_id, sequence_id=None, - partition_key=None, # TODO -- get partition key + partition_key=spawn_parent_pointer_model.sequence_id, # TODO -- get partition key ), event_time=datetime.datetime.now(), event_type="spawn_start", diff --git a/examples/multi-modal-chatbot/test.py b/examples/multi-modal-chatbot/test.py new file mode 100644 index 000000000..e69de29bb