WIP for new step list UI

DAGWorks-Inc · Aug 21, 2024 · 9e4f568 · 9e4f568
1 parent 96a4fd6
commit 9e4f568
Show file tree

Hide file tree

Showing 59 changed files with 345 additions and 908 deletions.
diff --git a/burr/cli/demo_data.py b/burr/cli/demo_data.py
@@ -194,14 +194,14 @@ def generate_rag_data(
 def generate_all(
     data_dir: Optional[str] = None, s3_bucket: Optional[str] = None, unique_app_names: bool = False
 ):
-    logger.info("Generating chatbot data")
-    generate_chatbot_data(
-        data_dir=data_dir, s3_bucket=s3_bucket, use_traces=False, unique_app_names=unique_app_names
-    )
     logger.info("Generating chatbot data with traces")
     generate_chatbot_data(
         data_dir=data_dir, s3_bucket=s3_bucket, use_traces=True, unique_app_names=unique_app_names
     )
+    logger.info("Generating chatbot data")
+    generate_chatbot_data(
+        data_dir=data_dir, s3_bucket=s3_bucket, use_traces=False, unique_app_names=unique_app_names
+    )
     logger.info("Generating counter data")
     generate_counter_data(data_dir=data_dir, s3_bucket=s3_bucket, unique_app_names=unique_app_names)
     logger.info("Generating RAG data")

diff --git a/burr/tracking/client.py b/burr/tracking/client.py
@@ -497,6 +497,7 @@ def do_log_attributes(
                 span_id=span.uid if span is not None else None,
                 value=serde.serialize(attribute, **self.serde_kwargs),
                 tags=tags,
+                time_logged=system.now(),
             )
             self._append_write_line(attribute_model)
 

diff --git a/burr/tracking/common/models.py b/burr/tracking/common/models.py
@@ -204,6 +204,7 @@ class AttributeModel(IdentifyingModel):
     ]  # It doesn't have to relate to a span, it can be at the level of an action as well
     value: Union[dict, str, int, float, bool, list, None]
     tags: Dict[str, str]
+    time_logged: Optional[datetime.datetime] = None
     type: str = "attribute"
 
     @property

diff --git a/burr/tracking/s3client.py b/burr/tracking/s3client.py
@@ -12,6 +12,7 @@
 
 import pydantic
 
+from burr import system
 from burr.common import types as burr_types
 from burr.core import Action, ApplicationGraph, State, serde
 from burr.integrations.base import require_plugin
@@ -137,6 +138,7 @@ def do_log_attributes(
                 span_id=span.uid if span is not None else None,
                 value=serde.serialize(attribute, **self.serde_kwargs),
                 tags=tags,
+                time_logged=system.now(),
             )
             self.submit_log_event(attribute_model, app_id=app_id, partition_key=partition_key)
 

diff --git a/burr/tracking/server/demo_data/demo_chatbot/chat-1-giraffe/graph.json b/burr/tracking/server/demo_data/demo_chatbot/chat-1-giraffe/graph.json