databar 2.0.0__tar.gz → 2.0.2__tar.gz

{databar-2.0.0/src/databar.egg-info → databar-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databar
-Version: 2.0.0
+Version: 2.0.2
 Summary: Official Databar.ai Python SDK and CLI — connect to enrichments, waterfalls, and tables via api.databar.ai
 Author-email: "Databar.ai Team" <info@databar.ai>
 License: MIT License
@@ -139,7 +139,7 @@ for param in enrichment.params:
 
 # Run single enrichment (async — returns task)
 task = client.run_enrichment(123, {"email": "alice@example.com"})
-data = client.poll_task(task.request_id)
+data = client.poll_task(task.task_id)
 
 # Run single enrichment (sync convenience wrapper)
 data = client.run_enrichment_sync(123, {"email": "alice@example.com"})

{databar-2.0.0 → databar-2.0.2}/README.md

@@ -84,7 +84,7 @@ for param in enrichment.params:
 
 # Run single enrichment (async — returns task)
 task = client.run_enrichment(123, {"email": "alice@example.com"})
-data = client.poll_task(task.request_id)
+data = client.poll_task(task.task_id)
 
 # Run single enrichment (sync convenience wrapper)
 data = client.run_enrichment_sync(123, {"email": "alice@example.com"})

{databar-2.0.0 → databar-2.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "databar"
-version = "2.0.0"
+version = "2.0.2"
 description = "Official Databar.ai Python SDK and CLI — connect to enrichments, waterfalls, and tables via api.databar.ai"
 readme = "README.md"
 license = { file = "LICENSE" }

{databar-2.0.0 → databar-2.0.2}/src/databar/__init__.py

@@ -45,6 +45,7 @@ from .models import (
     EnrichmentSummary,
     InsertOptions,
     InsertRow,
+    RunResponse,
     Table,
     TableEnrichment,
     TaskResponse,
@@ -56,7 +57,7 @@ from .models import (
     WaterfallEnrichment,
 )
 
-__version__ = "2.0.0"
+__version__ = "2.0.2"
 __all__ = [
     "DatabarClient",
     # exceptions
@@ -78,6 +79,7 @@ __all__ = [
     "ChoiceItem",
     "Choices",
     "ChoicesResponse",
+    "RunResponse",
     "TaskResponse",
     "TaskStatus",
     "Waterfall",

{databar-2.0.0 → databar-2.0.2}/src/databar/cli/_auth.py

@@ -10,6 +10,7 @@ Key resolution order (same as MCP server):
 from __future__ import annotations
 
 import os
+import shutil
 from pathlib import Path
 
 import typer
@@ -87,10 +88,23 @@ def login(
     success(f"API key saved to {CONFIG_FILE}")
     console.print("[dim]Tip: You can also set DATABAR_API_KEY as an environment variable.[/dim]")
 
+    # Check if the databar binary is on PATH and print a hint if not
+    if shutil.which("databar") is None:
+        import sys
+        bin_dir = Path(sys.executable).parent
+        console.print(
+            f"\n[yellow]Note:[/yellow] The [bold]databar[/bold] command is not on your PATH.\n"
+            f"Add this to your shell profile ([dim]~/.zshrc[/dim] or [dim]~/.bashrc[/dim]):\n\n"
+            f"  [bold]export PATH=\"{bin_dir}:$PATH\"[/bold]\n\n"
+            f"Then restart your terminal, or run:\n\n"
+            f"  [bold]source ~/.zshrc[/bold]\n\n"
+            f"Until then, use the full path: [bold]{bin_dir}/databar[/bold]"
+        )
+
 
 @app.command("whoami")
 def whoami(
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f", help="Output format.")
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f", help="Output format.")
 ) -> None:
     """Show current user info and credit balance."""
     client = get_client()

{databar-2.0.0 → databar-2.0.2}/src/databar/cli/enrichments.py

@@ -28,7 +28,7 @@ app = typer.Typer(help="Search and run data enrichments.")
 @app.command("list")
 def list_enrichments(
     query: Optional[str] = typer.Option(None, "--query", "-q", help="Search query."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List available enrichments."""
     client = get_client()
@@ -59,7 +59,7 @@ def list_enrichments(
 @app.command("get")
 def get_enrichment(
     enrichment_id: int = typer.Argument(..., help="Enrichment ID."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Get full details for an enrichment including parameters."""
     client = get_client()
@@ -99,11 +99,20 @@ def get_enrichment(
         output(rows, OutputFormat.TABLE, table_columns=["name", "type"])
 
 
+@app.command("info", hidden=True)
+def info_enrichment(
+    enrichment_id: int = typer.Argument(..., help="Enrichment ID."),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
+) -> None:
+    """Alias for 'get'. Get full details for an enrichment including parameters."""
+    get_enrichment(enrichment_id, fmt)
+
+
 @app.command("run")
 def run_enrichment(
     enrichment_id: int = typer.Argument(..., help="Enrichment ID."),
     params_json: str = typer.Option(..., "--params", "-p", help='JSON params, e.g. \'{"email":"alice@example.com"}\''),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     raw: bool = typer.Option(False, "--raw", help="Print raw result without formatting."),
 ) -> None:
     """Run a single enrichment and wait for results."""
@@ -132,7 +141,7 @@ def run_enrichment(
 def bulk_enrichment(
     enrichment_id: int = typer.Argument(..., help="Enrichment ID."),
     input_file: Path = typer.Option(..., "--input", "-i", help="CSV file with one row per input.", exists=True),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     out: Optional[Path] = typer.Option(None, "--out", "-o", help="Output file (for CSV format)."),
 ) -> None:
     """Run a bulk enrichment from a CSV input file."""
@@ -159,7 +168,7 @@ def param_choices(
     query: Optional[str] = typer.Option(None, "--query", "-q", help="Filter choices."),
     page: int = typer.Option(1, "--page", help="Page number."),
     limit: int = typer.Option(50, "--limit", help="Results per page."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List available choices for a select/mselect enrichment parameter."""
     client = get_client()

{databar-2.0.0 → databar-2.0.2}/src/databar/cli/tables.py

@@ -33,7 +33,7 @@ app = typer.Typer(help="Manage tables and rows.")
 
 @app.command("list")
 def list_tables(
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List all tables in your workspace."""
     client = get_client()
@@ -59,7 +59,7 @@ def list_tables(
 def create_table(
     name: Optional[str] = typer.Option(None, "--name", "-n", help="Table name."),
     columns: Optional[str] = typer.Option(None, "--columns", "-c", help="Comma-separated column names."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Create a new empty table."""
     col_list = [c.strip() for c in columns.split(",")] if columns else None
@@ -79,7 +79,7 @@ def create_table(
 @app.command("columns")
 def get_columns(
     table_uuid: str = typer.Argument(..., help="Table UUID."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List columns defined on a table."""
     client = get_client()
@@ -106,7 +106,7 @@ def get_rows(
     table_uuid: str = typer.Argument(..., help="Table UUID."),
     page: int = typer.Option(1, "--page", help="Page number."),
     per_page: int = typer.Option(1000, "--per-page", help="Rows per page (max 1000)."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     out: Optional[Path] = typer.Option(None, "--out", "-o", help="Output file (for CSV format)."),
 ) -> None:
     """Get rows from a table."""
@@ -143,7 +143,7 @@ def insert_rows(
     input_file: Optional[Path] = typer.Option(None, "--input", "-i", help="CSV file.", exists=True),
     allow_new_columns: bool = typer.Option(False, "--allow-new-columns", help="Auto-create unknown columns."),
     dedupe_keys: Optional[str] = typer.Option(None, "--dedupe-keys", help="Comma-separated column names for deduplication."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Insert rows into a table."""
     raw_rows = _load_rows(data_json, input_file)
@@ -183,7 +183,7 @@ def patch_rows(
     data_json: Optional[str] = typer.Option(None, "--data", "-d", help='JSON array: [{"id":"<uuid>","fields":{...}}]'),
     input_file: Optional[Path] = typer.Option(None, "--input", "-i", help="CSV file (must have an 'id' column).", exists=True),
     no_overwrite: bool = typer.Option(False, "--no-overwrite", help="Only fill empty cells; keep existing values."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Update existing rows by row UUID."""
     raw_rows = _load_rows(data_json, input_file)
@@ -219,7 +219,7 @@ def upsert_rows(
     key_col: str = typer.Option(..., "--key-col", "-k", help="Column name to match on (e.g. 'email')."),
     data_json: Optional[str] = typer.Option(None, "--data", "-d", help="JSON array of row objects."),
     input_file: Optional[Path] = typer.Option(None, "--input", "-i", help="CSV file.", exists=True),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Insert or update rows matched by a key column."""
     raw_rows = _load_rows(data_json, input_file)
@@ -253,7 +253,7 @@ def upsert_rows(
 @app.command("enrichments")
 def table_enrichments(
     table_uuid: str = typer.Argument(..., help="Table UUID."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List enrichments configured on a table."""
     client = get_client()
@@ -277,7 +277,7 @@ def add_enrichment(
     table_uuid: str = typer.Argument(..., help="Table UUID."),
     enrichment_id: int = typer.Option(..., "--enrichment-id", "-e", help="Enrichment ID to add."),
     mapping_json: str = typer.Option(..., "--mapping", "-m", help='JSON mapping of param → column, e.g. \'{"email": "email_col"}\''),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Add an enrichment to a table with a column mapping."""
     try:

{databar-2.0.0 → databar-2.0.2}/src/databar/cli/tasks.py

@@ -19,7 +19,7 @@ app = typer.Typer(help="Check the status of async tasks.")
 @app.command("get")
 def get_task(
     task_id: str = typer.Argument(..., help="Task ID returned by a run or bulk-run call."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     poll: bool = typer.Option(
         False,
         "--poll",

{databar-2.0.0 → databar-2.0.2}/src/databar/cli/waterfalls.py

@@ -27,7 +27,7 @@ app = typer.Typer(help="Search and run waterfall enrichments.")
 @app.command("list")
 def list_waterfalls(
     query: Optional[str] = typer.Option(None, "--query", "-q", help="Filter by name/description."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """List available waterfall enrichments."""
     client = get_client()
@@ -64,7 +64,7 @@ def list_waterfalls(
 @app.command("get")
 def get_waterfall(
     identifier: str = typer.Argument(..., help="Waterfall identifier."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
 ) -> None:
     """Get details for a specific waterfall."""
     client = get_client()
@@ -103,13 +103,22 @@ def get_waterfall(
         output(rows, OutputFormat.TABLE, table_columns=["id", "name", "price"])
 
 
+@app.command("info", hidden=True)
+def info_waterfall(
+    identifier: str = typer.Argument(..., help="Waterfall identifier."),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
+) -> None:
+    """Alias for 'get'. Get details for a specific waterfall."""
+    get_waterfall(identifier, fmt)
+
+
 @app.command("run")
 def run_waterfall(
     identifier: str = typer.Argument(..., help="Waterfall identifier."),
     params_json: str = typer.Option(..., "--params", "-p", help='JSON params, e.g. \'{"linkedin_url":"https://..."}\''),
     providers: Optional[str] = typer.Option(None, "--providers", help="Comma-separated provider IDs (default: all)."),
     email_verifier: Optional[int] = typer.Option(None, "--email-verifier", help="Email verifier enrichment ID."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     raw: bool = typer.Option(False, "--raw", help="Print raw result without formatting."),
 ) -> None:
     """Run a waterfall enrichment and wait for results."""
@@ -151,7 +160,7 @@ def bulk_waterfall(
     input_file: Path = typer.Option(..., "--input", "-i", help="CSV file with one row per input.", exists=True),
     providers: Optional[str] = typer.Option(None, "--providers", help="Comma-separated provider IDs (default: all)."),
     email_verifier: Optional[int] = typer.Option(None, "--email-verifier", help="Email verifier enrichment ID."),
-    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "-f"),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
     out: Optional[Path] = typer.Option(None, "--out", "-o", help="Output file (for CSV format)."),
 ) -> None:
     """Run a waterfall enrichment in bulk from a CSV input file."""

{databar-2.0.0 → databar-2.0.2}/src/databar/client.py

@@ -42,6 +42,7 @@ from .models import (
     EnrichmentSummary,
     InsertOptions,
     InsertRow,
+    RunResponse,
     Table,
     TableEnrichment,
     TaskResponse,
@@ -270,31 +271,31 @@ class DatabarClient:
         data = self._request("GET", f"/enrichments/{enrichment_id}")
         return Enrichment.model_validate(data)
 
-    def run_enrichment(self, enrichment_id: int, params: Dict[str, Any]) -> TaskResponse:
+    def run_enrichment(self, enrichment_id: int, params: Dict[str, Any]) -> RunResponse:
         """Submit an enrichment run. Returns a task — use poll_task() or run_enrichment_sync()."""
         data = self._request("POST", f"/enrichments/{enrichment_id}/run", json={"params": params})
-        return TaskResponse.model_validate(data)
+        return RunResponse.model_validate(data)
 
     def run_enrichment_bulk(
         self, enrichment_id: int, params: List[Dict[str, Any]]
-    ) -> TaskResponse:
+    ) -> RunResponse:
         """Submit a bulk enrichment run for multiple inputs."""
         data = self._request("POST", f"/enrichments/{enrichment_id}/bulk-run", json={"params": params})
-        return TaskResponse.model_validate(data)
+        return RunResponse.model_validate(data)
 
     def run_enrichment_sync(
         self, enrichment_id: int, params: Dict[str, Any]
     ) -> Any:
         """Submit and poll an enrichment, returning final data when complete."""
         task = self.run_enrichment(enrichment_id, params)
-        return self.poll_task(task.request_id)
+        return self.poll_task(task.task_id)
 
     def run_enrichment_bulk_sync(
         self, enrichment_id: int, params: List[Dict[str, Any]]
     ) -> Any:
         """Submit and poll a bulk enrichment, returning final data when complete."""
         task = self.run_enrichment_bulk(enrichment_id, params)
-        return self.poll_task(task.request_id)
+        return self.poll_task(task.task_id)
 
     def get_param_choices(
         self,
@@ -335,7 +336,7 @@ class DatabarClient:
         params: Dict[str, Any],
         enrichments: Optional[List[int]] = None,
         email_verifier: Optional[int] = None,
-    ) -> TaskResponse:
+    ) -> RunResponse:
         """
         Submit a waterfall run.
 
@@ -351,7 +352,7 @@ class DatabarClient:
             payload["email_verifier"] = email_verifier
 
         data = self._request("POST", f"/waterfalls/{identifier}/run", json=payload)
-        return TaskResponse.model_validate(data)
+        return RunResponse.model_validate(data)
 
     def run_waterfall_bulk(
         self,
@@ -359,7 +360,7 @@ class DatabarClient:
         params: List[Dict[str, Any]],
         enrichments: Optional[List[int]] = None,
         email_verifier: Optional[int] = None,
-    ) -> TaskResponse:
+    ) -> RunResponse:
         """Submit a bulk waterfall run for multiple inputs."""
         if not enrichments:
             waterfall = self.get_waterfall(identifier)
@@ -370,7 +371,7 @@ class DatabarClient:
             payload["email_verifier"] = email_verifier
 
         data = self._request("POST", f"/waterfalls/{identifier}/bulk-run", json=payload)
-        return TaskResponse.model_validate(data)
+        return RunResponse.model_validate(data)
 
     def run_waterfall_sync(
         self,
@@ -381,7 +382,7 @@ class DatabarClient:
     ) -> Any:
         """Submit and poll a waterfall, returning final data when complete."""
         task = self.run_waterfall(identifier, params, enrichments, email_verifier)
-        return self.poll_task(task.request_id)
+        return self.poll_task(task.task_id)
 
     def run_waterfall_bulk_sync(
         self,
@@ -392,7 +393,7 @@ class DatabarClient:
     ) -> Any:
         """Submit and poll a bulk waterfall, returning final data when complete."""
         task = self.run_waterfall_bulk(identifier, params, enrichments, email_verifier)
-        return self.poll_task(task.request_id)
+        return self.poll_task(task.task_id)
 
     # -----------------------------------------------------------------------
     # Tables

{databar-2.0.0 → databar-2.0.2}/src/databar/models.py

@@ -104,8 +104,24 @@ class TaskStatus(str, Enum):
     GONE = "gone"
 
 
+class RunResponse(BaseModel):
+    """Returned by all /run and /bulk-run endpoints. Contains the task_id to poll."""
+
+    task_id: str = Field(description="Unique identifier of the submitted task.")
+    status: str = Field(default="processing")
+
+
 class TaskResponse(BaseModel):
-    request_id: str = Field(description="Unique identifier of the request.")
+    """Returned by GET /v1/tasks/{task_id}.
+
+    The backend currently uses 'request_id' as the field name; this model
+    accepts both 'task_id' and 'request_id' so it works before and after
+    the backend renames the field.
+    """
+
+    task_id: str = Field(
+        description="Unique identifier of the task.",
+    )
     status: str = Field(
         description="Current status: processing, completed, failed, or gone."
     )
@@ -114,6 +130,13 @@ class TaskResponse(BaseModel):
         description="Resulting data once completed.",
     )
     error: Optional[Union[str, List[str]]] = None
+    credits_spent: float = 0
+
+    @classmethod
+    def model_validate(cls, obj: Any, **kwargs: Any) -> "TaskResponse":
+        if isinstance(obj, dict) and "task_id" not in obj and "request_id" in obj:
+            obj = {**obj, "task_id": obj["request_id"]}
+        return super().model_validate(obj, **kwargs)
 
 
 # ===========================================================================

{databar-2.0.0 → databar-2.0.2/src/databar.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databar
-Version: 2.0.0
+Version: 2.0.2
 Summary: Official Databar.ai Python SDK and CLI — connect to enrichments, waterfalls, and tables via api.databar.ai
 Author-email: "Databar.ai Team" <info@databar.ai>
 License: MIT License
@@ -139,7 +139,7 @@ for param in enrichment.params:
 
 # Run single enrichment (async — returns task)
 task = client.run_enrichment(123, {"email": "alice@example.com"})
-data = client.poll_task(task.request_id)
+data = client.poll_task(task.task_id)
 
 # Run single enrichment (sync convenience wrapper)
 data = client.run_enrichment_sync(123, {"email": "alice@example.com"})

{databar-2.0.0 → databar-2.0.2}/tests/test_cli.py

@@ -110,7 +110,7 @@ def _client_mock(**method_overrides):
     m.get_table_enrichments.return_value = [TableEnrichment(id=5, name="My Enrichment")]
     m.add_enrichment.return_value = {"status": "ok"}
     m.run_table_enrichment.return_value = {"status": "triggered"}
-    m.get_task.return_value = TaskResponse(request_id="t1", status="completed", data={"result": "ok"})
+    m.get_task.return_value = TaskResponse(task_id="t1", status="completed", data={"result": "ok"})
     m.poll_task.return_value = {"result": "ok"}
 
     for method, return_value in method_overrides.items():
@@ -329,4 +329,4 @@ def test_task_get_poll(monkeypatch):
 def test_version_flag():
     result = runner.invoke(app, ["--version"])
     assert result.exit_code == 0
-    assert "2.0.0" in result.output
+    assert "2.0.2" in result.output

{databar-2.0.0 → databar-2.0.2}/tests/test_client.py

@@ -29,6 +29,7 @@ from databar.models import (
     BatchUpdateResponse,
     BatchUpdateRow,
     InsertRow,
+    RunResponse,
     UpsertResponse,
     UpsertRow,
 )
@@ -150,7 +151,7 @@ def test_get_enrichment(client: DatabarClient, httpx_mock: HTTPXMock):
 def test_run_enrichment_returns_task(client: DatabarClient, httpx_mock: HTTPXMock):
     httpx_mock.add_response(url=f"{BASE_URL}/enrichments/1/run", json=task_payload("processing"))
     task = client.run_enrichment(1, {"email": "test@example.com"})
-    assert task.request_id == "task-123"
+    assert task.task_id == "task-123"
     assert task.status == "processing"
 
 
@@ -163,7 +164,7 @@ def test_run_enrichment_sync(client: DatabarClient, httpx_mock: HTTPXMock):
 
 def test_run_enrichment_sync_failed(client: DatabarClient, httpx_mock: HTTPXMock):
     httpx_mock.add_response(url=f"{BASE_URL}/enrichments/1/run", json=task_payload("processing"))
-    httpx_mock.add_response(url=f"{BASE_URL}/tasks/task-123", json={"request_id": "task-123", "status": "failed", "data": None, "error": "upstream error"})
+    httpx_mock.add_response(url=f"{BASE_URL}/tasks/task-123", json={"task_id": "task-123", "status": "failed", "data": None, "error": "upstream error"})
     with pytest.raises(DatabarTaskFailedError, match="upstream error"):
         client.run_enrichment_sync(1, {"email": "test@example.com"})
 
@@ -193,7 +194,7 @@ def test_run_waterfall_auto_resolves_providers(client: DatabarClient, httpx_mock
     httpx_mock.add_response(url=f"{BASE_URL}/waterfalls/email_getter", json=waterfall_payload())
     httpx_mock.add_response(url=f"{BASE_URL}/waterfalls/email_getter/run", json=task_payload("processing"))
     task = client.run_waterfall("email_getter", {"linkedin_url": "https://linkedin.com/in/alice"})
-    assert task.request_id == "task-123"
+    assert task.task_id == "task-123"
     req = httpx_mock.get_requests()[-1]
     body = json.loads(req.content)
     assert body["enrichments"] == [10, 11]