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

lucidicai/__init__.py

@@ -111,7 +111,7 @@ __all__ = [
 def init(
     session_name: Optional[str] = None,
     session_id: Optional[str] = None,
-    lucidic_api_key: Optional[str] = None,
+    api_key: Optional[str] = None,
     agent_id: Optional[str] = None,
     task: Optional[str] = None,
     providers: Optional[List[ProviderType]] = [],
@@ -128,7 +128,7 @@ def init(
     Args:
         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.
+        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.
         providers: List of provider types ("openai", "anthropic", "langchain", "pydantic_ai").
@@ -142,27 +142,29 @@ def init(
         InvalidOperationError: If the client is already initialized.
         APIKeyVerificationError: If the API key is invalid.
     """
-    if lucidic_api_key is None:
-        lucidic_api_key = os.getenv("LUCIDIC_API_KEY", None)
-        if lucidic_api_key is None:
-            raise APIKeyVerificationError("Make sure to either pass your API key into lai.init() or set the LUCIDIC_API_KEY environment variable.")
-    if agent_id is None:
-        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.")
-
+    
     # get current client which will be NullClient if never lai is never initialized
     client = Client()
-    # ff not yet initialized or still the NullClient -> creaet a real client when init is called
+    # if not yet initialized or still the NullClient -> creaet 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)
-    
-    if not production_monitoring:
-        production_monitoring = os.getenv("LUCIDIC_PRODUCTION_MONITORING", False)
-        if production_monitoring == "True":
-            production_monitoring = True
-        else:
-            production_monitoring = False
+        if api_key is None:
+            api_key = os.getenv("LUCIDIC_API_KEY", None)
+            if api_key is None:
+                raise APIKeyVerificationError("Make sure to either pass your API key into lai.init() or set the LUCIDIC_API_KEY environment variable.")
+        if agent_id is None:
+            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.")
+        client = Client(api_key=api_key, agent_id=agent_id)
+    else:
+        # Already initialized, this is a re-init
+        api_key = api_key or os.getenv("LUCIDIC_API_KEY", None)
+        agent_id = agent_id or os.getenv("LUCIDIC_AGENT_ID", None)
+        client.agent_id = agent_id
+        if api_key is not None and agent_id is not None and (api_key != client.api_key or agent_id != client.agent_id):
+            client.set_api_key(api_key)
+            client.agent_id = agent_id
+        
     
     # Handle auto_end with environment variable support
     if auto_end is None:
@@ -170,14 +172,14 @@ def init(
     
     # Set up providers
     _setup_providers(client, providers)
-    session_id = client.init_session(
+    real_session_id = client.init_session(
         session_name=session_name,
         mass_sim_id=mass_sim_id,
         task=task,
         rubrics=rubrics,
         tags=tags,
         production_monitoring=production_monitoring,
-        custom_session_id=session_id,
+        session_id=session_id,
     )
     if masking_function:
         client.masking_function = masking_function
@@ -186,20 +188,20 @@ def init(
     client.auto_end = auto_end
     
     logger.info("Session initialized successfully")
-    return session_id
+    return real_session_id
 
 
 def continue_session(
     session_id: str,
-    lucidic_api_key: Optional[str] = None,
+    api_key: Optional[str] = None,
     agent_id: Optional[str] = None,
     providers: Optional[List[ProviderType]] = [],
     masking_function = None,
     auto_end: Optional[bool] = True,
 ):
-    if lucidic_api_key is None:
-        lucidic_api_key = os.getenv("LUCIDIC_API_KEY", None)
-        if lucidic_api_key is None:
+    if api_key is None:
+        api_key = os.getenv("LUCIDIC_API_KEY", None)
+        if api_key is None:
             raise APIKeyVerificationError("Make sure to either pass your API key into lai.init() or set the LUCIDIC_API_KEY environment variable.")
     if agent_id is None:
         agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
@@ -211,7 +213,7 @@ def continue_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)
+        client = Client(api_key=api_key, agent_id=agent_id)
     
     # Handle auto_end with environment variable support
     if auto_end is None:
@@ -284,8 +286,10 @@ def end_session(
 
 def reset_sdk() -> None:
     """
-    Reset the SDK.
+    DEPRECATED: Reset the SDK.
     """
+    return
+
     client = Client()
     if not client.initialized:
         return
@@ -342,7 +346,7 @@ signal.signal(signal.SIGTERM, _signal_handler)
 def create_mass_sim(
     mass_sim_name: str,
     total_num_sessions: int,
-    lucidic_api_key: Optional[str] = None,
+    api_key: Optional[str] = None,
     agent_id: Optional[str] = None,
     task: Optional[str] = None,
     tags: Optional[list] = None
@@ -353,7 +357,7 @@ def create_mass_sim(
     Args:
         mass_sim_name: Name of the mass simulation.
         total_num_sessions: Total intended number of sessions. More sessions can be added later.
-        lucidic_api_key: API key for authentication. If not provided, will use the LUCIDIC_API_KEY environment variable.
+        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.
         tags: Tags for the mass simulation.
@@ -361,9 +365,9 @@ def create_mass_sim(
     Returns:
         mass_sim_id: ID of the created mass simulation. Pass this to lai.init() to create a new session in the mass sim.
     """
-    if lucidic_api_key is None:
-        lucidic_api_key = os.getenv("LUCIDIC_API_KEY", None)
-        if lucidic_api_key is None:
+    if api_key is None:
+        api_key = os.getenv("LUCIDIC_API_KEY", None)
+        if api_key is None:
             raise APIKeyVerificationError("Make sure to either pass your API key into lai.init() or set the LUCIDIC_API_KEY environment variable.")
     if agent_id is None:
         agent_id = os.getenv("LUCIDIC_AGENT_ID", None)
@@ -373,7 +377,7 @@ def create_mass_sim(
         client = Client()
     except LucidicNotInitializedError:
         client = Client( # TODO: fail hard if incorrect API key or agent ID provided and wrong, fail silently if not provided
-            lucidic_api_key=lucidic_api_key,
+            api_key=api_key,
             agent_id=agent_id,
         )
     mass_sim_id = client.init_mass_sim(mass_sim_name=mass_sim_name, total_num_sims=total_num_sessions, task=task, tags=tags)  # TODO: change total_num_sims to total_num_sessions everywhere

lucidicai/client.py

@@ -13,7 +13,7 @@ from .errors import APIKeyVerificationError, InvalidOperationError, LucidicNotIn
 from .telemetry.base_provider import BaseProvider 
 from .session import Session
 from .singleton import singleton, clear_singletons
-
+from .lru import LRUCache
 
 NETWORK_RETRIES = 3
 
@@ -22,14 +22,16 @@ NETWORK_RETRIES = 3
 class Client:
     def __init__(
         self,
-        lucidic_api_key: str,
+        api_key: str,
         agent_id: str,
     ):
         self.base_url = "https://analytics.lucidic.ai/api" if not (os.getenv("LUCIDIC_DEBUG", 'False') == 'True') else "http://localhost:8000/api"
         self.initialized = False
         self.session = None
+        self.previous_sessions = LRUCache(500)  # For LRU cache of previously initialized sessions
+        self.custom_session_id_translations = LRUCache(500) # For translations of custom session IDs to real session IDs
         self.providers = []
-        self.api_key = lucidic_api_key
+        self.api_key = api_key
         self.agent_id = agent_id
         self.masking_function = None
         self.auto_end = False  # Default to False until explicitly set during init
@@ -42,13 +44,17 @@ class Client:
         )
         adapter = HTTPAdapter(max_retries=retry_cfg, pool_connections=20, pool_maxsize=100)
         self.request_session.mount("https://", adapter)
-        self.request_session.headers.update({"Authorization": f"Api-Key {self.api_key}", "User-Agent": "lucidic-sdk/1.1"})
+        self.set_api_key(api_key)
         self.prompts = dict()
+
+    def set_api_key(self, api_key: str):
+        self.api_key = api_key
+        self.request_session.headers.update({"Authorization": f"Api-Key {self.api_key}", "User-Agent": "lucidic-sdk/1.1"})
         try:
-            self.verify_api_key(self.base_url, lucidic_api_key)
+            self.verify_api_key(self.base_url, api_key)
         except APIKeyVerificationError:
             raise APIKeyVerificationError("Invalid API Key")
-    
+
     def clear(self):
         self.undo_overrides()
         clear_singletons()
@@ -69,7 +75,7 @@ class Client:
     def undo_overrides(self):
         for provider in self.providers:
             provider.undo_override()
-    
+
     def init_session(
         self,
         session_name: str,
@@ -78,30 +84,69 @@ class Client:
         rubrics: Optional[list] = None,
         tags: Optional[list] = None,
         production_monitoring: Optional[bool] = False,
-        custom_session_id: Optional[str] = None,
+        session_id: Optional[str] = None,
     ) -> None:
+        if session_id:
+            # Check if it's a known session ID, maybe custom and maybe real
+            if session_id in self.custom_session_id_translations:
+                session_id = self.custom_session_id_translations[session_id]
+            # Check if it's the same as the current session
+            if self.session and self.session.session_id == session_id:
+                return self.session.session_id
+            # Check if it's a previous session that we have saved
+            if session_id in self.previous_sessions:
+                if self.session:
+                    self.previous_sessions[self.session.session_id] = self.session
+                self.session = self.previous_sessions.pop(session_id)  # Remove from previous sessions because it's now the current session
+                return self.session.session_id
+
+        # Either there's no session ID, or we don't know about the old session
+        # We need to go to the backend in both cases
+        request_data = {
+            "agent_id": self.agent_id,
+            "session_name": session_name,
+            "task": task,
+            "mass_sim_id": mass_sim_id,
+            "rubrics": rubrics,
+            "tags": tags,
+            "session_id": session_id
+        }
+        data = self.make_request('initsession', 'POST', request_data)
+        real_session_id = data["session_id"]
+        if session_id and session_id != real_session_id:
+            self.custom_session_id_translations[session_id] = real_session_id
+        
+        if self.session:
+            self.previous_sessions[self.session.session_id] = self.session
+
         self.session = Session(
             agent_id=self.agent_id,
+            session_id=real_session_id,
             session_name=session_name,
             mass_sim_id=mass_sim_id,
             task=task,
             rubrics=rubrics,
             tags=tags,
-            production_monitoring=production_monitoring,
-            custom_session_id=custom_session_id
         )
         self.initialized = True
         return self.session.session_id
 
     def continue_session(self, session_id: str):
+        if session_id in self.custom_session_id_translations:
+            session_id = self.custom_session_id_translations[session_id]
+        if self.session and self.session.session_id == session_id:
+            return self.session.session_id
+        if self.session:
+            self.previous_sessions[self.session.session_id] = self.session
+        data = self.make_request('continuesession', 'POST', {"session_id": session_id})
+        real_session_id = data["session_id"]
+        if session_id != real_session_id:
+            self.custom_session_id_translations[session_id] = real_session_id
         self.session = Session(
             agent_id=self.agent_id,
-            session_id=session_id
+            session_id=real_session_id
         )
-        if self.session.session_id != session_id:
-            # Custom session ID provided
-            self.session.custom_session_id = session_id
-        self.initialized = True
+        logger.info(f"Session {data.get('session_name', '')} continuing...")
         return self.session.session_id
 
     def init_mass_sim(self, **kwargs) -> str:

lucidicai/event.py

@@ -16,7 +16,6 @@ class Event:
         self.is_finished = False
         self.init_event(**kwargs)
 
-    # TODO: this is really bad, clean this up later
     def init_event(self, **kwargs) -> None:
         from .client import Client
         request_data = self._build_request_data(**kwargs)
@@ -41,7 +40,7 @@ class Event:
 
     def _build_request_data(self, **kwargs) -> dict:
         from .client import Client
-        num_new_screenshots = len(kwargs.get("screenshots", []))
+        num_new_screenshots = len(kwargs.get("screenshots", []) or [])
         return {
             "description": Client().mask(kwargs.get("description", None)),
             "result": Client().mask(kwargs.get("result", None)),

lucidicai/lru.py

@@ -0,0 +1,19 @@
+from collections import OrderedDict
+
+class LRUCache(OrderedDict):
+    def __init__(self, capacity: int):
+        super().__init__()
+        self.capacity = capacity
+
+    def __getitem__(self, key):
+        if key not in self:
+            raise KeyError(key)
+        self.move_to_end(key)  # Mark as recently used
+        return super().__getitem__(key)
+
+    def __setitem__(self, key, value):
+        if key in self:
+            self.move_to_end(key)  # Update position
+        super().__setitem__(key, value)
+        if len(self) > self.capacity:
+            self.popitem(last=False)  # Evict least recently used item

lucidicai/session.py

@@ -15,11 +15,12 @@ logger = logging.getLogger("Lucidic")
 class Session:
     def __init__(
         self, 
-        agent_id: str, 
+        agent_id: str,
+        session_id = None,
         **kwargs
     ):
         self.agent_id = agent_id
-        self.session_id = None
+        self.session_id = session_id
         self.step_history = dict()
         self._active_step: Optional[str] = None  # Step ID, not Step object
         self.event_history = dict()
@@ -30,33 +31,6 @@ class Session:
         self.session_eval = None
         self.session_eval_reason = None
         self.has_gif = None
-        if kwargs.get("session_id", None) is None:  # The kwarg, not the attribute
-            self.init_session(**kwargs)
-        else:
-            self.continue_session(kwargs["session_id"])
-
-    def init_session(self, **kwargs) -> None:
-        from .client import Client
-        request_data = {
-            "agent_id": self.agent_id,
-            "session_name": kwargs.get("session_name", None),
-            "task": kwargs.get("task", None),
-            "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),
-            "custom_session_id": kwargs.get("custom_session_id", None)
-        }
-        data = Client().make_request('initsession', 'POST', request_data)
-        self.session_id = data["session_id"]
-    
-    def continue_session(self, session_id: str) -> None:
-        from .client import Client
-        self.session_id = session_id
-        data = Client().make_request('continuesession', 'POST', {"session_id": session_id})
-        self.session_id = data["session_id"]
-        logger.info(f"Session {data['session_name']} continuing...")
-        return self.session_id
 
     @property   
     def active_step(self) -> Optional[Step]:

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

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

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

@@ -1,13 +1,14 @@
-lucidicai/__init__.py,sha256=8_qfdx62xyvrdyONXCHZ4iKDWCvmyncp1T5Epb1DIv8,20664
+lucidicai/__init__.py,sha256=-eSSR9UaMubE0KpbesR2jLU0xUn37oucMNxjgKaCfkA,20738
 lucidicai/action.py,sha256=sPRd1hTIVXDqnvG9ZXWEipUFh0bsXcE0Fm7RVqmVccM,237
-lucidicai/client.py,sha256=3yBVppvwBwesca_pZSKgTUDihzZe5JhVgh1AALCpJ_Q,6620
+lucidicai/client.py,sha256=MZuNd1S6MTwNRDJ8cUrZfnTZgBBHFwTO2Pc_-wbVNEM,8981
 lucidicai/constants.py,sha256=_u0z3M4geZgS1g-CrOZUVjtcew8l70dKQnpVQvlXh9w,2172
 lucidicai/decorators.py,sha256=oqXyfHk9f9UmeaIquuU8mtzed1qZtO_-svwadpoat6g,13950
 lucidicai/errors.py,sha256=gTg0bdzjuTcUnakRbZnxjngO4gZnRLVwRHRglpZZJsM,970
-lucidicai/event.py,sha256=SjPSGNBuW6YLDJUhY16IvQgE9lrg3bypgWOK9IKVJWI,2567
+lucidicai/event.py,sha256=EF-KNPMBHGdNnhchu5HF7QnYJSMofUaq-QuGL0j_OAE,2521
 lucidicai/image_upload.py,sha256=6SRudg-BpInM2gzMx1Yf1Rz_Zyh8inwoJ7U4pBw7ruY,3807
+lucidicai/lru.py,sha256=PXiDSoUCOxjamG1QlQx6pDbQCm8h5hKAnnr_NI0PEgE,618
 lucidicai/model_pricing.py,sha256=o1yWCaF5Qxj4tloXxVG3SZXcTMKtk56J_Nfdo8M4uR0,11947
-lucidicai/session.py,sha256=mUPwXcGW5tFsK6Udl4i2jndV0STBYHbPyWqA8GxBdHM,5285
+lucidicai/session.py,sha256=NFkWJioqGcRxHjfqwWw-1DZSiScNpnUlY1gtGGApqSQ,4060
 lucidicai/singleton.py,sha256=gfT3XdWLXSIWMqDXbY6-pnesMZ8RGRitaEPhIsgrRPw,1272
 lucidicai/state.py,sha256=4Tb1X6l2or6w_e62FYSuEeghAv3xXm5gquKwzCpvdok,235
 lucidicai/step.py,sha256=_oBIyTBZBvNkUkYHIrwWd75KMSlMtR9Ws2Lo71Lyff8,2522
@@ -44,7 +45,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.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,,
+lucidicai-1.3.1.dist-info/METADATA,sha256=fMRzVo9mOgHvcC6PGrH5zdyLaMRrz3KBch6fMkHtgEE,902
+lucidicai-1.3.1.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
+lucidicai-1.3.1.dist-info/top_level.txt,sha256=vSSdM3lclF4I5tyVC0xxUk8eIRnnYXMe1hW-eO91HUo,10
+lucidicai-1.3.1.dist-info/RECORD,,