feat: Add Original Column generator as a default feature generator (#153)

Added the original column generator to the api defaults file.

Author: ErikLagerSB

agentune/api/defaults.py

@@ -19,6 +19,9 @@
 from agentune.analyze.feature.gen.insightful_text_generator.insightful_text_generator import (
     ConversationQueryFeatureGenerator,
 )
+from agentune.analyze.feature.gen.original_column_generator.original_column_generator import (
+    OriginalColumnsGenerator,
+)
 from agentune.analyze.feature.select import FeatureSelector
 from agentune.analyze.feature.select.base import EnrichedFeatureSelector
 from agentune.analyze.feature.select.linear_pairwise import LinearPairWiseFeatureSelector
@@ -90,7 +93,7 @@ def conversation_query_feature_generator(self) -> ConversationQueryFeatureGenera
         return ConversationQueryFeatureGenerator.default(self.run_context._llm_context)
 
     def feature_generators(self) -> tuple[FeatureGenerator, ...]:
-        return (self.conversation_query_feature_generator(), )
+        return (self.conversation_query_feature_generator(), OriginalColumnsGenerator())
 
     def feature_selector(self) -> FeatureSelector | EnrichedFeatureSelector:
         return LinearPairWiseFeatureSelector()

tests/agentune/api/test_ops.py

@@ -12,7 +12,6 @@
 from agentune.analyze.feature.problem import ProblemDescription
 from agentune.analyze.run.analysis.base import (
     AnalyzeComponents,
-    NoFeaturesFoundError,
 )
 from agentune.api.base import RunContext
 from agentune.api.data import BoundTable
@@ -43,9 +42,6 @@ async def test_e2e_flow_synthetic(input_data_csv_path: Path, tmp_path: Path) ->
         split_input = await input.split()
         problem_description = ProblemDescription('target', 'Test synthetic data problem')
 
-        with pytest.raises(NoFeaturesFoundError): # Default generators can't do anything with this synthetic data
-            await ctx.ops.analyze(problem_description, split_input)
-
         components = AnalyzeComponents(
             generators=(ToySyncFeatureGenerator(), ToyAsyncFeatureGenerator(), ToySyncFeatureGenerator(), ToyAsyncFeatureGenerator()),
             selector=ToyAsyncEnrichedFeatureSelector()