lucidicai 1.2.21__py3-none-any.whl → 1.2.22__py3-none-any.whl

lucidicai/__init__.py

@@ -315,6 +315,7 @@ def _auto_end_session():
         client = Client()
         if hasattr(client, 'auto_end') and client.auto_end and client.session and not client.session.is_finished:
             logger.info("Auto-ending active session on exit")
+            client.auto_end = False  # To avoid repeating auto-end on exit
             end_session()
     except Exception as e:
         logger.debug(f"Error during auto-end session: {e}")

lucidicai/event.py

@@ -14,37 +14,48 @@ class Event:
         self.event_id = None
         self.screenshots = []
         self.is_finished = False
-        self.init_event()
-        self.update_event(**kwargs)
+        self.init_event(**kwargs)
 
-
-    def init_event(self) -> None:
+    # TODO: this is really bad, clean this up later
+    def init_event(self, **kwargs) -> None:
         from .client import Client
-        request_data = {
-            "step_id": self.step_id,
-            # TODO: get rid of these in backend API interface?
-            # "description": description,
-            # "result": result
-        }
+        request_data = self._build_request_data(**kwargs)
+        if self.step_id:
+            request_data['step_id'] = self.step_id
+        else:
+            request_data['session_id'] = self.session_id
         data = Client().make_request('initevent', 'POST', request_data)
         self.event_id = data["event_id"]
+        self.step_id = data["step_id"]
+        self._upload_screenshots(**kwargs)
+        return self.event_id
 
     def update_event(self, **kwargs) -> None:
         from .client import Client
-        if 'screenshots' in kwargs and kwargs['screenshots'] is not None:
-            for i in range(len(kwargs['screenshots'])):
-                presigned_url, bucket_name, object_key = get_presigned_url(Client().agent_id, session_id=self.session_id, event_id=self.event_id, nthscreenshot=len(self.screenshots))
-                upload_image_to_s3(presigned_url, kwargs['screenshots'][i], "JPEG")
-                self.screenshots.append(kwargs['screenshots'][i])
         if 'is_finished' in kwargs:
             self.is_finished = kwargs['is_finished']
-        request_data = {
-            "event_id": self.event_id,
+        request_data = self._build_request_data(**kwargs)
+        request_data['event_id'] = self.event_id
+        Client().make_request('updateevent', 'PUT', request_data)
+        self._upload_screenshots(**kwargs)
+
+    def _build_request_data(self, **kwargs) -> dict:
+        from .client import Client
+        num_new_screenshots = len(kwargs.get("screenshots", []))
+        return {
             "description": Client().mask(kwargs.get("description", None)),
             "result": Client().mask(kwargs.get("result", None)),
             "is_finished": self.is_finished, 
             "cost_added": kwargs.get("cost_added", None),
             "model": kwargs.get("model", None), 
-            "nscreenshots": len(self.screenshots)
+            "nscreenshots": len(self.screenshots) + num_new_screenshots,
+            "duration": kwargs.get("duration", None)
         }
-        Client().make_request('updateevent', 'PUT', request_data)
+    
+    def _upload_screenshots(self, **kwargs) -> None:
+        from .client import Client
+        if 'screenshots' in kwargs and kwargs['screenshots'] is not None:
+            for i in range(len(kwargs['screenshots'])):
+                presigned_url, bucket_name, object_key = get_presigned_url(Client().agent_id, session_id=self.session_id, event_id=self.event_id, nthscreenshot=len(self.screenshots))
+                upload_image_to_s3(presigned_url, kwargs['screenshots'][i], "JPEG")
+                self.screenshots.append(kwargs['screenshots'][i])

lucidicai/session.py

@@ -110,14 +110,12 @@ class Session:
 
     def create_event(self, **kwargs):
         # Get step_id from kwargs or active step
-        temp_step_created = False
         if 'step_id' in kwargs and kwargs['step_id'] is not None:
             step_id = kwargs['step_id']
         elif self._active_step:
             step_id = self._active_step
         else:
-            step_id = self.create_step()
-            temp_step_created = True
+            step_id = None
         kwargs.pop('step_id', None)
         event = Event(
             session_id=self.session_id,
@@ -126,9 +124,6 @@ class Session:
         )
         self.event_history[event.event_id] = event
         self._active_event = event
-        if temp_step_created:
-            self.update_step(step_id=step_id, is_finished=True)
-            self._active_step = None
         return event.event_id
 
     def update_event(self, **kwargs):

lucidicai/telemetry/lucidic_span_processor.py

@@ -239,6 +239,12 @@ class LucidicSpanProcessor(SpanProcessor):
             # Calculate cost
             cost = self._calculate_cost(attributes)
             
+            # Calculate duration in seconds
+            duration_seconds = None
+            if span.start_time and span.end_time:
+                duration_ns = span.end_time - span.start_time
+                duration_seconds = duration_ns / 1_000_000_000
+            
             # Check success
             is_successful = span.status.status_code != StatusCode.ERROR
             
@@ -247,7 +253,8 @@ class LucidicSpanProcessor(SpanProcessor):
                 'description': description,
                 'result': formatted_result,
                 'model': model,
-                'is_finished': True
+                'is_finished': True,
+                'duration': duration_seconds
             }
             
             if images:
