lucidicai 1.3.1__tar.gz → 1.3.2__tar.gz

{lucidicai-1.3.1 → lucidicai-1.3.2}/PKG-INFO

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

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai/__init__.py

@@ -117,6 +117,7 @@ def init(
     providers: Optional[List[ProviderType]] = [],
     production_monitoring: Optional[bool] = False,
     mass_sim_id: Optional[str] = None,
+    experiment_id: Optional[str] = None,
     rubrics: Optional[list] = None,
     tags: Optional[list] = None,
     masking_function = None,
@@ -133,6 +134,7 @@ def init(
         task: Task description.
         providers: List of provider types ("openai", "anthropic", "langchain", "pydantic_ai").
         mass_sim_id: Optional mass simulation ID, if session is to be part of a mass simulation.
+        experiment_id: Optional experiment ID, if session is to be part of an experiment.
         rubrics: Optional rubrics for evaluation, list of strings.
         tags: Optional tags for the session, list of strings.
         masking_function: Optional function to mask sensitive data.
@@ -180,6 +182,7 @@ def init(
         tags=tags,
         production_monitoring=production_monitoring,
         session_id=session_id,
+        experiment_id=experiment_id,
     )
     if masking_function:
         client.masking_function = masking_function
@@ -487,7 +490,9 @@ def create_event(
     result: Optional[str] = None,
     cost_added: Optional[float] = None, 
     model: Optional[str] = None,
-    screenshots: Optional[List[str]] = None
+    screenshots: Optional[List[str]] = None,
+    function_name: Optional[str] = None,
+    arguments: Optional[dict] = None,
 ) -> str:
     """
     Create a new event in the current step. Current step must not be finished.
@@ -498,6 +503,8 @@ def create_event(
         cost_added: Cost added by the event.
         model: Model used for the event.
         screenshots: List of screenshots encoded in base64.
+        function_name: Name of the function that created the event.
+        arguments: Arguments of the function that created the event.
     """
 
     client = Client()
@@ -512,7 +519,9 @@ def update_event(
     result: Optional[str] = None,
     cost_added: Optional[float] = None, 
     model: Optional[str] = None,
-    screenshots: Optional[List[str]] = None
+    screenshots: Optional[List[str]] = None,
+    function_name: Optional[str] = None,
+    arguments: Optional[dict] = None,
 ) -> None:
     """
     Update the event with the given ID in the current step.
@@ -524,6 +533,8 @@ def update_event(
         cost_added: Cost added by the event.
         model: Model used for the event.
         screenshots: List of screenshots encoded in base64.
+        function_name: Name of the function that created the event.
+        arguments: Arguments of the function that created the event.
     """
     client = Client()
     if not client.session:
@@ -537,7 +548,9 @@ def end_event(
     result: Optional[str] = None,
     cost_added: Optional[float] = None, 
     model: Optional[str] = None,
-    screenshots: Optional[List[str]] = None
+    screenshots: Optional[List[str]] = None,
+    function_name: Optional[str] = None,
+    arguments: Optional[dict] = None,
 ) -> None:
     """
     End the latest event in the current step.
@@ -548,6 +561,9 @@ def end_event(
         result: Result of the event.
         cost_added: Cost added by the event.
         model: Model used for the event.
+        screenshots: List of screenshots encoded in base64.
+        function_name: Name of the function that created the event.
+        arguments: Arguments of the function that created the event.
     """
     client = Client()
     if not client.session:

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai/client.py

@@ -85,6 +85,7 @@ class Client:
         tags: Optional[list] = None,
         production_monitoring: Optional[bool] = False,
         session_id: Optional[str] = None,
+        experiment_id: Optional[str] = None,
     ) -> None:
         if session_id:
             # Check if it's a known session ID, maybe custom and maybe real
@@ -107,6 +108,7 @@ class Client:
             "session_name": session_name,
             "task": task,
             "mass_sim_id": mass_sim_id,
+            "experiment_id": experiment_id,
             "rubrics": rubrics,
             "tags": tags,
             "session_id": session_id
@@ -124,6 +126,7 @@ class Client:
             session_id=real_session_id,
             session_name=session_name,
             mass_sim_id=mass_sim_id,
+            experiment_id=experiment_id,
             task=task,
             rubrics=rubrics,
             tags=tags,

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai/decorators.py

@@ -5,6 +5,7 @@ import inspect
 import json
 import logging
 from typing import Any, Callable, Optional, TypeVar, Union
+from collections.abc import Iterable
 
 from .client import Client
 from .errors import LucidicNotInitializedError
@@ -219,27 +220,44 @@ def event(
             
             # Build event description from inputs if not provided
             event_desc = description
+            function_name = func.__name__
+            
+            # Get function signature
+            sig = inspect.signature(func)
+            bound_args = sig.bind(*args, **kwargs)
+            bound_args.apply_defaults()
+
+            def serialize(value):
+                if isinstance(value, str):
+                    return value
+                if isinstance(value, int):
+                    return value
+                if isinstance(value, float):
+                    return value
+                if isinstance(value, bool):
+                    return value
+                if isinstance(value, dict):
+                    return {k: serialize(v) for k, v in value.items()}
+                if isinstance(value, Iterable):
+                    return [serialize(v) for v in value]
+                return str(value)
+
+            # Construct JSONable object of args
+            args_dict = {
+                param_name: serialize(param_value)  # Recursive - maybe change later
+                for param_name, param_value in bound_args.arguments.items()
+            }
+            
             if not event_desc:
-                # Get function signature
-                sig = inspect.signature(func)
-                bound_args = sig.bind(*args, **kwargs)
-                bound_args.apply_defaults()
-                
-                # Create string representation of inputs
-                input_parts = []
-                for param_name, param_value in bound_args.arguments.items():
-                    try:
-                        input_parts.append(f"{param_name}={repr(param_value)}")
-                    except Exception:
-                        input_parts.append(f"{param_name}=<{type(param_value).__name__}>")
-                
-                event_desc = f"{func.__name__}({', '.join(input_parts)})"
+                event_desc = f"Function {function_name}({json.dumps(args_dict)})"
             
             # Create the event
             event_id = create_event(
                 description=event_desc,
                 model=model,
-                cost_added=cost_added
+                cost_added=cost_added,
+                function_name=function_name,
+                arguments=args_dict, 
             )
             tok = _current_event.set(event_id)
             try:

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai/event.py

@@ -48,7 +48,9 @@ class Event:
             "cost_added": kwargs.get("cost_added", None),
             "model": kwargs.get("model", None), 
             "nscreenshots": len(self.screenshots) + num_new_screenshots,
-            "duration": kwargs.get("duration", None)
+            "duration": kwargs.get("duration", None),
+            "function_name": kwargs.get("function_name", None),
+            "arguments": kwargs.get("arguments", None),
         }
     
     def _upload_screenshots(self, **kwargs) -> None:

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai/session.py

@@ -31,7 +31,7 @@ class Session:
         self.session_eval = None
         self.session_eval_reason = None
         self.has_gif = None
-
+        
     @property   
     def active_step(self) -> Optional[Step]:
         """Get the active step object"""

{lucidicai-1.3.1 → lucidicai-1.3.2}/lucidicai.egg-info/PKG-INFO

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

{lucidicai-1.3.1 → lucidicai-1.3.2}/setup.py

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