discovery-engine-api 0.2.104__tar.gz → 0.2.106__tar.gz

{discovery_engine_api-0.2.104 → discovery_engine_api-0.2.106}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.2.104
+Version: 0.2.106
 Summary: Python SDK for Disco API
 Project-URL: Homepage, https://www.leap-labs.com
 Project-URL: Documentation, https://disco.leap-labs.com/llms-full.txt
@@ -37,7 +37,7 @@ Description-Content-Type: text/markdown
 
 # Disco Python SDK
 
-Find novel, statistically validated patterns in tabular data — feature interactions, subgroup effects, and conditional relationships that correlation analysis and LLMs miss.
+Find novel, statistically validated patterns in tabular data — feature interactions, subgroup effects, and conditional relationships that humans and agents miss.
 
 ## Installation
 
@@ -124,7 +124,7 @@ result = await engine.discover(
 
 ### Running in the Background
 
-Runs take 3–15 minutes. While waiting, the SDK logs progress automatically:
+Runs take a few minutes. While waiting, the SDK logs progress automatically:
 
 ```
 Waiting for run abc123 to complete...

{discovery_engine_api-0.2.104 → discovery_engine_api-0.2.106}/README.md

@@ -1,6 +1,6 @@
 # Disco Python SDK
 
-Find novel, statistically validated patterns in tabular data — feature interactions, subgroup effects, and conditional relationships that correlation analysis and LLMs miss.
+Find novel, statistically validated patterns in tabular data — feature interactions, subgroup effects, and conditional relationships that humans and agents miss.
 
 ## Installation
 
@@ -87,7 +87,7 @@ result = await engine.discover(
 
 ### Running in the Background
 
-Runs take 3–15 minutes. While waiting, the SDK logs progress automatically:
+Runs take a few minutes. While waiting, the SDK logs progress automatically:
 
 ```
 Waiting for run abc123 to complete...

{discovery_engine_api-0.2.104 → discovery_engine_api-0.2.106}/discovery/__init__.py

@@ -1,6 +1,6 @@
 """Disco Python SDK."""
 
-__version__ = "0.2.104"
+__version__ = "0.2.106"
 
 from discovery.client import Engine
 from discovery.types import (

{discovery_engine_api-0.2.104 → discovery_engine_api-0.2.106}/discovery/client.py

@@ -239,6 +239,15 @@ class Engine:
             cls._raise_for_status(response)
             data = response.json()
 
+        # Server may return a fresh key directly (short-circuit when OTP isn't
+        # required) — mirror Engine.signup() and use it if present. Only fall
+        # back to the OTP prompt when the server explicitly asks for verification.
+        if data.get("key"):
+            engine = cls(api_key=data["key"], quiet=quiet)
+            if not quiet:
+                print(f"Logged in. Tier: {data.get('tier', 'free_tier')}")
+            return engine
+
         if data.get("status") != "verification_required":
             raise ValueError(f"Unexpected login response: {data}")
 
@@ -437,25 +446,24 @@ class Engine:
         file_size_mb: float,
         num_columns: int,
         analysis_depth: int = 2,
-        visibility: str = "public",
-        use_llms: bool = False,
     ) -> Dict[str, Any]:
-        """Estimate the credit cost for an analysis run.
+        """Estimate the credits required for an analysis run.
 
-        Works with or without authentication. If authenticated, the response
-        includes your current credit balance and whether you have enough.
+        Returns ``required_credits`` for public (always 0) and private, with
+        private split by whether LLMs are enabled (``without_llms`` /
+        ``with_llms``). Works with or without authentication; when
+        authenticated, the response also includes ``account.available_credits``.
 
         Args:
             file_size_mb: Size of the data file in megabytes.
             num_columns: Number of columns in the dataset.
-            analysis_depth: Depth iterations (1=fast, higher=deeper).
-            visibility: "public" (free, results published) or "private" (costs credits).
-            use_llms: Slower and more expensive, but you get smarter pre-processing,
-                literature context and novelty assessment. Default False. Public runs
-                always use LLMs.
+            analysis_depth: Depth iterations (1=fast, higher=deeper). Used to
+                compute the private-run cost.
 
         Returns:
-            Dict with ``cost``, ``limits``, and ``account`` info.
+            Dict with ``required_credits`` (public / private.without_llms /
+            private.with_llms), ``limits`` (including per-visibility
+            ``max_depth``), ``note``, and ``account`` info when authenticated.
         """
         client = await self._get_dashboard_client()
         response = await client.post(
@@ -464,8 +472,6 @@ class Engine:
                 "file_size_mb": file_size_mb,
                 "num_columns": num_columns,
                 "analysis_depth": analysis_depth,
-                "visibility": visibility,
-                "use_llms": use_llms,
             },
         )
         self._raise_for_status(response)
@@ -740,6 +746,38 @@ class Engine:
         )
         return finalize_data
 
+    @staticmethod
+    def _looks_like_url(file: Any) -> bool:
+        """True iff ``file`` is a string that names an http(s) URL.
+
+        Lets the SDK match the MCP ``file_url`` contract — agents can pass URLs
+        directly to ``upload_file()`` / ``discover()`` instead of needing to
+        download the file first.
+        """
+        return isinstance(file, str) and (file.startswith("http://") or file.startswith("https://"))
+
+    async def _upload_from_url(self, url: str) -> Dict[str, Any]:
+        """Hand a remote URL to the server's from-url endpoint and return the
+        same finalize-shaped dict that ``_upload_file_direct`` produces.
+
+        The server downloads the file directly — nothing passes through the
+        SDK process or the model context.
+        """
+        client = await self._get_dashboard_client()
+        response = await client.post(
+            "/api/data/upload/from-url",
+            json={"url": url},
+            timeout=httpx.Timeout(connect=30.0, read=300.0, write=300.0, pool=30.0),
+        )
+        if response.status_code >= 400:
+            try:
+                error_data = response.json()
+                error_message = error_data.get("error", response.text)
+            except Exception:
+                error_message = response.text
+            raise ValueError(f"URL upload error ({response.status_code}): {error_message}")
+        return response.json()
+
     def _prepare_upload(
         self,
         file: Union[str, Path, "pd.DataFrame"],
@@ -806,10 +844,27 @@ class Engine:
         creating a run (e.g. to choose a target column).  Pass the result
         to ``run_async(upload_result=...)`` to avoid re-uploading.
 
+        ``file`` may be a local path, a pandas DataFrame, or an http(s) URL.
+        URLs are downloaded server-side — nothing passes through the SDK or
+        the model context.
+
         Returns:
             Dict with ``file`` (key, name, size, fileHash) and ``columns``
             (list of dicts with ``name``, ``type``, ``enabled``).
         """
+        if self._looks_like_url(file):
+            self._log(f"Uploading from URL: {file}")
+            result = await self._upload_from_url(file)  # type: ignore[arg-type]
+            if not result.get("ok"):
+                errors = result.get("issues", {}).get("errors", [])
+                error_msg = errors[0].get("message") if errors else "Upload failed"
+                raise ValueError(f"Upload failed: {error_msg}")
+            return {
+                "file": result["file"],
+                "columns": result.get("columns", []),
+                "rowCount": result.get("rowCount"),
+            }
+
         file_source, filename, mime_type, _ = self._prepare_upload(
             file=file,
             title=title,
@@ -892,6 +947,17 @@ class Engine:
             self._log(
                 f"Creating run from pre-uploaded file (depth: {analysis_depth}, target: {target_column})..."
             )
+        elif self._looks_like_url(file):
+            self._log(f"Uploading from URL: {file}")
+            raw_result = await self._upload_from_url(file)  # type: ignore[arg-type]
+
+            if not raw_result.get("ok"):
+                errors = raw_result.get("issues", {}).get("errors", [])
+                error_msg = errors[0].get("message") if errors else "Upload failed"
+                raise ValueError(f"Upload failed: {error_msg}")
+
+            uploaded_file = raw_result["file"]
+            columns = raw_result.get("columns", [])
         else:
             file_source, filename, mime_type, _ = self._prepare_upload(
                 file=file,
@@ -980,7 +1046,12 @@ class Engine:
 
             if wait:
                 return await self.get_results(run_id)
-            return EngineResult(run_id=run_id, status="completed", report_id=report_id)
+            return EngineResult(
+                run_id=run_id,
+                status="completed",
+                report_id=report_id,
+                report_url=f"{self.dashboard_url}/reports/{report_id}",
+            )
 
         run_id = result_data["run_id"]
         self._log(f"Run created: {run_id}")

{discovery_engine_api-0.2.104 → discovery_engine_api-0.2.106}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "discovery-engine-api"
-version = "0.2.104"
+version = "0.2.106"
 description = "Python SDK for Disco API"
 readme = "README.md"
 requires-python = ">=3.10"