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

discovery/__init__.py

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

discovery/client.py

@@ -36,6 +36,9 @@ class Engine:
     # This points to the Modal-deployed FastAPI API
     _DEFAULT_BASE_URL = "https://leap-labs-production--discovery-api.modal.run"
 
+    # Dashboard URL for web UI and /api/* endpoints
+    _DEFAULT_DASHBOARD_URL = "https://disco.leap-labs.com"
+
     def __init__(self, api_key: str):
         """
         Initialize the Discovery Engine.
@@ -49,8 +52,13 @@ class Engine:
         # Use DISCOVERY_API_URL env var if set (for testing/custom deployments),
         # otherwise use the production default
         self.base_url = os.getenv("DISCOVERY_API_URL", self._DEFAULT_BASE_URL).rstrip("/")
+        # Dashboard URL for /api/* endpoints and web UI links
+        self.dashboard_url = os.getenv(
+            "DISCOVERY_DASHBOARD_URL", self._DEFAULT_DASHBOARD_URL
+        ).rstrip("/")
         self._organization_id: Optional[str] = None
         self._client: Optional[httpx.AsyncClient] = None
+        self._dashboard_client: Optional[httpx.AsyncClient] = None
         self._org_fetched = False
 
     async def _ensure_organization_id(self) -> str:
@@ -119,11 +127,25 @@ class Engine:
 
         return client
 
+    async def _get_dashboard_client(self) -> httpx.AsyncClient:
+        """Get or create the HTTP client for dashboard API calls."""
+        if self._dashboard_client is None:
+            headers = {"Authorization": f"Bearer {self.api_key}"}
+            self._dashboard_client = httpx.AsyncClient(
+                base_url=self.dashboard_url,
+                headers=headers,
+                timeout=60.0,
+            )
+        return self._dashboard_client
+
     async def close(self):
-        """Close the HTTP client."""
+        """Close the HTTP clients."""
         if self._client:
             await self._client.aclose()
             self._client = None
+        if self._dashboard_client:
+            await self._dashboard_client.aclose()
+            self._dashboard_client = None
 
     async def __aenter__(self):
         """Async context manager entry."""
@@ -365,10 +387,11 @@ class Engine:
         Returns:
             EngineResult with complete analysis data
         """
-        client = await self._get_client_with_org()
+        # Use dashboard client for /api/* endpoints (hosted on Next.js dashboard, not Modal API)
+        dashboard_client = await self._get_dashboard_client()
 
         # Call dashboard API for results
-        response = await client.get(f"/api/runs/{run_id}/results")
+        response = await dashboard_client.get(f"/api/runs/{run_id}/results")
         response.raise_for_status()
 
         data = response.json()
@@ -501,8 +524,6 @@ class Engine:
         Returns:
             EngineResult with run_id and (if wait=True) complete results
         """
-        client = await self._get_client_with_org()
-
         # Prepare file for upload
         if pd is not None and isinstance(file, pd.DataFrame):
             # Convert DataFrame to CSV in memory
@@ -556,8 +577,10 @@ class Engine:
         print(
             f"🚀 Uploading file and creating run (depth: {depth_iterations}, target: {target_column})..."
         )
+        # Use dashboard client for /api/* endpoints (hosted on Next.js dashboard, not Modal API)
+        dashboard_client = await self._get_dashboard_client()
         # 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 = await dashboard_client.post("/api/reports/create", files=files, data=data)
         response.raise_for_status()
 
         result_data = response.json()
@@ -574,8 +597,8 @@ class Engine:
             print(f"ℹ️  Duplicate report found (run_id: {run_id})")
 
             # Construct dashboard URL for the processing page
-            dashboard_url = f"{self.base_url}/reports/new/{run_id}/processing"
-            print(f"🔗 View progress: {dashboard_url}")
+            progress_url = f"{self.dashboard_url}/reports/new/{run_id}/processing"
+            print(f"🔗 View progress: {progress_url}")
 
             # If wait is True, fetch the full results for the existing report
             if wait:
@@ -592,8 +615,8 @@ class Engine:
         print(f"✓ Run created: {run_id}")
 
         # Construct dashboard URL for the processing page
-        dashboard_url = f"{self.base_url}/reports/new/{run_id}/processing"
-        print(f"🔗 View progress: {dashboard_url}")
+        progress_url = f"{self.dashboard_url}/reports/new/{run_id}/processing"
+        print(f"🔗 View progress: {progress_url}")
 
         if wait:
             # Wait for completion and return full results

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.1.47
+Version: 0.1.57
 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.57.dist-info/RECORD

@@ -0,0 +1,6 @@
+discovery/__init__.py,sha256=SaMyN0LNagF7B8YbpxeoiimRgWaRRgWseBlhvY2jkCE,586
+discovery/client.py,sha256=4qRYvnf7oU0HUC5M15S8fKbQjyw4C_qQ7g4ExJEeRwo,32174
+discovery/types.py,sha256=4Z3gKdxWnOpymEjBGCzAeUGjwRT2A0aCpmuwctbE4w0,6008
+discovery_engine_api-0.1.57.dist-info/METADATA,sha256=maCzXl4-4CVOupFobK6Ubx3XzdnH12EUXbv7eg7kLlw,12675
+discovery_engine_api-0.1.57.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+discovery_engine_api-0.1.57.dist-info/RECORD,,

discovery_engine_api-0.1.47.dist-info/RECORD

@@ -1,6 +0,0 @@
-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,,