discovery-engine-api 0.2.78__tar.gz → 0.2.80__tar.gz

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.2.78
+Version: 0.2.80
 Summary: Python SDK for the Discovery Engine API
 Project-URL: Homepage, https://www.leap-labs.com
 Project-URL: Documentation, https://disco.leap-labs.com/llms-full.txt

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/discovery/__init__.py

@@ -1,6 +1,6 @@
 """Discovery Engine Python SDK."""
 
-__version__ = "0.2.78"
+__version__ = "0.2.80"
 
 from discovery.client import Engine
 from discovery.types import (

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/discovery/client.py

@@ -395,7 +395,7 @@ class Engine:
             num_columns: Number of columns in the dataset.
             num_rows: Number of rows (improves time estimate accuracy).
             depth_iterations: Depth iterations (1=fast, higher=deeper).
-            visibility: "public" (free, depth=1) or "private" (costs credits).
+            visibility: "public" (free, results published) or "private" (costs credits).
 
         Returns:
             Dict with ``cost``, ``time_estimate``, ``limits``, and ``account`` info.
@@ -445,6 +445,15 @@ class Engine:
             status=data["status"],
             job_id=data.get("job_id"),
             job_status=data.get("job_status"),
+            queue_position=data.get("queue_position"),
+            current_step=data.get("current_step", {}).get("status")
+            if data.get("current_step")
+            else None,
+            current_step_message=data.get("current_step", {}).get("message")
+            if data.get("current_step")
+            else None,
+            estimated_seconds=data.get("estimated_seconds"),
+            estimated_wait_seconds=data.get("estimated_wait_seconds"),
             error_message=data.get("error_message"),
         )
 
@@ -463,6 +472,7 @@ class Engine:
         """
         start_time = time.time()
         last_status = None
+        last_step = None
         poll_count = 0
         consecutive_errors = 0
         max_consecutive_errors = 3
@@ -500,18 +510,53 @@ class Engine:
             elapsed = time.time() - start_time
             poll_count += 1
 
-            if result.status != last_status or poll_count % 3 == 0:
-                status_msg = f"Status: {result.status}"
-                if result.job_status:
-                    status_msg += f" (job: {result.job_status})"
-                if elapsed > 0:
-                    status_msg += f" | Elapsed: {elapsed:.1f}s"
+            if result.status != "completed" and (
+                result.status != last_status
+                or result.current_step != last_step
+                or poll_count % 3 == 0
+            ):
+                if result.job_status == "pending":
+                    pos = result.queue_position
+                    pos_str = f" (position {pos} in queue)" if pos is not None else ""
+                    wait_str = ""
+                    if result.estimated_wait_seconds is not None:
+                        mins = max(1, round(result.estimated_wait_seconds / 60))
+                        wait_str = f" | Est. wait: ~{mins} min"
+                    status_msg = f"Status: waiting{pos_str}{wait_str} | Upgrade at disco.leap-labs.com/account for priority processing"
+                else:
+                    step_str = ""
+                    if result.current_step and result.current_step not in (
+                        "pending",
+                        "completed",
+                        "failed",
+                    ):
+                        msg = (
+                            f" — {result.current_step_message}"
+                            if result.current_step_message
+                            else ""
+                        )
+                        step_str = f" ({result.current_step}{msg})"
+                    eta_str = ""
+                    if result.estimated_seconds is not None and elapsed > 0:
+                        remaining = max(0, result.estimated_seconds - elapsed)
+                        if remaining > 0:
+                            eta_str = f" | ETA: ~{max(1, round(remaining / 60))} min"
+                    status_msg = (
+                        f"Status: {result.status}{step_str} | Elapsed: {elapsed:.1f}s{eta_str}"
+                    )
                 self._log(f"  {status_msg}")
 
             last_status = result.status
+            last_step = result.current_step
 
             if result.status == "completed":
-                self._log(f"Run completed in {elapsed:.1f}s")
+                n = len(result.patterns)
+                pattern_str = f"{n} pattern{'s' if n != 1 else ''} found"
+                self._log(f"Run completed in {elapsed:.1f}s | {pattern_str}")
+                for hint in result.hints:
+                    self._log(hint)
+                if result.report_url:
+                    self._log(f"Report: {result.report_url}")
                 return result
             elif result.status == "failed":
                 error_msg = result.error_message or "Unknown error"
@@ -731,11 +776,11 @@ class Engine:
         Returns:
             EngineResult with run_id and (if wait=True) complete results.
         """
