lumera 0.6.0__tar.gz → 0.7.3__tar.gz

{lumera-0.6.0 → lumera-0.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lumera
-Version: 0.6.0
+Version: 0.7.3
 Summary: SDK for building on Lumera platform
 Requires-Python: >=3.11
 Requires-Dist: requests

{lumera-0.6.0 → lumera-0.7.3}/lumera/__init__.py

@@ -5,10 +5,10 @@ This SDK provides helpers for agents running within the Lumera Notebook environm
 to interact with the Lumera API and define dynamic user interfaces.
 """
 
-__version__ = "0.5.0"
+__version__ = "0.7.0"
 
 # Import new modules (as modules, not individual functions)
-from . import exceptions, llm, locks, pb, storage
+from . import automations, exceptions, llm, locks, pb, storage
 from ._utils import (
     LumeraAPIError,
     RecordNotUniqueError,
@@ -34,7 +34,7 @@ from .sdk import (
     create_record,
     delete_collection,
     delete_record,
-    get_agent_run,
+    get_automation_run,
     get_collection,
     get_record,
     get_record_by_external_id,
@@ -42,8 +42,9 @@ from .sdk import (
     list_records,
     query_sql,
     replay_hook,
-    run_agent,
+    run_automation,
     save_to_lumera,
+    update_automation_run,
     update_collection,
     update_record,
     upload_lumera_file,
@@ -73,8 +74,9 @@ __all__ = [
     # Other operations
     "replay_hook",
     "query_sql",
-    "run_agent",
-    "get_agent_run",
+    "run_automation",
+    "get_automation_run",
+    "update_automation_run",
     "upload_lumera_file",
     "save_to_lumera",
     # Type definitions
@@ -89,6 +91,7 @@ __all__ = [
     "RecordNotFoundError",
     "LockHeldError",
     # New modules (use as lumera.pb, lumera.storage, etc.)
+    "automations",
     "pb",
     "storage",
     "llm",

{lumera-0.6.0 → lumera-0.7.3}/lumera/_utils.py

@@ -117,10 +117,10 @@ def _utcnow_iso() -> str:
     )
 
 
-def _default_provenance(agent_id: str, run_id: str | None) -> dict[str, Any]:
+def _default_provenance(automation_id: str, run_id: str | None) -> dict[str, Any]:
     recorded_at = _utcnow_iso()
-    env_agent_id = os.getenv("LUMERA_AGENT_ID", "").strip()
-    agent_id = (agent_id or "").strip() or env_agent_id
+    env_automation_id = os.getenv("LUMERA_AUTOMATION_ID", "").strip()
+    automation_id = (automation_id or "").strip() or env_automation_id
 
     run_id = (run_id or "").strip() or os.getenv("LUMERA_RUN_ID", "").strip()
 
@@ -132,8 +132,8 @@ def _default_provenance(agent_id: str, run_id: str | None) -> dict[str, Any]:
         "recorded_at": recorded_at,
     }
 
-    if agent_id:
-        payload["agent"] = {"id": agent_id}
+    if automation_id:
+        payload["agent"] = {"id": automation_id}  # Backend still expects "agent" key
 
     if run_id:
         payload["agent_run"] = {"id": run_id}
@@ -284,7 +284,7 @@ def _ensure_mapping(payload: Mapping[str, Any] | None, *, name: str) -> dict[str
     return dict(payload)
 
 
-def _prepare_agent_inputs(
+def _prepare_automation_inputs(
     inputs: Mapping[str, Any] | str | None,
 ) -> dict[str, Any] | None:
     if inputs is None:
@@ -345,7 +345,7 @@ def _upload_file_to_presigned(upload_url: str, path: pathlib.Path, content_type:
         resp.raise_for_status()
 
 
-def _upload_agent_files(
+def _upload_automation_files(
     run_id: str | None,
     files: Mapping[str, str | os.PathLike[str] | Sequence[str | os.PathLike[str]]],
     *,
@@ -370,7 +370,7 @@ def _upload_agent_files(
             size = file_path.stat().st_size
 
             body: dict[str, Any] = {
-                "scope": "agent_run",
+                "scope": "automation_run",
                 "filename": filename,
                 "content_type": content_type,
                 "size": size,
@@ -631,7 +631,7 @@ def _upload_session_file(file_path: str, session_id: str) -> dict:
     return {"name": filename, "notebook_path": notebook_path}
 
 
-def _upload_agent_run_file(file_path: str, run_id: str) -> dict:
+def _upload_automation_run_file(file_path: str, run_id: str) -> dict:
     token = get_lumera_token()
     path = pathlib.Path(file_path).expanduser().resolve()
     if not path.is_file():
@@ -644,7 +644,7 @@ def _upload_agent_run_file(file_path: str, run_id: str) -> dict:
     headers = {"Authorization": f"token {token}", "Content-Type": "application/json"}
 
     resp = requests.post(
-        f"{API_BASE}/agent-runs/{run_id}/files/upload-url",
+        f"{API_BASE}/automation-runs/{run_id}/files/upload-url",
         json={"filename": filename, "content_type": mimetype, "size": size},
         headers=headers,
         timeout=30,
@@ -742,10 +742,10 @@ __all__ = [
     "_is_sequence",
     "_normalize_mount_path",
     "_pretty_size",
-    "_prepare_agent_inputs",
+    "_prepare_automation_inputs",
     "_record_mutation",
-    "_upload_agent_files",
-    "_upload_agent_run_file",
+    "_upload_automation_files",
+    "_upload_automation_run_file",
     "_upload_file_to_presigned",
     "_upload_document",
     "_upload_session_file",