@@ -288,6 +295,12 @@ class LucidicSpanProcessor(SpanProcessor):
             # Calculate cost
             cost = self._calculate_cost(attributes)
             
+            # Calculate duration in seconds
+            duration_seconds = None
+            if span.start_time and span.end_time:
+                duration_ns = span.end_time - span.start_time
+                duration_seconds = duration_ns / 1_000_000_000
+            
             # Check success
             is_successful = span.status.status_code != StatusCode.ERROR
             
@@ -295,7 +308,8 @@ class LucidicSpanProcessor(SpanProcessor):
             update_kwargs = {
                 'event_id': event_id,
                 'result': result,
-                'is_finished': True
+                'is_finished': True,
+                'duration': duration_seconds
             }
             
             if cost is not None:

{lucidicai-1.2.21.dist-info → lucidicai-1.2.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lucidicai
-Version: 1.2.21
+Version: 1.2.22
 Summary: Lucidic AI Python SDK
 Author: Andy Liang
 Author-email: andy@lucidic.ai

{lucidicai-1.2.21.dist-info → lucidicai-1.2.22.dist-info}/RECORD

@@ -1,13 +1,13 @@
-lucidicai/__init__.py,sha256=sUfrleiPQPuXq5e0UnkXPjuTShiVoHnPtl_O0nzPcd8,20589
+lucidicai/__init__.py,sha256=8_qfdx62xyvrdyONXCHZ4iKDWCvmyncp1T5Epb1DIv8,20664
 lucidicai/action.py,sha256=sPRd1hTIVXDqnvG9ZXWEipUFh0bsXcE0Fm7RVqmVccM,237
 lucidicai/client.py,sha256=3yBVppvwBwesca_pZSKgTUDihzZe5JhVgh1AALCpJ_Q,6620
 lucidicai/constants.py,sha256=_u0z3M4geZgS1g-CrOZUVjtcew8l70dKQnpVQvlXh9w,2172
 lucidicai/decorators.py,sha256=oqXyfHk9f9UmeaIquuU8mtzed1qZtO_-svwadpoat6g,13950
 lucidicai/errors.py,sha256=gTg0bdzjuTcUnakRbZnxjngO4gZnRLVwRHRglpZZJsM,970
-lucidicai/event.py,sha256=iPbNBRb2ZFohBsrRQZHzfa9GbxCsgFIw3M8w4gFrSi4,2000
+lucidicai/event.py,sha256=SjPSGNBuW6YLDJUhY16IvQgE9lrg3bypgWOK9IKVJWI,2567
 lucidicai/image_upload.py,sha256=6SRudg-BpInM2gzMx1Yf1Rz_Zyh8inwoJ7U4pBw7ruY,3807
 lucidicai/model_pricing.py,sha256=o1yWCaF5Qxj4tloXxVG3SZXcTMKtk56J_Nfdo8M4uR0,11947
-lucidicai/session.py,sha256=T_4z3CLKMp5DOKSVpdWwMZkZG9ndXU6oiI5Al6DHSTY,5501
+lucidicai/session.py,sha256=mUPwXcGW5tFsK6Udl4i2jndV0STBYHbPyWqA8GxBdHM,5285
 lucidicai/singleton.py,sha256=gfT3XdWLXSIWMqDXbY6-pnesMZ8RGRitaEPhIsgrRPw,1272
 lucidicai/state.py,sha256=4Tb1X6l2or6w_e62FYSuEeghAv3xXm5gquKwzCpvdok,235
 lucidicai/step.py,sha256=_oBIyTBZBvNkUkYHIrwWd75KMSlMtR9Ws2Lo71Lyff8,2522
@@ -33,7 +33,7 @@ lucidicai/telemetry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 lucidicai/telemetry/base_provider.py,sha256=nrZVr4Y9xcAiMn4uAN3t3k6DlHNTvlXrA4qQg7lANOQ,544
 lucidicai/telemetry/litellm_bridge.py,sha256=mOdjEfvP--ToDv8snoOMU4pRQx_Yg4s2o3BMTMzeRK8,14979
 lucidicai/telemetry/lucidic_exporter.py,sha256=h2GOnEk22Fpeke4Zc7SSk391yr0joUApVwolV5Q4hz4,10818
-lucidicai/telemetry/lucidic_span_processor.py,sha256=Hf15USiY2smfr29HXwmMRy7SNLwuLJg5wCUdLKjOp3Y,29420
+lucidicai/telemetry/lucidic_span_processor.py,sha256=9EKD4fvnEQIoPBLhGCAHRxsLAWmJNGFnb1NpwRUlMOI,30048
 lucidicai/telemetry/openai_agents_instrumentor.py,sha256=__wIbeglMnEEf4AGTQ--FXeWCKmz2yy8SBupwprEdZA,12694
 lucidicai/telemetry/opentelemetry_converter.py,sha256=xOHCqoTyO4hUkL6k7fxy84PbljPpYep6ET9ZqbkJehc,17665
 lucidicai/telemetry/otel_handlers.py,sha256=HqGYIWJI_Vp8So2-HMpPjnrgTBSgBHHLDu01z_sq-Qk,14646
@@ -44,7 +44,7 @@ lucidicai/telemetry/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 lucidicai/telemetry/utils/image_storage.py,sha256=4Z59ZpVexr7-lcExfr8GsqXe0y2VZmr8Yjwa-3DeOxU,1457
 lucidicai/telemetry/utils/text_storage.py,sha256=L62MMJ8E23TDqDTUv2aRntdKMCItsXV7XjY6cFwx2DE,1503
 lucidicai/telemetry/utils/universal_image_interceptor.py,sha256=zPfVsMjtKxJP2n2OOjKbtPiQJTZ0sf5_28GWprOnJP4,12185
-lucidicai-1.2.21.dist-info/METADATA,sha256=J9vZ_O1IedeJFTUJu7VynU7-RgADkqXDLSIOE8r_Oms,903
-lucidicai-1.2.21.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-lucidicai-1.2.21.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
-lucidicai-1.2.21.dist-info/RECORD,,
+lucidicai-1.2.22.dist-info/METADATA,sha256=jFsbWQNg8hHEu2sHwDzC8xOlRdgUFo1KsL4QjvJDN50,903
+lucidicai-1.2.22.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+lucidicai-1.2.22.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
+lucidicai-1.2.22.dist-info/RECORD,,