feat: add --max-abstractions flag to control number of identified abstractions

main.py

@@ -40,6 +40,8 @@ def main():
     parser.add_argument("--language", default="english", help="Language for the generated tutorial (default: english)")
     # Add use_cache parameter to control LLM caching
     parser.add_argument("--no-cache", action="store_true", help="Disable LLM response caching (default: caching enabled)")
+    # Add max_abstraction_num parameter to control the number of abstractions
+    parser.add_argument("--max-abstractions", type=int, default=10, help="Maximum number of abstractions to identify (default: 20)")
 
     args = parser.parse_args()
 
@@ -68,6 +70,9 @@ def main():
         
         # Add use_cache flag (inverse of no-cache flag)
         "use_cache": not args.no_cache,
+        
+        # Add max_abstraction_num parameter
+        "max_abstraction_num": args.max_abstractions,
 
         # Outputs will be populated by the nodes
         "files": [],