utilities.rb in tailwind_dsl-0.0.31

- old
+ new

@@ -33,12 +33,19 @@
         # Use GPT3 to extract component models in supervised fashion
         helpers.batch_extraction(
           components,
           target_component_model_path,
           batch_size: 1,
-          use_prompt: true,
-          filter_design_system: 'tui',
+          filter: {
+            design_system: 'tui',
+            exclude_group_key: [
+              'application_ui.application_shells.multi_column',
+              'application_ui.application_shells.sidebar',
+              'application_ui.element.avatars',
+              'application_ui.element.badges'
+            ]
+          },
           extract_handler: TailwindDsl::Etl::Extractors::DataExtractor
         )
         # puts target_component_path
         # /Users/davidcruwys/dev/kgems/tailwind_dsl/.components
 
@@ -77,16 +84,15 @@
   end
 
   # Extracts the next component data using GPT3
   #
   # This needs to be supervised and verified, so it will only do a few models at a time
-  def batch_extraction(components, target_folder, batch_size: 1, use_prompt: false, filter_design_system: nil, extract_handler: nil)
+  def batch_extraction(components, target_folder, batch_size: 1, filter: nil, extract_handler: nil)
     extraction = TailwindDsl::Etl::Extractors::BatchExtraction.new(
       components,
       target_folder,
       batch_size: batch_size,
-      use_prompt: use_prompt,
-      filter_design_system: filter_design_system,
+      filter: filter,
       extract_handler: extract_handler)
     extraction.extract
   end
 end
 