lucidicai 1.2.17__py3-none-any.whl → 1.2.18__py3-none-any.whl

lucidicai/__init__.py

@@ -105,7 +105,8 @@ __all__ = [
 
 
 def init(
-    session_name: str,
+    session_name: Optional[str] = None,
+    session_id: Optional[str] = None,
     lucidic_api_key: Optional[str] = None,
     agent_id: Optional[str] = None,
     task: Optional[str] = None,
@@ -121,7 +122,8 @@ def init(
     Initialize the Lucidic client.
     
     Args:
-        session_name: The name of the session.
+        session_name: The display name of the session.
+        session_id: Custom ID of the session. If not provided, a random ID will be generated.
         lucidic_api_key: API key for authentication. If not provided, will use the LUCIDIC_API_KEY environment variable.
         agent_id: Agent ID. If not provided, will use the LUCIDIC_AGENT_ID environment variable.
         task: Task description.
@@ -171,6 +173,7 @@ def init(
         rubrics=rubrics,
         tags=tags,
         production_monitoring=production_monitoring,
+        custom_session_id=session_id,
     )
     if masking_function:
         client.masking_function = masking_function
@@ -198,15 +201,13 @@ def continue_session(
         agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
         if agent_id is None:
             raise APIKeyVerificationError("Lucidic agent ID not specified. Make sure to either pass your agent ID into lai.init() or set the LUCIDIC_AGENT_ID environment variable.")
-    try:
-        client = Client()
-        if client.session:
-            raise InvalidOperationError("[Lucidic] Session already in progress. Please call lai.end_session() first.")
-    except LucidicNotInitializedError:
-        client = Client(
-            lucidic_api_key=lucidic_api_key,
-            agent_id=agent_id,
-        )
+    
+    client = Client()
+    if client.session:
+        raise InvalidOperationError("[Lucidic] Session already in progress. Please call lai.end_session() or lai.reset_sdk() first.")
+    # if not yet initialized or still the NullClient -> create a real client when init is called
+    if not getattr(client, 'initialized', False):
+        client = Client(lucidic_api_key=lucidic_api_key, agent_id=agent_id)
     
     # Handle auto_end with environment variable support
     if auto_end is None:

lucidicai/client.py

@@ -77,7 +77,8 @@ class Client:
         task: Optional[str] = None,
         rubrics: Optional[list] = None,
         tags: Optional[list] = None,
-        production_monitoring: Optional[bool] = False
+        production_monitoring: Optional[bool] = False,
+        custom_session_id: Optional[str] = None,
     ) -> None:
         self.session = Session(
             agent_id=self.agent_id,
@@ -86,7 +87,8 @@ class Client:
             task=task,
             rubrics=rubrics,
             tags=tags,
-            production_monitoring=production_monitoring
+            production_monitoring=production_monitoring,
+            custom_session_id=custom_session_id
         )
         self.initialized = True
         return self.session.session_id
@@ -96,6 +98,9 @@ class Client:
             agent_id=self.agent_id,
             session_id=session_id
         )
+        if self.session.session_id != session_id:
+            # Custom session ID provided
+            self.session.custom_session_id = session_id
         self.initialized = True
         return self.session.session_id
 

lucidicai/session.py

@@ -44,7 +44,8 @@ class Session:
             "mass_sim_id": kwargs.get("mass_sim_id", None),
             "rubrics": kwargs.get("rubrics", None),
             "tags": kwargs.get("tags", None),
-            "production_monitoring": kwargs.get("production_monitoring", False)
+            "production_monitoring": kwargs.get("production_monitoring", False),
+            "custom_session_id": kwargs.get("custom_session_id", None)
         }
         data = Client().make_request('initsession', 'POST', request_data)
         self.session_id = data["session_id"]

{lucidicai-1.2.17.dist-info → lucidicai-1.2.18.dist-info}/METADATA

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

{lucidicai-1.2.17.dist-info → lucidicai-1.2.18.dist-info}/RECORD

@@ -1,13 +1,13 @@
-lucidicai/__init__.py,sha256=lj8_84KeLYcJVrqUnkuyeFKUfwv1L5qY85gtpMxEsIM,19765
+lucidicai/__init__.py,sha256=qu1NTg-te86yT2j8BVz2NSi_Zs2G842K_YpxlMs9m6Q,20036
 lucidicai/action.py,sha256=sPRd1hTIVXDqnvG9ZXWEipUFh0bsXcE0Fm7RVqmVccM,237
-lucidicai/client.py,sha256=_EHUB--u-hvdhKTks4jpqvA18-GBwJfnyjoPZOXkA04,6374
+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=2j5G4drwVhmdQkk9-armYQj1ctmLm2H_rNfPsfV_8wA,2004
 lucidicai/image_upload.py,sha256=6SRudg-BpInM2gzMx1Yf1Rz_Zyh8inwoJ7U4pBw7ruY,3807
 lucidicai/model_pricing.py,sha256=Q_Kd8OtKPqtvZ4pIh__ztKb3RXe-cX-8xgrM-AIUj9E,12189
-lucidicai/session.py,sha256=_67V6Be9Js-JEdCfiY_wkr4VUkQOdwZKUGUwf5mGXps,5429
+lucidicai/session.py,sha256=T_4z3CLKMp5DOKSVpdWwMZkZG9ndXU6oiI5Al6DHSTY,5501
 lucidicai/singleton.py,sha256=gfT3XdWLXSIWMqDXbY6-pnesMZ8RGRitaEPhIsgrRPw,1272
 lucidicai/state.py,sha256=4Tb1X6l2or6w_e62FYSuEeghAv3xXm5gquKwzCpvdok,235
 lucidicai/step.py,sha256=_oBIyTBZBvNkUkYHIrwWd75KMSlMtR9Ws2Lo71Lyff8,2522
@@ -43,7 +43,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.17.dist-info/METADATA,sha256=SljtwfhzcGtEwnsx6JAmqDmYY9KniQi5mY3KH3xhR6M,577
-lucidicai-1.2.17.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-lucidicai-1.2.17.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
-lucidicai-1.2.17.dist-info/RECORD,,
+lucidicai-1.2.18.dist-info/METADATA,sha256=1Ltw_Uxf-WrD3qPqVzBSlk0vaTzJrFdv-SgObiA6GJY,577
+lucidicai-1.2.18.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+lucidicai-1.2.18.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
+lucidicai-1.2.18.dist-info/RECORD,,