datalegion-cli 1.6.1__tar.gz → 1.7.0__tar.gz

{datalegion_cli-1.6.1 → datalegion_cli-1.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datalegion-cli
-Version: 1.6.1
+Version: 1.7.0
 Summary: CLI for the Data Legion API — agent-friendly, fully async
 Project-URL: Homepage, https://www.datalegion.ai
 Project-URL: Documentation, https://www.datalegion.ai/docs

{datalegion_cli-1.6.1 → datalegion_cli-1.7.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datalegion-cli"
-version = "1.6.1"
+version = "1.7.0"
 description = "CLI for the Data Legion API — agent-friendly, fully async"
 readme = "README.md"
 requires-python = ">=3.11"

{datalegion_cli-1.6.1 → datalegion_cli-1.7.0}/src/datalegion_cli/commands/company.py

@@ -103,7 +103,7 @@ async def enrich(
 @async_command
 async def search(
     query: Annotated[str, typer.Argument(help="SQL SELECT query against the companies table.")],
-    limit: LimitOpt = 10,
+    limit: LimitOpt = 1,
     titlecase: TitleOpt = False,
     include_fields: FieldsInclude = None,
     exclude_fields: FieldsExclude = None,
@@ -138,7 +138,7 @@ async def search(
 @async_command
 async def discover(
     query: Annotated[str, typer.Argument(help="Natural language company search query.")],
-    limit: LimitOpt = 10,
+    limit: LimitOpt = 1,
     titlecase: TitleOpt = False,
     include_fields: FieldsInclude = None,
     exclude_fields: FieldsExclude = None,

{datalegion_cli-1.6.1 → datalegion_cli-1.7.0}/src/datalegion_cli/commands/person.py

@@ -129,7 +129,7 @@ async def enrich(
 @async_command
 async def search(
     query: Annotated[str, typer.Argument(help="SQL SELECT query against the people table.")],
-    limit: LimitOpt = 10,
+    limit: LimitOpt = 1,
     titlecase: TitleOpt = False,
     include_fields: FieldsInclude = None,
     exclude_fields: FieldsExclude = None,
@@ -163,7 +163,7 @@ async def search(
 @async_command
 async def discover(
     query: Annotated[str, typer.Argument(help="Natural language search query.")],
-    limit: LimitOpt = 10,
+    limit: LimitOpt = 1,
     titlecase: TitleOpt = False,
     include_fields: FieldsInclude = None,
     exclude_fields: FieldsExclude = None,

{datalegion_cli-1.6.1 → datalegion_cli-1.7.0}/tests/test_commands.py

@@ -490,3 +490,29 @@ class TestSearchWarning:
         assert result.exit_code == 0
         assert json.loads(result.stdout) == {"matches": [], "total": 0}
         assert f"search warning: {warning}" in result.stderr
+
+
+class TestSearchLimitDefault:
+    """search and discover send limit 1 unless --limit says otherwise (the API's default too)."""
+
+    @pytest.mark.parametrize(
+        "args",
+        [
+            ["person", "search", "SELECT * FROM people WHERE city = 'austin'"],
+            ["person", "discover", "engineers in austin"],
+            ["company", "search", "SELECT * FROM companies WHERE industry = 'banking'"],
+            ["company", "discover", "banks in austin"],
+        ],
+    )
+    def test_the_default_limit_is_one(self, monkeypatch, args):
+        _mock_config(monkeypatch)
+        sent: list[dict] = []
+
+        async def capture(self, method, endpoint, *, payload=None, params=None):
+            sent.append(payload or {})
+            return {"matches": [], "total": 0}
+
+        with patch("datalegion_cli.client.LegionClient.request", capture):
+            assert runner.invoke(app, args).exit_code == 0
+            assert runner.invoke(app, [*args, "--limit", "25"]).exit_code == 0
+        assert [p["limit"] for p in sent] == [1, 25]