-        # Warn if depth > 1 with public visibility (server will cap to 1)
+        # Public runs are always depth=1 (server enforces this)
         if depth_iterations > 1 and visibility == "public":
             self._log(
-                "Public runs are limited to depth 1. "
-                "Set visibility='private' for deeper analysis (requires credits)."
+                "Public runs use depth=1. Upgrade your plan at disco.leap-labs.com/subscribe "
+                "to see more patterns and get priority processing."
             )
             depth_iterations = 1
 
@@ -751,9 +796,7 @@ class Engine:
                 title=title,
                 log=True,
             )
-            self._log(
-                f"Uploading file and creating run (depth: {depth_iterations}, target: {target_column})..."
-            )
+            self._log("Uploading file and creating run...")
 
             # Step 1: Upload file
             self._log("  Uploading to storage...")
@@ -1004,9 +1047,6 @@ class Engine:
         hidden_deep_count = data.get("hidden_deep_count", 0)
         hidden_deep_novel_count = data.get("hidden_deep_novel_count", 0)
 
-        for hint in hints:
-            self._log(hint)
-
         return EngineResult(
             run_id=data["run_id"],
             report_id=data.get("report_id"),
@@ -1023,6 +1063,15 @@ class Engine:
             feature_importance=feature_importance,
             job_id=data.get("job_id"),
             job_status=data.get("job_status"),
+            queue_position=data.get("queue_position"),
+            current_step=data.get("current_step", {}).get("status")
+            if data.get("current_step")
+            else None,
+            current_step_message=data.get("current_step", {}).get("message")
+            if data.get("current_step")
+            else None,
+            estimated_seconds=data.get("estimated_seconds"),
+            estimated_wait_seconds=data.get("estimated_wait_seconds"),
             error_message=data.get("error_message"),
             report_url=report_url,
             hints=hints,

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/discovery/errors.py

@@ -39,7 +39,7 @@ class InsufficientCreditsError(DiscoveryError):
         super().__init__(
             message,
             suggestion=(
-                "Run with visibility='public' (free, depth=1) or purchase credits "
+                "Run with visibility='public' (free, results published) or purchase credits "
                 "with engine.purchase_credits()."
             ),
         )

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/discovery/types.py

@@ -202,6 +202,11 @@ class EngineResult:
     # Job tracking
     job_id: Optional[str] = None
     job_status: Optional[str] = None
+    queue_position: Optional[int] = None
+    current_step: Optional[str] = None
+    current_step_message: Optional[str] = None
+    estimated_seconds: Optional[int] = None
+    estimated_wait_seconds: Optional[int] = None
     error_message: Optional[str] = None
 
     # Shareable link to the interactive web report
@@ -221,4 +226,9 @@ class RunStatus:
     status: str
     job_id: Optional[str] = None
     job_status: Optional[str] = None
+    queue_position: Optional[int] = None
+    current_step: Optional[str] = None
+    current_step_message: Optional[str] = None
+    estimated_seconds: Optional[int] = None
+    estimated_wait_seconds: Optional[int] = None
     error_message: Optional[str] = None

{discovery_engine_api-0.2.78 → discovery_engine_api-0.2.80}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "discovery-engine-api"
-version = "0.2.78"
+version = "0.2.80"
 description = "Python SDK for the Discovery Engine API"
 readme = "README.md"
 requires-python = ">=3.10"