lucidicai 1.2.17__tar.gz → 1.2.18__tar.gz

{lucidicai-1.2.17 → lucidicai-1.2.18}/PKG-INFO

@@ -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 → lucidicai-1.2.18}/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-1.2.17 → lucidicai-1.2.18}/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-1.2.17 → lucidicai-1.2.18}/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 → lucidicai-1.2.18}/lucidicai.egg-info/PKG-INFO

@@ -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 → lucidicai-1.2.18}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="lucidicai",
-    version="1.2.17",
+    version="1.2.18",
     packages=find_packages(),
     install_requires=[
         "requests>=2.25.1",