databar 2.1.0__tar.gz → 2.2.0__tar.gz

{databar-2.1.0/src/databar.egg-info → databar-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databar
-Version: 2.1.0
+Version: 2.2.0
 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
@@ -55,7 +55,7 @@ Dynamic: license-file
 
 # Databar Python SDK
 
-Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, and manage tables via `api.databar.ai/v1`.
+Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, flows, and manage tables via `api.databar.ai/v1`.
 
 [![PyPI](https://img.shields.io/pypi/v/databar-ai)](https://pypi.org/project/databar-ai/)
 [![Python](https://img.shields.io/pypi/pyversions/databar-ai)](https://pypi.org/project/databar-ai/)
@@ -154,7 +154,8 @@ data = client.run_enrichment_sync(123, {"email": "alice@example.com"})
 # Run with pagination (for list-style enrichments)
 data = client.run_enrichment_sync(123, {"query": "CEO"}, pages=3)
 
-# Bulk run
+# Bulk run — results are aligned to inputs: one element per input, in input
+# order, with None for inputs that returned no data (len(data) == len(inputs)).
 data = client.run_enrichment_bulk_sync(123, [
     {"email": "alice@example.com"},
     {"email": "bob@example.com"},
@@ -192,6 +193,26 @@ results = client.run_waterfall_bulk_sync(
 )
 ```
 
+### Flows
+
+```python
+# List saved flows
+flows = client.list_flows()
+
+# Inspect a flow's declared inputs
+flow = client.get_flow("flow-uuid")
+for inp in flow.inputs:
+    print(f"  {inp.id} (required={inp.required}): {inp.description}")
+
+# Run a flow (submit + poll in one call).
+# inputs maps each flow input id → value.
+result = client.run_flow_sync("flow-uuid", {"email": "alice@example.com"})
+
+# Or submit and poll manually
+task = client.run_flow("flow-uuid", {"email": "alice@example.com"})
+result = client.poll_task(task.task_id)
+```
+
 ### Tables
 
 ```python
@@ -427,6 +448,21 @@ databar waterfall run email_getter --params '{"linkedin_url": "https://linkedin.
 databar waterfall bulk email_getter --input leads.csv --out results.csv
 ```
 
+### Flows
+
+```bash
+# List saved flows
+databar flow list
+databar flow list --query "buyer"
+
+# Get flow details (declared inputs)
+databar flow get <flow-id>
+
+# Run a flow (--inputs maps each flow input id → value)
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}'
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}' --format json
+```
+
 ### Tables
 
 ```bash

{databar-2.1.0 → databar-2.2.0}/README.md

@@ -1,6 +1,6 @@
 # Databar Python SDK
 
-Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, and manage tables via `api.databar.ai/v1`.
+Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, flows, and manage tables via `api.databar.ai/v1`.
 
 [![PyPI](https://img.shields.io/pypi/v/databar-ai)](https://pypi.org/project/databar-ai/)
 [![Python](https://img.shields.io/pypi/pyversions/databar-ai)](https://pypi.org/project/databar-ai/)
@@ -99,7 +99,8 @@ data = client.run_enrichment_sync(123, {"email": "alice@example.com"})
 # Run with pagination (for list-style enrichments)
 data = client.run_enrichment_sync(123, {"query": "CEO"}, pages=3)
 
-# Bulk run
+# Bulk run — results are aligned to inputs: one element per input, in input
+# order, with None for inputs that returned no data (len(data) == len(inputs)).
 data = client.run_enrichment_bulk_sync(123, [
     {"email": "alice@example.com"},
     {"email": "bob@example.com"},
@@ -137,6 +138,26 @@ results = client.run_waterfall_bulk_sync(
 )
 ```
 
+### Flows
+
+```python
+# List saved flows
+flows = client.list_flows()
+
+# Inspect a flow's declared inputs
+flow = client.get_flow("flow-uuid")
+for inp in flow.inputs:
+    print(f"  {inp.id} (required={inp.required}): {inp.description}")
+
+# Run a flow (submit + poll in one call).
+# inputs maps each flow input id → value.
+result = client.run_flow_sync("flow-uuid", {"email": "alice@example.com"})
+
+# Or submit and poll manually
+task = client.run_flow("flow-uuid", {"email": "alice@example.com"})
+result = client.poll_task(task.task_id)
+```
+
 ### Tables
 
 ```python
@@ -372,6 +393,21 @@ databar waterfall run email_getter --params '{"linkedin_url": "https://linkedin.
 databar waterfall bulk email_getter --input leads.csv --out results.csv
 ```
 
+### Flows
+
+```bash
+# List saved flows
+databar flow list
+databar flow list --query "buyer"
+
+# Get flow details (declared inputs)
+databar flow get <flow-id>
+
+# Run a flow (--inputs maps each flow input id → value)
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}'
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}' --format json
+```
+
 ### Tables
 
 ```bash

{databar-2.1.0 → databar-2.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "databar"
-version = "2.1.0"
+version = "2.2.0"
 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.1.0 → databar-2.2.0}/src/databar/__init__.py

@@ -59,6 +59,10 @@ from .models import (
     # Waterfalls
     Waterfall,
     WaterfallEnrichment,
+    # Flows
+    Flow,
+    FlowInput,
+    FlowOutput,
     # Tables
     Table,
     Column,
@@ -95,7 +99,7 @@ from .models import (
     Folder,
 )
 
-__version__ = "2.1.0"
+__version__ = "2.2.0"
 __all__ = [
     "DatabarClient",
     # exceptions
@@ -131,6 +135,10 @@ __all__ = [
     # waterfalls
     "Waterfall",
     "WaterfallEnrichment",
+    # flows
+    "Flow",
+    "FlowInput",
+    "FlowOutput",
     # tables
     "Table",
     "Column",

{databar-2.1.0 → databar-2.2.0}/src/databar/cli/_guide.py

@@ -128,6 +128,16 @@ databar waterfall run <identifier> --params '{"key": "value"}' --format json
 databar waterfall bulk <identifier> --input data.csv --out results.csv
 ```
 
+### Flows
+```bash
+databar flow list --format json
+databar flow get <flow-id> --format json                     # declared inputs
+databar flow run <flow-id> --inputs '{"input_id": "value"}' --format json
+```
+
+NOTE: `flow run` uses `--inputs` (maps each flow input id → value), NOT `--params`.
+See `flow get` for the list of declared inputs.
+
 ### Tables
 ```bash
 databar table list --format json
@@ -174,6 +184,13 @@ result = client.run_enrichment_sync(123, {"email": "alice@example.com"})
 result = client.run_waterfall_sync("email_getter", {"linkedin_url": "..."})
 # waterfall.identifier (also .slug) → slug like "email_getter"
 
+# Flows
+flows = client.list_flows()
+flow  = client.get_flow("flow-uuid")
+# flow.id (also .identifier) → flow UUID
+# flow.inputs[i].id → input key to use in the run inputs dict
+result = client.run_flow_sync("flow-uuid", {"email": "alice@example.com"})
+
 # Tables
 tables = client.list_tables()
 # table.identifier (also .id, .uuid) → UUID string
@@ -201,6 +218,7 @@ client.create_rows(table.identifier, [InsertRow(fields={"email": "alice@example.
 
 - `Table`: `.id`, `.uuid` → `.identifier`
 - `Waterfall`: `.slug` → `.identifier`
+- `Flow`: `.identifier` → `.id`
 - `EnrichmentParam`: `.slug` → `.name`, `.label` → `.description`, `.required` → `.is_required`
 - `EnrichmentResponseField`: `.slug` → `.name`
 

{databar-2.1.0 → databar-2.2.0}/src/databar/cli/app.py

@@ -10,7 +10,7 @@ import typer
 
 from databar import __version__
 
-from . import enrichments, tables, tasks, waterfalls
+from . import enrichments, flows, tables, tasks, waterfalls
 from ._auth import app as auth_app
 
 app = typer.Typer(
@@ -29,6 +29,7 @@ app.add_typer(auth_app, name=None)  # merged at root level
 # Subcommand groups
 app.add_typer(enrichments.app, name="enrich")
 app.add_typer(waterfalls.app, name="waterfall")
+app.add_typer(flows.app, name="flow")
 app.add_typer(tables.app, name="table")
 app.add_typer(tasks.app, name="task")
 

databar-2.2.0/src/databar/cli/flows.py

@@ -0,0 +1,122 @@
+"""
+CLI commands for flows.
+
+  databar flow list [--query] [--format]
+  databar flow get  <flow-id> [--format]
+  databar flow run  <flow-id> --inputs '{"email":"a@b.com"}' [--format] [--raw]
+"""
+
+from __future__ import annotations
+
+import json
+from typing import Optional
+
+import typer
+
+from databar.exceptions import DatabarError
+
+from ._auth import get_client
+from ._output import OutputFormat, console, error, info, output
+
+app = typer.Typer(help="List and run saved flows.")
+
+
+@app.command("list")
+def list_flows(
+    query: Optional[str] = typer.Option(None, "--query", "-q", help="Filter by name/description."),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
+) -> None:
+    """List saved flows in the workspace."""
+    client = get_client()
+    try:
+        flows = client.list_flows()
+    except DatabarError as e:
+        error(str(e))
+    finally:
+        client.close()
+
+    if query:
+        q = query.lower()
+        flows = [
+            f for f in flows
+            if q in f.name.lower() or q in f.description.lower() or q in f.id.lower()
+        ]
+
+    if not flows:
+        info("No flows found.")
+        return
+
+    rows = [
+        {
+            "id": f.id,
+            "name": f.name,
+            "inputs": len(f.inputs),
+            "description": f.description[:60] + ("…" if len(f.description) > 60 else ""),
+        }
+        for f in flows
+    ]
+    output(rows, fmt, table_columns=["id", "name", "inputs", "description"])
+
+
+@app.command("get")
+def get_flow(
+    flow_id: str = typer.Argument(..., help="Flow id."),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
+) -> None:
+    """Get details for a specific flow."""
+    client = get_client()
+    try:
+        f = client.get_flow(flow_id)
+    except DatabarError as exc:
+        error(str(exc))
+    finally:
+        client.close()
+
+    if fmt == OutputFormat.JSON:
+        output(f.model_dump(), fmt)
+        return
+
+    console.print(f"\n[bold cyan]Flow:[/bold cyan] {f.name} ({f.id})")
+    console.print(f"\n{f.description}\n")
+
+    if f.inputs:
+        console.print("[bold]Inputs:[/bold]")
+        rows = [
+            {
+                "id": inp.id,
+                "required": "yes" if inp.required else "no",
+                "type": inp.type,
+                "description": inp.description,
+            }
+            for inp in f.inputs
+        ]
+        output(rows, OutputFormat.TABLE, table_columns=["id", "required", "type", "description"])
+
+
+@app.command("run")
+def run_flow(
+    flow_id: str = typer.Argument(..., help="Flow id."),
+    inputs_json: str = typer.Option(..., "--inputs", "-i", help='JSON inputs, e.g. \'{"email":"a@b.com"}\''),
+    fmt: OutputFormat = typer.Option(OutputFormat.TABLE, "--format", "--output", "-f"),
+    raw: bool = typer.Option(False, "--raw", help="Print raw result without formatting."),
+) -> None:
+    """Run a flow and wait for results."""
+    try:
+        inputs = json.loads(inputs_json)
+    except json.JSONDecodeError as e:
+        error(f"Invalid JSON for --inputs: {e}")
+
+    client = get_client()
+    try:
+        info(f"Running flow '{flow_id}'…")
+        result = client.run_flow_sync(flow_id, inputs)
+    except DatabarError as exc:
+        error(str(exc))
+    finally:
+        client.close()
+
+    if raw:
+        console.print(result)
+        return
+
+    output(result, fmt)

{databar-2.1.0 → databar-2.2.0}/src/databar/client.py

@@ -15,6 +15,7 @@ Endpoint groups:
   - Enrichments: list_enrichments, get_enrichment, run_enrichment[_bulk][_sync],
                  get_param_choices
   - Waterfalls:  list_waterfalls, get_waterfall, run_waterfall[_bulk][_sync]
+  - Flows:       list_flows, get_flow, run_flow[_sync]
   - Tables:      create_table, list_tables, delete_table, rename_table,
                  get_columns, create_column, rename_column, delete_column,
                  get_table_enrichments, add_enrichment, run_table_enrichment,
@@ -70,6 +71,7 @@ from .models import (
     ExporterListResponse,
     ExporterParam,
     ExporterResponseField,
+    Flow,
     Folder,
     InsertOptions,
     InsertRow,
@@ -265,6 +267,11 @@ class DatabarClient:
         Returns the task's data payload on success (status completed or partially_completed).
         Raises DatabarTaskFailedError, DatabarGoneError or DatabarTimeoutError otherwise.
 
+        For bulk runs the payload is a list aligned to the inputs: one element
+        per input in input order, with ``None`` for inputs that returned no data
+        (so ``len(data) == number of inputs`` and ``data[i]`` ↔ ``input[i]``). A
+        single run returns the bare result object.
+
         Task data is stored for 24 hours. After that the status becomes 'gone'.
         """
         for _ in range(self._max_poll_attempts):
@@ -383,6 +390,11 @@ class DatabarClient:
         """
         Submit a bulk enrichment run for multiple inputs.
 
+        The polled result is aligned to the inputs: one element per input, in the
+        same order as ``params``, with ``None`` for inputs that returned no data.
+        So ``len(result) == len(params)`` and ``result[i]`` is the result for
+        ``params[i]`` — join results back to inputs by position.
+
         Args:
             enrichment_id: The enrichment to run.
             params: List of per-row input parameter dicts.
@@ -410,7 +422,11 @@ class DatabarClient:
         params: List[Dict[str, Any]],
         pages: Optional[int] = None,
     ) -> Any:
-        """Submit and poll a bulk enrichment, returning final data when complete."""
+        """Submit and poll a bulk enrichment, returning final data when complete.
+
+        Returns a list aligned to ``params``: one element per input in input
+        order, ``None`` for misses (``len(result) == len(params)``).
+        """
         task = self.run_enrichment_bulk(enrichment_id, params, pages=pages)
         return self.poll_task(task.task_id)
 
@@ -478,7 +494,12 @@ class DatabarClient:
         enrichments: Optional[List[int]] = None,
         email_verifier: Optional[int] = None,
     ) -> RunResponse:
-        """Submit a bulk waterfall run for multiple inputs."""
+        """Submit a bulk waterfall run for multiple inputs.
+
+        The polled result is aligned to the inputs: one element per input, in the
+        same order as ``params``, with ``None`` for inputs that returned no data
+        (``len(result) == len(params)``, ``result[i]`` ↔ ``params[i]``).
+        """
         if not enrichments:
             waterfall = self.get_waterfall(identifier)
             enrichments = [e.id for e in waterfall.available_enrichments]
@@ -508,10 +529,45 @@ class DatabarClient:
         enrichments: Optional[List[int]] = None,
         email_verifier: Optional[int] = None,
     ) -> Any:
-        """Submit and poll a bulk waterfall, returning final data when complete."""
+        """Submit and poll a bulk waterfall, returning final data when complete.
+
+        Returns a list aligned to ``params``: one element per input in input
+        order, ``None`` for misses (``len(result) == len(params)``).
+        """
         task = self.run_waterfall_bulk(identifier, params, enrichments, email_verifier)
         return self.poll_task(task.task_id)
 
+    # -----------------------------------------------------------------------
+    # Flows
+    # -----------------------------------------------------------------------
+
+    def list_flows(self) -> List[Flow]:
+        """List all saved flows in the workspace."""
+        data = self._request("GET", "/flows")
+        return [Flow.model_validate(f) for f in data]
+
+    def get_flow(self, flow_id: str) -> Flow:
+        """Get details for a specific flow, including its declared inputs."""
+        data = self._request("GET", f"/flows/{flow_id}")
+        return Flow.model_validate(data)
+
+    def run_flow(self, flow_id: str, inputs: Dict[str, str]) -> RunResponse:
+        """
+        Submit a flow run. Returns a task — use poll_task() or run_flow_sync().
+
+        Args:
+            flow_id: The flow UUID (from list_flows / get_flow).
+            inputs: Maps each flow input id → value. See get_flow() for the
+                list of declared inputs.
+        """
+        data = self._request("POST", f"/flows/{flow_id}/run", json={"inputs": inputs})
+        return RunResponse.model_validate(data)
+
+    def run_flow_sync(self, flow_id: str, inputs: Dict[str, str]) -> Any:
+        """Submit and poll a flow, returning final data when complete."""
+        task = self.run_flow(flow_id, inputs)
+        return self.poll_task(task.task_id)
+
     # -----------------------------------------------------------------------
     # Tables — CRUD
     # -----------------------------------------------------------------------

{databar-2.1.0 → databar-2.2.0}/src/databar/models.py

@@ -261,7 +261,12 @@ class TaskResponse(BaseModel):
     )
     data: Optional[Union[List[Any], Dict[str, Any]]] = Field(
         default=None,
-        description="Resulting data once completed.",
+        description=(
+            "Resulting data once completed. For bulk runs this is a list aligned to "
+            "the inputs: one element per input in input order, with null for inputs "
+            "that returned no data (len(data) == number of inputs). A single run "
+            "returns the bare result object."
+        ),
     )
     error: Optional[Union[str, List[str]]] = None
     credits_spent: float = 0
@@ -315,6 +320,62 @@ class Waterfall(BaseModel):
         return self.identifier
 
 
+# ===========================================================================
+# Flows
+# ===========================================================================
+
+
+class FlowInput(BaseModel):
+    """A declared input of a flow.
+
+    Fields: id, description, type, required.
+
+    The ``id`` is the key to use in the inputs dict passed to run_flow().
+    """
+
+    id: str = Field(description="Input id — the key to use in the run_flow() inputs dict.")
+    description: str
+    type: str
+    required: bool
+
+
+class FlowOutput(BaseModel):
+    """A field produced by a flow node.
+
+    Fields: id, response_field_id.
+    """
+
+    id: str
+    response_field_id: Optional[int] = None
+
+
+class Flow(BaseModel):
+    """A saved workspace flow (multi-step enrichment pipeline).
+
+    Fields: id, name, description, inputs, outputs, created_at, updated_at.
+
+    Property aliases: .identifier → .id.
+
+    Usage::
+
+        flow = client.get_flow("flow-uuid")
+        result = client.run_flow_sync(flow.id, {"email": "alice@example.com"})
+    """
+
+    id: str = Field(description="Flow UUID. Use this when calling get_flow()/run_flow().")
+    name: str
+    description: str
+    inputs: List[FlowInput] = Field(default_factory=list)
+    outputs: List[FlowOutput] = Field(default_factory=list)
+    created_at: str = ""
+    updated_at: str = ""
+
+    @property
+    def identifier(self) -> str:
+        """Alias for id."""
+        return self.id
+
+
 # ===========================================================================
 # Tables
 # ===========================================================================

{databar-2.1.0 → databar-2.2.0/src/databar.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databar
-Version: 2.1.0
+Version: 2.2.0
 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
@@ -55,7 +55,7 @@ Dynamic: license-file
 
 # Databar Python SDK
 
-Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, and manage tables via `api.databar.ai/v1`.
+Official Python SDK and CLI for [Databar.ai](https://databar.ai) — run data enrichments, waterfall lookups, flows, and manage tables via `api.databar.ai/v1`.
 
 [![PyPI](https://img.shields.io/pypi/v/databar-ai)](https://pypi.org/project/databar-ai/)
 [![Python](https://img.shields.io/pypi/pyversions/databar-ai)](https://pypi.org/project/databar-ai/)
@@ -154,7 +154,8 @@ data = client.run_enrichment_sync(123, {"email": "alice@example.com"})
 # Run with pagination (for list-style enrichments)
 data = client.run_enrichment_sync(123, {"query": "CEO"}, pages=3)
 
-# Bulk run
+# Bulk run — results are aligned to inputs: one element per input, in input
+# order, with None for inputs that returned no data (len(data) == len(inputs)).
 data = client.run_enrichment_bulk_sync(123, [
     {"email": "alice@example.com"},
     {"email": "bob@example.com"},
@@ -192,6 +193,26 @@ results = client.run_waterfall_bulk_sync(
 )
 ```
 
+### Flows
+
+```python
+# List saved flows
+flows = client.list_flows()
+
+# Inspect a flow's declared inputs
+flow = client.get_flow("flow-uuid")
+for inp in flow.inputs:
+    print(f"  {inp.id} (required={inp.required}): {inp.description}")
+
+# Run a flow (submit + poll in one call).
+# inputs maps each flow input id → value.
+result = client.run_flow_sync("flow-uuid", {"email": "alice@example.com"})
+
+# Or submit and poll manually
+task = client.run_flow("flow-uuid", {"email": "alice@example.com"})
+result = client.poll_task(task.task_id)
+```
+
 ### Tables
 
 ```python
@@ -427,6 +448,21 @@ databar waterfall run email_getter --params '{"linkedin_url": "https://linkedin.
 databar waterfall bulk email_getter --input leads.csv --out results.csv
 ```
 
+### Flows
+
+```bash
+# List saved flows
+databar flow list
+databar flow list --query "buyer"
+
+# Get flow details (declared inputs)
+databar flow get <flow-id>
+
+# Run a flow (--inputs maps each flow input id → value)
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}'
+databar flow run <flow-id> --inputs '{"email": "alice@example.com"}' --format json
+```
+
 ### Tables
 
 ```bash

{databar-2.1.0 → databar-2.2.0}/src/databar.egg-info/SOURCES.txt

@@ -18,6 +18,7 @@ src/databar/cli/_onboard.py
 src/databar/cli/_output.py
 src/databar/cli/app.py
 src/databar/cli/enrichments.py
+src/databar/cli/flows.py
 src/databar/cli/tables.py
 src/databar/cli/tasks.py
 src/databar/cli/waterfalls.py

{databar-2.1.0 → databar-2.2.0}/tests/test_cli.py

@@ -28,6 +28,8 @@ from databar.models import (
     Enrichment,
     EnrichmentParam,
     EnrichmentSummary,
+    Flow,
+    FlowInput,
     Table,
     TableEnrichment,
     TaskResponse,
@@ -88,6 +90,18 @@ def _client_mock(**method_overrides):
     m.run_waterfall_sync.return_value = [{"email": "alice@example.com"}]
     m.run_waterfall_bulk_sync.return_value = [{"email": "alice@example.com"}]
 
+    m.list_flows.return_value = [
+        Flow(
+            id="flow-uuid-1",
+            name="Find buyer",
+            description="Enrich a lead",
+            inputs=[FlowInput(id="email", description="Email", type="text", required=True)],
+            outputs=[],
+        )
+    ]
+    m.get_flow.return_value = m.list_flows.return_value[0]
+    m.run_flow_sync.return_value = {"full_name": "Alice"}
+
     m.list_tables.return_value = [
         Table(identifier="tbl-1", name="My Table", created_at="2024-01-01", updated_at="2024-01-01")
     ]
@@ -241,6 +255,36 @@ def test_waterfall_run(monkeypatch):
     mock.run_waterfall_sync.assert_called_once()
 
 
+# ===========================================================================
+# Flows
+# ===========================================================================
+
+
+def test_flow_list(monkeypatch):
+    mock = _client_mock()
+    monkeypatch.setattr("databar.cli.flows.get_client", lambda: mock)
+    result = invoke(["flow", "list"])
+    assert result.exit_code == 0
+    assert "flow-uuid-1" in result.output
+    assert "Find buyer" in result.output
+
+
+def test_flow_get(monkeypatch):
+    mock = _client_mock()
+    monkeypatch.setattr("databar.cli.flows.get_client", lambda: mock)
+    result = invoke(["flow", "get", "flow-uuid-1"])
+    assert result.exit_code == 0
+    assert "Find buyer" in result.output
+
+
+def test_flow_run(monkeypatch):
+    mock = _client_mock()
+    monkeypatch.setattr("databar.cli.flows.get_client", lambda: mock)
+    result = invoke(["flow", "run", "flow-uuid-1", "--inputs", '{"email":"alice@example.com"}'])
+    assert result.exit_code == 0
+    mock.run_flow_sync.assert_called_once_with("flow-uuid-1", {"email": "alice@example.com"})
+
+
 # ===========================================================================
 # Tables
 # ===========================================================================
@@ -329,4 +373,4 @@ def test_task_get_poll(monkeypatch):
 def test_version_flag():
     result = runner.invoke(app, ["--version"])
     assert result.exit_code == 0
-    assert "2.0.9" in result.output
+    assert "2.2.0" in result.output

{databar-2.1.0 → databar-2.2.0}/tests/test_client.py

@@ -38,6 +38,7 @@ from .conftest import (
     BASE_URL,
     enrichment_payload,
     enrichment_summary_payload,
+    flow_payload,
     table_payload,
     task_payload,
     user_payload,
@@ -200,6 +201,43 @@ def test_run_waterfall_auto_resolves_providers(client: DatabarClient, httpx_mock
     assert body["enrichments"] == [10, 11]
 
 
+# ===========================================================================
+# Flows
+# ===========================================================================
+
+
+def test_list_flows(client: DatabarClient, httpx_mock: HTTPXMock):
+    httpx_mock.add_response(url=f"{BASE_URL}/flows", json=[flow_payload()])
+    result = client.list_flows()
+    assert len(result) == 1
+    assert result[0].id == "flow-uuid-1"
+    assert result[0].identifier == "flow-uuid-1"
+    assert result[0].inputs[0].id == "email"
+
+
+def test_get_flow(client: DatabarClient, httpx_mock: HTTPXMock):
+    httpx_mock.add_response(url=f"{BASE_URL}/flows/flow-uuid-1", json=flow_payload())
+    f = client.get_flow("flow-uuid-1")
+    assert f.name == "Find buyer"
+    assert f.inputs[0].required is True
+
+
+def test_run_flow_returns_task(client: DatabarClient, httpx_mock: HTTPXMock):
+    httpx_mock.add_response(url=f"{BASE_URL}/flows/flow-uuid-1/run", json=task_payload("processing"))
+    task = client.run_flow("flow-uuid-1", {"email": "alice@example.com"})
+    assert task.task_id == "task-123"
+    req = httpx_mock.get_requests()[-1]
+    body = json.loads(req.content)
+    assert body == {"inputs": {"email": "alice@example.com"}}
+
+
+def test_run_flow_sync(client: DatabarClient, httpx_mock: HTTPXMock):
+    httpx_mock.add_response(url=f"{BASE_URL}/flows/flow-uuid-1/run", json=task_payload("processing"))
+    httpx_mock.add_response(url=f"{BASE_URL}/tasks/task-123", json=task_payload("completed", data={"full_name": "Alice"}))
+    result = client.run_flow_sync("flow-uuid-1", {"email": "alice@example.com"})
+    assert result == {"full_name": "Alice"}
+
+
 # ===========================================================================
 # Tables
 # ===========================================================================