chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,80 @@
|
||||
import asyncio
|
||||
import json
|
||||
|
||||
from leann.cli import LeannCLI
|
||||
|
||||
|
||||
def test_cli_ask_accepts_positional_query(tmp_path, monkeypatch):
|
||||
monkeypatch.chdir(tmp_path)
|
||||
|
||||
cli = LeannCLI()
|
||||
parser = cli.create_parser()
|
||||
|
||||
args = parser.parse_args(["ask", "my-docs", "Where are prompts configured?"])
|
||||
|
||||
assert args.command == "ask"
|
||||
assert args.index_name == "my-docs"
|
||||
assert args.query == "Where are prompts configured?"
|
||||
|
||||
|
||||
def test_cli_ask_parses_metadata_filters_flag():
|
||||
cli = LeannCLI()
|
||||
parser = cli.create_parser()
|
||||
|
||||
filters_json = '{"chapter": {"<=": 5}, "genre": {"==": "fiction"}}'
|
||||
args = parser.parse_args(
|
||||
["ask", "my-docs", "Summarize early chapters", "--metadata-filters", filters_json]
|
||||
)
|
||||
|
||||
assert args.command == "ask"
|
||||
assert args.metadata_filters == filters_json
|
||||
# The raw string parses to the expected dict so downstream consumers can rely on it.
|
||||
assert json.loads(args.metadata_filters) == {
|
||||
"chapter": {"<=": 5},
|
||||
"genre": {"==": "fiction"},
|
||||
}
|
||||
|
||||
|
||||
def test_cli_ask_metadata_filters_default_is_none():
|
||||
cli = LeannCLI()
|
||||
parser = cli.create_parser()
|
||||
|
||||
args = parser.parse_args(["ask", "my-docs", "any query"])
|
||||
|
||||
assert args.metadata_filters is None
|
||||
|
||||
|
||||
def test_cli_ask_rejects_invalid_metadata_filters_json(tmp_path, monkeypatch, capsys):
|
||||
monkeypatch.chdir(tmp_path)
|
||||
|
||||
cli = LeannCLI()
|
||||
parser = cli.create_parser()
|
||||
# Set up an empty index dir so the existence check passes and we reach the JSON parser.
|
||||
index_dir = tmp_path / ".leann" / "indexes" / "my-docs"
|
||||
index_dir.mkdir(parents=True)
|
||||
(index_dir / "documents.leann.meta.json").write_text("{}")
|
||||
|
||||
args = parser.parse_args(["ask", "my-docs", "any query", "--metadata-filters", "not-json"])
|
||||
|
||||
asyncio.run(cli.ask_questions(args))
|
||||
|
||||
captured = capsys.readouterr()
|
||||
assert "--metadata-filters is not valid JSON" in captured.out
|
||||
|
||||
|
||||
def test_cli_ask_rejects_non_object_metadata_filters(tmp_path, monkeypatch, capsys):
|
||||
monkeypatch.chdir(tmp_path)
|
||||
|
||||
cli = LeannCLI()
|
||||
parser = cli.create_parser()
|
||||
index_dir = tmp_path / ".leann" / "indexes" / "my-docs"
|
||||
index_dir.mkdir(parents=True)
|
||||
(index_dir / "documents.leann.meta.json").write_text("{}")
|
||||
|
||||
# A valid JSON value that is not an object (dict) — must be rejected.
|
||||
args = parser.parse_args(["ask", "my-docs", "any query", "--metadata-filters", "[1, 2, 3]"])
|
||||
|
||||
asyncio.run(cli.ask_questions(args))
|
||||
|
||||
captured = capsys.readouterr()
|
||||
assert "--metadata-filters must be a JSON object" in captured.out
|
||||
Reference in New Issue
Block a user