wehub-resource-sync
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
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
|