fix: wire max_discovered_files through centralized config

nikolay-e · nikolay-e · commit 1907219be73b · 2026-02-28T12:13:33.000+01:00
diff --git a/src/treemapper/diffctx/__init__.py b/src/treemapper/diffctx/__init__.py
@@ -38,7 +38,7 @@
 _MAX_EXPANSION_FILES = LIMITS.max_expansion_files
 _OVERHEAD_PER_FRAGMENT = LIMITS.overhead_per_fragment
 _FALLBACK_MAX_FILES = 10_000
-_MAX_DISCOVERED_FILES = 200
+_MAX_DISCOVERED_FILES = LIMITS.max_discovered_files
 
 _SEMANTIC_KINDS = frozenset(
     {
diff --git a/src/treemapper/diffctx/config/limits.py b/src/treemapper/diffctx/config/limits.py
@@ -9,6 +9,7 @@ class AlgorithmLimits:
     max_fragments: int = 200
     max_generated_fragments: int = 5
     max_candidate_files: int = 5000
+    max_discovered_files: int = 200
     skip_expensive_threshold: int = 2000
     rare_identifier_threshold: int = 3
     max_expansion_files: int = 20

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@`
`38`	`38`	`_MAX_EXPANSION_FILES = LIMITS.max_expansion_files`
`39`	`39`	`_OVERHEAD_PER_FRAGMENT = LIMITS.overhead_per_fragment`
`40`	`40`	`_FALLBACK_MAX_FILES = 10_000`
`41`		`-_MAX_DISCOVERED_FILES = 200`
	`41`	`+_MAX_DISCOVERED_FILES = LIMITS.max_discovered_files`
`42`	`42`
`43`	`43`	`_SEMANTIC_KINDS = frozenset(`
`44`	`44`	`{`