BlackVortexAIandClaude Fable 5.1
ccd946e7a5
The LUT install directory is checked before it is created, so a linked parent never causes a directory outside models/; installs check for an existing file before downloading, run at most two in parallel with one per entry, and parse off the event loop. Autocomplete searches each selected dataset once and truncates terms. Raster masks are validated against a 32 megapixel budget before decoding, and text logs validate the name before creating the log directory. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
34 lines
1.7 KiB
Python
34 lines
1.7 KiB
Python
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from py.util.completion.service import MAX_SELECTED_DATASETS, MAX_TERM_LENGTH, CompletionService
|
|
|
|
|
|
class CompletionLimitTests(unittest.TestCase):
|
|
def test_repeated_and_padded_dataset_selections_search_each_dataset_once(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
(root / "a.csv").write_text("tag,category\nalpha,general\n", encoding="utf-8")
|
|
(root / "b.csv").write_text("tag,category\nalpha2,general\n", encoding="utf-8")
|
|
service = CompletionService(root)
|
|
selected = ["a.csv"] * 500 + ["missing.csv", "b.csv", "a.csv"]
|
|
paths = service.resolve_paths(selected)
|
|
self.assertEqual([path.name for path in paths], ["a.csv"], "duplicates collapse and the slice limit applies before b.csv")
|
|
self.assertEqual([path.name for path in service.resolve_paths(["b.csv", "a.csv", "b.csv"])], ["b.csv", "a.csv"])
|
|
self.assertLessEqual(len(service.resolve_paths(["a.csv", "b.csv"] * 100)), MAX_SELECTED_DATASETS)
|
|
results = service.search("al", 20, ["a.csv", "a.csv", "b.csv"])
|
|
self.assertEqual(sorted(item["insert_text"] for item in results), ["alpha", "alpha2"])
|
|
|
|
def test_search_term_is_truncated(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
(root / "a.csv").write_text("tag,category\nalpha,general\n", encoding="utf-8")
|
|
service = CompletionService(root)
|
|
self.assertEqual(service.search("al" + "x" * 10_000, 5), [])
|
|
self.assertEqual(len("x" * MAX_TERM_LENGTH), MAX_TERM_LENGTH)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|