lumera 0.9.9__py3-none-any.whl → 0.10.0__py3-none-any.whl

lumera/automations.py

@@ -57,18 +57,76 @@ __all__ = [
     "create",
     "update",
     "upsert",
-    # Log streaming and download
+    # Log functions
     "stream_logs",
+    "get_logs",
     "get_log_download_url",
     # Classes
     "Run",
     "Automation",
+    "LogsResponse",
 ]
 
 from ._utils import LumeraAPIError, _api_request
 from .sdk import get_automation_run as _get_automation_run
 from .sdk import run_automation as _run_automation
 
+# ============================================================================
+# LogsResponse Class
+# ============================================================================
+
+
+class LogsResponse:
+    """Response from fetching automation run logs.
+
+    Attributes:
+        data: Raw log content as a string (NDJSON format).
+        offset: Byte offset where this chunk starts.
+        size: Number of bytes in this chunk.
+        total_size: Total size of the log file.
+        has_more: True if there are more logs after this chunk.
+        source: Where logs came from ("live" or "archived").
+        truncated: True if logs were truncated at storage time (>50MB).
+    """
+
+    def __init__(self, data: dict[str, Any]) -> None:
+        self._data = data
+
+    @property
+    def data(self) -> str:
+        return self._data.get("data", "")
+
+    @property
+    def offset(self) -> int:
+        return self._data.get("offset", 0)
+
+    @property
+    def size(self) -> int:
+        return self._data.get("size", 0)
+
+    @property
+    def total_size(self) -> int:
+        return self._data.get("total_size", 0)
+
+    @property
+    def has_more(self) -> bool:
+        return self._data.get("has_more", False)
+
+    @property
+    def source(self) -> str:
+        return self._data.get("source", "")
+
+    @property
+    def truncated(self) -> bool:
+        return self._data.get("truncated", False)
+
+    def __repr__(self) -> str:
+        return (
+            f"LogsResponse(offset={self.offset}, size={self.size}, "
+            f"total_size={self.total_size}, has_more={self.has_more})"
+        )
+
+
 # ============================================================================
 # Run Class
 # ============================================================================
@@ -246,6 +304,66 @@ class Run:
             raise ValueError("Cannot get log URL without run id")
         return get_log_download_url(self.id)
 
+    def logs(
+        self,
+        *,
+        offset: int = 0,
+        limit: int = 1024 * 1024,
+        all: bool = False,
+    ) -> LogsResponse:
+        """Fetch logs for this run.
+
+        Works for both live (running) and archived (completed) runs.
+        Returns raw log data as a string (NDJSON format).
+
+        Args:
+            offset: Byte offset to start from. Negative values read from end
+                    (e.g., -1048576 = last 1MB).
+            limit: Maximum bytes to return (default 1MB).
+            all: If True, fetch all logs at once. Returns 400 if logs > 10MB.
+
+        Returns:
+            A LogsResponse object with data, offset, size, total_size, has_more,
+            source ("live" or "archived"), and truncated flag.
+
+        Raises:
+            ValueError: If the run has no ID.
+            LumeraAPIError: If logs are not available or request fails.
+
+        Example:
+            >>> run = automations.get_run("run_id")
+            >>> resp = run.logs()
+            >>> print(resp.data)  # Raw NDJSON log content
+            >>> while resp.has_more:
+            ...     resp = run.logs(offset=resp.offset + resp.size)
+            ...     print(resp.data)
+        """
+        if not self.id:
+            raise ValueError("Cannot fetch logs without run id")
+        return get_logs(self.id, offset=offset, limit=limit, all=all)
+
+    def stream_logs(self, *, timeout: float = 30) -> Iterator[str]:
+        """Stream logs from this run.
+
+        Works for both live (running) and archived (completed) runs.
+        For live runs, streams in real-time as logs are produced.
+        For archived runs, streams the entire log from S3.
+
+        Args:
+            timeout: HTTP connection timeout in seconds.
+
+        Yields:
+            Log lines as strings (raw NDJSON lines).
+
+        Example:
+            >>> run = automations.run("automation_id", inputs={})
+            >>> for line in run.stream_logs():
+            ...     print(line)
+        """
+        if not self.id:
+            raise ValueError("Cannot stream logs without run id")
+        return stream_logs(self.id, timeout=timeout)
+
     def to_dict(self) -> dict[str, Any]:
         """Return the underlying data dict."""
         return self._data.copy()
@@ -795,17 +913,19 @@ def delete(automation_id: str) -> None:
 
 
 def stream_logs(run_id: str, *, timeout: float = 30) -> Iterator[str]:
-    """Stream live logs from a running automation.
+    """Stream logs from an automation run.
 
+    Works for both live (running) and archived (completed) runs.
     Connects to the server-sent events endpoint and yields log lines
-    as they arrive. Stops when the run completes.
+    as they arrive. For live runs, streams in real-time. For archived
+    runs, streams the entire log from storage.
 
     Args:
         run_id: The run ID to stream logs from.
         timeout: HTTP connection timeout in seconds.
 
     Yields:
-        Log lines as strings.
+        Log lines as strings (raw NDJSON lines).
 
     Example:
         >>> for line in automations.stream_logs("run_id"):
