discovery-engine-api 0.1.50__tar.gz → 0.1.57__tar.gz

{discovery_engine_api-0.1.50 → discovery_engine_api-0.1.57}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.1.50
+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.50 → discovery_engine_api-0.1.57}/discovery/__init__.py

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

{discovery_engine_api-0.1.50 → discovery_engine_api-0.1.57}/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.50 → discovery_engine_api-0.1.57}/pyproject.toml

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

{discovery_engine_api-0.1.50 → discovery_engine_api-0.1.57}/tests/test_client.py

@@ -521,7 +521,8 @@ class TestGetResults:
         mock_httpx_client.get = AsyncMock(return_value=mock_response)
         mock_httpx_client.headers = {}
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        # get_results uses dashboard client for /api/* endpoints
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             result = await client.get_results(run_id)
 
         assert isinstance(result, EngineResult)
@@ -541,7 +542,8 @@ class TestGetResults:
         mock_httpx_client.get = AsyncMock(return_value=mock_response)
         mock_httpx_client.headers = {}
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        # get_results uses dashboard client for /api/* endpoints
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             result = await client.get_results("run-123")
 
         # Check pattern parsing
@@ -668,7 +670,7 @@ class TestRunAsync:
         # Create a DataFrame
         df = pd.DataFrame({"age": [30, 40], "price": [100, 150]})
 
-        # Mock the single /api/reports/create endpoint response
+        # Mock the single /api/reports/create endpoint response (uses dashboard client)
         mock_response_create = MagicMock()
         mock_response_create.json.return_value = {"run_id": sample_run["id"]}
         mock_response_create.raise_for_status = MagicMock()
@@ -676,7 +678,7 @@ class TestRunAsync:
         mock_httpx_client.post = AsyncMock(return_value=mock_response_create)
         mock_httpx_client.headers = {}
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             result = await client.run_async(
                 file=df,
                 target_column="price",
@@ -707,7 +709,7 @@ class TestRunAsync:
         # Create a DataFrame
         df = pd.DataFrame({"age": [30, 40], "price": [100, 150]})
 
-        # Mock the single /api/reports/create endpoint response
+        # Mock the single /api/reports/create endpoint response (uses dashboard client)
         mock_response_create = MagicMock()
         mock_response_create.json.return_value = {"run_id": sample_run["id"]}
         mock_response_create.raise_for_status = MagicMock()
@@ -717,7 +719,7 @@ class TestRunAsync:
 
         completed_result = EngineResult(**sample_results)
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             with patch.object(client, "wait_for_completion", return_value=completed_result):
                 result = await client.run_async(
                     file=df,
@@ -754,7 +756,7 @@ class TestRunAsync:
         mock_httpx_client.post = AsyncMock(side_effect=error)
         mock_httpx_client.headers = {}
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             with pytest.raises(httpx.HTTPStatusError):
                 await client.run_async(
                     file=df,
@@ -783,7 +785,7 @@ class TestRun:
         # Create a DataFrame
         df = pd.DataFrame({"age": [30, 40], "price": [100, 150]})
 
-        # Mock the single /api/reports/create endpoint response
+        # Mock the single /api/reports/create endpoint response (uses dashboard client)
         mock_response_create = MagicMock()
         mock_response_create.json.return_value = {"run_id": sample_run["id"]}
         mock_response_create.raise_for_status = MagicMock()
@@ -791,7 +793,7 @@ class TestRun:
         mock_httpx_client.post = AsyncMock(return_value=mock_response_create)
         mock_httpx_client.headers = {}
 
-        with patch.object(client, "_get_client_with_org", return_value=mock_httpx_client):
+        with patch.object(client, "_get_dashboard_client", return_value=mock_httpx_client):
             result = client.run(
                 file=df,
                 target_column="price",
@@ -824,12 +826,18 @@ class TestContextManager:
     @pytest.mark.asyncio
     async def test_close_method(self, client, mock_httpx_client):
         """Test explicit close method."""
+        mock_dashboard_client = AsyncMock(spec=httpx.AsyncClient)
+        mock_dashboard_client.aclose = AsyncMock()
+
         client._client = mock_httpx_client
+        client._dashboard_client = mock_dashboard_client
 
         await client.close()
 
         assert client._client is None
+        assert client._dashboard_client is None
         mock_httpx_client.aclose.assert_called_once()
+        mock_dashboard_client.aclose.assert_called_once()
 
 
 class TestGetClientWithOrg: