jason/mempalace
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: align cmd_compress dict keys with compression_stats() return values (#569)

Browse Source 
* fix: align cmd_compress dict keys with compression_stats() return values

* test: align compress test mocks with actual compression_stats() keys

* fix: address review — add Total: assertion, move stats key test to test_dialect.py
This commit is contained in:
Arnold Wender
2026-04-12 01:16:31 +02:00
committed by GitHub
parent 9c4b7302cc
commit 89c0a58271
3 changed files with 30 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mempalace/cli.py
+4 -4
View File
@@ -349,7 +349,7 @@ def cmd_compress(args):
stats = dialect.compression_stats(doc, compressed)
total_original += stats["original_chars"]
total_compressed += stats["compressed_chars"]
total_compressed += stats["summary_chars"]
compressed_entries.append((doc_id, compressed, meta, stats))
@@ -359,7 +359,7 @@ def cmd_compress(args):
source = Path(meta.get("source_file", "?")).name
print(f" [{wing_name}/{room_name}] {source}")
print(
f" {stats['original_tokens']}t -> {stats['compressed_tokens']}t ({stats['ratio']:.1f}x)"
f" {stats['original_tokens_est']}t -> {stats['summary_tokens_est']}t ({stats['size_ratio']:.1f}x)"
)
print(f" {compressed}")
print()
@@ -370,8 +370,8 @@ def cmd_compress(args):
comp_col = client.get_or_create_collection("mempalace_compressed")
for doc_id, compressed, meta, stats in compressed_entries:
comp_meta = dict(meta)
comp_meta["compression_ratio"] = round(stats["ratio"], 1)
comp_meta["original_tokens"] = stats["original_tokens"]
comp_meta["compression_ratio"] = round(stats["size_ratio"], 1)
comp_meta["original_tokens"] = stats["original_tokens_est"]
comp_col.upsert(
ids=[doc_id],
documents=[compressed],