discovery-engine-api 0.1.33__py3-none-any.whl → 0.1.47__py3-none-any.whl

discovery/__init__.py

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

discovery/client.py

@@ -32,9 +32,9 @@ from discovery.types import (
 class Engine:
     """Engine for the Discovery Engine API."""
 
-    # Production dashboard URL (can be overridden via DISCOVERY_API_URL env var for testing)
-    # The SDK calls the dashboard API which handles all report creation and credit deduction
-    _DEFAULT_BASE_URL = "https://disco.leap-labs.com"
+    # Production API URL (can be overridden via DISCOVERY_API_URL env var for testing)
+    # This points to the Modal-deployed FastAPI API
+    _DEFAULT_BASE_URL = "https://leap-labs-production--discovery-api.modal.run"
 
     def __init__(self, api_key: str):
         """
@@ -296,7 +296,7 @@ class Engine:
         dataset_id: str,
         target_column_id: str,
         task: str = "regression",
-        mode: str = "fast",
+        depth_iterations: int = 1,
         visibility: str = "public",
         timeseries_groups: Optional[List[Dict[str, Any]]] = None,
         target_column_override: Optional[str] = None,
@@ -311,7 +311,7 @@ class Engine:
             dataset_id: Dataset ID
             target_column_id: Target column ID
             task: Task type (regression, binary_classification, multiclass_classification)
-            mode: Analysis mode ("fast" or "deep")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             visibility: Dataset visibility ("public" or "private")
             timeseries_groups: Optional list of timeseries column groups
             target_column_override: Optional override for target column name
@@ -327,7 +327,7 @@ class Engine:
         payload = {
             "run_target_column_id": target_column_id,
             "task": task,
-            "mode": mode,
+            "depth_iterations": depth_iterations,
             "visibility": visibility,
             "auto_report_use_llm_evals": auto_report_use_llm_evals,
         }
@@ -458,7 +458,7 @@ class Engine:
         self,
         file: Union[str, Path, "pd.DataFrame"],
         target_column: str,
-        mode: str = "fast",
+        depth_iterations: int = 1,
         title: Optional[str] = None,
         description: Optional[str] = None,
         column_descriptions: Optional[Dict[str, str]] = None,
@@ -483,7 +483,7 @@ class Engine:
         Args:
             file: File path, Path object, or pandas DataFrame
             target_column: Name of the target column
-            mode: Analysis mode ("fast" or "deep", default: "fast")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             title: Optional dataset title
             description: Optional dataset description
             column_descriptions: Optional dict mapping column names to descriptions
@@ -535,7 +535,7 @@ class Engine:
         files = {"file": (filename, file_content, mime_type)}
         data: Dict[str, Any] = {
             "target_column": target_column,
-            "mode": mode,
+            "depth_iterations": str(depth_iterations),
             "visibility": visibility,
         }
 
@@ -553,7 +553,9 @@ class Engine:
             data["timeseries_groups"] = json.dumps(timeseries_groups)
 
         # Call dashboard API to create report
-        print(f"🚀 Uploading file and creating run (mode: {mode}, target: {target_column})...")
+        print(
+            f"🚀 Uploading file and creating run (depth: {depth_iterations}, target: {target_column})..."
+        )
         # httpx automatically handles multipart/form-data when both files and data are provided
         response = await client.post("/api/reports/create", files=files, data=data)
         response.raise_for_status()
@@ -607,7 +609,7 @@ class Engine:
         self,
         file: Union[str, Path, "pd.DataFrame"],
         target_column: str,
-        mode: str = "fast",
+        depth_iterations: int = 1,
         title: Optional[str] = None,
         description: Optional[str] = None,
         column_descriptions: Optional[Dict[str, str]] = None,
@@ -631,7 +633,7 @@ class Engine:
         Args:
             file: File path, Path object, or pandas DataFrame
             target_column: Name of the target column
-            mode: Analysis mode ("fast" or "deep", default: "fast")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             title: Optional dataset title
             description: Optional dataset description
             column_descriptions: Optional dict mapping column names to descriptions
@@ -653,7 +655,7 @@ class Engine:
         coro = self.run_async(
             file,
             target_column,
-            mode,
+            depth_iterations,
             title=title,
             description=description,
             column_descriptions=column_descriptions,

{discovery_engine_api-0.1.33.dist-info → discovery_engine_api-0.1.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.1.33
+Version: 0.1.47
 Summary: Python SDK for the Discovery Engine API
 Project-URL: Homepage, https://github.com/leap-laboratories/discovery
 Project-URL: Documentation, https://github.com/leap-laboratories/discovery

discovery_engine_api-0.1.47.dist-info/RECORD

@@ -0,0 +1,6 @@
+discovery/__init__.py,sha256=fkx9RJUVloq7PZZKXI1CDnw93YDMn5FfuIyHuzzrrzo,586
+discovery/client.py,sha256=sv1AoRzipSlgx1H6GRab4EfT8DlOm5xzKKNeTdkQiO8,30976
+discovery/types.py,sha256=4Z3gKdxWnOpymEjBGCzAeUGjwRT2A0aCpmuwctbE4w0,6008
+discovery_engine_api-0.1.47.dist-info/METADATA,sha256=jHGgVtlj1Fc9Fhl84yh2NMpMLrLR7_fGJ96oAoyK9OE,12675
+discovery_engine_api-0.1.47.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+discovery_engine_api-0.1.47.dist-info/RECORD,,

discovery_engine_api-0.1.33.dist-info/RECORD

@@ -1,6 +0,0 @@
-discovery/__init__.py,sha256=Yh_QSiN6TummOyq-mzcsBMDX1j5Ek5LOKra9MQf-KJo,586
-discovery/client.py,sha256=Io1vGqiEGrPPwOFxBDMaFbKlARj_Xg0izyWoRmoQy-c,30802
-discovery/types.py,sha256=4Z3gKdxWnOpymEjBGCzAeUGjwRT2A0aCpmuwctbE4w0,6008
-discovery_engine_api-0.1.33.dist-info/METADATA,sha256=UK34g4h6SZ3JZmnQvdiW8vbfEA_aaC1WJs3nPOWHzYI,12675
-discovery_engine_api-0.1.33.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-discovery_engine_api-0.1.33.dist-info/RECORD,,