From 2d5d01ea85442a10338c8427c25583d741d649b3 Mon Sep 17 00:00:00 2001
From: Eva Lott <evalott100@gmail.com>
Date: Mon, 11 Sep 2023 09:52:37 +0100
Subject: [PATCH] Made seq_nums optional

---
 event_model/__init__.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/event_model/__init__.py b/event_model/__init__.py
index 896500ad..766ce657 100644
--- a/event_model/__init__.py
+++ b/event_model/__init__.py
@@ -1996,6 +1996,11 @@ def compose_resource(
     )
 
 
+# A dict of Tuple[str, StreamRange] where the string is the StreamDatum uuid
+
+_stream_datum_seq_nums: Dict[str, StreamRange] = {}
+
+
 @dataclass
 class ComposeStreamDatum:
     stream_resource: StreamResource
@@ -2004,12 +2009,18 @@ class ComposeStreamDatum:
     def __call__(
         self,
         data_keys: List[str],
-        seq_nums: StreamRange,
         indices: StreamRange,
+        seq_nums: Optional[StreamRange] = None,
         descriptor: Optional[EventDescriptor] = None,
         validate: bool = True,
     ) -> StreamDatum:
         resource_uid = self.stream_resource["uid"]
+
+        # If the seq_nums aren't passed in then the bluesky
+        # bundler will keep track of them
+        if not seq_nums:
+            seq_nums = StreamRange(start=0, stop=0)
+
         doc = StreamDatum(
             stream_resource=resource_uid,
             uid=f"{resource_uid}/{next(self.counter)}",