@@ -825,7 +945,7 @@ def stream_logs(run_id: str, *, timeout: float = 30) -> Iterator[str]:
     if not token:
         raise ValueError("LUMERA_TOKEN environment variable is required")
 
-    url = f"{base_url}/automation-runs/{run_id}/logs/live"
+    url = f"{base_url}/automation-runs/{run_id}/logs?stream=true"
     headers = {
         "Authorization": f"token {token}",
         "Accept": "text/event-stream",
@@ -902,3 +1022,51 @@ def get_log_download_url(run_id: str) -> str:
     if isinstance(result, dict) and "url" in result:
         return result["url"]
     raise RuntimeError("Unexpected response: no download URL returned")
+
+
+def get_logs(
+    run_id: str,
+    *,
+    offset: int = 0,
+    limit: int = 1024 * 1024,
+    all: bool = False,
+) -> LogsResponse:
+    """Fetch logs for an automation run.
+
+    Works for both live (running) and archived (completed) runs.
+    Returns raw log data as a string (NDJSON format).
+
+    Args:
+        run_id: The run ID to get logs for.
+        offset: Byte offset to start from. Negative values read from end
+                (e.g., -1048576 = last 1MB).
+        limit: Maximum bytes to return (default 1MB).
+        all: If True, fetch all logs at once. Returns 400 if logs > 10MB.
+
+    Returns:
+        A LogsResponse object with data, offset, size, total_size, has_more,
+        source ("live" or "archived"), and truncated flag.
+
+    Raises:
+        ValueError: If run_id is empty.
+        LumeraAPIError: If logs are not available or request fails.
+
+    Example:
+        >>> resp = automations.get_logs("run_id")
+        >>> print(resp.data)  # Raw NDJSON log content
+        >>> while resp.has_more:
+        ...     resp = automations.get_logs("run_id", offset=resp.offset + resp.size)
+        ...     print(resp.data)
+    """
+    run_id = run_id.strip()
+    if not run_id:
+        raise ValueError("run_id is required")
+
+    params: dict[str, Any] = {"offset": offset, "limit": limit}
+    if all:
+        params["all"] = "true"
+
+    result = _api_request("GET", f"automation-runs/{run_id}/logs", params=params)
+    if isinstance(result, dict):
+        return LogsResponse(result)
+    raise RuntimeError("Unexpected response from logs endpoint")

{lumera-0.9.9.dist-info → lumera-0.10.0.dist-info}/METADATA

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

{lumera-0.9.9.dist-info → lumera-0.10.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 lumera/__init__.py,sha256=5FlY5dSJ1WNM4ko7wgmcajO8G2voBGn4S19E91_WdqE,2687
 lumera/_utils.py,sha256=b-l3Ebh4n2pC-9T5mR6h4hPf_Wl48VDlHES0pLo1zKE,25766
-lumera/automations.py,sha256=UKKtgmsYWIzJvMnIf0K5ywXCtgIu8kb9DmsAX_upubM,27397
+lumera/automations.py,sha256=KPP_rD7WKmBs865jiKoonZJjdTno-FSAU7hajPFyqs0,32851
 lumera/email.py,sha256=lk8KUsRw1ZvxgM0FPQXH-jVKUQA5f0zLv88jlc3IWlA,5056
 lumera/exceptions.py,sha256=bNsx4iYaroAAGsYxErfELC2B5ZJ3w5lVa1kKdIx5s9g,2173
 lumera/files.py,sha256=xMJmLTSaQQDttM3AMmpOWc6soh4lvCCKBreV0fXWHQw,3159
@@ -13,7 +13,7 @@ lumera/storage.py,sha256=fWkscTvKDzQ-5tsfA1lREO2qgtjJ4Yvxj3hvYNLKiW0,10527
 lumera/webhooks.py,sha256=L_Q5YHBJKQNpv7G9Nq0QqlGMRch6x9ptlwu1xD2qwUc,8661
 lumera/integrations/__init__.py,sha256=LnJmAnFB_p3YMKyeGVdDP4LYlJ85XFNQFAxGo6zF7CI,937
 lumera/integrations/google.py,sha256=QkbBbbDh3I_OToPDFqcivU6sWy2UieHBxZ_TPv5rqK0,11862
-lumera-0.9.9.dist-info/METADATA,sha256=NFyzKojtDJrZktYxFL-a5QM85V9XmWwJk3kawbSTR20,1611
-lumera-0.9.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-lumera-0.9.9.dist-info/top_level.txt,sha256=HgfK4XQkpMTnM2E5iWM4kB711FnYqUY9dglzib3pWlE,7
-lumera-0.9.9.dist-info/RECORD,,
+lumera-0.10.0.dist-info/METADATA,sha256=uWvSDuD868zVICFyVUppHMIrWe6A-JKeyxFRurjxieU,1612
+lumera-0.10.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+lumera-0.10.0.dist-info/top_level.txt,sha256=HgfK4XQkpMTnM2E5iWM4kB711FnYqUY9dglzib3pWlE,7
+lumera-0.10.0.dist-info/RECORD,,